ctd

agent.py

@@ -13,6 +13,15 @@ load_dotenv()
 # Initialize tracing when module is imported
 trace_provider = None
 
+MANAGER_PROMPT = """You are a helpful assistant tasked with answering questions using a set of tools.
+Now, I will ask you a question. Report your thoughts, and finish your answer with the following template:
+FINAL ANSWER: [YOUR FINAL ANSWER].
+YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
+If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise.
+If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise.
+If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
+Your answer should only start with "FINAL ANSWER: ", then follows with the answer. """
+
 
 def initialize_tracing(enabled=True, provider="langfuse"):
     """
@@ -40,7 +49,12 @@ def get_agent():
         model_id="Qwen/Qwen2.5-Coder-32B-Instruct", provider="together"
     )
     llm_deepseek = InferenceClientModel(
-        "deepseek-ai/DeepSeek-R1", provider="together", max_tokens=8096
+        "deepseek-ai/DeepSeek-R1",
+        provider="together",
+        max_tokens=8096,
+        # "Qwen/Qwen3-235B-A22B-FP8",
+        # provider="together",
+        # max_tokens=8096,
     )
 
     # Create web agent
@@ -50,6 +64,7 @@ def get_agent():
         max_steps=3,
         name="Web_Agent",
         description="A web agent that can search the web and visit webpages.",
+        verbosity_level=1,
     )
 
     # Create manager agent
@@ -57,9 +72,11 @@ def get_agent():
         tools=[],
         managed_agents=[web_agent],
         model=llm_deepseek,
-        max_steps=3,
+        max_steps=5,
         planning_interval=10,
         additional_authorized_imports=["pandas", "numpy"],
+        verbosity_level=1,
+        description=MANAGER_PROMPT,
     )
     return manager_agent