Kackle commited on
Commit
704d900
·
verified ·
1 Parent(s): 8a1c7db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -9
app.py CHANGED
@@ -373,9 +373,9 @@ class SlpMultiAgent:
373
 
374
  # Simplified research agent
375
  research_agent = CodeAgent(
376
- tools=[GoogleSearchTool(), KnowledgeBaseTool(), WikipediaSearchTool(),WebScrapeTool()],
377
  model=model if not isinstance(model, GeminiModel) else OpenAIServerModel(model_id="gpt-3.5-turbo", temperature=0.0, max_tokens=400),
378
- additional_authorized_imports=["re", "math", "json", "pandas", "numpy", "requests", "time", "os", "tempfile", "uuid"],
379
  max_steps=2,
380
  name="ResearchAgent",
381
  verbosity_level=0,
@@ -385,7 +385,7 @@ class SlpMultiAgent:
385
  solver_agent = CodeAgent(
386
  tools=[MathTool(), GoogleSearchTool()],
387
  model=model if not isinstance(model, GeminiModel) else OpenAIServerModel(model_id="gpt-3.5-turbo", temperature=0.0, max_tokens=400),
388
- additional_authorized_imports=["re", "math", "json", "pandas", "numpy", "requests", "time", "os", "tempfile", "uuid"],
389
  max_steps=2,
390
  name="SolverAgent",
391
  verbosity_level=0,
@@ -393,12 +393,8 @@ class SlpMultiAgent:
393
  )
394
 
395
  manager_agent = CodeAgent(
396
- model=OpenAIServerModel(
397
- model_id="gpt-3.5-turbo",
398
- temperature=0.0,
399
- max_tokens=500
400
- ),
401
- tools=[GoogleSearchTool(), MathTool(), FileProcessorTool(), KnowledgeBaseTool(), WikipediaSearchTool(),WebScrapeTool()],
402
  managed_agents=[research_agent, solver_agent],
403
  name="ManagerAgent",
404
  description="Manager with comprehensive tool access and agent coordination.",
 
373
 
374
  # Simplified research agent
375
  research_agent = CodeAgent(
376
+ tools=[GoogleSearchTool(), KnowledgeBaseTool()],
377
  model=model if not isinstance(model, GeminiModel) else OpenAIServerModel(model_id="gpt-3.5-turbo", temperature=0.0, max_tokens=400),
378
+ additional_authorized_imports=["re", "json"],
379
  max_steps=2,
380
  name="ResearchAgent",
381
  verbosity_level=0,
 
385
  solver_agent = CodeAgent(
386
  tools=[MathTool(), GoogleSearchTool()],
387
  model=model if not isinstance(model, GeminiModel) else OpenAIServerModel(model_id="gpt-3.5-turbo", temperature=0.0, max_tokens=400),
388
+ additional_authorized_imports=["math", "re"],
389
  max_steps=2,
390
  name="SolverAgent",
391
  verbosity_level=0,
 
393
  )
394
 
395
  manager_agent = CodeAgent(
396
+ model=model if not isinstance(model, GeminiModel) else OpenAIServerModel(model_id="gpt-3.5-turbo", temperature=0.0, max_tokens=400),
397
+ tools=[GoogleSearchTool(), MathTool(), FileProcessorTool(), KnowledgeBaseTool()],
 
 
 
 
398
  managed_agents=[research_agent, solver_agent],
399
  name="ManagerAgent",
400
  description="Manager with comprehensive tool access and agent coordination.",