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

Correct usage of the summarize function

Browse Source
This commit is contained in:
sabaimran
2024-10-09 15:58:05 -07:00
parent c91678078d
commit 4fbaef10e9
2 changed files with 17 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -783,7 +783,7 @@ async def chat(
async for result in send_llm_response(response_log): async for result in send_llm_response(response_log):
yield result yield result
else: else:
response_log = await generate_summary_from_files( async for response in generate_summary_from_files(
q=q, q=q,
user=user, user=user,
file_filters=file_filters, file_filters=file_filters,
@@ -791,7 +791,9 @@ async def chat(
subscribed=subscribed, subscribed=subscribed,
send_status_func=partial(send_event, ChatEvent.STATUS), send_status_func=partial(send_event, ChatEvent.STATUS),
send_response_func=partial(send_llm_response), send_response_func=partial(send_llm_response),
) ):
yield response
await sync_to_async(save_to_conversation_log)( await sync_to_async(save_to_conversation_log)(
q, q,
response_log, response_log,

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

@@ -245,13 +245,24 @@ async def execute_information_collection(
) )
) )
else: else:
response_log = await generate_summary_from_files( async for response in generate_summary_from_files(
q=query, q=query,
user=user, user=user,
file_filters=file_filters, file_filters=file_filters,
meta_log=conversation_history, meta_log=conversation_history,
subscribed=subscribed, subscribed=subscribed,
send_status_func=send_status_func, send_status_func=send_status_func,
):
if isinstance(response, dict) and ChatEvent.STATUS in response:
yield response[ChatEvent.STATUS]
else:
response_log = response
previous_iterations.append(
InformationCollectionIteration(
data_source=this_iteration.data_source,
query=this_iteration.query,
context=response_log,
)
) )
else: else:
iteration = MAX_ITERATIONS iteration = MAX_ITERATIONS