Dkapsis's picture
manager tools
ed20377
import os
from smolagents.utils import encode_image_base64, make_image_url
from smolagents import OpenAIServerModel, CodeAgent, InferenceClientModel
import app_agents.web_agent as web_agent
import app_tools.text_inspector
import app_tools.visual_qa
manager_agent = CodeAgent(
model=InferenceClientModel("deepseek-ai/DeepSeek-R1", provider="together", max_tokens=8096),
tools=[app_tools.text_inspector.TextInspectorTool(), app_tools.visual_qa.VisualQATool()],
planning_interval=4,
verbosity_level=2,
managed_agents=[web_agent.web_agent],
max_steps=10,
# grammar=DEFAULT_CODEAGENT_REGEX_GRAMMAR,
additional_authorized_imports=[
"requests",
"zipfile",
"os",
"pandas",
"numpy",
"sympy",
"json",
"bs4",
"pubchempy",
"xml",
"yahoo_finance",
"Bio",
"sklearn",
"scipy",
"pydub",
"io",
"PIL",
"chess",
"PyPDF2",
"pptx",
"torch",
"datetime",
"csv",
"fractions",
],
)