Update run.py

run.py

@@ -282,12 +282,12 @@ def create_agent(model_name="o1"):
     model_params = MODEL_CONFIGS[model_name]["litellm_params"]
     model_params.setdefault("custom_role_conversions", custom_role_conversions)
     model_params.setdefault("max_completion_tokens", 8192)
-
     # (Optional: adjust parameters here to lower temperature for more factual answers.)
+
     model = LiteLLMModel(**model_params)
     logger.info(f"Initialized LiteLLMModel with model_name={model_name}")
-
     text_limit = 100000
+
     browser = SimpleTextBrowser(**BROWSER_CONFIG)
     logger.info("Initialized SimpleTextBrowser with custom configuration.")
 
@@ -329,16 +329,13 @@ def create_agent(model_name="o1"):
         managed_agents=[text_webbrowser_agent],
     )
     logger.info("Initialized Manager CodeAgent.")
-
     return manager_agent
 
 def main():
     args = parse_args()
     logger.info(f"Received question: {args.question} with model_id={args.model_id}")
-
     agent = create_agent(model_name=args.model_id)
     answer = agent.run(args.question)
-
     if isinstance(answer, str):
         print(f"Got this answer: {answer}")
     else:
@@ -346,8 +343,7 @@ def main():
         for chunk in answer:
             result += chunk
         print(f"Got this answer: {result}")
-
     logger.info("Agent has completed processing the question.")
 
 if __name__ == "__main__":
-    main()
+    main()