From b4ba5bdec53c8e9b475bdf40e441673bd019d231 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=D0=9C=D0=B0=D1=80=D0=B0=D1=82=20=D0=A4=D0=B0=D0=B7=D1=8B?= =?UTF-8?q?=D0=BB=D0=BE=D0=B2?= Date: Fri, 15 May 2026 08:39:47 +0000 Subject: [PATCH] add vector_store --- vector_store.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) create mode 100644 vector_store.py diff --git a/vector_store.py b/vector_store.py new file mode 100644 index 0000000..93b19f6 --- /dev/null +++ b/vector_store.py @@ -0,0 +1,13 @@ +from langchain_qdrant import QdrantStore +from langchain_ollama import OllamaEmbeddings + +class QdrantStore: + def __init__(self, url="http://localhost:6333", collection_name="rag_collection"): + self.client = QdrantStore(url=url, collection_name=collection_name) + self.embeddings = OllamaEmbeddings(model="nomic-embed-text") + + def add_documents(self, docs): + self.client.add_documents(docs, embeddings=self.embeddings) + + def search(self, query, limit=5): + return self.client.search(query, limit=limit, embeddings=self.embeddings)