113 lines
4.5 KiB
Python
113 lines
4.5 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
CLI для агента с рефлексией и try/except retry на LangGraph
|
|
"""
|
|
|
|
import argparse
|
|
import sys
|
|
from graph import build_reflection_graph
|
|
from state import ReflectState
|
|
|
|
DEMO_QUESTION = "Объясни студенту разницу между tool и resource в MCP"
|
|
|
|
def run_agent(question: str, max_rounds: int = 2, max_retries: int = 3, verbose: bool = True):
|
|
"""Запускает агента с рефлексией и retry"""
|
|
if verbose:
|
|
print("\n" + "="*80)
|
|
print(f"🤖 ЗАПУСК АГЕНТА С РЕФЛЕКСИЕЙ И TRY/EXCEPT RETRY")
|
|
print(f"📋 Вопрос: {question}")
|
|
print(f"🔄 Макс. раундов рефлексии: {max_rounds}")
|
|
print(f"🔄 Макс. попыток при ошибках: {max_retries}")
|
|
print("="*80)
|
|
|
|
# Создаём граф
|
|
app = build_reflection_graph(max_rounds=max_rounds, max_retries=max_retries)
|
|
|
|
# Состояние
|
|
initial_state: ReflectState = {
|
|
"question": question,
|
|
"draft": "",
|
|
"critique": "",
|
|
"verdict": "needs_revision",
|
|
"round": 1,
|
|
"max_rounds": max_rounds,
|
|
"error": None
|
|
}
|
|
|
|
# Запуск
|
|
final_state = app.invoke(initial_state)
|
|
|
|
# Вывод
|
|
if verbose:
|
|
print("\n" + "="*80)
|
|
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(f"⚠️ Достигнут лимит раундов ({max_rounds})")
|
|
|
|
return final_state
|
|
|
|
def interactive_mode():
|
|
"""Интерактивный режим"""
|
|
print("\n🎮 Интерактивный режим")
|
|
print("Введите 'exit' для выхода\n")
|
|
|
|
while True:
|
|
question = input("Ваш вопрос: ").strip()
|
|
if question.lower() == 'exit':
|
|
break
|
|
if not question:
|
|
continue
|
|
|
|
max_rounds = input("Макс. раундов рефлексии (по умолч. 2): ").strip()
|
|
max_rounds = int(max_rounds) if max_rounds.isdigit() else 2
|
|
|
|
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 агент с рефлексией и try/except retry",
|
|
formatter_class=argparse.RawDescriptionHelpFormatter,
|
|
epilog="""
|
|
Примеры:
|
|
python cli.py --demo # Запуск на демо-вопросе
|
|
python cli.py -q "Что такое AI?" # Свой вопрос
|
|
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=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="Тихий режим (только ответ)")
|
|
|
|
args = parser.parse_args()
|
|
|
|
if args.interactive:
|
|
interactive_mode()
|
|
elif args.demo or (not args.question):
|
|
run_agent(DEMO_QUESTION, args.max_rounds, args.max_retries, verbose=not args.quiet)
|
|
elif args.question:
|
|
run_agent(args.question, args.max_rounds, args.max_retries, verbose=not args.quiet)
|
|
else:
|
|
parser.print_help()
|
|
|
|
if __name__ == "__main__":
|
|
main() |