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)],
        planning_interval=3,
        max_steps=20,
    )