NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2025-07-14 01:41: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')
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]]
"""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):
@ -101,8 +84,9 @@ class BaseAgent(BaseModel):
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.
Optional[types.Content]: The content to return to the 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
"""Callback signature that is invoked after the agent run.
@ -111,8 +95,9 @@ class BaseAgent(BaseModel):
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.
Optional[types.Content]: The content to return to the user.
When the content is present, the provided content will be used as agent
response and appended to event history as agent response.
"""
@final