Add Litellm support (#1493)

2024-07-17 23:49:48 -07:00
parent fb5a3bfd95
commit c411dc294e
6 changed files with 899 additions and 3 deletions
@@ -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