Spaces:
Running
Running
File size: 1,035 Bytes
fafbbd5 fdad76e 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 31 32 |
from langchain_gigachat.chat_models import GigaChat
from langchain_core.messages import HumanMessage
import os
api_key = os.environ.get("api_key")
def ask_gigachat(user_message, credentials=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 |