From 14e453039d3320f286b4b02f5a3d3489a7532683 Mon Sep 17 00:00:00 2001 From: Debanjum Date: Fri, 1 Nov 2024 18:03:52 -0700 Subject: [PATCH] Add prompt tracing, agent personality to infer webpage urls chat actor --- src/khoj/processor/tools/online_search.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/src/khoj/processor/tools/online_search.py b/src/khoj/processor/tools/online_search.py index a6f6e7f7..c6fc7c20 100644 --- a/src/khoj/processor/tools/online_search.py +++ b/src/khoj/processor/tools/online_search.py @@ -162,10 +162,9 @@ async def read_webpages( ): "Infer web pages to read from the query and extract relevant information from them" logger.info(f"Inferring web pages to read") - if send_status_func: - async for event in send_status_func(f"**Inferring web pages to read**"): - yield {ChatEvent.STATUS: event} - urls = await infer_webpage_urls(query, conversation_history, location, user, query_images) + urls = await infer_webpage_urls( + query, conversation_history, location, user, query_images, agent=agent, tracer=tracer + ) # Get the top 10 web pages to read urls = urls[:max_webpages_to_read]