sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2024-12-02 20:03:01 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 7

Fix to continue showing retrieved documents in train of thought

Browse source
This commit is contained in:
Debanjum Singh Solanky 2024-10-09 21:19:04 -07:00
parent a6f6e4f418
commit 6ad85e2275
1 changed files with 12 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -21,6 +21,7 @@ from khoj.routers.helpers import (
from khoj.utils.helpers import ( from khoj.utils.helpers import (
ConversationCommand, ConversationCommand,
function_calling_description_for_llm, function_calling_description_for_llm,
is_none_or_empty,
timer, timer,
) )
from khoj.utils.rawconfig import LocationData from khoj.utils.rawconfig import LocationData
@ -189,6 +190,17 @@ async def execute_information_collection(
defiltered_query = result[2] defiltered_query = result[2]
this_iteration.context = compiled_references this_iteration.context = compiled_references
if not is_none_or_empty(compiled_references):
try:
headings = "\n- " + "\n- ".join(set([c.get("compiled", c).split("\n")[0] for c in compiled_references]))
# Strip only leading # from headings
headings = headings.replace("#", "")
async for result in send_status_func(f"**Found Relevant Notes**: {headings}"):
yield result
except Exception as e:
# TODO Get correct type for compiled across research notes extraction
logger.error(f"Error extracting references: {e}", exc_info=True)
elif this_iteration.data_source == ConversationCommand.Online: elif this_iteration.data_source == ConversationCommand.Online:
async for result in search_online( async for result in search_online(
this_iteration.query, this_iteration.query,