sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2024-11-27 17:35:07 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

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
Show stats Download patch file Download diff file Expand all files Collapse all files

17
src/khoj/routers/api_chat.py
View file

@ -717,16 +717,17 @@ 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:
pending_research = False if research_result.summarizedResult:
# if research_result.onlineContext: pending_research = False
# researched_results += str(research_result.onlineContext) # if research_result.onlineContext:
# online_results.update(research_result.onlineContext) # researched_results += str(research_result.onlineContext)
# online_results.update(research_result.onlineContext)
# if research_result.context: # if research_result.context:
# researched_results += str(research_result.context) # researched_results += str(research_result.context)
# compiled_references.extend(research_result.context) # compiled_references.extend(research_result.context)
researched_results += research_result.summarizedResult researched_results += research_result.summarizedResult
else: else:
yield research_result yield research_result

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}")