add query thread polaris tools
parent
9332ebfe05
commit
3187c887b8
@ -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 ThreadPolarisEpollTimeTool(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]:
|
||||
pod = tool_parameters.get('pod', '.*')
|
||||
start_time = tool_parameters.get("startTime")
|
||||
end_time = tool_parameters.get("endTime")
|
||||
params = {
|
||||
'metricName': 'Thread Polaris Metrics - 北极星指标(线程) - 各类型耗时折线图 - Epoll',
|
||||
'params': {
|
||||
'pod': pod
|
||||
},
|
||||
'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 ThreadPolarisFileTimeTool(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]:
|
||||
pod = tool_parameters.get('pod', '.*')
|
||||
start_time = tool_parameters.get("startTime")
|
||||
end_time = tool_parameters.get("endTime")
|
||||
params = {
|
||||
'metricName': 'Thread Polaris Metrics - 北极星指标(线程) - 各类型耗时折线图 - File',
|
||||
'params': {
|
||||
'pod': pod
|
||||
},
|
||||
'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 ThreadPolarisFutexTimeTool(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]:
|
||||
pod = tool_parameters.get('pod', '.*')
|
||||
start_time = tool_parameters.get("startTime")
|
||||
end_time = tool_parameters.get("endTime")
|
||||
params = {
|
||||
'metricName': 'Thread Polaris Metrics - 北极星指标(线程) - 各类型耗时折线图 - Futex',
|
||||
'params': {
|
||||
'pod': pod
|
||||
},
|
||||
'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 ThreadPolarisIdleTimeTool(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]:
|
||||
pod = tool_parameters.get('pod', '.*')
|
||||
start_time = tool_parameters.get("startTime")
|
||||
end_time = tool_parameters.get("endTime")
|
||||
params = {
|
||||
'metricName': 'Thread Polaris Metrics - 北极星指标(线程) - 各类型耗时折线图 - Idle',
|
||||
'params': {
|
||||
'pod': pod
|
||||
},
|
||||
'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 ThreadPolarisNetTimeTool(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]:
|
||||
pod = tool_parameters.get('pod', '.*')
|
||||
start_time = tool_parameters.get("startTime")
|
||||
end_time = tool_parameters.get("endTime")
|
||||
params = {
|
||||
'metricName': 'Thread Polaris Metrics - 北极星指标(线程) - 各类型耗时折线图 - Net',
|
||||
'params': {
|
||||
'pod': pod
|
||||
},
|
||||
'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 ThreadPolarisOncpuTimeTool(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]:
|
||||
pod = tool_parameters.get('pod', '.*')
|
||||
start_time = tool_parameters.get("startTime")
|
||||
end_time = tool_parameters.get("endTime")
|
||||
params = {
|
||||
'metricName': 'Thread Polaris Metrics - 北极星指标(线程) - 各类型耗时折线图 - OnCPU',
|
||||
'params': {
|
||||
'pod': pod
|
||||
},
|
||||
'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 ThreadPolarisOtherTimeTool(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]:
|
||||
pod = tool_parameters.get('pod', '.*')
|
||||
start_time = tool_parameters.get("startTime")
|
||||
end_time = tool_parameters.get("endTime")
|
||||
params = {
|
||||
'metricName': 'Thread Polaris Metrics - 北极星指标(线程) - 各类型耗时折线图 - Other',
|
||||
'params': {
|
||||
'pod': pod
|
||||
},
|
||||
'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 ThreadPolarisRunqueueTimeTool(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]:
|
||||
pod = tool_parameters.get('pod', '.*')
|
||||
start_time = tool_parameters.get("startTime")
|
||||
end_time = tool_parameters.get("endTime")
|
||||
params = {
|
||||
'metricName': 'Thread Polaris Metrics - 北极星指标(线程) - 各类型耗时折线图 - Runqueue',
|
||||
'params': {
|
||||
'pod': pod
|
||||
},
|
||||
'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