[openai_api_base support] - ft/Added openai OPENAI_API_BASE llm config support (#1737)

2024-08-25 18:55:14 +08:00
parent 3fb4f2655b
commit b80925e857
4 changed files with 14 additions and 3 deletions
--- a/mem0/llms/openai.py
+++ b/mem0/llms/openai.py
@@ -22,7 +22,8 @@ class OpenAILLM(LLMBase):
            )
        else:
            api_key = os.getenv("OPENAI_API_KEY") or self.config.api_key
-            self.client = OpenAI(api_key=api_key)
+            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):
        """