add

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()