feat: add some host metrics query tools (#2)
* feat: add some host metrics query tools * fix: wrong paramspull/17608/head
MisluNotFound
1 year ago
committed by
GitHub
parent
d35b3c5bb7
commit
ff2c59a881
@ -0,0 +1,46 @@
|
|||||||
|
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
|
||||||
|
from libs.apo_utils import APOUtils
|
||||||
|
|
||||||
|
|
||||||
|
class HostCoreNumTool(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]:
|
||||||
|
node = tool_parameters.get("node")
|
||||||
|
job = tool_parameters.get("job")
|
||||||
|
start_time = tool_parameters.get("startTime")
|
||||||
|
end_time = tool_parameters.get("endTime")
|
||||||
|
params = {
|
||||||
|
'metricName': "宿主机监控指标 - Quick CPU / Mem / Disk - CPU Cores",
|
||||||
|
'params': {
|
||||||
|
'node': node,
|
||||||
|
**({'job': job} if job else {})
|
||||||
|
},
|
||||||
|
'startTime': start_time,
|
||||||
|
'endTime': end_time,
|
||||||
|
'step': APOUtils.get_step(start_time, end_time),
|
||||||
|
}
|
||||||
|
resp = requests.post(dify_config.APO_BACKEND_URL + '/api/metric/query', json=params)
|
||||||
|
list = resp.json()['result']
|
||||||
|
list = json.dumps({
|
||||||
|
'type': 'metric',
|
||||||
|
'display': True,
|
||||||
|
'unit': list['unit'],
|
||||||
|
'data': {
|
||||||
|
'timeseries': list['timeseries']
|
||||||
|
}
|
||||||
|
})
|
||||||
|
yield self.create_text_message(list)
|
||||||
@ -0,0 +1,45 @@
|
|||||||
|
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
|
||||||
|
from libs.apo_utils import APOUtils
|
||||||
|
|
||||||
|
|
||||||
|
class HostCpuCoreUsageTool(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]:
|
||||||
|
node = tool_parameters.get("node")
|
||||||
|
job = tool_parameters.get("job")
|
||||||
|
start_time = tool_parameters.get("startTime")
|
||||||
|
end_time = tool_parameters.get("endTime")
|
||||||
|
params = {
|
||||||
|
'metricName': "宿主机监控指标 - Basic CPU / Mem / Net / Disk - CPU Basic - CPU Usage",
|
||||||
|
'params': {
|
||||||
|
'node': node,
|
||||||
|
},
|
||||||
|
'startTime': start_time,
|
||||||
|
'endTime': end_time,
|
||||||
|
'step': APOUtils.get_step(start_time, end_time),
|
||||||
|
}
|
||||||
|
resp = requests.post(dify_config.APO_BACKEND_URL + '/api/metric/query', json=params)
|
||||||
|
list = resp.json()['result']
|
||||||
|
list = json.dumps({
|
||||||
|
'type': 'metric',
|
||||||
|
'display': True,
|
||||||
|
'unit': list['unit'],
|
||||||
|
'data': {
|
||||||
|
'timeseries': list['timeseries']
|
||||||
|
}
|
||||||
|
})
|
||||||
|
yield self.create_text_message(list)
|
||||||
@ -0,0 +1,46 @@
|
|||||||
|
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
|
||||||
|
from libs.apo_utils import APOUtils
|
||||||
|
|
||||||
|
|
||||||
|
class HostCpuPressureTool(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]:
|
||||||
|
node = tool_parameters.get("node")
|
||||||
|
job = tool_parameters.get("job")
|
||||||
|
start_time = tool_parameters.get("startTime")
|
||||||
|
end_time = tool_parameters.get("endTime")
|
||||||
|
params = {
|
||||||
|
'metricName': "宿主机监控指标 - Quick CPU / Mem / Disk - Pressure - CPU",
|
||||||
|
'params': {
|
||||||
|
'node': node,
|
||||||
|
**({'job': job} if job else {})
|
||||||
|
},
|
||||||
|
'startTime': start_time,
|
||||||
|
'endTime': end_time,
|
||||||
|
'step': APOUtils.get_step(start_time, end_time),
|
||||||
|
}
|
||||||
|
resp = requests.post(dify_config.APO_BACKEND_URL + '/api/metric/query', json=params)
|
||||||
|
list = resp.json()['result']
|
||||||
|
list = json.dumps({
|
||||||
|
'type': 'metric',
|
||||||
|
'display': True,
|
||||||
|
'unit': list['unit'],
|
||||||
|
'data': {
|
||||||
|
'timeseries': list['timeseries']
|
||||||
|
}
|
||||||
|
})
|
||||||
|
yield self.create_text_message(list)
|
||||||
@ -0,0 +1,44 @@
|
|||||||
|
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
|
||||||
|
from libs.apo_utils import APOUtils
|
||||||
|
|
||||||
|
|
||||||
|
class HostCpuUsageTool(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]:
|
||||||
|
node = tool_parameters.get("node")
|
||||||
|
start_time = tool_parameters.get("startTime")
|
||||||
|
end_time = tool_parameters.get("endTime")
|
||||||
|
params = {
|
||||||
|
'metricName': "宿主机监控指标 - Quick CPU / Mem / Disk - CPU Busy",
|
||||||
|
'params': {
|
||||||
|
'node': node,
|
||||||
|
},
|
||||||
|
'startTime': start_time,
|
||||||
|
'endTime': end_time,
|
||||||
|
'step': APOUtils.get_step(start_time, end_time),
|
||||||
|
}
|
||||||
|
resp = requests.post(dify_config.APO_BACKEND_URL + '/api/metric/query', json=params)
|
||||||
|
list = resp.json()['result']
|
||||||
|
list = json.dumps({
|
||||||
|
'type': 'metric',
|
||||||
|
'display': True,
|
||||||
|
'unit': list['unit'],
|
||||||
|
'data': {
|
||||||
|
'timeseries': list['timeseries']
|
||||||
|
}
|
||||||
|
})
|
||||||
|
yield self.create_text_message(list)
|
||||||
@ -0,0 +1,57 @@
|
|||||||
|
identity:
|
||||||
|
name: 查询宿主机CPU繁忙率
|
||||||
|
author: APO
|
||||||
|
label:
|
||||||
|
en_US: Query host's CPU usage
|
||||||
|
zh_Hans: 查询宿主机CPU繁忙率
|
||||||
|
pt_BR: Query host's CPU usage
|
||||||
|
description:
|
||||||
|
human:
|
||||||
|
en_US: Query host's CPU usage
|
||||||
|
zh_Hans: 查询宿主机CPU繁忙率
|
||||||
|
pt_BR: Query host's CPU usage
|
||||||
|
llm: Query host's CPU usage
|
||||||
|
display:
|
||||||
|
type: metric
|
||||||
|
title: 宿主机监控指标 - Quick CPU / Mem / Disk - CPU Busy
|
||||||
|
unit: percent
|
||||||
|
parameters:
|
||||||
|
- name: node
|
||||||
|
type: string
|
||||||
|
required: true
|
||||||
|
label:
|
||||||
|
en_US: Host Node
|
||||||
|
zh_Hans: 宿主机节点
|
||||||
|
pt_BR: Host Node
|
||||||
|
human_description:
|
||||||
|
en_US: Host node identifier
|
||||||
|
zh_Hans: 宿主机节点标识
|
||||||
|
pt_BR: Host node identifier
|
||||||
|
llm_description: Unique identifier for host node
|
||||||
|
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
|
||||||
@ -0,0 +1,46 @@
|
|||||||
|
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
|
||||||
|
from libs.apo_utils import APOUtils
|
||||||
|
|
||||||
|
|
||||||
|
class HostIoPressureTool(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]:
|
||||||
|
node = tool_parameters.get("node")
|
||||||
|
job = tool_parameters.get("job")
|
||||||
|
start_time = tool_parameters.get("startTime")
|
||||||
|
end_time = tool_parameters.get("endTime")
|
||||||
|
params = {
|
||||||
|
'metricName': "宿主机监控指标 - Quick CPU / Mem / Disk - Pressure - I/O",
|
||||||
|
'params': {
|
||||||
|
'node': node,
|
||||||
|
**({'job': job} if job else {})
|
||||||
|
},
|
||||||
|
'startTime': start_time,
|
||||||
|
'endTime': end_time,
|
||||||
|
'step': APOUtils.get_step(start_time, end_time),
|
||||||
|
}
|
||||||
|
resp = requests.post(dify_config.APO_BACKEND_URL + '/api/metric/query', json=params)
|
||||||
|
list = resp.json()['result']
|
||||||
|
list = json.dumps({
|
||||||
|
'type': 'metric',
|
||||||
|
'display': True,
|
||||||
|
'unit': list['unit'],
|
||||||
|
'data': {
|
||||||
|
'timeseries': list['timeseries']
|
||||||
|
}
|
||||||
|
})
|
||||||
|
yield self.create_text_message(list)
|
||||||
@ -0,0 +1,46 @@
|
|||||||
|
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
|
||||||
|
from libs.apo_utils import APOUtils
|
||||||
|
|
||||||
|
|
||||||
|
class HostMemPressureTool(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]:
|
||||||
|
node = tool_parameters.get("node")
|
||||||
|
job = tool_parameters.get("job")
|
||||||
|
start_time = tool_parameters.get("startTime")
|
||||||
|
end_time = tool_parameters.get("endTime")
|
||||||
|
params = {
|
||||||
|
'metricName': "宿主机监控指标 - Quick CPU / Mem / Disk - Pressure - Mem",
|
||||||
|
'params': {
|
||||||
|
'node': node,
|
||||||
|
**({'job': job} if job else {})
|
||||||
|
},
|
||||||
|
'startTime': start_time,
|
||||||
|
'endTime': end_time,
|
||||||
|
'step': APOUtils.get_step(start_time, end_time),
|
||||||
|
}
|
||||||
|
resp = requests.post(dify_config.APO_BACKEND_URL + '/api/metric/query', json=params)
|
||||||
|
list = resp.json()['result']
|
||||||
|
list = json.dumps({
|
||||||
|
'type': 'metric',
|
||||||
|
'display': True,
|
||||||
|
'unit': list['unit'],
|
||||||
|
'data': {
|
||||||
|
'timeseries': list['timeseries']
|
||||||
|
}
|
||||||
|
})
|
||||||
|
yield self.create_text_message(list)
|
||||||
@ -0,0 +1,46 @@
|
|||||||
|
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
|
||||||
|
from libs.apo_utils import APOUtils
|
||||||
|
|
||||||
|
|
||||||
|
class HostRamTotalTool(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]:
|
||||||
|
node = tool_parameters.get("node")
|
||||||
|
job = tool_parameters.get("job")
|
||||||
|
start_time = tool_parameters.get("startTime")
|
||||||
|
end_time = tool_parameters.get("endTime")
|
||||||
|
params = {
|
||||||
|
'metricName': "宿主机监控指标 - Quick CPU / Mem / Disk - RAM Total",
|
||||||
|
'params': {
|
||||||
|
'node': node,
|
||||||
|
**({'job': job} if job else {})
|
||||||
|
},
|
||||||
|
'startTime': start_time,
|
||||||
|
'endTime': end_time,
|
||||||
|
'step': APOUtils.get_step(start_time, end_time),
|
||||||
|
}
|
||||||
|
resp = requests.post(dify_config.APO_BACKEND_URL + '/api/metric/query', json=params)
|
||||||
|
list = resp.json()['result']
|
||||||
|
list = json.dumps({
|
||||||
|
'type': 'metric',
|
||||||
|
'display': True,
|
||||||
|
'unit': list['unit'],
|
||||||
|
'data': {
|
||||||
|
'timeseries': list['timeseries']
|
||||||
|
}
|
||||||
|
})
|
||||||
|
yield self.create_text_message(list)
|
||||||
@ -0,0 +1,46 @@
|
|||||||
|
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
|
||||||
|
from libs.apo_utils import APOUtils
|
||||||
|
|
||||||
|
|
||||||
|
class HostRamUsageTool(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]:
|
||||||
|
node = tool_parameters.get("node")
|
||||||
|
job = tool_parameters.get("job")
|
||||||
|
start_time = tool_parameters.get("startTime")
|
||||||
|
end_time = tool_parameters.get("endTime")
|
||||||
|
params = {
|
||||||
|
'metricName': "宿主机监控指标 - Quick CPU / Mem / Disk - RAM Used - B",
|
||||||
|
'params': {
|
||||||
|
'node': node,
|
||||||
|
**({'job': job} if job else {})
|
||||||
|
},
|
||||||
|
'startTime': start_time,
|
||||||
|
'endTime': end_time,
|
||||||
|
'step': APOUtils.get_step(start_time, end_time),
|
||||||
|
}
|
||||||
|
resp = requests.post(dify_config.APO_BACKEND_URL + '/api/metric/query', json=params)
|
||||||
|
list = resp.json()['result']
|
||||||
|
list = json.dumps({
|
||||||
|
'type': 'metric',
|
||||||
|
'display': True,
|
||||||
|
'unit': list['unit'],
|
||||||
|
'data': {
|
||||||
|
'timeseries': list['timeseries']
|
||||||
|
}
|
||||||
|
})
|
||||||
|
yield self.create_text_message(list)
|
||||||
@ -0,0 +1,44 @@
|
|||||||
|
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
|
||||||
|
from libs.apo_utils import APOUtils
|
||||||
|
|
||||||
|
|
||||||
|
class HostRootFsTotalTool(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]:
|
||||||
|
node = tool_parameters.get("node")
|
||||||
|
start_time = tool_parameters.get("startTime")
|
||||||
|
end_time = tool_parameters.get("endTime")
|
||||||
|
params = {
|
||||||
|
'metricName': "宿主机监控指标 - Quick CPU / Mem / Disk - RootFS Total",
|
||||||
|
'params': {
|
||||||
|
'node': node,
|
||||||
|
},
|
||||||
|
'startTime': start_time,
|
||||||
|
'endTime': end_time,
|
||||||
|
'step': APOUtils.get_step(start_time, end_time),
|
||||||
|
}
|
||||||
|
resp = requests.post(dify_config.APO_BACKEND_URL + '/api/metric/query', json=params)
|
||||||
|
list = resp.json()['result']
|
||||||
|
list = json.dumps({
|
||||||
|
'type': 'metric',
|
||||||
|
'display': True,
|
||||||
|
'unit': list['unit'],
|
||||||
|
'data': {
|
||||||
|
'timeseries': list['timeseries']
|
||||||
|
}
|
||||||
|
})
|
||||||
|
yield self.create_text_message(list)
|
||||||
@ -0,0 +1,46 @@
|
|||||||
|
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
|
||||||
|
from libs.apo_utils import APOUtils
|
||||||
|
|
||||||
|
|
||||||
|
class HostRootFsUsageTool(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]:
|
||||||
|
node = tool_parameters.get("node")
|
||||||
|
job = tool_parameters.get("job")
|
||||||
|
start_time = tool_parameters.get("startTime")
|
||||||
|
end_time = tool_parameters.get("endTime")
|
||||||
|
params = {
|
||||||
|
'metricName': "宿主机监控指标 - Quick CPU / Mem / Disk - Root FS Used",
|
||||||
|
'params': {
|
||||||
|
'node': node,
|
||||||
|
**({'job': job} if job else {})
|
||||||
|
},
|
||||||
|
'startTime': start_time,
|
||||||
|
'endTime': end_time,
|
||||||
|
'step': APOUtils.get_step(start_time, end_time),
|
||||||
|
}
|
||||||
|
resp = requests.post(dify_config.APO_BACKEND_URL + '/api/metric/query', json=params)
|
||||||
|
list = resp.json()['result']
|
||||||
|
list = json.dumps({
|
||||||
|
'type': 'metric',
|
||||||
|
'display': True,
|
||||||
|
'unit': list['unit'],
|
||||||
|
'data': {
|
||||||
|
'timeseries': list['timeseries']
|
||||||
|
}
|
||||||
|
})
|
||||||
|
yield self.create_text_message(list)
|
||||||
@ -0,0 +1,46 @@
|
|||||||
|
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
|
||||||
|
from libs.apo_utils import APOUtils
|
||||||
|
|
||||||
|
|
||||||
|
class HostSwapTotal(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]:
|
||||||
|
node = tool_parameters.get("node")
|
||||||
|
job = tool_parameters.get("job")
|
||||||
|
start_time = tool_parameters.get("startTime")
|
||||||
|
end_time = tool_parameters.get("endTime")
|
||||||
|
params = {
|
||||||
|
'metricName': "宿主机监控指标 - Quick CPU / Mem / Disk - SWAP Total",
|
||||||
|
'params': {
|
||||||
|
'node': node,
|
||||||
|
**({'job': job} if job else {})
|
||||||
|
},
|
||||||
|
'startTime': start_time,
|
||||||
|
'endTime': end_time,
|
||||||
|
'step': APOUtils.get_step(start_time, end_time),
|
||||||
|
}
|
||||||
|
resp = requests.post(dify_config.APO_BACKEND_URL + '/api/metric/query', json=params)
|
||||||
|
list = resp.json()['result']
|
||||||
|
list = json.dumps({
|
||||||
|
'type': 'metric',
|
||||||
|
'display': True,
|
||||||
|
'unit': list['unit'],
|
||||||
|
'data': {
|
||||||
|
'timeseries': list['timeseries']
|
||||||
|
}
|
||||||
|
})
|
||||||
|
yield self.create_text_message(list)
|
||||||
@ -0,0 +1,46 @@
|
|||||||
|
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
|
||||||
|
from libs.apo_utils import APOUtils
|
||||||
|
|
||||||
|
|
||||||
|
class HostSwapUsageTool(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]:
|
||||||
|
node = tool_parameters.get("node")
|
||||||
|
job = tool_parameters.get("job")
|
||||||
|
start_time = tool_parameters.get("startTime")
|
||||||
|
end_time = tool_parameters.get("endTime")
|
||||||
|
params = {
|
||||||
|
'metricName': "宿主机监控指标 - Quick CPU / Mem / Disk - SWAP Used",
|
||||||
|
'params': {
|
||||||
|
'node': node,
|
||||||
|
**({'job': job} if job else {})
|
||||||
|
},
|
||||||
|
'startTime': start_time,
|
||||||
|
'endTime': end_time,
|
||||||
|
'step': APOUtils.get_step(start_time, end_time),
|
||||||
|
}
|
||||||
|
resp = requests.post(dify_config.APO_BACKEND_URL + '/api/metric/query', json=params)
|
||||||
|
list = resp.json()['result']
|
||||||
|
list = json.dumps({
|
||||||
|
'type': 'metric',
|
||||||
|
'display': True,
|
||||||
|
'unit': list['unit'],
|
||||||
|
'data': {
|
||||||
|
'timeseries': list['timeseries']
|
||||||
|
}
|
||||||
|
})
|
||||||
|
yield self.create_text_message(list)
|
||||||
@ -0,0 +1,46 @@
|
|||||||
|
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
|
||||||
|
from libs.apo_utils import APOUtils
|
||||||
|
|
||||||
|
|
||||||
|
class HostSysLoad(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]:
|
||||||
|
node = tool_parameters.get("node")
|
||||||
|
job = tool_parameters.get("job")
|
||||||
|
start_time = tool_parameters.get("startTime")
|
||||||
|
end_time = tool_parameters.get("endTime")
|
||||||
|
params = {
|
||||||
|
'metricName': "宿主机监控指标 - Quick CPU / Mem / Disk - Sys Load",
|
||||||
|
'params': {
|
||||||
|
'node': node,
|
||||||
|
**({'job': job} if job else {})
|
||||||
|
},
|
||||||
|
'startTime': start_time,
|
||||||
|
'endTime': end_time,
|
||||||
|
'step': APOUtils.get_step(start_time, end_time),
|
||||||
|
}
|
||||||
|
resp = requests.post(dify_config.APO_BACKEND_URL + '/api/metric/query', json=params)
|
||||||
|
list = resp.json()['result']
|
||||||
|
list = json.dumps({
|
||||||
|
'type': 'metric',
|
||||||
|
'display': True,
|
||||||
|
'unit': list['unit'],
|
||||||
|
'data': {
|
||||||
|
'timeseries': list['timeseries']
|
||||||
|
}
|
||||||
|
})
|
||||||
|
yield self.create_text_message(list)
|
||||||
@ -0,0 +1,46 @@
|
|||||||
|
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
|
||||||
|
from libs.apo_utils import APOUtils
|
||||||
|
|
||||||
|
|
||||||
|
class HostSysProcBusyTool(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]:
|
||||||
|
node = tool_parameters.get("node")
|
||||||
|
job = tool_parameters.get("job")
|
||||||
|
start_time = tool_parameters.get("startTime")
|
||||||
|
end_time = tool_parameters.get("endTime")
|
||||||
|
params = {
|
||||||
|
'metricName': "宿主机监控指标 - Basic CPU / Mem / Net / Disk - CPU Basic - Busy System",
|
||||||
|
'params': {
|
||||||
|
'node': node,
|
||||||
|
**({'job': job} if job else {})
|
||||||
|
},
|
||||||
|
'startTime': start_time,
|
||||||
|
'endTime': end_time,
|
||||||
|
'step': APOUtils.get_step(start_time, end_time),
|
||||||
|
}
|
||||||
|
resp = requests.post(dify_config.APO_BACKEND_URL + '/api/metric/query', json=params)
|
||||||
|
list = resp.json()['result']
|
||||||
|
list = json.dumps({
|
||||||
|
'type': 'metric',
|
||||||
|
'display': True,
|
||||||
|
'unit': list['unit'],
|
||||||
|
'data': {
|
||||||
|
'timeseries': list['timeseries']
|
||||||
|
}
|
||||||
|
})
|
||||||
|
yield self.create_text_message(list)
|
||||||
@ -0,0 +1,46 @@
|
|||||||
|
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
|
||||||
|
from libs.apo_utils import APOUtils
|
||||||
|
|
||||||
|
|
||||||
|
class HostUptimeTool(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]:
|
||||||
|
node = tool_parameters.get("node")
|
||||||
|
job = tool_parameters.get("job")
|
||||||
|
start_time = tool_parameters.get("startTime")
|
||||||
|
end_time = tool_parameters.get("endTime")
|
||||||
|
params = {
|
||||||
|
'metricName': "宿主机监控指标 - Quick CPU / Mem / Disk - Uptime",
|
||||||
|
'params': {
|
||||||
|
'node': node,
|
||||||
|
**({'job': job} if job else {})
|
||||||
|
},
|
||||||
|
'startTime': start_time,
|
||||||
|
'endTime': end_time,
|
||||||
|
'step': APOUtils.get_step(start_time, end_time),
|
||||||
|
}
|
||||||
|
resp = requests.post(dify_config.APO_BACKEND_URL + '/api/metric/query', json=params)
|
||||||
|
list = resp.json()['result']
|
||||||
|
list = json.dumps({
|
||||||
|
'type': 'metric',
|
||||||
|
'display': True,
|
||||||
|
'unit': list['unit'],
|
||||||
|
'data': {
|
||||||
|
'timeseries': list['timeseries']
|
||||||
|
}
|
||||||
|
})
|
||||||
|
yield self.create_text_message(list)
|
||||||
Loading…
Reference in New Issue