Update auto_swarm_builder.py

pull/1101/head
CI-DEV 1 week ago committed by GitHub
parent a602dff844
commit dc134dd49d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -358,14 +358,7 @@ class AutoSwarmBuilder:
raise e raise e
def build_llm_agent(self, config: BaseModel): def build_llm_agent(self, config: BaseModel):
# return OpenAIFunctionCaller( # Legacy OpenAIFunctionCaller code removed - now using LiteLLM for multi-provider support
# system_prompt=BOSS_SYSTEM_PROMPT,
# api_key=os.getenv("OPENAI_API_KEY"),
# temperature=0.5,
# base_model=config,
# model_name=self.model_name,
# max_tokens=self.max_tokens,
# )
return LiteLLM( return LiteLLM(
model_name=self.model_name, model_name=self.model_name,
system_prompt=BOSS_SYSTEM_PROMPT, system_prompt=BOSS_SYSTEM_PROMPT,

Loading…
Cancel
Save