examples/basic_usage.py - master - angrybricks/dementor

examples/basic_usage.py
68 строк2 KB
MaksPV
Я просто сделаю коммит
30 сен 2025, 19:37
30 сен 2025, 19:379f30974
"""
Пример базового использования Study Assistant Agent в виде интерактивного чата.

Команды:
  /exit  — выйти из чата
  /reset — начать новый диалог (новый thread)
"""

import os
import sys
import uuid
from pathlib import Path

# Ensure project root is on sys.path when running this script directly
PROJECT_ROOT = str(Path(__file__).resolve().parents[1])
if PROJECT_ROOT not in sys.path:
    sys.path.insert(0, PROJECT_ROOT)
from core.agent import StudyAssistantAgent


def main():
    # Инициализация агента (автоматически загружает настройки из .env файла)
    # try:
    agent = StudyAssistantAgent()
    # except ValueError as e:
    #     print(f"Ошибка инициализации: {e}")
    #     print("Убедитесь, что в .env файле указан GIGACHAT_CREDENTIALS")
    #     return

    print("=== Study Assistant Agent — Chat Mode ===")
    print("Введите ваш запрос. Команды: /exit для выхода, /reset для нового диалога.")

    # Один thread_id на сессию, можно сбросить командой
    thread_id = f"chat_{uuid.uuid4().hex[:8]}"

    try:
        while True:
            try:
                user_input = input("you> ").strip()
            except EOFError:
                print()
                break

            if not user_input:
                continue

            if user_input.lower() in {"/exit", "exit", "quit", ":q"}:
                break

            if user_input.lower() == "/reset":
                thread_id = f"chat_{uuid.uuid4().hex[:8]}"
                print("Диалог сброшен. Начат новый thread.")
                continue

            try:
                response = agent.chat(user_input, thread_id=thread_id)
            except Exception as e:
                print(f"agent> Ошибка: {e}")
                continue

            print(f"agent> {response}")

    except KeyboardInterrupt:
        print("\nВыход…")


if __name__ == "__main__":
    main()