[improvement]: Graph memory support for non-structured models. (#1823)

2024-09-07 13:26:43 -07:00
parent a972d2fb07
commit 51c4f2aae8
5 changed files with 265 additions and 14 deletions
@@ -20,7 +20,8 @@ class LlmFactory:
        "litellm": "mem0.llms.litellm.LiteLLM",
        "azure_openai": "mem0.llms.azure_openai.AzureOpenAILLM",
        "openai_structured": "mem0.llms.openai_structured.OpenAIStructuredLLM",
-        "anthropic": "mem0.llms.anthropic.AnthropicLLM"
+        "anthropic": "mem0.llms.anthropic.AnthropicLLM",
+        "azure_openai_structured": "mem0.llms.azure_openai_structured.AzureOpenAIStructuredLLM"
    }

    @classmethod