File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -526,12 +526,8 @@ def get_fast_api_app(
526526 "DevServer not found, falling back to ApiServer. "
527527 "Dev-only endpoints will not be available."
528528 )
529- from .api_server import ApiServer
530-
531529 ServerClass = ApiServer
532530 else :
533- from .api_server import ApiServer
534-
535531 ServerClass = ApiServer
536532
537533 adk_web_server = ServerClass (
Original file line number Diff line number Diff line change @@ -540,6 +540,31 @@ def _create_test_client(
540540 return TestClient (app )
541541
542542
543+ def test_web_reload_agents_uses_module_api_server_import (
544+ mock_session_service ,
545+ mock_artifact_service ,
546+ mock_memory_service ,
547+ mock_agent_loader ,
548+ mock_eval_sets_manager ,
549+ mock_eval_set_results_manager ,
550+ ):
551+ """Regression test for web reload mode with DevServer available."""
552+ with patch ("google.adk.cli.api_server.Observer" ) as mock_observer_class :
553+ _create_test_client (
554+ mock_session_service ,
555+ mock_artifact_service ,
556+ mock_memory_service ,
557+ mock_agent_loader ,
558+ mock_eval_sets_manager ,
559+ mock_eval_set_results_manager ,
560+ reload_agents = True ,
561+ )
562+
563+ observer = mock_observer_class .return_value
564+ observer .schedule .assert_called_once ()
565+ observer .start .assert_called_once ()
566+
567+
543568def test_agent_with_bigquery_analytics_plugin (
544569 tmp_path ,
545570 mock_session_service ,
You can’t perform that action at this time.
0 commit comments