apunkt/rowboat
1
0
Fork 0
mirror of https://github.com/rowboatlabs/rowboat.git synced 2026-05-16 18:25:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

added web_search tool to agents

Browse source
This commit is contained in:
arkml 2025-04-14 13:08:45 +05:30 • committed by Ramnique Singh
parent aad14804ea
commit 41c3f2cfc6
1 changed files with 10 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

17
apps/rowboat_agents/src/graph/swarm_wrapper.py
View file

@ -13,7 +13,7 @@ from .helpers.instructions import (
add_rag_instructions_to_agent add_rag_instructions_to_agent
) )
from agents import Agent as NewAgent, Runner, FunctionTool, RunContextWrapper, ModelSettings from agents import Agent as NewAgent, Runner, FunctionTool, RunContextWrapper, ModelSettings, WebSearchTool
# Add import for OpenAI functionality # Add import for OpenAI functionality
from src.utils.common import common_logger as logger, generate_openai_output from src.utils.common import common_logger as logger, generate_openai_output
from typing import Any from typing import Any
@ -221,14 +221,17 @@ def get_agents(agent_configs, tool_configs, complete_request):
"type": "function", "type": "function",
"function": tool_config "function": tool_config
}) })
tool = FunctionTool( if tool_name == "web_search":
name=tool_name, tool = WebSearchTool()
description=tool_config["description"], else:
params_json_schema=tool_config["parameters"], tool = FunctionTool(
strict_json_schema=False, name=tool_name,
description=tool_config["description"],
params_json_schema=tool_config["parameters"],
strict_json_schema=False,
on_invoke_tool=lambda ctx, args, _tool_name=tool_name, _tool_config=tool_config, _complete_request=complete_request: on_invoke_tool=lambda ctx, args, _tool_name=tool_name, _tool_config=tool_config, _complete_request=complete_request:
catch_all(ctx, args, _tool_name, _tool_config, _complete_request) catch_all(ctx, args, _tool_name, _tool_config, _complete_request)
) )
new_tools.append(tool) new_tools.append(tool)
logger.debug(f"Added tool {tool_name} to agent {agent_config['name']}") logger.debug(f"Added tool {tool_name} to agent {agent_config['name']}")
print(f"Added tool {tool_name} to agent {agent_config['name']}") print(f"Added tool {tool_name} to agent {agent_config['name']}")