From 524f670925f1f7bb10928700d153084b8bb3f05f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=D0=94=D0=B0=D0=BD=D0=B8=D0=B8=D0=BB=20=D0=92=D0=B8=D0=BA?= =?UTF-8?q?=D1=82=D0=BE=D1=80=D0=BE=D0=B2?= Date: Thu, 2 Jul 2026 08:35:09 +0000 Subject: [PATCH] =?UTF-8?q?fix:=20rag=5Fstore.py=20=E2=80=94=20=D0=90?= =?UTF-8?q?=D0=B3=D0=B5=D0=BD=D1=82=20=D1=81=20RAG-=D0=BF=D0=B0=D0=BC?= =?UTF-8?q?=D1=8F=D1=82=D1=8C=D1=8E?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- rag_store.py | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) create mode 100644 rag_store.py diff --git a/rag_store.py b/rag_store.py new file mode 100644 index 0000000..36b31f5 --- /dev/null +++ b/rag_store.py @@ -0,0 +1,23 @@ +import os +from langchain_openai import OpenAIEmbeddings +from langchain_qdrant import Qdrant +from langchain_core.documents import Document + +class KnowledgeBase: + def __init__(self, collection_name: str = "knowledge", host: str = "localhost", port: int = 6333): + self.embeddings = OpenAIEmbeddings( + model="text-embedding-3-small", + base_url="https://openrouter.ai/api/v1", + api_key=os.getenv("OPENAI_API_KEY"), + ) + self.vector_store = Qdrant( + collection_name=collection_name, + url=f"http://{host}:{port}", + embedding_function=self.embeddings, + ) + + def add_documents(self, documents): + self.vector_store.add_documents(documents) + + def similarity_search(self, query: str, k: int = 3): + return self.vector_store.similarity_search(query, k=k) \ No newline at end of file