agent_second/chat_sample.py - master - vitbur/ai_agents

agent_second/chat_sample.py
58 строк3 KB
Vitaly Buravlev
added german grammar checker
06 июл 2026, 13:37
06 июл 2026, 13:371624b2b
import os

from graph import grammar_agent


def print_agent_result(title: str, final_state: dict):
    """Красиво форматирует и выводит результат работы агента в консоль."""
    print("=" * 60)
    print(f"📌 ТЕСТ: {title}")
    print("=" * 60)
    print(f"📝 Исходный текст: '{final_state.get('text_content') or 'Картинка'}'")
    print(f"📊 Оценка грамматики: {final_state['score']}/5")
    print(f"❓ Есть ошибки: {'Да' if final_state['has_errors'] else 'Нет'}")

    if final_state["has_errors"]:
        print("\n❌ Найденные ошибки:")
        for err in final_state["errors_list"]:
            print(f"  - {err}")

        print("\n📖 Грамматические правила:")
        print(final_state["rules"])

        print("\n✅ Варианты исправления:")
        print(final_state["corrections"])
    else:
        print("\n🎉 Отличная работа! Ошибок не обнаружено.")
    print("=" * 60 + "\n")


if __name__ == "__main__":
    # Убедитесь, что у вас установлена переменная окружения OPENAI_API_KEY
    if not os.getenv("OPENAI_API_KEY"):
        print("⚠️ Ошибка: Пожалуйста, установите переменную окружения OPENAI_API_KEY перед запуском.")
        exit(1)

    print("🤖 Запуск тестирования LangGraph агента...\n")

    # --- ТЕСТ 1: Текст с явными ошибками ---
    # Ошибки: неправильный порядок слов, неверное окончание глагола, смешение языков
    bad_german_input = {
        "text_content": "Ich gehen nach Hause gestern und хочу кушать.",
        "image_path": None
    }

    print("⏳ Отправка текста с ошибками в LangGraph...")
    result_bad = grammar_agent.invoke(bad_german_input)
    print_agent_result("Немецкий текст с ошибками", result_bad)

    # --- ТЕСТ 2: Идеальный текст без ошибок ---
    # Проверим, что условный переход (Router Edge) работает правильно и завершает граф досрочно
    good_german_input = {
        "text_content": "Ich bin gestern nach Hause gegangen, weil ich Hunger hatte.",
        "image_path": None
    }

    print("⏳ Отправка правильного текста в LangGraph...")
    result_good = grammar_agent.invoke(good_german_input)
    print_agent_result("Идеальный немецкий текст", result_good)