ryan/simbarag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

RAG optimizations

Browse Source
This commit is contained in:
Ryan Chen
2026-01-11 09:36:36 -05:00
parent 07512409f1
commit 733ffae8cf
2 changed files with 23 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
services/raggr/blueprints/rag/logic.py
View File

@@ -8,7 +8,7 @@ from langchain_text_splitters import RecursiveCharacterTextSplitter
from .fetchers import PaperlessNGXService
embeddings = OpenAIEmbeddings(model="text-embedding-3-large")
embeddings = OpenAIEmbeddings(model="text-embedding-3-small")
vector_store = Chroma(
collection_name="simba_docs",
@@ -61,7 +61,7 @@ async def index_documents():
async def query_vector_store(query: str):
retrieved_docs = vector_store.similarity_search(query, k=2)
retrieved_docs = await vector_store.asimilarity_search(query, k=2)
serialized = "\n\n".join(
(f"Source: {doc.metadata}\nContent: {doc.page_content}")
for doc in retrieved_docs