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

Fix NOT NULL violation on empty splits and increase search results to k=6

Browse Source 
Empty documents after sanitization caused aadd_documents to issue a
DEFAULT VALUES insert. Guard with an emptiness check. Also increase
similarity search k from 2 to 6 so multi-word queries like full names
have better recall.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Ryan Chen
2026-05-31 07:16:28 -04:00
parent 1e6bc536b4
commit 9f51dc3cdb
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
blueprints/rag/logic.py
+2 -1
View File
@@ -325,6 +325,7 @@ async def sync_obsidian_documents() -> dict[str, int]:
if documents: if documents:
splits = text_splitter.split_documents(documents) splits = text_splitter.split_documents(documents)
splits = _sanitize_documents(splits) splits = _sanitize_documents(splits)
if splits:
vector_store = _get_vector_store() vector_store = _get_vector_store()
await vector_store.aadd_documents(documents=splits) await vector_store.aadd_documents(documents=splits)
@@ -336,7 +337,7 @@ async def sync_obsidian_documents() -> dict[str, int]:
async def query_vector_store(query: str): async def query_vector_store(query: str):
vector_store = _get_vector_store() vector_store = _get_vector_store()
retrieved_docs = await vector_store.asimilarity_search(query, k=2) retrieved_docs = await vector_store.asimilarity_search(query, k=6)
serialized = "\n\n".join( serialized = "\n\n".join(
(f"Source: {doc.metadata}\nContent: {doc.page_content}") (f"Source: {doc.metadata}\nContent: {doc.page_content}")
for doc in retrieved_docs for doc in retrieved_docs