mirror of
https://github.com/imartinez/privateGPT.git
synced 2026-07-17 01:48:03 +00:00
15 lines
561 B
Python
15 lines
561 B
Python
from private_gpt.components.chat.models.chat_config_models import ToolSpec
|
|
from private_gpt.components.tools.tool_names import (
|
|
WEB_FETCH_TOOL_NAME,
|
|
resolve_internal_tool_name,
|
|
)
|
|
|
|
|
|
def test_resolve_internal_tool_name_accepts_web_extract_legacy_alias() -> None:
|
|
assert resolve_internal_tool_name("web_extract_v1") == WEB_FETCH_TOOL_NAME
|
|
|
|
|
|
def test_tool_spec_original_name_canonicalizes_web_extract_legacy_alias() -> None:
|
|
tool = ToolSpec(name="web_extract", type="web_extract_v1")
|
|
assert tool.get_original_tool_name() == WEB_FETCH_TOOL_NAME
|