add preview api
fuwx
1 year ago
parent
a6767d746d
commit
ab453fa9d6
@ -1,88 +0,0 @@
|
|||||||
import json
|
|
||||||
from collections.abc import Generator
|
|
||||||
from typing import Any, Optional
|
|
||||||
|
|
||||||
import requests
|
|
||||||
|
|
||||||
from configs import dify_config
|
|
||||||
from core.tools.builtin_tool.tool import BuiltinTool
|
|
||||||
from core.tools.entities.tool_entities import ToolInvokeMessage
|
|
||||||
|
|
||||||
|
|
||||||
class FaultLogTool(BuiltinTool):
|
|
||||||
def _invoke(
|
|
||||||
self,
|
|
||||||
user_id: str,
|
|
||||||
tool_parameters: dict[str, Any],
|
|
||||||
conversation_id: Optional[str] = None,
|
|
||||||
app_id: Optional[str] = None,
|
|
||||||
message_id: Optional[str] = None,
|
|
||||||
) -> Generator[ToolInvokeMessage, None, None]:
|
|
||||||
service = tool_parameters.get("service")
|
|
||||||
endpoint = tool_parameters.get("endpoint")
|
|
||||||
entryService = tool_parameters.get("entryService")
|
|
||||||
entryEndpoint = tool_parameters.get("entryEndpoint")
|
|
||||||
start_time = tool_parameters.get("startTime")
|
|
||||||
end_time = tool_parameters.get("endTime")
|
|
||||||
type = tool_parameters.get("type")
|
|
||||||
|
|
||||||
params = {
|
|
||||||
"anormalTypes": "app,container,infra,network,error,appInstance",
|
|
||||||
"endTime": end_time,
|
|
||||||
"endpoint": endpoint,
|
|
||||||
"entryEndpoint": entryEndpoint,
|
|
||||||
"entryService": entryService,
|
|
||||||
"service": service,
|
|
||||||
"startTime": start_time,
|
|
||||||
"step": self.get_step(start_time, end_time),
|
|
||||||
"type": [type],
|
|
||||||
"withTopology": True,
|
|
||||||
}
|
|
||||||
|
|
||||||
url = "http://192.168.1.6:13680/api/nodeinfo"
|
|
||||||
res = {}
|
|
||||||
with requests.post(url, json=params, stream=True) as response:
|
|
||||||
# 逐行读取流式响应
|
|
||||||
for line in response.iter_lines():
|
|
||||||
if line: # 过滤空行
|
|
||||||
# 解码字节为字符串
|
|
||||||
decoded_line = line.decode('utf-8')
|
|
||||||
res = json.loads(decoded_line[5:].strip())
|
|
||||||
list = json.dumps({
|
|
||||||
'type': 'nodeinfo',
|
|
||||||
'display': True,
|
|
||||||
'data': res,
|
|
||||||
})
|
|
||||||
yield self.create_text_message(list)
|
|
||||||
|
|
||||||
def get_step(self, start_time, end_time):
|
|
||||||
time_diff = end_time - start_time
|
|
||||||
|
|
||||||
SECOND = 1000000 # microseconds
|
|
||||||
MINUTE = 60 * SECOND
|
|
||||||
HOUR = 60 * MINUTE
|
|
||||||
|
|
||||||
step = SECOND # default step is 1 second
|
|
||||||
|
|
||||||
if time_diff <= 15 * MINUTE:
|
|
||||||
step = 30 * SECOND
|
|
||||||
elif time_diff <= 30 * MINUTE:
|
|
||||||
step = 1 * MINUTE
|
|
||||||
elif time_diff <= 1 * HOUR:
|
|
||||||
step = 2 * MINUTE
|
|
||||||
elif time_diff <= 1.5 * HOUR:
|
|
||||||
step = 3 * MINUTE
|
|
||||||
elif time_diff <= 3 * HOUR:
|
|
||||||
step = 6 * MINUTE
|
|
||||||
elif time_diff <= 6 * HOUR:
|
|
||||||
step = 12 * MINUTE
|
|
||||||
elif time_diff <= 12 * HOUR:
|
|
||||||
step = 24 * MINUTE
|
|
||||||
elif time_diff <= 15 * HOUR:
|
|
||||||
step = 30 * MINUTE
|
|
||||||
elif time_diff <= 30 * HOUR:
|
|
||||||
step = 1 * HOUR
|
|
||||||
else:
|
|
||||||
step = ((time_diff + 30 * SECOND - 1) // (30 * SECOND)) * SECOND
|
|
||||||
|
|
||||||
return step
|
|
||||||
@ -1,108 +0,0 @@
|
|||||||
identity:
|
|
||||||
name: node_info
|
|
||||||
author: APO
|
|
||||||
label:
|
|
||||||
en_US: node_info
|
|
||||||
zh_Hans: 获取节点信息
|
|
||||||
pt_BR: node_info
|
|
||||||
description:
|
|
||||||
human:
|
|
||||||
en_US: A tool for getting fault log.
|
|
||||||
zh_Hans: 获取节点信息
|
|
||||||
pt_BR: A tool for getting fault log.
|
|
||||||
llm: A tool for getting fault log.
|
|
||||||
display:
|
|
||||||
type: info
|
|
||||||
unit: info
|
|
||||||
parameters:
|
|
||||||
- name: entryService
|
|
||||||
type: string
|
|
||||||
required: true
|
|
||||||
label:
|
|
||||||
en_US: entryService
|
|
||||||
zh_Hans: 入口服务名
|
|
||||||
pt_BR: entryService
|
|
||||||
human_description:
|
|
||||||
en_US: Time format in strftime standard.
|
|
||||||
zh_Hans: 拓扑入口服务名
|
|
||||||
pt_BR: Time format in strftime standard.
|
|
||||||
llm_description: toplogy service name
|
|
||||||
form: llm
|
|
||||||
- name: entryEndpoint
|
|
||||||
type: string
|
|
||||||
required: true
|
|
||||||
label:
|
|
||||||
en_US: entryEndpoint
|
|
||||||
zh_Hans: 入口服务端点
|
|
||||||
pt_BR: entryEndpoint
|
|
||||||
human_description:
|
|
||||||
en_US: Time format in strftime standard.
|
|
||||||
zh_Hans: 拓扑入口服务端点
|
|
||||||
pt_BR: Time format in strftime standard.
|
|
||||||
llm_description: toplogy endpoint
|
|
||||||
form: llm
|
|
||||||
- name: service
|
|
||||||
type: string
|
|
||||||
required: true
|
|
||||||
label:
|
|
||||||
en_US: service
|
|
||||||
zh_Hans: 服务名
|
|
||||||
pt_BR: service
|
|
||||||
human_description:
|
|
||||||
en_US: Time format in strftime standard.
|
|
||||||
zh_Hans: 查询指定服务的名称
|
|
||||||
pt_BR: Time format in strftime standard.
|
|
||||||
llm_description: Time format in strftime standard.
|
|
||||||
form: llm
|
|
||||||
- name: endpoint
|
|
||||||
type: string
|
|
||||||
required: true
|
|
||||||
label:
|
|
||||||
en_US: endpoint
|
|
||||||
zh_Hans: 服务端点
|
|
||||||
pt_BR: endpoint
|
|
||||||
human_description:
|
|
||||||
en_US: Time format in strftime standard.
|
|
||||||
zh_Hans: 指定服务的端点
|
|
||||||
pt_BR: Time format in strftime standard.
|
|
||||||
llm_description: toplogy endpoint
|
|
||||||
form: llm
|
|
||||||
- name: startTime
|
|
||||||
type: number
|
|
||||||
required: true
|
|
||||||
label:
|
|
||||||
en_US: startTime
|
|
||||||
zh_Hans: startTime
|
|
||||||
pt_BR: startTime
|
|
||||||
human_description:
|
|
||||||
en_US: Data query start time
|
|
||||||
zh_Hans: 开始时间 (微秒)
|
|
||||||
pt_BR: Data query start time
|
|
||||||
llm_description: Data query start time
|
|
||||||
form: llm
|
|
||||||
- name: endTime
|
|
||||||
type: number
|
|
||||||
required: true
|
|
||||||
label:
|
|
||||||
en_US: endTime
|
|
||||||
zh_Hans: endTime
|
|
||||||
pt_BR: endTime
|
|
||||||
human_description:
|
|
||||||
en_US: Data query end time
|
|
||||||
zh_Hans: 结束时间 (微秒)
|
|
||||||
pt_BR: Data query end time
|
|
||||||
llm_description: Data query start time
|
|
||||||
form: llm
|
|
||||||
- name: type
|
|
||||||
type: string
|
|
||||||
required: false
|
|
||||||
label:
|
|
||||||
en_US: type
|
|
||||||
zh_Hans: 查询数据类型(log, slow, error)
|
|
||||||
pt_BR: type
|
|
||||||
human_description:
|
|
||||||
en_US: Time format in strftime standard.
|
|
||||||
zh_Hans: 指定查询数据类型(log, slow, error)。
|
|
||||||
pt_BR: Time format in strftime standard.
|
|
||||||
llm_description: Time format in strftime standard.
|
|
||||||
form: llm
|
|
||||||
Loading…
Reference in New Issue