xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-06-09 23:36:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: lint error

Browse Source
This commit is contained in:
wangcham
2026-02-09 01:01:20 +08:00
parent d2d7892325
commit c2574bdd3a
3 changed files with 103 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

94
src/langbot/pkg/api/http/controller/groups/monitoring.py
View File

@@ -348,9 +348,22 @@ class MonitoringRouterGroup(group.RouterGroup):
end_time=end_time,
limit=limit,
)
headers = ['id', 'timestamp', 'bot_id', 'bot_name', 'pipeline_id', 'pipeline_name',
'runner_name', 'message_content', 'message_text', 'session_id', 'status', 'level',
'platform', 'user_id']
headers = [
'id',
'timestamp',
'bot_id',
'bot_name',
'pipeline_id',
'pipeline_name',
'runner_name',
'message_content',
'message_text',
'session_id',
'status',
'level',
'platform',
'user_id',
]
elif export_type == 'llm-calls':
data = await self.ap.monitoring_service.export_llm_calls(
bot_ids=bot_ids if bot_ids else None,
@@ -359,18 +372,46 @@ class MonitoringRouterGroup(group.RouterGroup):
end_time=end_time,
limit=limit,
)
headers = ['id', 'timestamp', 'model_name', 'input_tokens', 'output_tokens',
'total_tokens', 'duration_ms', 'cost', 'status', 'bot_id', 'bot_name',
'pipeline_id', 'pipeline_name', 'session_id', 'message_id', 'error_message']
headers = [
'id',
'timestamp',
'model_name',
'input_tokens',
'output_tokens',
'total_tokens',
'duration_ms',
'cost',
'status',
'bot_id',
'bot_name',
'pipeline_id',
'pipeline_name',
'session_id',
'message_id',
'error_message',
]
elif export_type == 'embedding-calls':
data = await self.ap.monitoring_service.export_embedding_calls(
start_time=start_time,
end_time=end_time,
limit=limit,
)
headers = ['id', 'timestamp', 'model_name', 'prompt_tokens', 'total_tokens',
'duration_ms', 'input_count', 'status', 'error_message', 'knowledge_base_id',
'query_text', 'session_id', 'message_id', 'call_type']
headers = [
'id',
'timestamp',
'model_name',
'prompt_tokens',
'total_tokens',
'duration_ms',
'input_count',
'status',
'error_message',
'knowledge_base_id',
'query_text',
'session_id',
'message_id',
'call_type',
]
elif export_type == 'errors':
data = await self.ap.monitoring_service.export_errors(
bot_ids=bot_ids if bot_ids else None,
@@ -379,8 +420,19 @@ class MonitoringRouterGroup(group.RouterGroup):
end_time=end_time,
limit=limit,
)
headers = ['id', 'timestamp', 'error_type', 'error_message', 'bot_id', 'bot_name',
'pipeline_id', 'pipeline_name', 'session_id', 'message_id', 'stack_trace']
headers = [
'id',
'timestamp',
'error_type',
'error_message',
'bot_id',
'bot_name',
'pipeline_id',
'pipeline_name',
'session_id',
'message_id',
'stack_trace',
]
elif export_type == 'sessions':
data = await self.ap.monitoring_service.export_sessions(
bot_ids=bot_ids if bot_ids else None,
@@ -389,9 +441,19 @@ class MonitoringRouterGroup(group.RouterGroup):
end_time=end_time,
limit=limit,
)
headers = ['session_id', 'bot_id', 'bot_name', 'pipeline_id', 'pipeline_name',
'message_count', 'start_time', 'last_activity', 'is_active',
'platform', 'user_id']
headers = [
'session_id',
'bot_id',
'bot_name',
'pipeline_id',
'pipeline_name',
'message_count',
'start_time',
'last_activity',
'is_active',
'platform',
'user_id',
]
else:
return self.error(message=f'Invalid export type: {export_type}', code=400)
@@ -417,6 +479,8 @@ class MonitoringRouterGroup(group.RouterGroup):
# Return as file download
response = await quart.make_response(csv_content)
response.headers['Content-Type'] = 'text/csv; charset=utf-8'
response.headers['Content-Disposition'] = f'attachment; filename="monitoring-{export_type}-{int(datetime.datetime.now().timestamp())}.csv"'
response.headers['Content-Disposition'] = (
f'attachment; filename="monitoring-{export_type}-{int(datetime.datetime.now().timestamp())}.csv"'
)
return response, 200

9
src/langbot/pkg/api/http/service/monitoring.py
View File

@@ -820,6 +820,7 @@ class MonitoringService:
try:
import json
message_chain = json.loads(message_content)
if not isinstance(message_chain, list):
return message_content
@@ -909,7 +910,9 @@ class MonitoringService:
'pipeline_name': row[0].pipeline_name if isinstance(row, tuple) else row.pipeline_name,
'runner_name': row[0].runner_name if isinstance(row, tuple) else row.runner_name,
'message_content': row[0].message_content if isinstance(row, tuple) else row.message_content,
'message_text': self._extract_message_text(row[0].message_content if isinstance(row, tuple) else row.message_content),
'message_text': self._extract_message_text(
row[0].message_content if isinstance(row, tuple) else row.message_content
),
'session_id': row[0].session_id if isinstance(row, tuple) else row.session_id,
'status': row[0].status if isinstance(row, tuple) else row.status,
'level': row[0].level if isinstance(row, tuple) else row.level,
@@ -1108,7 +1111,9 @@ class MonitoringService:
'pipeline_name': row[0].pipeline_name if isinstance(row, tuple) else row.pipeline_name,
'message_count': row[0].message_count if isinstance(row, tuple) else row.message_count,
'start_time': self._format_timestamp(row[0].start_time if isinstance(row, tuple) else row.start_time),
'last_activity': self._format_timestamp(row[0].last_activity if isinstance(row, tuple) else row.last_activity),
'last_activity': self._format_timestamp(
row[0].last_activity if isinstance(row, tuple) else row.last_activity
),
'is_active': str(row[0].is_active if isinstance(row, tuple) else row.is_active),
'platform': row[0].platform if isinstance(row, tuple) else row.platform,
'user_id': row[0].user_id if isinstance(row, tuple) else row.user_id,