From 67902b5da7c34d28e65e2f4fd6b8b75ead1fe292 Mon Sep 17 00:00:00 2001
From: Yeuoly <45712896+Yeuoly@users.noreply.github.com>
Date: Sat, 4 May 2024 16:17:15 +0800
Subject: [PATCH] fix: agent log timezone (#4076)

---
 api/services/agent_service.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/api/services/agent_service.py b/api/services/agent_service.py
index a25a48dc35..ba5fd93326 100644
--- a/api/services/agent_service.py
+++ b/api/services/agent_service.py
@@ -1,3 +1,6 @@
+import pytz
+from flask_login import current_user
+
 from core.app.app_config.easy_ui_based_app.agent.manager import AgentConfigManager
 from core.tools.tool_manager import ToolManager
 from extensions.ext_database import db
@@ -46,11 +49,13 @@ class AgentService:
         else:
             executor = 'Unknown'
 
+        timezone = pytz.timezone(current_user.timezone)
+
         result = {
             'meta': {
                 'status': 'success',
                 'executor': executor,
-                'start_time': message.created_at.isoformat(),
+                'start_time': message.created_at.astimezone(timezone).isoformat(),
                 'elapsed_time': message.provider_response_latency,
                 'total_tokens': message.answer_tokens + message.message_tokens,
                 'agent_mode': app_model.app_model_config.agent_mode_dict.get('strategy', 'react'),