opt deep research

tests/test_agents.py

@@ -338,18 +338,16 @@ async def test_deep_research_agent():
     from src.agent.deep_research.deep_research_agent import DeepResearchAgent, PLAN_FILENAME, REPORT_FILENAME
     from src.utils import llm_provider
 
-    # llm = llm_provider.get_llm_model(
-    #     provider="azure_openai",
-    #     model_name="gpt-4o",
-    #     temperature=0.5,
-    #     base_url=os.getenv("AZURE_OPENAI_ENDPOINT", ""),
-    #     api_key=os.getenv("AZURE_OPENAI_API_KEY", ""),
-    # )
-
     llm = llm_provider.get_llm_model(
-        provider="bedrock",
+        provider="openai",
+        model_name="gpt-4o",
+        temperature=0.5
     )
 
+    # llm = llm_provider.get_llm_model(
+    #     provider="bedrock",
+    # )
+
     mcp_server_config = {
         "mcpServers": {
             "desktop-commander": {
@@ -364,9 +362,8 @@ async def test_deep_research_agent():
 
     browser_config = {"headless": False, "window_width": 1280, "window_height": 1100, "use_own_browser": False}
     agent = DeepResearchAgent(llm=llm, browser_config=browser_config, mcp_server_config=mcp_server_config)
-
     research_topic = "Impact of Microplastics on Marine Ecosystems"
-    task_id_to_resume = None  # Set this to resume a previous task ID
+    task_id_to_resume = "815460fb-337a-4850-8fa4-a5f2db301a89"  # Set this to resume a previous task ID
 
     print(f"Starting research on: {research_topic}")
 
@@ -374,8 +371,9 @@ async def test_deep_research_agent():
         # Call run and wait for the final result dictionary
         result = await agent.run(research_topic,
                                  task_id=task_id_to_resume,
-                                 save_dir="./tmp/downloads",
-                                 max_parallel_browsers=1)
+                                 save_dir="./tmp/deep_research",
+                                 max_parallel_browsers=1,
+                                 )
 
         print("\n--- Research Process Ended ---")
         print(f"Status: {result.get('status')}")

tests/test_llm_api.py

@@ -141,13 +141,19 @@ def test_ibm_model():
     test_llm(config, "Describe this image", "assets/examples/test.png")
 
 
+def test_qwen_model():
+    config = LLMConfig(provider="alibaba", model_name="qwen3-30b-a3b")
+    test_llm(config, "How many 'r's are in the word 'strawberry'?")
+
+
 if __name__ == "__main__":
     # test_openai_model()
     # test_google_model()
-    test_azure_openai_model()
+    # test_azure_openai_model()
     # test_deepseek_model()
     # test_ollama_model()
-    # test_deepseek_r1_model()
+    test_deepseek_r1_model()
     # test_deepseek_r1_ollama_model()
     # test_mistral_model()
     # test_ibm_model()
+    # test_qwen_model()