Update tools/tools_on_modal_labs.py

tools/tools_on_modal_labs.py

@@ -82,7 +82,8 @@ def run_research_workflow(queryInput: QueryInput) -> str:
     return result
 
 NEBIUS_API_KEY = os.getenv("NEBIUS_API_KEY")
-AGENT_MODEL = os.getenv("AGENT_MODEL", "meta-llama/Meta-Llama-3.1-8B-Instruct")
+AGENT_MODEL = os.getenv("AGENT_MODEL", "Qwen/Qwen2.5-Coder-32B-Instruct")
+print(f"Using LLM model: {AGENT_MODEL}")
 
 # Load an LLM
 llm = NebiusLLM(
@@ -140,19 +141,20 @@ async def review_report(ctx: Context, review: str) -> str:
 # - The WriteAgent is responsible for writing the report.
 # - The ReviewAgent is responsible for reviewing the report.
 
-# The ResearchAgent uses the DuckDuckGoSearchAPIWrapper to search the web.
+# The ResearchAgent uses the Tavily Client to search the web.
 
 research_agent = FunctionAgent(
     name="ResearchAgent",
     description=(
-        "A research agent that searches the web using TavilyClient Search."
-        "It must not exceed 3 searches total, and must avoid repeating the same query. "
-        "Once sufficient information is collected, it should hand off to the WriteAgent."
+        "Useful for searching the web for information on a given topic and recording notes on the topic."
+        "It must not exceed 3 searches total and must avoid repeating the same query. "
+        "Once sufficient information is collected, you should hand off to the WriteAgent."
     ),
     system_prompt=(
-        "You are the ResearchAgent. Your goal is to gather sufficient information on the topic. "
-        "Only perform at most 2 distinct searches. If you have enough information or have reached 2 searches, "
-        "handoff to the WriteAgent. Avoid infinite loops! If search throws an error, stop further work and skip WriteAgent and ReviewAgent and return."
+        "You are the ResearchAgent. Your goal is to search the web for information on a given topic and record notes on the topic."
+        "Only perform at most 3 distinct searches. If you have enough information or have reached 3 searches, "
+        "handoff to the WriteAgent. Avoid infinite loops! If the search throws an error, stop further work and skip WriteAgent and ReviewAgent and return."
+        "You should have at least some notes on a topic before handing off control to the WriteAgent."
         "Respect invocation limits and cooldown periods."
     ),
     llm=llm,
@@ -160,7 +162,7 @@ research_agent = FunctionAgent(
         search_web,
         save_research
     ],
-    max_iterations=3,  # Limit to 2 iterations to prevent infinite loops
+    max_iterations=3,  # Limit to 3 iterations to prevent infinite loops
     cooldown=5,  # Cooldown to prevent rapid re-querying
     can_handoff_to=["WriteAgent"]
 )