Add AWS Bedrock support (#1482)

2024-07-18 03:08:10 +05:30
parent 4e5d34103f
commit 1e7618dfa4
12 changed files with 454 additions and 19 deletions
--- a/mem0/utils/factory.py
+++ b/mem0/utils/factory.py
@@ -12,7 +12,8 @@ class LlmFactory:
        "ollama": "mem0.llms.ollama.py.OllamaLLM",
        "openai": "mem0.llms.openai.OpenAILLM",
        "groq": "mem0.llms.groq.GroqLLM",
-        "together": "mem0.llms.together.TogetherLLM"
+        "together": "mem0.llms.together.TogetherLLM",
+        "aws_bedrock": "mem0.llms.aws_bedrock.AWSBedrockLLM"
    }

    @classmethod