main.py

from langchain_gigachat import GigaChat
from langchain.agents import create_agent
from langchain.tools import tool
from langchain_core.chat_history import InMemoryChatMessageHistory

AGENT = GigaChat(
    credentials='key',
    model='GigaChat-2',
    verify_ssl_certs=False
)

SYSTEM_PROMPT = """
    Ты ИИ-ассистент, который обрабатывает заявки
    Список твоих задач: квалификация лидов, отработка возражений, 
    донесение выгод, ответы на все вопросы по продукции,
    подобрать подходящую мебель, сделать расчёт, дожим рассылками для тех, кто перестал отвечать
"""

PRODUCT_KATALOG = {
    "диван": 12345,
    "стул": 5000,
    "стол": 10421,
    "лампа": 100
}
korzina = []
@tool
def vozr(text: str) -> str:
    """Отвечай на возражения и претензии культурно и прилично.
    Пытайся преподнести наш продукт как что-то лучшее."""
    return f"Отвечай на возражения{text} и претензии{text} культурно и прилично. Пытайся преподнести наш продукт как что-то лучшее."
@tool
def somnenie(text: str) -> str:
    """Если вдруг твой клиент выражает сомнение в покупке услуги, уважительно и приятно донеси ему, что эта услуга стоит своей покупки
    Общайся уважительно."""
    return f"Если вдруг твой клиент выражает сомнение{text} в покупке услуги, уважительно и приятно донеси ему, что эта услуга стоит своей покупки.Общайся уважительно."

@tool
def raschet(text: str) -> str:
    """
    Делай расчёт стоимости товаров о которых пишет твой покупатель
    """
    return f"Делай расчёт стоимости товаров, которые хочет купить покупатель, описывая это тебе{text}"

@tool
def add_korzina(text: str) -> str:
    """Добавляет товар в корзину по его названию.
    Пример вызова: add_korzina("диван")"""
    product_name = text.strip().lower()

    if product_name in PRODUCT_KATALOG:
        price = PRODUCT_KATALOG[product_name]
        korzina.append({"name": product_name, "price": price})
        return f"Товар {product_name} добавлен в корзину. Цена - {price} рублей"
    else:
        return f"Товара {product_name} нет в наличии"

@tool
def verifi(text: str) -> str:
    """Вызывается, если пользователь пишет, что хочет уже оплачивать заказ"""
    return f"Вы уверены в том, что хотите платить?"
@tool
def oplata(text: str) -> str:
    """Вызывается после того, как пользователь изъявляет желание об оформлении покупки."""
    return (f"Выведи стоимость каждого товара, а после сумму всех вместе\n"
            "Реквизиты: СБЕР - 1234 5678 1234 5678")

tools = [vozr, somnenie, raschet, add_korzina, verifi, oplata]

chat = InMemoryChatMessageHistory()

AGENT = create_agent(model=AGENT, tools=tools, system_prompt=SYSTEM_PROMPT)

while True:
    question = str(input("Пользователь: "))
    chat.add_user_message(question)
    full_messages = chat.messages
    inputs = {"messages": full_messages}

    response_content = ""
    for chunk in AGENT.stream(inputs, stream_mode="updates"):
        if 'model' in chunk and 'messages' in chunk['model']:
            ai_message = chunk['model']['messages'][0]
            if ai_message.content:
                response_content = ai_message.content
                print("Ассистент:", response_content)

    if response_content:
        chat.add_ai_message(response_content)