StanSava commited on
Commit
21a0c2b
·
1 Parent(s): d6e5a14

upgrade packages for gpt-5.1

Browse files
agents/orchestrator_agent/orchestrator_agent.py CHANGED
@@ -9,7 +9,7 @@ from .get_orchestrator_prompt import get_orchestrator_prompt
9
  from .managed_agents import get_web_search_agent, get_supabase_agent
10
 
11
  llm_provider = LLMProviderType.OPENAI
12
- llm_model = LLMModelType.openai.gpt_4o
13
 
14
  @tool
15
  def final_answer_callback(project_id: str, project_name: str, team: List[EmployeeTrainingPlan]) -> AnalysisResult:
@@ -54,7 +54,8 @@ class OrchestratorAgent:
54
  tools=[final_answer_callback],
55
  managed_agents=[get_web_search_agent(model), get_supabase_agent(model)],
56
  name="orchestrator_agent",
57
- description="An agent that orchestrates the training-plan analysis process."
 
58
  )
59
 
60
  def analyze_and_plan(self, user_prompt: str):
 
9
  from .managed_agents import get_web_search_agent, get_supabase_agent
10
 
11
  llm_provider = LLMProviderType.OPENAI
12
+ llm_model = LLMModelType.openai.gpt_5_1
13
 
14
  @tool
15
  def final_answer_callback(project_id: str, project_name: str, team: List[EmployeeTrainingPlan]) -> AnalysisResult:
 
54
  tools=[final_answer_callback],
55
  managed_agents=[get_web_search_agent(model), get_supabase_agent(model)],
56
  name="orchestrator_agent",
57
+ description="An agent that orchestrates the training-plan analysis process.",
58
+ stream_outputs=True
59
  )
60
 
61
  def analyze_and_plan(self, user_prompt: str):
llm_provider.py CHANGED
@@ -97,7 +97,12 @@ class LLMProvider:
97
  model_id=self.model_id,
98
  token=self.api_key
99
  )
100
- elif self.provider in [LLMProviderType.OPENAI, LLMProviderType.OPENROUTER, LLMProviderType.CLAUDE, LLMProviderType.GEMINI]:
 
 
 
 
 
101
  api_base = BASE_URL_MAP.get(self.provider)
102
  if not api_base:
103
  raise ValueError(f"Base URL not configured for provider: {self.provider}")
 
97
  model_id=self.model_id,
98
  token=self.api_key
99
  )
100
+ elif self.provider in [
101
+ LLMProviderType.OPENAI,
102
+ LLMProviderType.OPENROUTER,
103
+ LLMProviderType.CLAUDE,
104
+ LLMProviderType.GEMINI,
105
+ ]:
106
  api_base = BASE_URL_MAP.get(self.provider)
107
  if not api_base:
108
  raise ValueError(f"Base URL not configured for provider: {self.provider}")