80 lines
2.4 KiB
Python
80 lines
2.4 KiB
Python
|
"""工具表：从 tools/list 动态构建，新增工具无需改代码。"""
|
||
|
|
|
||
|
|
from .mcp_client import McpClient
|
||
|
|
|
||
|
|
# 只读工具白名单：审查阶段的受控子循环只允许调用这些（不做任何写库操作）
|
||
|
|
READ_ONLY_TOOLS = frozenset(
|
||
|
|
{
|
||
|
|
"query_entities",
|
||
|
|
"get_text_content",
|
||
|
|
"get_dimensions",
|
||
|
|
"get_block_references",
|
||
|
|
"get_polylines_info",
|
||
|
|
"get_entity_detail",
|
||
|
|
"get_layers",
|
||
|
|
"get_layouts",
|
||
|
|
"get_drawing_info",
|
||
|
|
"detect_drawing_frames",
|
||
|
|
"find_duplicate_entities",
|
||
|
|
"find_zero_length",
|
||
|
|
"zoom_to_window",
|
||
|
|
"zoom_to_entity",
|
||
|
|
"get_viewport_screenshot",
|
||
|
|
}
|
||
|
|
)
|
||
|
|
|
||
|
|
# 写类工具前缀/名称：调用时不重试，且默认禁止 LLM 在审查阶段触发
|
||
|
|
_WRITE_PREFIXES = ("add_", "delete_", "draw_", "set_")
|
||
|
|
_WRITE_EXACT = {"set_entity_properties", "highlight_entities"}
|
||
|
|
|
||
|
|
|
||
|
|
def is_write_tool(name: str) -> bool:
|
||
|
|
return name.startswith(_WRITE_PREFIXES) or name in _WRITE_EXACT
|
||
|
|
|
||
|
|
|
||
|
|
class CadTools:
|
||
|
|
"""工具注册表 + 调用代理。"""
|
||
|
|
|
||
|
|
def __init__(self, client=None):
|
||
|
|
self.client = client or McpClient()
|
||
|
|
self._by_name = {}
|
||
|
|
self.refresh()
|
||
|
|
|
||
|
|
def refresh(self):
|
||
|
|
tools = self.client.list_tools(use_cache=False)
|
||
|
|
self._by_name = {t["name"]: t for t in tools}
|
||
|
|
return tools
|
||
|
|
|
||
|
|
@property
|
||
|
|
def names(self):
|
||
|
|
return sorted(self._by_name)
|
||
|
|
|
||
|
|
def has(self, name):
|
||
|
|
return name in self._by_name
|
||
|
|
|
||
|
|
def describe(self, name):
|
||
|
|
return self._by_name.get(name)
|
||
|
|
|
||
|
|
def openai_tools(self, names=None):
|
||
|
|
"""转换为 OpenAI function 工具格式；names 为可选的名称白名单集合。"""
|
||
|
|
out = []
|
||
|
|
for name, tool in self._by_name.items():
|
||
|
|
if names is not None and name not in names:
|
||
|
|
continue
|
||
|
|
out.append(
|
||
|
|
{
|
||
|
|
"type": "function",
|
||
|
|
"function": {
|
||
|
|
"name": name,
|
||
|
|
"description": tool.get("description", ""),
|
||
|
|
"parameters": tool.get("inputSchema")
|
||
|
|
or {"type": "object", "properties": {}},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
)
|
||
|
|
return out
|
||
|
|
|
||
|
|
def call(self, name, arguments=None, *, is_write=None):
|
||
|
|
if is_write is None:
|
||
|
|
is_write = is_write_tool(name)
|
||
|
|
return self.client.call_tool(name, arguments, is_write=is_write)
|