From 05d9d5608d865633f86f6674ad74d8054c028362 Mon Sep 17 00:00:00 2001 From: Alyssa Coghlan Date: Mon, 3 Mar 2025 22:00:32 +1000 Subject: [PATCH] Simplify inferred tool names Closes #22 --- src/lmstudio/json_api.py | 2 +- tests/test_inference.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/lmstudio/json_api.py b/src/lmstudio/json_api.py index e438323..448a8a2 100644 --- a/src/lmstudio/json_api.py +++ b/src/lmstudio/json_api.py @@ -1030,7 +1030,7 @@ def from_callable( """Derive a tool function definition from the given callable.""" if name is None: try: - name = f"{f.__module__}.{f.__qualname__}" + name = f.__name__ except Exception as exc: raise LMStudioValueError( f"Could not extract tool name from {f!r}" diff --git a/tests/test_inference.py b/tests/test_inference.py index 445e7b5..c0544a8 100644 --- a/tests/test_inference.py +++ b/tests/test_inference.py @@ -113,7 +113,7 @@ def log_adding_two_integers(a: int, b: int) -> int: def test_tool_def_from_callable() -> None: default_def = ToolFunctionDef.from_callable(log_adding_two_integers) assert default_def == ToolFunctionDef( - name=f"{__name__}.{log_adding_two_integers.__name__}", + name=log_adding_two_integers.__name__, description="Log adding two integers together.", parameters=ADDITION_TOOL_SPEC["parameters"], implementation=log_adding_two_integers, @@ -132,7 +132,7 @@ def test_parse_tools() -> None: expected_implementations = { "add": log_adding_two_integers, "add_as_tool_def": log_adding_two_integers, - "tests.test_inference.log_adding_two_integers": log_adding_two_integers, + "log_adding_two_integers": log_adding_two_integers, } expected_names = list(expected_implementations.keys()) expected_param_schemas = 3 * [