NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2025-07-14 09:51:25 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update description of before/after_agent_callback to clarify the behavior.

Browse Source 
PiperOrigin-RevId: 754207459
This commit is contained in:
Wei Sun (Jack) 2025-05-02 18:57:15 -07:00 committed by Copybara-Service
parent 60d23c028f
commit 6595cda037
1 changed files with 6 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
src/google/adk/agents/base_agent.py
View File

@ -38,26 +38,9 @@ if TYPE_CHECKING:
tracer = trace.get_tracer('gcp.vertex.agent') tracer = trace.get_tracer('gcp.vertex.agent')
BeforeAgentCallback = Callable[[CallbackContext], Optional[types.Content]] BeforeAgentCallback = Callable[[CallbackContext], Optional[types.Content]]
"""Callback signature that is invoked before the agent run.
Args:
callback_context: MUST be named 'callback_context' (enforced).
Returns:
The content to return to the user. When set, the agent run will be skipped and
the provided content will be returned to user.
"""
AfterAgentCallback = Callable[[CallbackContext], Optional[types.Content]] AfterAgentCallback = Callable[[CallbackContext], Optional[types.Content]]
"""Callback signature that is invoked after the agent run.
Args:
callback_context: MUST be named 'callback_context' (enforced).
Returns:
The content to return to the user. When set, the provided content will be
appended to event history as agent response.
"""
class BaseAgent(BaseModel): class BaseAgent(BaseModel):
@ -101,8 +84,9 @@ class BaseAgent(BaseModel):
callback_context: MUST be named 'callback_context' (enforced). callback_context: MUST be named 'callback_context' (enforced).
Returns: Returns:
The content to return to the user. When set, the agent run will be skipped Optional[types.Content]: The content to return to the user.
and the provided content will be returned to user. When the content is present, the agent run will be skipped and the
provided content will be returned to user.
""" """
after_agent_callback: Optional[AfterAgentCallback] = None after_agent_callback: Optional[AfterAgentCallback] = None
"""Callback signature that is invoked after the agent run. """Callback signature that is invoked after the agent run.
@ -111,8 +95,9 @@ class BaseAgent(BaseModel):
callback_context: MUST be named 'callback_context' (enforced). callback_context: MUST be named 'callback_context' (enforced).
Returns: Returns:
The content to return to the user. When set, the provided content will be Optional[types.Content]: The content to return to the user.
appended to event history as agent response. When the content is present, the provided content will be used as agent
response and appended to event history as agent response.
""" """
@final @final