Support Ollama models (#1596)

2024-08-02 23:45:45 +05:30
parent 3eff82082e
commit 44aa16a0f8
8 changed files with 188 additions and 30 deletions
--- a/mem0/utils/factory.py
+++ b/mem0/utils/factory.py
@@ -17,6 +17,7 @@ class LlmFactory:
        "together": "mem0.llms.together.TogetherLLM",
        "aws_bedrock": "mem0.llms.aws_bedrock.AWSBedrockLLM",
        "litellm": "mem0.llms.litellm.LiteLLM",
+        "ollama": "mem0.llms.ollama.OllamaLLM",
    }

    @classmethod