24
24
from ..agents .llm_agent import Agent
25
25
from ..artifacts .base_artifact_service import BaseArtifactService
26
26
from ..artifacts .in_memory_artifact_service import InMemoryArtifactService
27
+ from ..memory .base_memory_service import BaseMemoryService
28
+ from ..memory .in_memory_memory_service import InMemoryMemoryService
27
29
from ..runners import Runner
28
30
from ..sessions .base_session_service import BaseSessionService
29
31
from ..sessions .in_memory_session_service import InMemorySessionService
@@ -142,11 +144,15 @@ async def _generate_inferences_from_root_agent(
142
144
session_id : Optional [str ] = None ,
143
145
session_service : Optional [BaseSessionService ] = None ,
144
146
artifact_service : Optional [BaseArtifactService ] = None ,
147
+ memory_service : Optional [BaseMemoryService ] = None ,
145
148
) -> list [Invocation ]:
146
149
"""Scrapes the root agent given the list of Invocations."""
147
150
if not session_service :
148
151
session_service = InMemorySessionService ()
149
152
153
+ if not memory_service :
154
+ memory_service = InMemoryMemoryService ()
155
+
150
156
app_name = (
151
157
initial_session .app_name if initial_session else "EvaluationGenerator"
152
158
)
@@ -168,6 +174,7 @@ async def _generate_inferences_from_root_agent(
168
174
agent = root_agent ,
169
175
artifact_service = artifact_service ,
170
176
session_service = session_service ,
177
+ memory_service = memory_service ,
171
178
)
172
179
173
180
# Reset agent state for each query
0 commit comments