ADK changes

PiperOrigin-RevId: 755201925
This commit is contained in:
Shangjie Chen
2025-05-05 21:57:51 -07:00
committed by Copybara-Service
parent 6dec235c13
commit 905c20dad6
12 changed files with 86 additions and 69 deletions

View File

@@ -186,7 +186,7 @@ class Runner:
root_agent = self.agent
if new_message:
self._append_new_message_to_session(
await self._append_new_message_to_session(
session,
new_message,
invocation_context,
@@ -199,7 +199,7 @@ class Runner:
self.session_service.append_event(session=session, event=event)
yield event
def _append_new_message_to_session(
async def _append_new_message_to_session(
self,
session: Session,
new_message: types.Content,
@@ -225,7 +225,7 @@ class Runner:
if part.inline_data is None:
continue
file_name = f'artifact_{invocation_context.invocation_id}_{i}'
self.artifact_service.save_artifact(
await self.artifact_service.save_artifact(
app_name=self.app_name,
user_id=session.user_id,
session_id=session.id,