phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-09 21:18:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix agentic_system utils

Browse source
This commit is contained in:
Ashwin Bharambe 2024-08-24 22:56:43 -07:00
parent 8efe614719
commit 830252257b
2 changed files with 11 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

6
llama_toolchain/tools/custom/datatypes.py
View file

@ -54,9 +54,9 @@ class CustomTool:
}
)
def get_tool_definition(self) -> AgenticSystemToolDefinition:
return AgenticSystemToolDefinition(
tool_name=self.get_name(),
def get_tool_definition(self) -> FunctionCallToolDefinition:
return FunctionCallToolDefinition(
function_name=self.get_name(),
description=self.get_description(),
parameters=self.get_params_definition(),
)