Add embedder docs and config changes (#1684)

2024-08-12 16:09:01 +05:30
parent 464a188662
commit b245309242
12 changed files with 130 additions and 28 deletions
--- a/mem0/llms/openai.py
+++ b/mem0/llms/openai.py
@@ -14,10 +14,11 @@ class OpenAILLM(LLMBase):
        if not self.config.model:
            self.config.model="gpt-4o"

-        if os.environ.get("OPENROUTER_API_KEY"):
+        if os.environ.get("OPENROUTER_API_KEY"): # Use OpenRouter
            self.client = OpenAI(api_key=os.environ.get("OPENROUTER_API_KEY"), base_url=self.config.openrouter_base_url)
        else:
-            self.client = OpenAI()
+            api_key = os.getenv("OPENAI_API_KEY") or self.config.api_key
+            self.client = OpenAI(api_key=api_key)
    
    def _parse_response(self, response, tools):
        """