2828from ..artifacts .base_artifact_service import BaseArtifactService
2929from ..artifacts .in_memory_artifact_service import InMemoryArtifactService
3030from ..errors .not_found_error import NotFoundError
31+ from ..memory .base_memory_service import BaseMemoryService
3132from ..sessions .base_session_service import BaseSessionService
3233from ..sessions .in_memory_session_service import InMemorySessionService
3334from ..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