fix(llm): use empty str for llm config (#155)

TODO: we should better reuse the LLM API/configs
apache · Jan 8, 2025 · 2441d2b · 2441d2b
1 parent 820bfb2
commit 2441d2b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/hugegraph-llm/src/hugegraph_llm/models/embeddings/openai.py b/hugegraph-llm/src/hugegraph_llm/models/embeddings/openai.py
@@ -28,6 +28,7 @@ def __init__(
             api_key: Optional[str] = None,
             api_base: Optional[str] = None
     ):
+        api_key = api_key or ''
         self.client = OpenAI(api_key=api_key, base_url=api_base)
         self.aclient = AsyncOpenAI(api_key=api_key, base_url=api_base)
         self.embedding_model_name = model_name

diff --git a/hugegraph-llm/src/hugegraph_llm/models/llms/openai.py b/hugegraph-llm/src/hugegraph_llm/models/llms/openai.py
@@ -42,6 +42,7 @@ def __init__(
         max_tokens: int = 4096,
         temperature: float = 0.0,
     ) -> None:
+        api_key = api_key or ''
         self.client = OpenAI(api_key=api_key, base_url=api_base)
         self.aclient = AsyncOpenAI(api_key=api_key, base_url=api_base)
         self.model = model_name