fix: agent log timezone (#4076)

This commit is contained in:
Yeuoly 2024-05-04 16:17:15 +08:00 committed by GitHub
parent c0476c7881
commit 67902b5da7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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.app.app_config.easy_ui_based_app.agent.manager import AgentConfigManager
from core.tools.tool_manager import ToolManager from core.tools.tool_manager import ToolManager
from extensions.ext_database import db from extensions.ext_database import db
@ -46,11 +49,13 @@ class AgentService:
else: else:
executor = 'Unknown' executor = 'Unknown'
timezone = pytz.timezone(current_user.timezone)
result = { result = {
'meta': { 'meta': {
'status': 'success', 'status': 'success',
'executor': executor, 'executor': executor,
'start_time': message.created_at.isoformat(), 'start_time': message.created_at.astimezone(timezone).isoformat(),
'elapsed_time': message.provider_response_latency, 'elapsed_time': message.provider_response_latency,
'total_tokens': message.answer_tokens + message.message_tokens, 'total_tokens': message.answer_tokens + message.message_tokens,
'agent_mode': app_model.app_model_config.agent_mode_dict.get('strategy', 'react'), 'agent_mode': app_model.app_model_config.agent_mode_dict.get('strategy', 'react'),