mirror of
https://github.com/arc53/DocsGPT
synced 2024-11-17 21:26:26 +00:00
48 lines
1.8 KiB
Python
48 lines
1.8 KiB
Python
|
from langchain_community.vectorstores.qdrant import Qdrant
|
||
|
|
from application.vectorstore.base import BaseVectorStore
|
||
|
|
from application.core.settings import settings
|
||
|
|
from qdrant_client import models
|
||
|
|
|
||
|
|
|
||
|
|
class QdrantStore(BaseVectorStore):
|
||
|
|
def __init__(self, path: str = "", embeddings_key: str = "embeddings"):
|
||
|
|
self._filter = models.Filter(
|
||
|
|
must=[
|
||
|
|
models.FieldCondition(
|
||
|
|
key="metadata.store",
|
||
|
|
match=models.MatchValue(value=path.replace("application/indexes/", "").rstrip("/")),
|
||
|
|
)
|
||
|
|
]
|
||
|
|
)
|
||
|
|
|
||
|
|
self._docsearch = Qdrant.construct_instance(
|
||
|
|
["TEXT_TO_OBTAIN_EMBEDDINGS_DIMENSION"],
|
||
|
|
embedding=self._get_embeddings(settings.EMBEDDINGS_NAME, embeddings_key),
|
||
|
|
collection_name=settings.QDRANT_COLLECTION_NAME,
|
||
|
|
location=settings.QDRANT_LOCATION,
|
||
|
|
url=settings.QDRANT_URL,
|
||
|
|
port=settings.QDRANT_PORT,
|
||
|
|
grpc_port=settings.QDRANT_GRPC_PORT,
|
||
|
|
https=settings.QDRANT_HTTPS,
|
||
|
|
prefer_grpc=settings.QDRANT_PREFER_GRPC,
|
||
|
|
api_key=settings.QDRANT_API_KEY,
|
||
|
|
prefix=settings.QDRANT_PREFIX,
|
||
|
|
timeout=settings.QDRANT_TIMEOUT,
|
||
|
|
path=settings.QDRANT_PATH,
|
||
|
|
distance_func=settings.QDRANT_DISTANCE_FUNC,
|
||
|
|
)
|
||
|
|
|
||
|
|
def search(self, *args, **kwargs):
|
||
|
|
return self._docsearch.similarity_search(filter=self._filter, *args, **kwargs)
|
||
|
|
|
||
|
|
def add_texts(self, *args, **kwargs):
|
||
|
|
return self._docsearch.add_texts(*args, **kwargs)
|
||
|
|
|
||
|
|
def save_local(self, *args, **kwargs):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def delete_index(self, *args, **kwargs):
|
||
|
|
return self._docsearch.client.delete(
|
||
|
|
collection_name=settings.QDRANT_COLLECTION_NAME, points_selector=self._filter
|
||
|
|
)
|