Add support for anthropic (#1819)

2024-09-07 02:12:22 +05:30
parent 965f7a3735
commit d32ae1a0b1
13 changed files with 82 additions and 84 deletions
--- a/mem0/utils/factory.py
+++ b/mem0/utils/factory.py
@@ -20,6 +20,7 @@ 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"
    }

    @classmethod