From 8c8f4120f448f07db816c1283247e8d7256ad703 Mon Sep 17 00:00:00 2001
From: krescent <c54444263@gmail.com>
Date: Thu, 8 Aug 2024 17:36:15 +0800
Subject: [PATCH] Update factory.py (#1657)

---
 mem0/utils/factory.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mem0/utils/factory.py b/mem0/utils/factory.py
index a4ff4214..14aba4e0 100644
--- a/mem0/utils/factory.py
+++ b/mem0/utils/factory.py
@@ -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",