diff --git a/agency_swarm/agents/agent.py b/agency_swarm/agents/agent.py index d64c8ebd..fc09eca6 100644 --- a/agency_swarm/agents/agent.py +++ b/agency_swarm/agents/agent.py @@ -10,6 +10,7 @@ from openai.types.beta.assistant import ToolResources from astra_assistants import patch, OpenAI +from agency_swarm.agents.config import DEFAULT_MODEL from agency_swarm.tools import BaseTool, ToolFactory, Retrieval from agency_swarm.tools import FileSearch, CodeInterpreter from agency_swarm.tools.oai.FileSearch import FileSearchConfig @@ -19,8 +20,6 @@ from pydantic import BaseModel from openai.lib._parsing._completions import type_to_response_format_param -DEFAULT_MODEL = "gpt-4o-2024-08-06" - class ExampleMessage(TypedDict): role: Literal["user", "assistant"] content: str diff --git a/agency_swarm/agents/config.py b/agency_swarm/agents/config.py new file mode 100644 index 00000000..4d95d206 --- /dev/null +++ b/agency_swarm/agents/config.py @@ -0,0 +1 @@ +DEFAULT_MODEL = "gpt-4o-2024-08-06"