From abcc1fed311b81451c5412f785828983c8c6f407 Mon Sep 17 00:00:00 2001 From: Andy Lee Date: Sat, 9 Aug 2025 17:40:13 -0700 Subject: [PATCH] fix: type --- packages/leann-core/src/leann/chat.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/packages/leann-core/src/leann/chat.py b/packages/leann-core/src/leann/chat.py index e8b8fb2..4d65f36 100644 --- a/packages/leann-core/src/leann/chat.py +++ b/packages/leann-core/src/leann/chat.py @@ -685,7 +685,7 @@ class HFChat(LLMInterface): class OpenAIChat(LLMInterface): """LLM interface for OpenAI models.""" - def __init__(self, model: str = "gpt-4o", api_key: Union[str, None] = None): + def __init__(self, model: str = "gpt-4o", api_key: Optional[str] = None): self.model = model self.api_key = api_key or os.getenv("OPENAI_API_KEY")