klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-08 05:39:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

Handle when summarize result is not present, rename variable in for loop from query

Browse Source
This commit is contained in:
sabaimran
2024-10-09 17:57:08 -07:00
parent 03544efde2
commit 717d9da8d8
2 changed files with 14 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/khoj/routers/api_chat.py
View File

@@ -717,6 +717,7 @@ async def chat(
file_filters=conversation.file_filters if conversation else [], file_filters=conversation.file_filters if conversation else [],
): ):
if type(research_result) == InformationCollectionIteration: if type(research_result) == InformationCollectionIteration:
if research_result.summarizedResult:
pending_research = False pending_research = False
# if research_result.onlineContext: # if research_result.onlineContext:
# researched_results += str(research_result.onlineContext) # researched_results += str(research_result.onlineContext)

10
src/khoj/routers/research.py
View File

@@ -211,13 +211,13 @@ async def execute_information_collection(
direct_web_pages: Dict[str, Dict] = result # type: ignore direct_web_pages: Dict[str, Dict] = result # type: ignore
webpages = [] webpages = []
for query in direct_web_pages: for web_query in direct_web_pages:
if online_results.get(query): if online_results.get(web_query):
online_results[query]["webpages"] = direct_web_pages[query]["webpages"] online_results[web_query]["webpages"] = direct_web_pages[web_query]["webpages"]
else: else:
online_results[query] = {"webpages": direct_web_pages[query]["webpages"]} online_results[web_query] = {"webpages": direct_web_pages[web_query]["webpages"]}
for webpage in direct_web_pages[query]["webpages"]: for webpage in direct_web_pages[web_query]["webpages"]:
webpages.append(webpage["link"]) webpages.append(webpage["link"])
yield send_status_func(f"**Read web pages**: {webpages}") yield send_status_func(f"**Read web pages**: {webpages}")