Update factory.py (#1657)

2024-08-08 15:06:15 +05:30
parent e190445492
commit 8c8f4120f4
1 changed files with 1 additions and 1 deletions
@@ -11,7 +11,7 @@ def load_class(class_type):

 class LlmFactory:
    provider_to_class = {
-        "ollama": "mem0.llms.ollama.py.OllamaLLM",
+        "ollama": "mem0.llms.ollama.OllamaLLM",
        "openai": "mem0.llms.openai.OpenAILLM",
        "groq": "mem0.llms.groq.GroqLLM",
        "together": "mem0.llms.together.TogetherLLM",