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