|
40 | 40 | from ..auth.credential_service.in_memory_credential_service import InMemoryCredentialService |
41 | 41 | from ..runners import Runner |
42 | 42 | from .api_server import ApiServer |
| 43 | +from .dev_server import DevServer |
43 | 44 | from .service_registry import load_services_module |
44 | 45 | from .utils import envs |
45 | 46 | from .utils.agent_change_handler import AgentChangeEventHandler |
@@ -513,26 +514,10 @@ def get_fast_api_app( |
513 | 514 | # Build the Credential service |
514 | 515 | credential_service = InMemoryCredentialService() |
515 | 516 |
|
516 | | - # Conditionally import and instantiate the appropriate server class |
| 517 | + # Instantiate the appropriate server class based on web option |
517 | 518 | # If web=True, use DevServer (includes all endpoints: production + dev) |
518 | 519 | # If web=False, use ApiServer (production-safe endpoints only) |
519 | | - if web: |
520 | | - try: |
521 | | - from .dev_server import DevServer |
522 | | - |
523 | | - ServerClass = DevServer |
524 | | - except ImportError: |
525 | | - logger.warning( |
526 | | - "DevServer not found, falling back to ApiServer. " |
527 | | - "Dev-only endpoints will not be available." |
528 | | - ) |
529 | | - from .api_server import ApiServer |
530 | | - |
531 | | - ServerClass = ApiServer |
532 | | - else: |
533 | | - from .api_server import ApiServer |
534 | | - |
535 | | - ServerClass = ApiServer |
| 520 | + ServerClass = DevServer if web else ApiServer |
536 | 521 |
|
537 | 522 | adk_web_server = ServerClass( |
538 | 523 | agent_loader=agent_loader, |
|
0 commit comments