You've already forked adk-python
mirror of
https://github.com/encounter/adk-python.git
synced 2026-03-30 10:57:20 -07:00
Liang Wu
43f302ce1a
Also expand agent folder name to full spelling, e.g. ma --> multi_agent PiperOrigin-RevId: 795926902
80 lines
1.8 KiB
Python
80 lines
1.8 KiB
Python
from google.genai import types
|
|
|
|
|
|
async def before_agent_callback(callback_context):
|
|
print('@before_agent_callback')
|
|
return None
|
|
|
|
|
|
async def after_agent_callback(callback_context):
|
|
print('@after_agent_callback')
|
|
return None
|
|
|
|
|
|
async def before_model_callback(callback_context, llm_request):
|
|
print('@before_model_callback')
|
|
return None
|
|
|
|
|
|
async def after_model_callback(callback_context, llm_response):
|
|
print('@after_model_callback')
|
|
return None
|
|
|
|
|
|
def after_agent_callback1(callback_context):
|
|
print('@after_agent_callback1')
|
|
|
|
|
|
def after_agent_callback2(callback_context):
|
|
print('@after_agent_callback2')
|
|
# ModelContent (or Content with role set to 'model') must be returned.
|
|
# Otherwise, the event will be excluded from the context in the next turn.
|
|
return types.ModelContent(
|
|
parts=[
|
|
types.Part(
|
|
text='(stopped) after_agent_callback2',
|
|
),
|
|
],
|
|
)
|
|
|
|
|
|
def after_agent_callback3(callback_context):
|
|
print('@after_agent_callback3')
|
|
|
|
|
|
def before_agent_callback1(callback_context):
|
|
print('@before_agent_callback1')
|
|
|
|
|
|
def before_agent_callback2(callback_context):
|
|
print('@before_agent_callback2')
|
|
|
|
|
|
def before_agent_callback3(callback_context):
|
|
print('@before_agent_callback3')
|
|
|
|
|
|
def before_tool_callback1(tool, args, tool_context):
|
|
print('@before_tool_callback1')
|
|
|
|
|
|
def before_tool_callback2(tool, args, tool_context):
|
|
print('@before_tool_callback2')
|
|
|
|
|
|
def before_tool_callback3(tool, args, tool_context):
|
|
print('@before_tool_callback3')
|
|
|
|
|
|
def after_tool_callback1(tool, args, tool_context, tool_response):
|
|
print('@after_tool_callback1')
|
|
|
|
|
|
def after_tool_callback2(tool, args, tool_context, tool_response):
|
|
print('@after_tool_callback2')
|
|
return {'test': 'after_tool_callback2', 'response': tool_response}
|
|
|
|
|
|
def after_tool_callback3(tool, args, tool_context, tool_response):
|
|
print('@after_tool_callback3')
|