|
50 | 50 | from watchdog.observers import Observer |
51 | 51 |
|
52 | 52 | from . import agent_graph |
53 | | -from ..agents.base_agent import BaseAgent |
54 | 53 | from ..agents.live_request_queue import LiveRequest |
55 | 54 | from ..agents.live_request_queue import LiveRequestQueue |
56 | 55 | from ..agents.run_config import RunConfig |
57 | 56 | from ..agents.run_config import StreamingMode |
58 | | -from ..apps.app import App |
59 | 57 | from ..artifacts.base_artifact_service import BaseArtifactService |
60 | 58 | from ..auth.credential_service.base_credential_service import BaseCredentialService |
61 | 59 | from ..errors.not_found_error import NotFoundError |
@@ -307,17 +305,10 @@ async def get_runner_async(self, app_name: str) -> Runner: |
307 | 305 | envs.load_dotenv_for_agent(os.path.basename(app_name), self.agents_dir) |
308 | 306 | if app_name in self.runner_dict: |
309 | 307 | return self.runner_dict[app_name] |
310 | | - agent_or_app = self.agent_loader.load_agent(app_name) |
311 | | - agentic_app = None |
312 | | - if isinstance(agent_or_app, BaseAgent): |
313 | | - agentic_app = App( |
314 | | - name=app_name, |
315 | | - root_agent=agent_or_app, |
316 | | - ) |
317 | | - else: |
318 | | - agentic_app = agent_or_app |
| 308 | + root_agent = self.agent_loader.load_agent(app_name) |
319 | 309 | runner = Runner( |
320 | | - app=agentic_app, |
| 310 | + app_name=app_name, |
| 311 | + agent=root_agent, |
321 | 312 | artifact_service=self.artifact_service, |
322 | 313 | session_service=self.session_service, |
323 | 314 | memory_service=self.memory_service, |
@@ -606,10 +597,9 @@ async def add_session_to_eval_set( |
606 | 597 | invocations = evals.convert_session_to_eval_invocations(session) |
607 | 598 |
|
608 | 599 | # Populate the session with initial session state. |
609 | | - agent_or_app = self.agent_loader.load_agent(app_name) |
610 | | - if isinstance(agent_or_app, App): |
611 | | - agent_or_app = agent_or_app.root_agent |
612 | | - initial_session_state = create_empty_state(agent_or_app) |
| 600 | + initial_session_state = create_empty_state( |
| 601 | + self.agent_loader.load_agent(app_name) |
| 602 | + ) |
613 | 603 |
|
614 | 604 | new_eval_case = EvalCase( |
615 | 605 | eval_id=req.eval_id, |
|
0 commit comments