chore: Make all FastAPI endpoints async

PiperOrigin-RevId: 792886431
This commit is contained in:
Shangjie Chen
2025-08-08 21:58:34 -07:00
committed by Copybara-Service
parent 8f937b5175
commit 7f12387eb1
2 changed files with 14 additions and 14 deletions
+12 -12
View File
@@ -352,18 +352,18 @@ class AdkWebServer:
)
@app.get("/list-apps")
def list_apps() -> list[str]:
async def list_apps() -> list[str]:
return self.agent_loader.list_agents()
@app.get("/debug/trace/{event_id}", tags=[TAG_DEBUG])
def get_trace_dict(event_id: str) -> Any:
async def get_trace_dict(event_id: str) -> Any:
event_dict = trace_dict.get(event_id, None)
if event_dict is None:
raise HTTPException(status_code=404, detail="Trace not found")
return event_dict
@app.get("/debug/trace/session/{session_id}", tags=[TAG_DEBUG])
def get_session_trace(session_id: str) -> Any:
async def get_session_trace(session_id: str) -> Any:
spans = memory_exporter.get_finished_spans(session_id)
if not spans:
return []
@@ -461,7 +461,7 @@ class AdkWebServer:
response_model_exclude_none=True,
tags=[TAG_EVALUATION],
)
def create_eval_set(
async def create_eval_set(
app_name: str,
eval_set_id: str,
):
@@ -479,7 +479,7 @@ class AdkWebServer:
response_model_exclude_none=True,
tags=[TAG_EVALUATION],
)
def list_eval_sets(app_name: str) -> list[str]:
async def list_eval_sets(app_name: str) -> list[str]:
"""Lists all eval sets for the given app."""
try:
return self.eval_sets_manager.list_eval_sets(app_name)
@@ -532,7 +532,7 @@ class AdkWebServer:
response_model_exclude_none=True,
tags=[TAG_EVALUATION],
)
def list_evals_in_eval_set(
async def list_evals_in_eval_set(
app_name: str,
eval_set_id: str,
) -> list[str]:
@@ -551,7 +551,7 @@ class AdkWebServer:
response_model_exclude_none=True,
tags=[TAG_EVALUATION],
)
def get_eval(
async def get_eval(
app_name: str, eval_set_id: str, eval_case_id: str
) -> EvalCase:
"""Gets an eval case in an eval set."""
@@ -574,7 +574,7 @@ class AdkWebServer:
response_model_exclude_none=True,
tags=[TAG_EVALUATION],
)
def update_eval(
async def update_eval(
app_name: str,
eval_set_id: str,
eval_case_id: str,
@@ -605,7 +605,7 @@ class AdkWebServer:
"/apps/{app_name}/eval_sets/{eval_set_id}/evals/{eval_case_id}",
tags=[TAG_EVALUATION],
)
def delete_eval(app_name: str, eval_set_id: str, eval_case_id: str):
async def delete_eval(app_name: str, eval_set_id: str, eval_case_id: str):
try:
self.eval_sets_manager.delete_eval_case(
app_name, eval_set_id, eval_case_id
@@ -690,7 +690,7 @@ class AdkWebServer:
response_model_exclude_none=True,
tags=[TAG_EVALUATION],
)
def get_eval_result(
async def get_eval_result(
app_name: str,
eval_result_id: str,
) -> EvalSetResult:
@@ -709,7 +709,7 @@ class AdkWebServer:
response_model_exclude_none=True,
tags=[TAG_EVALUATION],
)
def list_eval_results(app_name: str) -> list[str]:
async def list_eval_results(app_name: str) -> list[str]:
"""Lists all eval results for the given app."""
return self.eval_set_results_manager.list_eval_set_results(app_name)
@@ -718,7 +718,7 @@ class AdkWebServer:
response_model_exclude_none=True,
tags=[TAG_EVALUATION],
)
def list_eval_metrics(app_name: str) -> list[MetricInfo]:
async def list_eval_metrics(app_name: str) -> list[MetricInfo]:
"""Lists all eval metrics for the given app."""
try:
from ..evaluation.metric_evaluator_registry import DEFAULT_METRIC_EVALUATOR_REGISTRY
@@ -85,9 +85,9 @@ class SseConnectionParams(BaseModel):
class StreamableHTTPConnectionParams(BaseModel):
"""Parameters for the MCP streamable HTTP connection.
"""Parameters for the MCP SSE connection.
See MCP streamable HTTP Client documentation for more details.
See MCP SSE Client documentation for more details.
https://github.com/modelcontextprotocol/python-sdk/blob/main/src/mcp/client/streamable_http.py
Attributes: