chore: reformat the code using autoformat.sh.

PiperOrigin-RevId: 764918729
This commit is contained in:
Liang Wu 2025-05-29 15:55:32 -07:00 committed by Copybara-Service
parent db4bc9809c
commit 41ba2d1c8a
2 changed files with 10 additions and 7 deletions

View File

@ -343,7 +343,9 @@ def from_function_with_options(
) )
# This snippet catches the case when type hints are stored as strings # This snippet catches the case when type hints are stored as strings
if isinstance(return_value.annotation, str): if isinstance(return_value.annotation, str):
return_value = return_value.replace(annotation=typing.get_type_hints(func)["return"]) return_value = return_value.replace(
annotation=typing.get_type_hints(func)['return']
)
declaration.response = ( declaration.response = (
_function_parameter_parse_util._parse_schema_from_parameter( _function_parameter_parse_util._parse_schema_from_parameter(

View File

@ -21,11 +21,11 @@ from unittest.mock import patch
from fastapi.testclient import TestClient from fastapi.testclient import TestClient
from google.adk.agents.base_agent import BaseAgent from google.adk.agents.base_agent import BaseAgent
from google.adk.agents.run_config import RunConfig from google.adk.agents.run_config import RunConfig
from google.adk.cli.fast_api import get_fast_api_app
from google.adk.evaluation.eval_case import EvalCase from google.adk.evaluation.eval_case import EvalCase
from google.adk.evaluation.eval_case import Invocation from google.adk.evaluation.eval_case import Invocation
from google.adk.evaluation.eval_set import EvalSet
from google.adk.evaluation.eval_result import EvalSetResult from google.adk.evaluation.eval_result import EvalSetResult
from google.adk.cli.fast_api import get_fast_api_app from google.adk.evaluation.eval_set import EvalSet
from google.adk.events import Event from google.adk.events import Event
from google.adk.runners import Runner from google.adk.runners import Runner
from google.adk.sessions.base_session_service import ListSessionsResponse from google.adk.sessions.base_session_service import ListSessionsResponse
@ -343,9 +343,7 @@ def mock_eval_set_results_manager():
class MockEvalSetResultsManager: class MockEvalSetResultsManager:
"""Mock eval set results manager.""" """Mock eval set results manager."""
def save_eval_set_result( def save_eval_set_result(self, app_name, eval_set_id, eval_case_results):
self, app_name, eval_set_id, eval_case_results
):
if app_name not in eval_set_results: if app_name not in eval_set_results:
eval_set_results[app_name] = {} eval_set_results[app_name] = {}
eval_set_result_id = f"{app_name}_{eval_set_id}_eval_result" eval_set_result_id = f"{app_name}_{eval_set_id}_eval_result"
@ -695,7 +693,10 @@ def test_run_eval(test_app, create_test_eval_set):
data = response.json() data = response.json()
assert isinstance(data, dict) assert isinstance(data, dict)
assert data["evalSetId"] == "test_eval_set_id" assert data["evalSetId"] == "test_eval_set_id"
assert data["evalSetResultId"] == f"{info['app_name']}_test_eval_set_id_eval_result" assert (
data["evalSetResultId"]
== f"{info['app_name']}_test_eval_set_id_eval_result"
)
assert len(data["evalCaseResults"]) == 1 assert len(data["evalCaseResults"]) == 1
verify_eval_case_result(data["evalCaseResults"][0]) verify_eval_case_result(data["evalCaseResults"][0])