From 8417a7f54d2c1cd27eb208c9f92ea3e9f694fd01 Mon Sep 17 00:00:00 2001 From: Sarah Wooders Date: Sat, 16 Aug 2025 13:41:48 -0700 Subject: [PATCH] fix: patch openai byok (#3966) --- letta/llm_api/openai.py | 1 + letta/schemas/providers/openai.py | 4 ++-- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/letta/llm_api/openai.py b/letta/llm_api/openai.py index b83c4de4..39852be8 100644 --- a/letta/llm_api/openai.py +++ b/letta/llm_api/openai.py @@ -42,6 +42,7 @@ from letta.utils import get_tool_call_id, smart_urljoin logger = get_logger(__name__) +# TODO: MOVE THIS TO OPENAI_CLIENT def openai_check_valid_api_key(base_url: str, api_key: Union[str, None]) -> None: if api_key: try: diff --git a/letta/schemas/providers/openai.py b/letta/schemas/providers/openai.py index ed9d5988..c01a9542 100644 --- a/letta/schemas/providers/openai.py +++ b/letta/schemas/providers/openai.py @@ -20,10 +20,10 @@ class OpenAIProvider(Provider): provider_type: Literal[ProviderType.openai] = Field(ProviderType.openai, description="The type of the provider.") provider_category: ProviderCategory = Field(ProviderCategory.base, description="The category of the provider (base or byok)") api_key: str = Field(..., description="API key for the OpenAI API.") - base_url: str = Field(..., description="Base URL for the OpenAI API.") + base_url: str = Field("https://api.openai.com/v1", description="Base URL for the OpenAI API.") async def check_api_key(self): - from letta.llm_api.openai import openai_check_valid_api_key + from letta.llm_api.openai import openai_check_valid_api_key # TODO: DO NOT USE THIS - old code path openai_check_valid_api_key(self.base_url, self.api_key)