add local api

Author: zhiwei

configs/config_qwen.toml

@@ -0,0 +1,91 @@
+# General Config
+tag = "gaia"
+concurrency = 4
+workdir = "workdir"
+log_path = "log.txt"
+download_path = "downloads_folder"
+use_local_proxy = false # true for local proxy, false for public proxy
+split = "validation"
+save_path = "dra.jsonl"
+
+# Tool Config
+[searcher_tool]
+engine = "Google"
+fallback_engines = ["DuckDuckGo", "Baidu", "Bing"]
+retry_delay = 10
+max_retries = 3
+lang = "en"
+country = "us"
+filter_year = 0
+num_results = 5
+fetch_content = false
+max_length = 50000
+
+[deep_researcher_tool]
+model_id = "Qwen"
+max_depth = 2
+max_insights = 20
+time_limit_seconds = 60
+max_follow_ups = 3
+summarizer_model_id = "Qwen"
+
+[browser_tool]
+model_id = "Qwen"
+headless = false
+disable_security = true
+extra_chromium_args = []
+chrome_instance_path = ""
+wss_url = ""
+cdp_url = ""
+use_proxy = false
+max_length = 50000
+[browser.proxy]
+server = "xxxx"
+username = "xxxx"
+password = "xxxx"
+
+[deep_analyzer_tool]
+analyzer_model_ids = ["Qwen"]
+summarizer_model_id = "Qwen"
+
+# Agent configs
+[agent]
+name = "dra"
+use_hierarchical_agent = true
+
+[agent.planning_agent_config]
+model_id = "Qwen"
+name = "planning_agent"
+description = "A planning agent that can plan the steps to complete the task."
+max_steps = 15
+template_path = "src/agent/planning_agent/prompts/planning_agent.yaml"
+tools = ["planning"]
+managed_agents = ["deep_analyzer_agent", "browser_use_agent", "deep_researcher_agent"]
+
+[agent.deep_analyzer_agent_config]
+model_id = "Qwen"
+name = "deep_analyzer_agent"
+description = "Performs systematic, step-by-step analysis..."
+max_steps = 3
+template_path = "src/agent/deep_analyzer_agent/prompts/deep_analyzer_agent.yaml"
+tools = ["deep_analyzer", "python_interpreter"]
+
+[agent.browser_use_agent_config]
+model_id = "Qwen"
+name = "browser_use_agent"
+description = "Searches relevant web pages and interacts with them."
+max_steps = 5
+template_path = "src/agent/browser_use_agent/prompts/browser_use_agent.yaml"
+tools = ["auto_browser_use", "python_interpreter"]
+
+[agent.deep_researcher_agent_config]
+model_id = "Qwen"
+name = "deep_researcher_agent"
+description = "Conducts extensive web searches."
+max_steps = 3
+template_path = "src/agent/deep_researcher_agent/prompts/deep_researcher_agent.yaml"
+tools = ["deep_researcher", "python_interpreter"]
+
+[dataset]
+name = "2023_all"
+path = "data/GAIA"

main.py

@@ -0,0 +1,47 @@
+import warnings
+warnings.simplefilter("ignore", DeprecationWarning)
+
+import sys
+from pathlib import Path
+import asyncio
+
+root = str(Path(__file__).resolve().parents[0])
+sys.path.append(root)
+
+from src.logger import logger
+from src.config import config
+from src.models import model_manager
+from src.agent import create_agent
+from src.utils import assemble_project_path
+
+
+async def main():
+    # Init config and logger
+    config.init_config(config_path=assemble_project_path("./configs/config_qwen.toml"))
+    logger.init_logger(config.log_path)
+    logger.info(f"Initializing logger: {config.log_path}")
+    logger.info(f"Load config: {config}")
+
+    # Registed models
+    model_manager.init_models(use_local_proxy=config.use_local_proxy)
+    logger.info("Registed models: %s", ", ".join(model_manager.registed_models.keys()))
+    
+    # Create agent
+    agent = create_agent()
+
+    # Default task
+    default_task = "Use deep_researcher_agent to search the latest papers on the topic of 'AI Agent' and then summarize it."
+
+    # Interactive loop
+    print("Enter your task (press Enter to use default, type 'exit' to quit):")
+    while True:
+        task = input(f">>> ") or default_task
+        if task.lower() in ["exit", "quit"]:
+            print("Exiting.")
+            break
+        res = await agent.run(task)
+        logger.info(f"Result: {res}")
+        print(f"Result:\n{res}\n")
+
+if __name__ == '__main__':
+    asyncio.run(main())