Support Azure OpenAI LLM (#1581)

2024-08-04 00:01:43 +09:00
parent 504a87d799
commit 81b4431c9b
5 changed files with 184 additions and 12 deletions
--- a/mem0/utils/factory.py
+++ b/mem0/utils/factory.py
@@ -18,6 +18,7 @@ class LlmFactory:
        "aws_bedrock": "mem0.llms.aws_bedrock.AWSBedrockLLM",
        "litellm": "mem0.llms.litellm.LiteLLM",
        "ollama": "mem0.llms.ollama.OllamaLLM",
+        "azure_openai": "mem0.llms.azure_openai.AzureOpenAILLM",
    }

    @classmethod