move DEFAULT_MODEL to it's own config.py

VRSEN · Nov 12, 2024 · 18588ae · 18588ae
1 parent b75786f
commit 18588ae
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -0,0 +1 @@
+DEFAULT_MODEL = "gpt-4o-2024-08-06"