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

Fix unique has for job id

Browse Source
This commit is contained in:
sabaimran
2024-04-30 20:08:05 +05:30
committed by Debanjum Singh Solanky
parent 311d58e1ed
commit ad4145e48c
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -185,7 +185,7 @@ async def agenerate_chat_response(*args):
return await loop.run_in_executor(executor, generate_chat_response, *args) return await loop.run_in_executor(executor, generate_chat_response, *args)
async def aget_relevant_information_sources(query: str, conversation_history: dict): async def aget_relevant_information_sources(query: str, conversation_history: dict, is_task: bool):
""" """
Given a query, determine which of the available tools the agent should use in order to answer appropriately. Given a query, determine which of the available tools the agent should use in order to answer appropriately.
""" """
@@ -216,7 +216,7 @@ async def aget_relevant_information_sources(query: str, conversation_history: di
logger.error(f"Invalid response for determining relevant tools: {response}") logger.error(f"Invalid response for determining relevant tools: {response}")
return tool_options return tool_options
final_response = [] final_response = [] if not is_task else [ConversationCommand.AutomatedTask]
for llm_suggested_tool in response: for llm_suggested_tool in response:
if llm_suggested_tool in tool_options.keys(): if llm_suggested_tool in tool_options.keys():
# Check whether the tool exists as a valid ConversationCommand # Check whether the tool exists as a valid ConversationCommand
@@ -926,7 +926,7 @@ async def create_automation(
job_metadata = json.dumps( job_metadata = json.dumps(
{"query_to_run": query_to_run, "scheduling_request": q, "subject": subject, "crontime": crontime_string} {"query_to_run": query_to_run, "scheduling_request": q, "subject": subject, "crontime": crontime_string}
) )
query_id = hashlib.md5(f"{query_to_run}".encode("utf-8")).hexdigest() query_id = hashlib.md5(f"{query_to_run}{crontime_string}".encode("utf-8")).hexdigest()
job_id = f"automation_{user.uuid}_{crontime_string}_{query_id}" job_id = f"automation_{user.uuid}_{crontime_string}_{query_id}"
job = await sync_to_async(state.scheduler.add_job)( job = await sync_to_async(state.scheduler.add_job)(
run_with_process_lock, run_with_process_lock,