Skip to content

Commit dc3f60c

Browse files
ankursharmascopybara-github
authored andcommitted
chore: Plumb memory service from LocalEvalService to EvaluationGenerator
Co-authored-by: Ankur Sharma <[email protected]> PiperOrigin-RevId: 834398581
1 parent 14e3802 commit dc3f60c

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/google/adk/evaluation/local_eval_service.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@
2828
from ..artifacts.base_artifact_service import BaseArtifactService
2929
from ..artifacts.in_memory_artifact_service import InMemoryArtifactService
3030
from ..errors.not_found_error import NotFoundError
31+
from ..memory.base_memory_service import BaseMemoryService
3132
from ..sessions.base_session_service import BaseSessionService
3233
from ..sessions.in_memory_session_service import InMemorySessionService
3334
from ..utils.feature_decorator import experimental
@@ -77,6 +78,7 @@ def __init__(
7778
eval_set_results_manager: Optional[EvalSetResultsManager] = None,
7879
session_id_supplier: Callable[[], str] = _get_session_id,
7980
user_simulator_provider: UserSimulatorProvider = UserSimulatorProvider(),
81+
memory_service: Optional[BaseMemoryService] = None,
8082
):
8183
self._root_agent = root_agent
8284
self._eval_sets_manager = eval_sets_manager
@@ -91,6 +93,7 @@ def __init__(
9193
self._eval_set_results_manager = eval_set_results_manager
9294
self._session_id_supplier = session_id_supplier
9395
self._user_simulator_provider = user_simulator_provider
96+
self._memory_service = memory_service
9497

9598
@override
9699
async def perform_inference(
@@ -408,6 +411,7 @@ async def _perform_inference_single_eval_item(
408411
session_id=session_id,
409412
session_service=self._session_service,
410413
artifact_service=self._artifact_service,
414+
memory_service=self._memory_service,
411415
)
412416
)
413417

0 commit comments

Comments
 (0)