zxmfke
61 lines
1.5 KiB
Python
61 lines
1.5 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Agent Selector - Allows users to choose different agent modes
|
|
"""
|
|
|
|
import asyncio
|
|
import sys
|
|
|
|
from app.agent import CoTAgent
|
|
from app.agent.manus import Manus
|
|
from app.logger import logger
|
|
|
|
|
|
async def run_cot_agent():
|
|
"""Run Chain of Thought agent"""
|
|
agent = CoTAgent()
|
|
prompt = input("Enter your question: ")
|
|
if not prompt.strip():
|
|
logger.warning("Empty question provided")
|
|
return
|
|
|
|
logger.warning("Processing your question...")
|
|
result = await agent.run(prompt)
|
|
logger.info("Question processing completed")
|
|
|
|
|
|
async def run_react_agent():
|
|
"""Run ReAct agent (Manus)"""
|
|
agent = Manus()
|
|
prompt = input("Enter your request: ")
|
|
if not prompt.strip():
|
|
logger.warning("Empty request provided")
|
|
return
|
|
|
|
logger.warning("Processing your request...")
|
|
await agent.run(prompt)
|
|
logger.info("Request processing completed")
|
|
|
|
|
|
async def main():
|
|
print("\nSelect the agent mode to use:")
|
|
print("1. ReAct mode - Can use tools to execute tasks")
|
|
print("2. Chain of Thought (CoT) mode - Shows detailed thinking process")
|
|
|
|
choice = input("\nEnter option (1 or 2): ").strip()
|
|
|
|
if choice == "1":
|
|
await run_react_agent()
|
|
elif choice == "2":
|
|
await run_cot_agent()
|
|
else:
|
|
print("Invalid choice, please enter 1 or 2")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
asyncio.run(main())
|
|
except KeyboardInterrupt:
|
|
logger.warning("Operation interrupted")
|
|
sys.exit(0)
|