From e287fb9a8932ecd4424f92b2b1371d85c0b865d8 Mon Sep 17 00:00:00 2001
From: Mitul Kataria <53906546+kmitul@users.noreply.github.com>
Date: Fri, 2 Aug 2024 04:50:16 +0900
Subject: [PATCH] Fixed import for EmbeddingBase class for ollama and
 huggingface embedders (#1548)

---
 mem0/embeddings/huggingface.py | 2 +-
 mem0/embeddings/ollama.py      | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/mem0/embeddings/huggingface.py b/mem0/embeddings/huggingface.py
index 7e7ca590..6b187d81 100644
--- a/mem0/embeddings/huggingface.py
+++ b/mem0/embeddings/huggingface.py
@@ -1,4 +1,4 @@
-from embedding.base import EmbeddingBase
+from mem0.embeddings.base import EmbeddingBase
 from sentence_transformers import SentenceTransformer
 
 
diff --git a/mem0/embeddings/ollama.py b/mem0/embeddings/ollama.py
index 649fe161..00289d4e 100644
--- a/mem0/embeddings/ollama.py
+++ b/mem0/embeddings/ollama.py
@@ -1,5 +1,5 @@
 import ollama
-from embedding.base import EmbeddingBase
+from mem0.embeddings.base import EmbeddingBase
 
 
 class OllamaEmbedding(EmbeddingBase):