NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2026-02-04 13:56:24 -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/cli/fast_api.py
+10 -10
View File
@@ -503,7 +503,7 @@ def get_fast_api_app(
"/apps/{app_name}/users/{user_id}/sessions/{session_id}/artifacts/{artifact_name}",
response_model_exclude_none=True,
)
def load_artifact(
async def load_artifact(
app_name: str,
user_id: str,
session_id: str,
@@ -511,7 +511,7 @@ def get_fast_api_app(
version: Optional[int] = Query(None),
) -> Optional[types.Part]:
app_name = agent_engine_id if agent_engine_id else app_name
artifact = artifact_service.load_artifact(
artifact = await artifact_service.load_artifact(
app_name=app_name,
user_id=user_id,
session_id=session_id,
@@ -526,7 +526,7 @@ def get_fast_api_app(
"/apps/{app_name}/users/{user_id}/sessions/{session_id}/artifacts/{artifact_name}/versions/{version_id}",
response_model_exclude_none=True,
)
def load_artifact_version(
async def load_artifact_version(
app_name: str,
user_id: str,
session_id: str,
@@ -534,7 +534,7 @@ def get_fast_api_app(
version_id: int,
) -> Optional[types.Part]:
app_name = agent_engine_id if agent_engine_id else app_name
artifact = artifact_service.load_artifact(
artifact = await artifact_service.load_artifact(
app_name=app_name,
user_id=user_id,
session_id=session_id,
@@ -549,11 +549,11 @@ def get_fast_api_app(
"/apps/{app_name}/users/{user_id}/sessions/{session_id}/artifacts",
response_model_exclude_none=True,
)
def list_artifact_names(
async def list_artifact_names(
app_name: str, user_id: str, session_id: str
) -> list[str]:
app_name = agent_engine_id if agent_engine_id else app_name
return artifact_service.list_artifact_keys(
return await artifact_service.list_artifact_keys(
app_name=app_name, user_id=user_id, session_id=session_id
)
@@ -561,11 +561,11 @@ def get_fast_api_app(
"/apps/{app_name}/users/{user_id}/sessions/{session_id}/artifacts/{artifact_name}/versions",
response_model_exclude_none=True,
)
def list_artifact_versions(
async def list_artifact_versions(
app_name: str, user_id: str, session_id: str, artifact_name: str
) -> list[int]:
app_name = agent_engine_id if agent_engine_id else app_name
return artifact_service.list_versions(
return await artifact_service.list_versions(
app_name=app_name,
user_id=user_id,
session_id=session_id,
@@ -575,11 +575,11 @@ def get_fast_api_app(
@app.delete(
"/apps/{app_name}/users/{user_id}/sessions/{session_id}/artifacts/{artifact_name}",
)
def delete_artifact(
async def delete_artifact(
app_name: str, user_id: str, session_id: str, artifact_name: str
):
app_name = agent_engine_id if agent_engine_id else app_name
artifact_service.delete_artifact(
await artifact_service.delete_artifact(
app_name=app_name,
user_id=user_id,
session_id=session_id,