github/genai-toolbox
1
0
Fork 0
mirror of https://github.com/googleapis/genai-toolbox.git synced 2026-02-18 11:02:26 -05:00
Code Issues Packages Projects Releases 14 Wiki Activity

update method names

Browse Source
This commit is contained in:
Twisha Bansal
2026-02-18 12:56:59 +05:30
parent acf56382d7
commit b38d36aeef
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
docs/en/samples/pre_post_processing/python/adk/agent.py
View File

@@ -23,7 +23,7 @@ SYSTEM_PROMPT = """
# Pre processing
async def before_tool_callback(
async def enfore_business_rules(
tool: ToolboxTool, args: Dict[str, Any], tool_context: ToolContext
) -> Optional[Dict[str, Any]]:
"""
@@ -45,7 +45,7 @@ async def before_tool_callback(
# Post processing
async def after_tool_callback(
async def enrich_response(
tool: ToolboxTool,
args: Dict[str, Any],
tool_context: ToolContext,
@@ -109,8 +109,8 @@ async def main():
instruction=SYSTEM_PROMPT,
tools=tools,
# add any pre and post processing callbacks
before_tool_callback=before_tool_callback,
after_tool_callback=after_tool_callback,
before_tool_callback=enfore_business_rules,
after_tool_callback=enrich_response,
)
app = App(root_agent=root_agent, name="my_agent")
runner = Runner(app=app, session_service=InMemorySessionService())