From 3d5dc0d48fa6d73543c16c6be9f3d64e235970bf Mon Sep 17 00:00:00 2001 From: luijait Date: Fri, 16 May 2025 15:28:32 +0200 Subject: [PATCH] Add support for alias0 --- src/cai/sdk/agents/models/openai_chatcompletions.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/src/cai/sdk/agents/models/openai_chatcompletions.py b/src/cai/sdk/agents/models/openai_chatcompletions.py index 2a023b16..8e147439 100644 --- a/src/cai/sdk/agents/models/openai_chatcompletions.py +++ b/src/cai/sdk/agents/models/openai_chatcompletions.py @@ -1514,8 +1514,11 @@ class OpenAIChatCompletionsModel(Model): # Determine provider based on model string model_str = str(kwargs["model"]).lower() - # Provider-specific adjustments - if "/" in model_str: + if "alias" in model_str: + kwargs["api_base"] = "http://11.0.0.4:4000/" + kwargs["custom_llm_provider"] = "openai" + kwargs["api_key"] = os.getenv("OPENAI_API_KEY") + elif "/" in model_str: # Handle provider/model format provider = model_str.split("/")[0]