Spaces:
Running
Running
File size: 1,016 Bytes
fafbbd5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
from langchain_gigachat.chat_models import GigaChat
from langchain_core.messages import HumanMessage
import os
def ask_gigachat(user_message, credentials=os.environ.get("API_KEY")):
# Инициализация GigaChat
giga = GigaChat(
model="GigaChat",
credentials=credentials,
verify_ssl_certs=False,
temperature=0.7,
MaxTokens=300,
profanity_check=True
)
# Создание списка сообщений
messages = [
{
"role": "system",
"content": 'Ты эксперт по кинематографу. Рекомендуй только лучшие фильмы, под запрос пользователя. Отвечай коротко и лаконично в формате "Название" - короткое описание'
},
HumanMessage(
content=user_message
)
]
# Получение ответа
response = giga.invoke(messages)
return response.content |