Skip to content

Commit e2d543d

Browse files
committed
fix(tool): revert the ToolDefParam stuff because we reverted the bad commit
1 parent 2684579 commit e2d543d

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/llama_stack_client/lib/agents/client_tool.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
)
2121

2222
from llama_stack_client.types import CompletionMessage, Message, ToolResponse
23-
from llama_stack_client.types.tool import Parameter, Tool
23+
from llama_stack_client.types.tool_def_param import Parameter, ToolDefParam
2424

2525

2626
class ClientTool:
@@ -61,8 +61,8 @@ def parameters_for_system_prompt(self) -> str:
6161
}
6262
)
6363

64-
def get_tool_definition(self) -> Tool:
65-
return Tool(
64+
def get_tool_definition(self) -> ToolDefParam:
65+
return ToolDefParam(
6666
name=self.get_name(),
6767
description=self.get_description(),
6868
parameters=list(self.get_params_definition().values()),

0 commit comments

Comments
 (0)