feat: Tools from metrics 0-14 (#3)
MisluNotFound
1 year ago
committed by
GitHub
parent
800506f3fa
commit
8afd983d98
@ -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 ClusterCpuTotalTool(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]:
|
||||
cluster = tool_parameters.get('cluster')
|
||||
start_time = tool_parameters.get("startTime")
|
||||
end_time = tool_parameters.get("endTime")
|
||||
params = {
|
||||
'metricName': '集群总览 - 总览 - CPU使用量 - Total',
|
||||
'params': {
|
||||
**({'cluster': cluster} if cluster 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,53 @@
|
||||
identity:
|
||||
name: 集群总可用CPU核数
|
||||
author: APO
|
||||
label:
|
||||
en_US: The total number of available CPU cores in the cluster
|
||||
zh_Hans: 集群总可用CPU核数
|
||||
description:
|
||||
human:
|
||||
en_US: The total number of available CPU cores in the cluster
|
||||
zh_Hans: 集群总可用CPU核数
|
||||
llm: The total number of available CPU cores in the cluster
|
||||
display:
|
||||
type: metric
|
||||
title: 集群总览 - 总览 - CPU使用量 - Total
|
||||
unit: "core"
|
||||
parameters:
|
||||
- name: cluster
|
||||
type: string
|
||||
required: False
|
||||
label:
|
||||
en_US: Cluster name
|
||||
zh_Hans: 集群名称
|
||||
human_description:
|
||||
en_US: Cluster name
|
||||
zh_Hans: 集群名称
|
||||
llm_description: Cluster name
|
||||
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,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 ClusterCpuUsageTool(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]:
|
||||
cluster = tool_parameters.get('cluster')
|
||||
start_time = tool_parameters.get("startTime")
|
||||
end_time = tool_parameters.get("endTime")
|
||||
params = {
|
||||
'metricName': '集群总览 - 总览 - 集群总CPU使用率 - Real Linux',
|
||||
'params': {
|
||||
**({'cluster': cluster} if cluster 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,53 @@
|
||||
identity:
|
||||
name: 实时集群CPU使用率
|
||||
author: APO
|
||||
label:
|
||||
en_US: Real-time cluster CPU usage
|
||||
zh_Hans: 实时集群CPU使用率
|
||||
description:
|
||||
human:
|
||||
en_US: Real-time cluster CPU usage
|
||||
zh_Hans: 实时集群CPU使用率
|
||||
llm: Real-time cluster CPU usage
|
||||
display:
|
||||
type: metric
|
||||
title: 集群总览 - 总览 - 集群总CPU使用率 - Real Linux
|
||||
unit: "percentunit"
|
||||
parameters:
|
||||
- name: cluster
|
||||
type: string
|
||||
required: False
|
||||
label:
|
||||
en_US: Cluster name
|
||||
zh_Hans: 集群名称
|
||||
human_description:
|
||||
en_US: Cluster name
|
||||
zh_Hans: 集群名称
|
||||
llm_description: Cluster name
|
||||
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,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 ClusterCpuUsageRealLinuxTool(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]:
|
||||
cluster = tool_parameters.get('cluster')
|
||||
start_time = tool_parameters.get("startTime")
|
||||
end_time = tool_parameters.get("endTime")
|
||||
params = {
|
||||
'metricName': '集群总览 - 总览 - CPU使用量 - Real Linux',
|
||||
'params': {
|
||||
**({'cluster': cluster} if cluster 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,53 @@
|
||||
identity:
|
||||
name: 实时集群CPU使用核数
|
||||
author: APO
|
||||
label:
|
||||
en_US: The number of CPU cores in real-time use in the cluster
|
||||
zh_Hans: 实时集群CPU使用核数
|
||||
description:
|
||||
human:
|
||||
en_US: The number of CPU cores in real-time use in the cluster
|
||||
zh_Hans: 实时集群CPU使用核数
|
||||
llm: The number of CPU cores in real-time use in the cluster
|
||||
display:
|
||||
type: metric
|
||||
title: 集群总览 - 总览 - CPU使用量 - Real Linux
|
||||
unit: "core"
|
||||
parameters:
|
||||
- name: cluster
|
||||
type: string
|
||||
required: False
|
||||
label:
|
||||
en_US: Cluster name
|
||||
zh_Hans: 集群名称
|
||||
human_description:
|
||||
en_US: Cluster name
|
||||
zh_Hans: 集群名称
|
||||
llm_description: Cluster name
|
||||
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,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 ClusterMemoryTotalTool(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]:
|
||||
cluster = tool_parameters.get('cluster')
|
||||
start_time = tool_parameters.get("startTime")
|
||||
end_time = tool_parameters.get("endTime")
|
||||
params = {
|
||||
'metricName': '集群总览 - 总览 - 内存使用量 - Total',
|
||||
'params': {
|
||||
**({'cluster': cluster} if cluster 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,53 @@
|
||||
identity:
|
||||
name: 集群总可用内存字节数
|
||||
author: APO
|
||||
label:
|
||||
en_US: The total number of available bytes of memory in the cluster
|
||||
zh_Hans: 集群总可用内存字节数
|
||||
description:
|
||||
human:
|
||||
en_US: The total number of available bytes of memory in the cluster
|
||||
zh_Hans: 集群总可用内存字节数
|
||||
llm: The total number of available bytes of memory in the cluster
|
||||
display:
|
||||
type: metric
|
||||
title: 集群总览 - 总览 - 内存使用量 - Total
|
||||
unit: "bytes"
|
||||
parameters:
|
||||
- name: cluster
|
||||
type: string
|
||||
required: False
|
||||
label:
|
||||
en_US: Cluster name
|
||||
zh_Hans: 集群名称
|
||||
human_description:
|
||||
en_US: Cluster name
|
||||
zh_Hans: 集群名称
|
||||
llm_description: Cluster name
|
||||
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,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 ClusterMemoryUsageTool(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]:
|
||||
cluster = tool_parameters.get('cluster')
|
||||
start_time = tool_parameters.get("startTime")
|
||||
end_time = tool_parameters.get("endTime")
|
||||
params = {
|
||||
'metricName': '集群总览 - 总览 - 集群总内存使用率 - Real Linux',
|
||||
'params': {
|
||||
**({'cluster': cluster} if cluster 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,53 @@
|
||||
identity:
|
||||
name: 实时集群内存使用率
|
||||
author: APO
|
||||
label:
|
||||
en_US: Real-time cluster memory usage
|
||||
zh_Hans: 实时集群内存使用率
|
||||
description:
|
||||
human:
|
||||
en_US: Real-time cluster memory usage
|
||||
zh_Hans: 实时集群内存使用率
|
||||
llm: Real-time cluster memory usage
|
||||
display:
|
||||
type: metric
|
||||
title: 集群总览 - 总览 - 集群总内存使用率 - Real Linux
|
||||
unit: "percentunit"
|
||||
parameters:
|
||||
- name: cluster
|
||||
type: string
|
||||
required: False
|
||||
label:
|
||||
en_US: Cluster name
|
||||
zh_Hans: 集群名称
|
||||
human_description:
|
||||
en_US: Cluster name
|
||||
zh_Hans: 集群名称
|
||||
llm_description: Cluster name
|
||||
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,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 ClusterMemoryUsageRealLinuxTool(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]:
|
||||
cluster = tool_parameters.get('cluster')
|
||||
start_time = tool_parameters.get("startTime")
|
||||
end_time = tool_parameters.get("endTime")
|
||||
params = {
|
||||
'metricName': '集群总览 - 总览 - 内存使用量 - Real Linux',
|
||||
'params': {
|
||||
**({'cluster': cluster} if cluster 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,53 @@
|
||||
identity:
|
||||
name: 实时集群内存使用字节数
|
||||
author: APO
|
||||
label:
|
||||
en_US: The number of bytes of memory in real-time use in the cluster
|
||||
zh_Hans: 实时集群内存使用字节数
|
||||
description:
|
||||
human:
|
||||
en_US: The number of bytes of memory in real-time use in the cluster
|
||||
zh_Hans: 实时集群内存使用字节数
|
||||
llm: The number of bytes of memory in real-time use in the cluster
|
||||
display:
|
||||
type: metric
|
||||
title: 集群总览 - 总览 - 内存使用量 - Real Linux
|
||||
unit: "bytes"
|
||||
parameters:
|
||||
- name: cluster
|
||||
type: string
|
||||
required: False
|
||||
label:
|
||||
en_US: Cluster name
|
||||
zh_Hans: 集群名称
|
||||
human_description:
|
||||
en_US: Cluster name
|
||||
zh_Hans: 集群名称
|
||||
llm_description: Cluster name
|
||||
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,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 ClusterNamespaceCountTool(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]:
|
||||
cadvisor_job_name = tool_parameters.get('cadvisor_job_name')
|
||||
start_time = tool_parameters.get("startTime")
|
||||
end_time = tool_parameters.get("endTime")
|
||||
params = {
|
||||
'metricName': '集群总览 - 总览 - 命名空间数量',
|
||||
'params': {
|
||||
**({'cadvisor_job_name': cadvisor_job_name} if cadvisor_job_name 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,53 @@
|
||||
identity:
|
||||
name: K8s集群中命名空间的数量
|
||||
author: APO
|
||||
label:
|
||||
en_US: Number of namespaces in the K8s cluster
|
||||
zh_Hans: K8s集群中命名空间的数量
|
||||
description:
|
||||
human:
|
||||
en_US: Number of namespaces in the K8s cluster
|
||||
zh_Hans: K8s集群中命名空间的数量
|
||||
llm: Number of namespaces in the K8s cluster
|
||||
display:
|
||||
type: metric
|
||||
title: 集群总览 - 总览 - 命名空间数量
|
||||
unit: ""
|
||||
parameters:
|
||||
- name: cadvisor_job_name
|
||||
type: string
|
||||
required: False
|
||||
label:
|
||||
en_US: cAdvisor job name
|
||||
zh_Hans: cAdvisor任务名称
|
||||
human_description:
|
||||
en_US: cAdvisor job name
|
||||
zh_Hans: cAdvisor任务名称
|
||||
llm_description: cAdvisor job name
|
||||
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,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 ClusterNetworkLinuxReceiveDroppedPacketsTool(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]:
|
||||
cluster = tool_parameters.get('cluster')
|
||||
start_time = tool_parameters.get("startTime")
|
||||
end_time = tool_parameters.get("endTime")
|
||||
params = {
|
||||
'metricName': '集群总览 - 总览 - 网络饱和 - 丢包数 - Linux Packets dropped (receive)',
|
||||
'params': {
|
||||
**({'cluster': cluster} if cluster 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,53 @@
|
||||
identity:
|
||||
name: Linux集群网络接收丢包数
|
||||
author: APO
|
||||
label:
|
||||
en_US: The number of dropped packets received by the Linux cluster network
|
||||
zh_Hans: Linux集群网络接收丢包数
|
||||
description:
|
||||
human:
|
||||
en_US: The number of dropped packets received by the Linux cluster network
|
||||
zh_Hans: Linux集群网络接收丢包数
|
||||
llm: The number of dropped packets received by the Linux cluster network
|
||||
display:
|
||||
type: metric
|
||||
title: 集群总览 - 总览 - 网络饱和 - 丢包数 - Linux Packets dropped (receive)
|
||||
unit: "short"
|
||||
parameters:
|
||||
- name: cluster
|
||||
type: string
|
||||
required: False
|
||||
label:
|
||||
en_US: Cluster name
|
||||
zh_Hans: 集群名称
|
||||
human_description:
|
||||
en_US: Cluster name
|
||||
zh_Hans: 集群名称
|
||||
llm_description: Cluster name
|
||||
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,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 ClusterNetworkLinuxTransmitDroppedPacketsTool(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]:
|
||||
cluster = tool_parameters.get('cluster')
|
||||
start_time = tool_parameters.get("startTime")
|
||||
end_time = tool_parameters.get("endTime")
|
||||
params = {
|
||||
'metricName': '集群总览 - 总览 - 网络饱和 - 丢包数 - Linux Packets dropped (transmit)',
|
||||
'params': {
|
||||
**({'cluster': cluster} if cluster 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,53 @@
|
||||
identity:
|
||||
name: Linux集群网络发送丢包数
|
||||
author: APO
|
||||
label:
|
||||
en_US: The number of dropped packets sent by the Linux cluster network
|
||||
zh_Hans: Linux集群网络发送丢包数
|
||||
description:
|
||||
human:
|
||||
en_US: The number of dropped packets sent by the Linux cluster network
|
||||
zh_Hans: Linux集群网络发送丢包数
|
||||
llm: The number of dropped packets sent by the Linux cluster network
|
||||
display:
|
||||
type: metric
|
||||
title: 集群总览 - 总览 - 网络饱和 - 丢包数 - Linux Packets dropped (transmit)
|
||||
unit: "short"
|
||||
parameters:
|
||||
- name: cluster
|
||||
type: string
|
||||
required: False
|
||||
label:
|
||||
en_US: Cluster name
|
||||
zh_Hans: 集群名称
|
||||
human_description:
|
||||
en_US: Cluster name
|
||||
zh_Hans: 集群名称
|
||||
llm_description: Cluster name
|
||||
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,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 ClusterNetworkReceiveBandwidthTool(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]:
|
||||
cluster = tool_parameters.get('cluster')
|
||||
start_time = tool_parameters.get("startTime")
|
||||
end_time = tool_parameters.get("endTime")
|
||||
params = {
|
||||
'metricName': '集群总览 - 总览 - 集群整体网络带宽(按设备统计) - 接收',
|
||||
'params': {
|
||||
**({'cluster': cluster} if cluster 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,53 @@
|
||||
identity:
|
||||
name: 集群网络接收带宽(按设备统计,排除虚拟设备)
|
||||
author: APO
|
||||
label:
|
||||
en_US: Cluster network receiving bandwidth (statistically by device, excluding virtual devices)
|
||||
zh_Hans: 集群网络接收带宽(按设备统计,排除虚拟设备)
|
||||
description:
|
||||
human:
|
||||
en_US: Cluster network receiving bandwidth (statistically by device, excluding virtual devices)
|
||||
zh_Hans: 集群网络接收带宽(按设备统计,排除虚拟设备)
|
||||
llm: Cluster network receiving bandwidth (statistically by device, excluding virtual devices)
|
||||
display:
|
||||
type: metric
|
||||
title: 集群总览 - 总览 - 集群整体网络带宽(按设备统计) - 接收
|
||||
unit: "bytes"
|
||||
parameters:
|
||||
- name: cluster
|
||||
type: string
|
||||
required: False
|
||||
label:
|
||||
en_US: Cluster name
|
||||
zh_Hans: 集群名称
|
||||
human_description:
|
||||
en_US: Cluster name
|
||||
zh_Hans: 集群名称
|
||||
llm_description: Cluster name
|
||||
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,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 ClusterNetworkSendBandwidthTool(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]:
|
||||
cluster = tool_parameters.get('cluster')
|
||||
start_time = tool_parameters.get("startTime")
|
||||
end_time = tool_parameters.get("endTime")
|
||||
params = {
|
||||
'metricName': '集群总览 - 总览 - 集群整体网络带宽(按设备统计) - 发送',
|
||||
'params': {
|
||||
**({'cluster': cluster} if cluster 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,53 @@
|
||||
identity:
|
||||
name: 集群网络发送带宽(按设备统计,排除虚拟设备)
|
||||
author: APO
|
||||
label:
|
||||
en_US: Cluster network sending bandwidth (statistically by device, excluding virtual devices)
|
||||
zh_Hans: 集群网络发送带宽(按设备统计,排除虚拟设备)
|
||||
description:
|
||||
human:
|
||||
en_US: Cluster network sending bandwidth (statistically by device, excluding virtual devices)
|
||||
zh_Hans: 集群网络发送带宽(按设备统计,排除虚拟设备)
|
||||
llm: Cluster network sending bandwidth (statistically by device, excluding virtual devices)
|
||||
display:
|
||||
type: metric
|
||||
title: 集群总览 - 总览 - 集群整体网络带宽(按设备统计) - 发送
|
||||
unit: "bytes"
|
||||
parameters:
|
||||
- name: cluster
|
||||
type: string
|
||||
required: False
|
||||
label:
|
||||
en_US: Cluster name
|
||||
zh_Hans: 集群名称
|
||||
human_description:
|
||||
en_US: Cluster name
|
||||
zh_Hans: 集群名称
|
||||
llm_description: Cluster name
|
||||
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,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 ClusterNodeCountTool(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]:
|
||||
cadvisor_job_name = tool_parameters.get('cadvisor_job_name')
|
||||
start_time = tool_parameters.get("startTime")
|
||||
end_time = tool_parameters.get("endTime")
|
||||
params = {
|
||||
'metricName': '集群总览 - 总览 - 节点数',
|
||||
'params': {
|
||||
**({'cadvisor_job_name': cadvisor_job_name} if cadvisor_job_name 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,53 @@
|
||||
identity:
|
||||
name: K8s集群中节点的数量
|
||||
author: APO
|
||||
label:
|
||||
en_US: The number of nodes in the K8s cluster
|
||||
zh_Hans: K8s集群中节点的数量
|
||||
description:
|
||||
human:
|
||||
en_US: The number of nodes in the K8s cluster
|
||||
zh_Hans: K8s集群中节点的数量
|
||||
llm: The number of nodes in the K8s cluster
|
||||
display:
|
||||
type: metric
|
||||
title: 集群总览 - 总览 - 节点数
|
||||
unit: ""
|
||||
parameters:
|
||||
- name: cadvisor_job_name
|
||||
type: string
|
||||
required: False
|
||||
label:
|
||||
en_US: cAdvisor job name
|
||||
zh_Hans: cAdvisor任务名称
|
||||
human_description:
|
||||
en_US: cAdvisor job name
|
||||
zh_Hans: cAdvisor任务名称
|
||||
llm_description: cAdvisor job name
|
||||
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,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 ClusterNodeCpuUsageLinuxTool(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]:
|
||||
cluster = tool_parameters.get('cluster')
|
||||
start_time = tool_parameters.get("startTime")
|
||||
end_time = tool_parameters.get("endTime")
|
||||
params = {
|
||||
'metricName': '集群总览 - 节点资源使用 - 节点CPU使用率 - Linux',
|
||||
'params': {
|
||||
**({'cluster': cluster} if cluster 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,53 @@
|
||||
identity:
|
||||
name: Linux节点CPU使用率
|
||||
author: APO
|
||||
label:
|
||||
en_US: The CPU usage rate of Linux nodes
|
||||
zh_Hans: Linux节点CPU使用率
|
||||
description:
|
||||
human:
|
||||
en_US: The CPU usage rate of Linux nodes
|
||||
zh_Hans: Linux节点CPU使用率
|
||||
llm: The CPU usage rate of Linux nodes
|
||||
display:
|
||||
type: metric
|
||||
title: 集群总览 - 节点资源使用 - 节点CPU使用率 - Linux
|
||||
unit: "percentunit"
|
||||
parameters:
|
||||
- name: cluster
|
||||
type: string
|
||||
required: False
|
||||
label:
|
||||
en_US: Cluster name
|
||||
zh_Hans: 集群名称
|
||||
human_description:
|
||||
en_US: Cluster name
|
||||
zh_Hans: 集群名称
|
||||
llm_description: Cluster name
|
||||
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,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 ClusterNodeMemoryUsageLinuxTool(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]:
|
||||
cluster = tool_parameters.get('cluster')
|
||||
start_time = tool_parameters.get("startTime")
|
||||
end_time = tool_parameters.get("endTime")
|
||||
params = {
|
||||
'metricName': '集群总览 - 节点资源使用 - 节点内存使用量 - Linux',
|
||||
'params': {
|
||||
**({'cluster': cluster} if cluster 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,53 @@
|
||||
identity:
|
||||
name: Linux节点内存使用字节数
|
||||
author: APO
|
||||
label:
|
||||
en_US: The number of bytes of memory used by Linux nodes
|
||||
zh_Hans: Linux节点内存使用字节数
|
||||
description:
|
||||
human:
|
||||
en_US: The number of bytes of memory used by Linux nodes
|
||||
zh_Hans: Linux节点内存使用字节数
|
||||
llm: The number of bytes of memory used by Linux nodes
|
||||
display:
|
||||
type: metric
|
||||
title: 集群总览 - 节点资源使用 - 节点内存使用量 - Linux
|
||||
unit: "bytes"
|
||||
parameters:
|
||||
- name: cluster
|
||||
type: string
|
||||
required: False
|
||||
label:
|
||||
en_US: Cluster name
|
||||
zh_Hans: 集群名称
|
||||
human_description:
|
||||
en_US: Cluster name
|
||||
zh_Hans: 集群名称
|
||||
llm_description: Cluster name
|
||||
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,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 ClusterRunningPodCountTool(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]:
|
||||
cadvisor_job_name = tool_parameters.get('cadvisor_job_name')
|
||||
start_time = tool_parameters.get("startTime")
|
||||
end_time = tool_parameters.get("endTime")
|
||||
params = {
|
||||
'metricName': '集群总览 - 总览 - 运行的Pod数',
|
||||
'params': {
|
||||
**({'cadvisor_job_name': cadvisor_job_name} if cadvisor_job_name 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,53 @@
|
||||
identity:
|
||||
name: K8s集群中运行中的Pod数量
|
||||
author: APO
|
||||
label:
|
||||
en_US: The number of running Pods in the K8s cluster
|
||||
zh_Hans: K8s集群中运行中的Pod数量
|
||||
description:
|
||||
human:
|
||||
en_US: The number of running Pods in the K8s cluster
|
||||
zh_Hans: K8s集群中运行中的Pod数量
|
||||
llm: The number of running Pods in the K8s cluster
|
||||
display:
|
||||
type: metric
|
||||
title: 集群总览 - 总览 - 运行的Pod数
|
||||
unit: ""
|
||||
parameters:
|
||||
- name: cadvisor_job_name
|
||||
type: string
|
||||
required: False
|
||||
label:
|
||||
en_US: cAdvisor job name
|
||||
zh_Hans: cAdvisor任务名称
|
||||
human_description:
|
||||
en_US: cAdvisor job name
|
||||
zh_Hans: cAdvisor任务名称
|
||||
llm_description: cAdvisor job name
|
||||
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,91 @@
|
||||
import json
|
||||
from pathlib import Path
|
||||
|
||||
import json
|
||||
from pathlib import Path
|
||||
from typing import List, Dict
|
||||
|
||||
from templates import YAML_TEMPLATE, PARAM_TEMPLATE, PYTHON_TEMPLATE
|
||||
|
||||
class ToolGenerator:
|
||||
def __init__(self, config_path: str):
|
||||
self.config = self.load_config(config_path)
|
||||
|
||||
def load_config(self, path: str) -> List[Dict]:
|
||||
with open(path, 'r', encoding='utf-8') as f:
|
||||
return json.load(f)
|
||||
|
||||
def generate_all(self, output_path: str):
|
||||
base_path = Path(output_path)
|
||||
for tool in self.config:
|
||||
self._generate_tool(tool, base_path)
|
||||
|
||||
def _generate_tool(self, tool: Dict, base_path: Path):
|
||||
tool_dir = base_path
|
||||
tool_dir.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
yaml_content = self._build_yaml(tool)
|
||||
with open(tool_dir / f"{tool['metric_name']}.yaml", 'w', encoding='utf-8') as f:
|
||||
f.write(yaml_content)
|
||||
|
||||
python_content = self._build_python(tool)
|
||||
with open(tool_dir / f"{tool['metric_name']}.py", 'w', encoding='utf-8') as f:
|
||||
f.write(python_content)
|
||||
|
||||
def _build_yaml(self, tool: Dict) -> str:
|
||||
params_block = "\n".join(
|
||||
PARAM_TEMPLATE.format(
|
||||
name=param["name"],
|
||||
type=param["type"],
|
||||
required=param["required"],
|
||||
label_en=param["describe_en"],
|
||||
label_zh=param["describe"],
|
||||
desc_en=param["describe_en"],
|
||||
desc_zh=param["describe"],
|
||||
llm_desc=param["describe_en"],
|
||||
form="llm"
|
||||
) for param in tool["params"]
|
||||
)
|
||||
|
||||
return YAML_TEMPLATE.format(
|
||||
tool_id=tool["describe"],
|
||||
en_label=tool["describe_en"],
|
||||
zh_label=tool["describe"],
|
||||
en_desc=tool["describe_en"],
|
||||
zh_desc=tool["describe"],
|
||||
llm_desc=tool["describe_en"],
|
||||
disp_type="metric",
|
||||
disp_title=tool["title"],
|
||||
disp_unit=tool["unit"],
|
||||
params_block=params_block
|
||||
)
|
||||
|
||||
def _build_python(self, tool: Dict) -> str:
|
||||
param_lines = "\n".join(
|
||||
f" {param['name']} = tool_parameters.get('{param['name']}')"
|
||||
for param in tool["params"]
|
||||
)
|
||||
|
||||
param_dict = []
|
||||
for param in tool["params"]:
|
||||
name = param.get("name", "").strip()
|
||||
if not name:
|
||||
continue
|
||||
|
||||
if param.get("required", False):
|
||||
param_dict.append(f" '{name}': {name}")
|
||||
else:
|
||||
param_dict.append(f" **({{'{name}': {name}}} if {name} else {{}})")
|
||||
|
||||
params_content = ",\n".join(param_dict)
|
||||
|
||||
return PYTHON_TEMPLATE.format(
|
||||
class_name=self._generate_class_name(tool["metric_name"]),
|
||||
param_lines=param_lines,
|
||||
param_dict=params_content,
|
||||
metric_name=tool["title"]
|
||||
)
|
||||
|
||||
def _generate_class_name(self, metric_name: str) -> str:
|
||||
cleaned = ''.join(c for c in metric_name.title() if c.isalnum())
|
||||
return f"{cleaned}Tool"
|
||||
@ -0,0 +1,101 @@
|
||||
YAML_TEMPLATE = """identity:
|
||||
name: {tool_id}
|
||||
author: APO
|
||||
label:
|
||||
en_US: {en_label}
|
||||
zh_Hans: {zh_label}
|
||||
description:
|
||||
human:
|
||||
en_US: {en_desc}
|
||||
zh_Hans: {zh_desc}
|
||||
llm: {llm_desc}
|
||||
display:
|
||||
type: {disp_type}
|
||||
title: {disp_title}
|
||||
unit: "{disp_unit}"
|
||||
parameters:
|
||||
{params_block}
|
||||
- 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
|
||||
"""
|
||||
|
||||
PARAM_TEMPLATE = """ - name: {name}
|
||||
type: {type}
|
||||
required: {required}
|
||||
label:
|
||||
en_US: {label_en}
|
||||
zh_Hans: {label_zh}
|
||||
human_description:
|
||||
en_US: {desc_en}
|
||||
zh_Hans: {desc_zh}
|
||||
llm_description: {llm_desc}
|
||||
form: {form}"""
|
||||
|
||||
PYTHON_TEMPLATE = """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 {class_name}(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]:
|
||||
{param_lines}
|
||||
start_time = tool_parameters.get("startTime")
|
||||
end_time = tool_parameters.get("endTime")
|
||||
params = {{
|
||||
'metricName': {metric_name!r},
|
||||
'params': {{
|
||||
{param_dict}
|
||||
}},
|
||||
'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,10 @@
|
||||
from generator import ToolGenerator
|
||||
|
||||
import os
|
||||
|
||||
def main():
|
||||
generator = ToolGenerator('./metrics_conf.json')
|
||||
generator.generate_all("../core/tools/builtin_tool/providers/apo_select/tools")
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Loading…
Reference in New Issue