Improvements to Graph Memory (#1779)

2024-08-29 22:17:08 -07:00
parent 28bc4fe05b
commit 822a8acedb
10 changed files with 246 additions and 79 deletions
--- a/mem0/utils/factory.py
+++ b/mem0/utils/factory.py
@@ -19,6 +19,7 @@ class LlmFactory:
        "aws_bedrock": "mem0.llms.aws_bedrock.AWSBedrockLLM",
        "litellm": "mem0.llms.litellm.LiteLLM",
        "azure_openai": "mem0.llms.azure_openai.AzureOpenAILLM",
+        "openai_structured": "mem0.llms.openai_structured.OpenAIStructuredLLM",
    }

    @classmethod