File size: 1,322 Bytes
2214088 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
from langchain_qdrant import QdrantVectorStore,RetrievalMode
from src.settings import settings
async def answer_query_from_existing_collection(vectorembedding,collection_name_:str):
vectorstore=QdrantVectorStore.from_existing_collection(
embedding=vectorembedding,
collection_name=collection_name_,
url=settings.QDRANT_URL,
api_key=settings.QDRANT_API_KEY,
retrieval_mode=RetrievalMode.DENSE
)
return vectorstore
async def upload_document_existing_collection(documents_,vector_embeddings,collection_name_):
vector_store=QdrantVectorStore.from_documents(
documents=documents_,
embedding=vector_embeddings,
url=settings.QDRANT_URL,
api_key=settings.QDRANT_API_KEY,
prefer_grpc=True,
collection_name=collection_name_,
retrieval_mode=RetrievalMode.DENSE,
timeout=None
)
async def upload_document_new_collection(vector_embeddings,collection_name_):
vector_store=QdrantVectorStore.from_documents(
documents=[],
embedding=vector_embeddings,
url=settings.QDRANT_URL,
api_key=settings.QDRANT_API_KEY,
prefer_grpc=True,
collection_name=collection_name_,
retrieval_mode=RetrievalMode.DENSE,
force_recreate=True,
timeout=None
) |