mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-24 07:28:03 +00:00
fix agents to run custom tools
This commit is contained in:
parent
9192a9bbb4
commit
2ad67529ef
4 changed files with 6 additions and 14 deletions
|
|
@ -531,11 +531,6 @@ class ChatAgent(ShieldRunnerMixin):
|
|||
log.info(f"{str(message)}")
|
||||
tool_call = message.tool_calls[0]
|
||||
|
||||
name = tool_call.tool_name
|
||||
if not isinstance(name, BuiltinTool) or name not in enabled_tools:
|
||||
yield message
|
||||
return
|
||||
|
||||
step_id = str(uuid.uuid4())
|
||||
yield AgentTurnResponseStreamChunk(
|
||||
event=AgentTurnResponseEvent(
|
||||
|
|
@ -691,10 +686,8 @@ async def execute_tool_call_maybe(
|
|||
|
||||
tool_call = message.tool_calls[0]
|
||||
name = tool_call.tool_name
|
||||
assert isinstance(name, BuiltinTool)
|
||||
|
||||
name = name.value
|
||||
|
||||
if isinstance(name, BuiltinTool):
|
||||
name = name.value
|
||||
result = await tool_runtime_api.invoke_tool(
|
||||
tool_name=name,
|
||||
args=dict(
|
||||
|
|
|
|||
|
|
@ -54,7 +54,6 @@ class TavilySearchToolRuntimeImpl(
|
|||
"https://api.tavily.com/search",
|
||||
json={"api_key": api_key, "query": args["query"]},
|
||||
)
|
||||
print(f"================= Tavily response: {response.json()}")
|
||||
|
||||
return ToolInvocationResult(
|
||||
content=json.dumps(self._clean_tavily_response(response.json()))
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue