Fixed environment variables priority in OpenAILLM (#1851)

2024-09-13 02:07:15 +08:00
parent f40a2e7603
commit 47a8e677e9
1 changed files with 1 additions and 1 deletions
--- a/mem0/llms/openai.py
+++ b/mem0/llms/openai.py
@@ -22,7 +22,7 @@ class OpenAILLM(LLMBase):
            )
        else:
            api_key = self.config.api_key or os.getenv("OPENAI_API_KEY")
-            base_url = self.config.openai_base_url or os.getenv("OPENAI_API_BASE")
+            base_url = os.getenv("OPENAI_API_BASE") or self.config.openai_base_url
            self.client = OpenAI(api_key=api_key, base_url=base_url)

    def _parse_response(self, response, tools):