Support async client (#1980)

2024-10-22 12:42:55 +05:30
parent c5d298eec8
commit fbf1d8c372
11 changed files with 213 additions and 58 deletions
--- a/mem0/llms/openai.py
+++ b/mem0/llms/openai.py
@@ -18,7 +18,9 @@ class OpenAILLM(LLMBase):
        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 or os.getenv("OPENROUTER_API_BASE") or "https://openrouter.ai/api/v1",
+                base_url=self.config.openrouter_base_url
+                or os.getenv("OPENROUTER_API_BASE")
+                or "https://openrouter.ai/api/v1",
            )
        else:
            api_key = self.config.api_key or os.getenv("OPENAI_API_KEY")