agents/manager_agent.py

""" "Manger Agent"""

import os

from smolagents import CodeAgent, OpenAIServerModel, DuckDuckGoSearchTool


MODEL_ID = os.getenv("MANAGER_MODEL_ID")
BASE_URL = os.getenv("MANGER_BASE_URL")
API_KEY = os.getenv("OPENAI_API_KEY")
API_BASE = "/".join([BASE_URL, "v1"])

model = OpenAIServerModel(model_id=MODEL_ID, api_base=API_BASE, api_key=API_KEY)


def manager_agent_factory() -> CodeAgent:
    """Manager Agent Instance"""
    return CodeAgent(
        model=model,
        tools=[DuckDuckGoSearchTool(verify=False)],
        additional_authorized_imports=["requests", "bs4"],
        planning_interval=3,
        max_steps=20,
    )


# managed template task / report

#   additional_authorized_imports=["requests"]


#   "tools": self.tools,
#                 "managed_agents": self.managed_agents,
#                 "authorized_imports": (
#                     "You can import from any package you want."
#                     if "*" in self.authorized_imports
#                     else str(self.authorized_imports)
#                 ),
#                 "custom_instructions": self.instructions,