[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/embeddings/openai.py
+++ b/mem0/embeddings/openai.py
@@ -15,7 +15,8 @@ class OpenAIEmbedding(EmbeddingBase):
        self.config.embedding_dims = self.config.embedding_dims or 1536

        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 embed(self, text):
        """