mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-10-28 19:06:36 +08:00
57 lines
2.2 KiB
Python
57 lines
2.2 KiB
Python
"""
|
||
验证工具存根的文档转义不会改变参数传递或注入可执行源码。
|
||
"""
|
||
|
||
import ast
|
||
import inspect
|
||
|
||
import pytest
|
||
|
||
from app.adapters.system.code.client import build_client
|
||
|
||
|
||
@pytest.mark.parametrize("description", [
|
||
'普通中文说明',
|
||
'包含三引号 """ 和反斜杠 \\ 的说明',
|
||
'首行\n """\nraise RuntimeError("不得执行")\n# 末行',
|
||
'包含回车\r、制表符\t和空字符\x00的说明',
|
||
])
|
||
def test_schema_documentation_cannot_change_generated_tool_behavior(description):
|
||
"""
|
||
不可信 Schema 文本只能成为文档，必填参数与可选参数的省略语义保持不变。
|
||
"""
|
||
schema = {
|
||
"type": "object",
|
||
"description": description,
|
||
"properties": {
|
||
"value": {"type": "string", "description": description},
|
||
"optional": {"type": ["string", "null"]},
|
||
},
|
||
"required": ["value"],
|
||
}
|
||
source = build_client(("example",), {"example": schema})
|
||
namespace = {}
|
||
exec(compile(source, "<generated-client>", "exec"), namespace)
|
||
namespace["_call"] = lambda name, arguments: (name, arguments)
|
||
tool = namespace["example"]
|
||
assert tool("输入") == ("example", {"value": "输入"})
|
||
assert tool("输入", optional=None) == ("example", {"value": "输入", "optional": None})
|
||
assert inspect.signature(tool).parameters["value"].default is inspect.Parameter.empty
|
||
assert "输入 Schema：" in tool.__doc__
|
||
node = next(item for item in ast.parse(source).body if isinstance(item, ast.FunctionDef) and item.name == "example")
|
||
literal = ast.get_source_segment(source, node.body[0])
|
||
assert literal.startswith('"""\n')
|
||
assert literal.endswith('\n """')
|
||
|
||
|
||
def test_untyped_tool_forwards_named_arguments_with_chinese_documentation():
|
||
"""
|
||
没有 Schema 时仍转发全部命名参数，生成文档遵守中文多行格式。
|
||
"""
|
||
source = build_client(("example",))
|
||
namespace = {}
|
||
exec(compile(source, "<generated-client>", "exec"), namespace)
|
||
namespace["_call"] = lambda name, arguments: (name, arguments)
|
||
assert namespace["example"](value=None) == ("example", {"value": None})
|
||
assert ":param kwargs:" in namespace["example"].__doc__
|