increased top-k limit for retrieve queries

codinglabsong · codinglabsong · commit f50735bf40cc · 2025-07-29T12:46:39.000-07:00
diff --git a/src/any_chatbot/tools.py b/src/any_chatbot/tools.py
@@ -28,7 +28,7 @@ def retrieve(
     ) -> Tuple[str, List[Document]]:
         retrieved_docs = vector_store.similarity_search(
             query,
-            k=2,
+            k=5,
             filter={"source_type": tag},
         )
         serialized = "\n\n".join(

Original file line number	Diff line number	Diff line change
`@@ -28,7 +28,7 @@ def retrieve(`
`28`	`28`	`) -> Tuple[str, List[Document]]:`
`29`	`29`	`retrieved_docs = vector_store.similarity_search(`
`30`	`30`	`query,`
`31`		`- k=2,`
	`31`	`+ k=5,`
`32`	`32`	`filter={"source_type": tag},`
`33`	`33`	`)`
`34`	`34`	`serialized = "\n\n".join(`