feat: fix timestamp (#1198)

Author: earayu

aperag/agent/stream_formatters.py

@@ -14,10 +14,9 @@
 
 """Stream message formatters for agent responses."""
 
+import time
 from typing import Any, Dict, List
 
-from aperag.utils.utils import now_unix_milliseconds
-
 from .i18n import ERROR_MESSAGES
 from .response_types import (
     AgentErrorResponse,
@@ -50,7 +49,7 @@ def format_i18n_error(error_key: str, language: str = "en-US", **kwargs) -> Agen
         type="error",
         id="error",
         data=error_message,
-        timestamp=now_unix_milliseconds(),
+        timestamp=int(time.time()),
     )
 
 
@@ -60,7 +59,7 @@ def format_stream_start(msg_id: str) -> AgentStartResponse:
     return AgentStartResponse(
         type="start",
         id=msg_id,
-        timestamp=now_unix_milliseconds(),
+        timestamp=int(time.time()),
     )
 
 
@@ -70,7 +69,7 @@ def format_stream_content(msg_id: str, content: str) -> AgentMessageResponse:
         type="message",
         id=msg_id,
         data=content,
-        timestamp=now_unix_milliseconds(),
+        timestamp=int(time.time()),
     )
 
 
@@ -88,7 +87,7 @@ def format_stream_end(
         id=msg_id,
         data=references,
         urls=urls,
-        timestamp=now_unix_milliseconds(),
+        timestamp=int(time.time()),
     )
 
 
@@ -98,7 +97,7 @@ def format_thinking(msg_id: str, content: str) -> AgentThinkingResponse:
         type="thinking",
         id=msg_id,
         data=content,
-        timestamp=now_unix_milliseconds(),
+        timestamp=int(time.time()),
     )
 
 
@@ -109,7 +108,7 @@ def format_tool_call_result(msg_id: str, data: str, tool_name: str, result: Any)
         data=data,
         tool_name=tool_name,
         result=result,
-        timestamp=now_unix_milliseconds(),
+        timestamp=int(time.time()),
     )
 
 
@@ -119,7 +118,7 @@ def format_agent_start_message(trace_id: str, language: str = "en-US") -> Dict[s
     return {
         "type": "start",
         "id": trace_id,
-        "timestamp": now_unix_milliseconds(),
+        "timestamp": int(time.time()),
     }
 
 
@@ -130,7 +129,7 @@ def format_agent_stop_message(trace_id: str, references: list = None, urls: list
         "id": trace_id,
         "data": references or [],
         "urls": urls or [],
-        "timestamp": now_unix_milliseconds(),
+        "timestamp": int(time.time()),
     }
 
 
@@ -140,7 +139,7 @@ def format_agent_thinking_message(trace_id: str, thinking_content: str) -> Dict[
         "type": "thinking",
         "id": trace_id,
         "data": thinking_content,
-        "timestamp": now_unix_milliseconds(),
+        "timestamp": int(time.time()),
     }
 
 
@@ -150,5 +149,5 @@ def format_agent_message(trace_id: str, content: str) -> Dict[str, Any]:
         "type": "message",
         "id": trace_id,
         "data": content,
-        "timestamp": now_unix_milliseconds(),
+        "timestamp": int(time.time()),
     }

aperag/service/chat_service.py

@@ -14,6 +14,7 @@
 
 import json
 import logging
+import time
 import uuid
 from typing import Any, AsyncGenerator, Dict, List, Optional
 
@@ -38,7 +39,6 @@
     stop_response,
     success_response,
 )
-from aperag.utils.utils import now_unix_milliseconds
 
 logger = logging.getLogger(__name__)
 
@@ -52,7 +52,7 @@ def format_stream_start(msg_id: str) -> Dict[str, Any]:
         return {
             "type": "start",
             "id": msg_id,
-            "timestamp": now_unix_milliseconds(),
+            "timestamp": int(time.time()),
         }
 
     @staticmethod
@@ -62,7 +62,7 @@ def format_stream_content(msg_id: str, content: str) -> Dict[str, Any]:
             "type": "message",
             "id": msg_id,
             "data": content,
-            "timestamp": now_unix_milliseconds(),
+            "timestamp": int(time.time()),
         }
 
     @staticmethod
@@ -84,7 +84,7 @@ def format_stream_end(
             "data": references,
             "memoryCount": memory_count,
             "urls": urls,
-            "timestamp": now_unix_milliseconds(),
+            "timestamp": int(time.time()),
         }
 
     @staticmethod
@@ -94,7 +94,7 @@ def format_complete_response(msg_id: str, content: str) -> Dict[str, Any]:
             "type": "message",
             "id": msg_id,
             "data": content,
-            "timestamp": now_unix_milliseconds(),
+            "timestamp": int(time.time()),
         }
 
     @staticmethod
@@ -104,7 +104,7 @@ def format_error(error: str) -> Dict[str, Any]:
             "type": "error",
             "id": str(uuid.uuid4()),
             "data": error,
-            "timestamp": now_unix_milliseconds(),
+            "timestamp": int(time.time()),
         }
 
 

aperag/utils/history.py

@@ -14,6 +14,7 @@
 
 import json
 import logging
+import time
 from abc import ABC, abstractmethod
 from typing import Any, Dict, List, Optional
 
@@ -26,7 +27,6 @@
     message_to_storage_dict,
     storage_dict_to_message,
 )
-from aperag.utils.utils import now_unix_milliseconds
 
 logger = logging.getLogger(__name__)
 
@@ -263,7 +263,7 @@ def success_response(message_id, data):
             "type": "message",
             "id": message_id,
             "data": data,
-            "timestamp": now_unix_milliseconds(),
+            "timestamp": int(time.time()),
         }
     )
 
@@ -274,7 +274,7 @@ def fail_response(message_id, error):
             "type": "error",
             "id": message_id,
             "data": error,
-            "timestamp": now_unix_milliseconds(),
+            "timestamp": int(time.time()),
         }
     )
 
@@ -284,7 +284,7 @@ def start_response(message_id):
         {
             "type": "start",
             "id": message_id,
-            "timestamp": now_unix_milliseconds(),
+            "timestamp": int(time.time()),
         }
     )
 
@@ -299,7 +299,7 @@ def references_response(message_id, references, memory_count=0, urls=[]):
             "data": references,
             "memoryCount": memory_count,
             "urls": urls,
-            "timestamp": now_unix_milliseconds(),
+            "timestamp": int(time.time()),
         }
     )
 
@@ -309,7 +309,7 @@ def stop_response(message_id):
         {
             "type": "stop",
             "id": message_id,
-            "timestamp": now_unix_milliseconds(),
+            "timestamp": int(time.time()),
         }
     )
 

frontend/src/pages/agent/$botId/chats/$chatId.tsx

@@ -73,7 +73,7 @@ export default () => {
 
   const onSubmit: ChatInputProps['onSubmit'] = useCallback(
     async (params) => {
-      const timestamp = new Date().getTime();
+      const timestamp = Math.floor(new Date().getTime() / 1000);
       const msg: ChatMessage = {
         type: 'message',
         role: 'human',

frontend/src/pages/bots/$botId/chats/$chatId.tsx

@@ -73,7 +73,7 @@ export default () => {
 
   const onSubmit: ChatInputProps['onSubmit'] = useCallback(
     async (params) => {
-      const timestamp = new Date().getTime();
+      const timestamp = Math.floor(new Date().getTime() / 1000);
       const msg: ChatMessage = {
         type: 'message',
         role: 'human',