diff --git a/cli.py b/cli.py index 8a2b4d1..e07bbcb 100644 --- a/cli.py +++ b/cli.py @@ -1,35 +1,37 @@ #!/usr/bin/env python3 """ -CLI для агента с retry через try/except -Демонстрирует цикл: generate → валидация → (retry при ошибке) → END +CLI для агента с рефлексией и try/except retry на LangGraph """ import argparse import sys -from graph import build_retry_graph +from graph import build_reflection_graph from state import ReflectState DEMO_QUESTION = "Объясни студенту разницу между tool и resource в MCP" -def run_agent(question: str, max_rounds: int = 3, verbose: bool = True): - """Запускает агента с retry-логикой""" +def run_agent(question: str, max_rounds: int = 2, max_retries: int = 3, verbose: bool = True): + """Запускает агента с рефлексией и retry""" if verbose: print("\n" + "="*80) - print(f"🤖 ЗАПУСК АГЕНТА С RETRY (TRY/EXCEPT)") + print(f"🤖 ЗАПУСК АГЕНТА С РЕФЛЕКСИЕЙ И TRY/EXCEPT RETRY") print(f"📋 Вопрос: {question}") - print(f"🔄 Макс. попыток: {max_rounds}") + print(f"🔄 Макс. раундов рефлексии: {max_rounds}") + print(f"🔄 Макс. попыток при ошибках: {max_retries}") print("="*80) # Создаём граф - app = build_retry_graph(max_rounds=max_rounds) + app = build_reflection_graph(max_rounds=max_rounds, max_retries=max_retries) # Состояние initial_state: ReflectState = { "question": question, "draft": "", - "error": None, + "critique": "", + "verdict": "needs_revision", "round": 1, - "max_rounds": max_rounds + "max_rounds": max_rounds, + "error": None } # Запуск @@ -38,25 +40,25 @@ def run_agent(question: str, max_rounds: int = 3, verbose: bool = True): # Вывод if verbose: print("\n" + "="*80) - if final_state["error"] is None: - print("✅ ФИНАЛЬНЫЙ РЕЗУЛЬТАТ") - print("="*80) - print(f"📝 Итоговый ответ (попытка {final_state['round']}):") - print("-"*80) - print(final_state["draft"]) - print("-"*80) - print(f"🎉 Успешно сгенерировано за {final_state['round']} попыток!") + print("✅ ФИНАЛЬНЫЙ РЕЗУЛЬТАТ") + print("="*80) + print(f"📝 Итоговый ответ (раунд {final_state['round']}):") + print("-"*80) + print(final_state["draft"]) + print("-"*80) + + if final_state.get("verdict") == "ok": + print("🎉 Ответ одобрен критиком!") + elif final_state.get("error"): + print(f"⚠️ Завершено с ошибкой: {final_state['error']}") else: - print("❌ ОШИБКА ГЕНЕРАЦИИ") - print("="*80) - print(f"Достигнут лимит попыток ({max_rounds})") - print(f"Последняя ошибка: {final_state['error']}") + print(f"⚠️ Достигнут лимит раундов ({max_rounds})") return final_state def interactive_mode(): """Интерактивный режим""" - print("\n🎮 Интерактивный режим агента с retry (try/except)") + print("\n🎮 Интерактивный режим") print("Введите 'exit' для выхода\n") while True: @@ -66,27 +68,32 @@ def interactive_mode(): if not question: continue - max_rounds = input("Макс. попыток (по умолч. 3): ").strip() - max_rounds = int(max_rounds) if max_rounds.isdigit() else 3 + max_rounds = input("Макс. раундов рефлексии (по умолч. 2): ").strip() + max_rounds = int(max_rounds) if max_rounds.isdigit() else 2 - run_agent(question, max_rounds, verbose=True) + max_retries = input("Макс. попыток при ошибках (по умолч. 3): ").strip() + max_retries = int(max_retries) if max_retries.isdigit() else 3 + + run_agent(question, max_rounds, max_retries, verbose=True) print("\n" + "─"*80 + "\n") def main(): parser = argparse.ArgumentParser( - description="LangGraph агент с retry через try/except", + description="LangGraph агент с рефлексией и try/except retry", formatter_class=argparse.RawDescriptionHelpFormatter, epilog=""" Примеры: python cli.py --demo # Запуск на демо-вопросе python cli.py -q "Что такое AI?" # Свой вопрос - python cli.py -q "..." --max-rounds 5 # 5 попыток + python cli.py -q "..." --max-rounds 3 # 3 раунда рефлексии + python cli.py -q "..." --max-retries 5 # 5 попыток при ошибках python cli.py --interactive # Интерактивный режим """ ) parser.add_argument("-q", "--question", help="Вопрос для агента") - parser.add_argument("--max-rounds", type=int, default=3, help="Макс. попыток (по умолч. 3)") + parser.add_argument("--max-rounds", type=int, default=2, help="Макс. раундов рефлексии (по умолч. 2)") + parser.add_argument("--max-retries", type=int, default=3, help="Макс. попыток при ошибках (по умолч. 3)") parser.add_argument("--demo", action="store_true", help="Запустить демо-вопрос") parser.add_argument("-i", "--interactive", action="store_true", help="Интерактивный режим") parser.add_argument("--quiet", action="store_true", help="Тихий режим (только ответ)") @@ -96,9 +103,9 @@ def main(): if args.interactive: interactive_mode() elif args.demo or (not args.question): - run_agent(DEMO_QUESTION, args.max_rounds, verbose=not args.quiet) + run_agent(DEMO_QUESTION, args.max_rounds, args.max_retries, verbose=not args.quiet) elif args.question: - run_agent(args.question, args.max_rounds, verbose=not args.quiet) + run_agent(args.question, args.max_rounds, args.max_retries, verbose=not args.quiet) else: parser.print_help()