add cli.py

cli.py

@@ -1,131 +1,106 @@
 #!/usr/bin/env python3
 """
-CLI для исследовательского агента с web search через Tavily
+CLI для агента с retry через try/except
-Строит бриф: план → пошаговый сбор данных → финальная сводка
+Демонстрирует цикл: generate → валидация → (retry при ошибке) → END
 """
 
 import argparse
 import sys
-from dotenv import load_dotenv
+from graph import build_retry_graph
-from graph import build_research_graph
+from state import ReflectState
-from state import BriefState
 
-# Загружаем переменные окружения (.env)
+DEMO_QUESTION = "Объясни студенту разницу между tool и resource в MCP"
-load_dotenv()
 
-# Демо-тема по умолчанию
+def run_agent(question: str, max_rounds: int = 3, verbose: bool = True):
-DEMO_TOPIC = "Как студенту безопасно подключать MCP к LangChain"
+    """Запускает агента с retry-логикой"""
-
-def run_researcher(topic: str, verbose: bool = True):
-    """Запускает исследовательского агента"""
     if verbose:
         print("\n" + "="*80)
-        print(f"🔬 ИССЛЕДОВАТЕЛЬСКИЙ АГЕНТ")
+        print(f"🤖 ЗАПУСК АГЕНТА С RETRY (TRY/EXCEPT)")
-        print(f"📚 Тема: {topic}")
+        print(f"📋 Вопрос: {question}")
+        print(f"🔄 Макс. попыток: {max_rounds}")
         print("="*80)
     
     # Создаём граф
-    app = build_research_graph()
+    app = build_retry_graph(max_rounds=max_rounds)
     
-    # Начальное состояние
+    # Состояние
-    initial_state: BriefState = {
+    initial_state: ReflectState = {
-        "topic": topic,
+        "question": question,
-        "outline": None,
+        "draft": "",
-        "step_index": 0,
+        "error": None,
-        "notes": [],
+        "round": 1,
-        "final_brief": None
+        "max_rounds": max_rounds
     }
     
-    # Запуск (с потоковым выводом для прогресса)
+    # Запуск
-    if verbose:
-        print("\n🚀 Начинаю исследование...")
-    
     final_state = app.invoke(initial_state)
     
+    # Вывод
     if verbose:
         print("\n" + "="*80)
-        print("✅ ИССЛЕДОВАНИЕ ЗАВЕРШЕНО")
+        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']} попыток!")
+        else:
+            print("❌ ОШИБКА ГЕНЕРАЦИИ")
+            print("="*80)
+            print(f"Достигнут лимит попыток ({max_rounds})")
+            print(f"Последняя ошибка: {final_state['error']}")
     
     return final_state
 
-def save_brief_to_file(brief: str, filename: str = "research_brief.md"):
+def interactive_mode():
-    """Сохраняет бриф в Markdown файл"""
+    """Интерактивный режим"""
-    try:
+    print("\n🎮 Интерактивный режим агента с retry (try/except)")
-        with open(filename, 'w', encoding='utf-8') as f:
+    print("Введите 'exit' для выхода\n")
-            f.write(brief)
+    
-        print(f"📁 Бриф сохранён в файл: {filename}")
+    while True:
-    except Exception as e:
+        question = input("Ваш вопрос: ").strip()
-        print(f"⚠️ Ошибка сохранения: {e}")
+        if question.lower() == 'exit':
+            break
+        if not question:
+            continue
+        
+        max_rounds = input("Макс. попыток (по умолч. 3): ").strip()
+        max_rounds = int(max_rounds) if max_rounds.isdigit() else 3
+        
+        run_agent(question, max_rounds, verbose=True)
+        print("\n" + "─"*80 + "\n")
 
 def main():
     parser = argparse.ArgumentParser(
-        description="Исследовательский агент для создания брифа с web search",
+        description="LangGraph агент с retry через try/except",
         formatter_class=argparse.RawDescriptionHelpFormatter,
         epilog="""
 Примеры:
-  python cli.py --demo                    # Демо-тема про MCP и LangChain
+  python cli.py --demo                    # Запуск на демо-вопросе
-  python cli.py -t "Экологичные технологии 2024"  # Своя тема
+  python cli.py -q "Что такое AI?"        # Свой вопрос
-  python cli.py -t "..." --save report.md        # Сохранить в файл
+  python cli.py -q "..." --max-rounds 5   # 5 попыток
   python cli.py --interactive             # Интерактивный режим
         """
     )
     
-    parser.add_argument("-t", "--topic", help="Тема исследования")
+    parser.add_argument("-q", "--question", help="Вопрос для агента")
-    parser.add_argument("--demo", action="store_true", help="Запустить на демо-теме")
+    parser.add_argument("--max-rounds", 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("-s", "--save", metavar="FILE", help="Сохранить бриф в файл")
+    parser.add_argument("--quiet", action="store_true", help="Тихий режим (только ответ)")
-    parser.add_argument("--quiet", action="store_true", help="Тихий режим (только итоговый бриф)")
     
     args = parser.parse_args()
     
-    # Определяем тему
-    topic = None
     if args.interactive:
-        print("\n🎓 ИНТЕРАКТИВНЫЙ РЕЖИМ")
+        interactive_mode()
-        topic = input("Введите тему исследования: ").strip()
+    elif args.demo or (not args.question):
-        if not topic:
+        run_agent(DEMO_QUESTION, args.max_rounds, verbose=not args.quiet)
-            print("❌ Тема не может быть пустой")
+    elif args.question:
-            sys.exit(1)
+        run_agent(args.question, args.max_rounds, verbose=not args.quiet)
-    elif args.demo or not args.topic:
-        topic = DEMO_TOPIC
-        if not args.quiet:
-            print(f"\n🎯 Запуск на демо-теме: {topic}")
     else:
-        topic = args.topic
+        parser.print_help()
 
-    # Запуск исследования
+if __name__ == "__main__":
-    try:
-        result = run_researcher(topic, verbose=not args.quiet)
-        
-        # Вывод брифа
-        if args.quiet:
-            print(result["final_brief"])
-        else:
-            print("\n" + result["final_brief"])
-        
-        # Сохранение в файл
-        if args.save:
-            save_brief_to_file(result["final_brief"], args.save)
-        elif not args.quiet and not args.save:
-            # Предлагаем сохранить
-            save_option = input("\n💾 Сохранить бриф в файл? (y/n): ").strip().lower()
-            if save_option == 'y':
-                filename = input("Имя файла (по умолч. research_brief.md): ").strip()
-                if not filename:
-                    filename = "research_brief.md"
-                save_brief_to_file(result["final_brief"], filename)
-        
-    except KeyboardInterrupt:
-        print("\n\n⚠️ Исследование прервано пользователем")
-        sys.exit(0)
-    except Exception as e:
-        print(f"\n❌ Ошибка: {e}")
-        print("\nУбедитесь, что:")
-        print("1. Установлен TAVILY_API_KEY в .env файле")
-        print("2. Установлен OPENAI_API_KEY в .env файле")
-        print("3. Установлены зависимости: pip install -r requirements.txt")
-        sys.exit(1)
-
-if __name__ "__main__":
     main()