Skip to content

Commit 200fa67

Browse files
authored
Simplify inferred tool names (#27)
Closes #22
1 parent f529c49 commit 200fa67

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

src/lmstudio/json_api.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1030,7 +1030,7 @@ def from_callable(
10301030
"""Derive a tool function definition from the given callable."""
10311031
if name is None:
10321032
try:
1033-
name = f"{f.__module__}.{f.__qualname__}"
1033+
name = f.__name__
10341034
except Exception as exc:
10351035
raise LMStudioValueError(
10361036
f"Could not extract tool name from {f!r}"

tests/test_inference.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,7 @@ def log_adding_two_integers(a: int, b: int) -> int:
113113
def test_tool_def_from_callable() -> None:
114114
default_def = ToolFunctionDef.from_callable(log_adding_two_integers)
115115
assert default_def == ToolFunctionDef(
116-
name=f"{__name__}.{log_adding_two_integers.__name__}",
116+
name=log_adding_two_integers.__name__,
117117
description="Log adding two integers together.",
118118
parameters=ADDITION_TOOL_SPEC["parameters"],
119119
implementation=log_adding_two_integers,
@@ -132,7 +132,7 @@ def test_parse_tools() -> None:
132132
expected_implementations = {
133133
"add": log_adding_two_integers,
134134
"add_as_tool_def": log_adding_two_integers,
135-
"tests.test_inference.log_adding_two_integers": log_adding_two_integers,
135+
"log_adding_two_integers": log_adding_two_integers,
136136
}
137137
expected_names = list(expected_implementations.keys())
138138
expected_param_schemas = 3 * [

0 commit comments

Comments
 (0)