From 4088d68af730f513f8aa0c0958bb76de7944d72c Mon Sep 17 00:00:00 2001 From: sallyom Date: Mon, 6 Jan 2025 15:29:21 -0500 Subject: [PATCH] update rag/app/manage_vectordb.py imports Signed-off-by: sallyom --- recipes/natural_language_processing/rag/app/manage_vectordb.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/recipes/natural_language_processing/rag/app/manage_vectordb.py b/recipes/natural_language_processing/rag/app/manage_vectordb.py index 82566abd..0ab7859e 100644 --- a/recipes/natural_language_processing/rag/app/manage_vectordb.py +++ b/recipes/natural_language_processing/rag/app/manage_vectordb.py @@ -2,11 +2,12 @@ from chromadb import HttpClient from chromadb.config import Settings import chromadb.utils.embedding_functions as embedding_functions -from langchain.embeddings.sentence_transformer import SentenceTransformerEmbeddings +from langchain_community.embeddings import SentenceTransformerEmbeddings from langchain_community.vectorstores import Milvus from pymilvus import MilvusClient from pymilvus import connections, utility + class VectorDB: def __init__(self, vector_vendor, host, port, collection_name, embedding_model): self.vector_vendor = vector_vendor