Add Litellm support (#1493)

2024-07-18 12:19:48 +05:30
parent fb5a3bfd95
commit c411dc294e
6 changed files with 899 additions and 3 deletions
--- a/mem0/utils/factory.py
+++ b/mem0/utils/factory.py
@@ -13,7 +13,8 @@ class LlmFactory:
        "openai": "mem0.llms.openai.OpenAILLM",
        "groq": "mem0.llms.groq.GroqLLM",
        "together": "mem0.llms.together.TogetherLLM",
-        "aws_bedrock": "mem0.llms.aws_bedrock.AWSBedrockLLM"
+        "aws_bedrock": "mem0.llms.aws_bedrock.AWSBedrockLLM",
+        "litellm": "mem0.llms.litellm.LiteLLM",
    }

    @classmethod