encounter/adk-python
0
0
Fork 0
mirror of https://github.com/encounter/adk-python.git synced 2026-07-09 18:19:28 -07:00
Code Issues Packages Projects Releases Wiki Activity

fix: Internal change

Browse Source 
PiperOrigin-RevId: 820159440
This commit is contained in:
Alexey Guseynov
2025-10-16 13:07:00 -07:00
committed by Copybara-Service
parent f51380f9ea
commit 1e1d63f34c
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/google/adk/agents/base_agent.py
+6 -6
View File
@@ -286,7 +286,7 @@ class BaseAgent(BaseModel):
with tracer.start_as_current_span(f'invoke_agent {self.name}') as span:
ctx = self._create_invocation_context(parent_context)
tracing.trace_agent_invocation(span, self, ctx)
if event := await self.__handle_before_agent_callback(ctx):
if event := await self._handle_before_agent_callback(ctx):
yield event
if ctx.end_invocation:
return
@@ -298,7 +298,7 @@ class BaseAgent(BaseModel):
if ctx.end_invocation:
return
if event := await self.__handle_after_agent_callback(ctx):
if event := await self._handle_after_agent_callback(ctx):
yield event
async with Aclosing(_run_with_trace()) as agen:
@@ -324,7 +324,7 @@ class BaseAgent(BaseModel):
with tracer.start_as_current_span(f'invoke_agent {self.name}') as span:
ctx = self._create_invocation_context(parent_context)
tracing.trace_agent_invocation(span, self, ctx)
if event := await self.__handle_before_agent_callback(ctx):
if event := await self._handle_before_agent_callback(ctx):
yield event
if ctx.end_invocation:
return
@@ -333,7 +333,7 @@ class BaseAgent(BaseModel):
async for event in agen:
yield event
if event := await self.__handle_after_agent_callback(ctx):
if event := await self._handle_after_agent_callback(ctx):
yield event
async with Aclosing(_run_with_trace()) as agen:
@@ -438,7 +438,7 @@ class BaseAgent(BaseModel):
return self.after_agent_callback
return [self.after_agent_callback]
async def __handle_before_agent_callback(
async def _handle_before_agent_callback(
self, ctx: InvocationContext
) -> Optional[Event]:
"""Runs the before_agent_callback if it exists.
@@ -496,7 +496,7 @@ class BaseAgent(BaseModel):
return None
async def __handle_after_agent_callback(
async def _handle_after_agent_callback(
self, invocation_context: InvocationContext
) -> Optional[Event]:
"""Runs the after_agent_callback if it exists.