refactor: rename agent_dir to agents_dir and rename app_id to app_name in fast_api.py to make it consistent among every endpoints

PiperOrigin-RevId: 763483339
This commit is contained in:
Xiang (Sean) Zhou
2025-05-26 11:51:19 -07:00
committed by Copybara-Service
parent 6b89ceb49a
commit be0786ea88
6 changed files with 32 additions and 32 deletions
@@ -36,8 +36,8 @@ def _sanitize_eval_set_result_name(eval_set_result_name: str) -> str:
class LocalEvalSetResultsManager(EvalSetResultsManager):
"""An EvalSetResult manager that stores eval set results locally on disk."""
def __init__(self, agent_dir: str):
self._agent_dir = agent_dir
def __init__(self, agents_dir: str):
self._agents_dir = agents_dir
@override
def save_eval_set_result(
@@ -108,4 +108,4 @@ class LocalEvalSetResultsManager(EvalSetResultsManager):
return eval_result_files
def _get_eval_history_dir(self, app_name: str) -> str:
return os.path.join(self._agent_dir, app_name, _ADK_EVAL_HISTORY_DIR)
return os.path.join(self._agents_dir, app_name, _ADK_EVAL_HISTORY_DIR)
@@ -182,8 +182,8 @@ def load_eval_set_from_file(
class LocalEvalSetsManager(EvalSetsManager):
"""An EvalSets manager that stores eval sets locally on disk."""
def __init__(self, agent_dir: str):
self._agent_dir = agent_dir
def __init__(self, agents_dir: str):
self._agents_dir = agents_dir
@override
def get_eval_set(self, app_name: str, eval_set_id: str) -> EvalSet:
@@ -216,7 +216,7 @@ class LocalEvalSetsManager(EvalSetsManager):
@override
def list_eval_sets(self, app_name: str) -> list[str]:
"""Returns a list of EvalSets that belong to the given app_name."""
eval_set_file_path = os.path.join(self._agent_dir, app_name)
eval_set_file_path = os.path.join(self._agents_dir, app_name)
eval_sets = []
for file in os.listdir(eval_set_file_path):
if file.endswith(_EVAL_SET_FILE_EXTENSION):
@@ -247,7 +247,7 @@ class LocalEvalSetsManager(EvalSetsManager):
def _get_eval_set_file_path(self, app_name: str, eval_set_id: str) -> str:
return os.path.join(
self._agent_dir,
self._agents_dir,
app_name,
eval_set_id + _EVAL_SET_FILE_EXTENSION,
)