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

Save separate user message time from Khoj response time in chat logs

Browse Source 
Previously user message time was being stored the same as Khoj
response time in conversation logs.
This commit is contained in:
Debanjum Singh Solanky
2024-04-18 01:26:15 +05:30
parent 5133b6e73b
commit a1e5195c8b
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -2,6 +2,7 @@ import functools
import json import json
import logging import logging
import math import math
from datetime import datetime
from typing import Dict, Optional from typing import Dict, Optional
from urllib.parse import unquote from urllib.parse import unquote
@@ -362,6 +363,7 @@ async def websocket_endpoint(
await send_rate_limit_message(e.detail) await send_rate_limit_message(e.detail)
break break
user_message_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
conversation_commands = [get_conversation_command(query=q, any_references=True)] conversation_commands = [get_conversation_command(query=q, any_references=True)]
await send_status_update(f"**👀 Understanding Query**: {q}") await send_status_update(f"**👀 Understanding Query**: {q}")
@@ -432,6 +434,7 @@ async def websocket_endpoint(
llm_response, llm_response,
user, user,
meta_log, meta_log,
user_message_time,
intent_type="reminder", intent_type="reminder",
client_application=websocket.user.client_app, client_application=websocket.user.client_app,
conversation_id=conversation_id, conversation_id=conversation_id,
@@ -531,6 +534,7 @@ async def websocket_endpoint(
image, image,
user, user,
meta_log, meta_log,
user_message_time,
intent_type=intent_type, intent_type=intent_type,
inferred_queries=[improved_image_prompt], inferred_queries=[improved_image_prompt],
client_application=websocket.user.client_app, client_application=websocket.user.client_app,
@@ -607,6 +611,7 @@ async def chat(
) -> Response: ) -> Response:
user: KhojUser = request.user.object user: KhojUser = request.user.object
q = unquote(q) q = unquote(q)
user_message_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
logger.info(f"Chat request by {user.username}: {q}") logger.info(f"Chat request by {user.username}: {q}")
await is_ready_to_chat(user) await is_ready_to_chat(user)
@@ -689,6 +694,7 @@ async def chat(
llm_response, llm_response,
user, user,
meta_log, meta_log,
user_message_time,
intent_type="reminder", intent_type="reminder",
client_application=request.user.client_app, client_application=request.user.client_app,
conversation_id=conversation_id, conversation_id=conversation_id,
@@ -765,6 +771,7 @@ async def chat(
image, image,
user, user,
meta_log, meta_log,
user_message_time,
intent_type=intent_type, intent_type=intent_type,
inferred_queries=[improved_image_prompt], inferred_queries=[improved_image_prompt],
client_application=request.user.client_app, client_application=request.user.client_app,