NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2026-02-04 22:06:23 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

ADK changes

Browse Source 
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
Download Patch File Download Diff File Expand all files Collapse all files
src/google/adk/agents/callback_context.py
+4 -4
View File
@@ -65,7 +65,7 @@ class CallbackContext(ReadonlyContext):
"""The user content that started this invocation. READONLY field."""
return self._invocation_context.user_content
def load_artifact(
async def load_artifact(
self, filename: str, version: Optional[int] = None
) -> Optional[types.Part]:
"""Loads an artifact attached to the current session.
@@ -80,7 +80,7 @@ class CallbackContext(ReadonlyContext):
"""
if self._invocation_context.artifact_service is None:
raise ValueError("Artifact service is not initialized.")
return self._invocation_context.artifact_service.load_artifact(
return await self._invocation_context.artifact_service.load_artifact(
app_name=self._invocation_context.app_name,
user_id=self._invocation_context.user_id,
session_id=self._invocation_context.session.id,
@@ -88,7 +88,7 @@ class CallbackContext(ReadonlyContext):
version=version,
)
def save_artifact(self, filename: str, artifact: types.Part) -> int:
async def save_artifact(self, filename: str, artifact: types.Part) -> int:
"""Saves an artifact and records it as delta for the current session.
Args:
@@ -100,7 +100,7 @@ class CallbackContext(ReadonlyContext):
"""
if self._invocation_context.artifact_service is None:
raise ValueError("Artifact service is not initialized.")
version = self._invocation_context.artifact_service.save_artifact(
version = await self._invocation_context.artifact_service.save_artifact(
app_name=self._invocation_context.app_name,
user_id=self._invocation_context.user_id,
session_id=self._invocation_context.session.id,