Movie_finder / funcs /try_gpt.py
xefr762's picture
Update funcs/try_gpt.py
e78a98f verified
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