embeddingf imports

siddhivelankar23 · Aug 20, 2024 · d6d18c7 · d6d18c7
1 parent d042f85
commit d6d18c7
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/comps/__init__.py b/comps/__init__.py
@@ -45,7 +45,8 @@
     FaqGenGateway,
     VisualQnAGateway,
 )
-
+# Embeddings
+from comps.embeddings.multimodal_embeddings.bridgetower import BridgeTowerEmbedding
 # Telemetry
 from comps.cores.telemetry.opea_telemetry import opea_telemetry
 

diff --git a/comps/cores/proto/docarray.py b/comps/cores/proto/docarray.py
@@ -95,7 +95,7 @@ class Config:
 
 
 class SearchedMultimodalDoc(BaseDoc):
-    retrieved_docs: DocList[TextImageDoc]
+    retrieved_docs: DocList[TextDoc]
     initial_query: str
     top_n: int = 1
     #metadata: DocList[dict]

diff --git a/comps/retrievers/langchain_multimodal/redis/retriever_redis.py b/comps/retrievers/langchain_multimodal/redis/retriever_redis.py
@@ -5,7 +5,7 @@
 import time
 from typing import Union
 
-from comps.embeddings.multimodal_embeddings.bridgetower.bridgetower_embedding import BridgeTowerEmbeddings
+from comps import BridgeTowerEmbedding
 from langchain_community.vectorstores import Redis
 from langsmith import traceable
 from multimodal_config import INDEX_NAME, REDIS_URL
@@ -93,6 +93,6 @@ def retrieve(
 
 if __name__ == "__main__":
 
-    embeddings = BridgeTowerEmbeddings()
+    embeddings = BridgeTowerEmbedding()
     vector_db = Redis(embedding=embeddings, index_name=INDEX_NAME, redis_url=REDIS_URL)
     opea_microservices["opea_service@retriever_redis"].start()