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/llms/configs.py
+++ b/mem0/llms/configs.py
@@ -14,7 +14,7 @@ class LlmConfig(BaseModel):
    @field_validator("config")
    def validate_config(cls, v, values):
        provider = values.data.get("provider")
-        if provider in ("openai", "ollama", "groq", "together", "aws_bedrock", "litellm"):
+        if provider in ("openai", "ollama", "groq", "together", "aws_bedrock", "litellm", "azure_openai"):
            return v
        else:
            raise ValueError(f"Unsupported LLM provider: {provider}")