enable parameter k to get web resources (opea-project#844)

Signed-off-by: letonghan <[email protected]>
lvliang-intel · Nov 1, 2024 · 9f68bd3 · 9f68bd3
1 parent 6dbb0a7
commit 9f68bd3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/comps/web_retrievers/chroma/langchain/retriever_chroma.py b/comps/web_retrievers/chroma/langchain/retriever_chroma.py
@@ -73,7 +73,7 @@ async def web_retrieve(input: EmbedDoc) -> SearchedDoc:
     embedding = input.embedding
 
     # Google Search the results, parse the htmls
-    search_results = get_urls(query)
+    search_results = get_urls(query=query, num_search_result=input.k)
     urls_to_look = []
     for res in search_results:
         if res.get("link", None):
@@ -94,7 +94,7 @@ async def web_retrieve(input: EmbedDoc) -> SearchedDoc:
     dump_docs(unique_documents)
 
     # Do the retrieval
-    search_res = await vector_db.asimilarity_search_by_vector(embedding=embedding)
+    search_res = await vector_db.asimilarity_search_by_vector(embedding=embedding, k=input.k)
 
     searched_docs = []