fix agents to run custom tools

This commit is contained in:
Dinesh Yeduguru 2024-12-20 22:02:00 -08:00
parent 9192a9bbb4
commit 2ad67529ef
4 changed files with 6 additions and 14 deletions

View file

@ -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(

View file

@ -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()))