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

Remove conversation command always in query, filter out inferred queries that were not with selected tool when going through tool selection iterations

Browse source
This commit is contained in:
sabaimran 2024-10-31 20:27:38 -07:00
parent 19241805ee
commit 21858acccc
2 changed files with 6 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
src/khoj/processor/conversation/utils.py
View file

@ -174,7 +174,8 @@ def construct_tool_chat_history(
"by": "khoj",
"intent": {
"type": "remember",
"inferred-queries": inferred_query_extractor(iteration),
# Only include inferred-queries for the specific tool that's being used in this iteration.
"inferred-queries": inferred_query_extractor(iteration) if iteration.tool == tool else [],
"query": iteration.query,
},
"message": iteration.summarizedResult,

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

@ -738,6 +738,10 @@ async def chat(
if mode not in conversation_commands:
conversation_commands.append(mode)
for cmd in conversation_commands:
await conversation_command_rate_limiter.update_and_check_if_valid(request, cmd)
q = q.replace(f"/{cmd.value}", "").strip()
if conversation_commands == [ConversationCommand.Research]:
async for research_result in execute_information_collection(
request=request,
@ -774,10 +778,6 @@ async def chat(
pending_research = False
for cmd in conversation_commands:
await conversation_command_rate_limiter.update_and_check_if_valid(request, cmd)
q = q.replace(f"/{cmd.value}", "").strip()
used_slash_summarize = conversation_commands == [ConversationCommand.Summarize]
file_filters = conversation.file_filters if conversation else []
# Skip trying to summarize if