sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2024-11-23 23:48:56 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 16

Do not exit if/else loop in research loop when notes not found

Browse source
This commit is contained in:
sabaimran 2024-10-31 13:51:10 -07:00
parent a13760640c
commit 559601dd0a
1 changed files with 12 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
src/khoj/routers/research.py
View file

@ -191,18 +191,18 @@ async def execute_information_collection(
document_results = result[0] document_results = result[0]
this_iteration.context += document_results this_iteration.context += document_results
if not is_none_or_empty(document_results): if not is_none_or_empty(document_results):
try: try:
distinct_files = {d["file"] for d in document_results} distinct_files = {d["file"] for d in document_results}
distinct_headings = set([d["compiled"].split("\n")[0] for d in document_results if "compiled" in d]) distinct_headings = set([d["compiled"].split("\n")[0] for d in document_results if "compiled" in d])
# Strip only leading # from headings # Strip only leading # from headings
headings_str = "\n- " + "\n- ".join(distinct_headings).replace("#", "") headings_str = "\n- " + "\n- ".join(distinct_headings).replace("#", "")
async for result in send_status_func( async for result in send_status_func(
f"**Found {len(distinct_headings)} Notes Across {len(distinct_files)} Files**: {headings_str}" f"**Found {len(distinct_headings)} Notes Across {len(distinct_files)} Files**: {headings_str}"
): ):
yield result yield result
except Exception as e: except Exception as e:
logger.error(f"Error extracting document references: {e}", exc_info=True) logger.error(f"Error extracting document references: {e}", exc_info=True)
elif this_iteration.tool == ConversationCommand.Online: elif this_iteration.tool == ConversationCommand.Online:
async for result in search_online( async for result in search_online(