)`inddlmZddlmZddlmZmZddlmZddl m Z de d eefd Z d efd Z d S) )Optional) APIRouter) RouteConfig create_router)logger)get_lora_route_config handler_typereturnc|dkrtddddgdS|dkrtd d d dgd St|S) aLGet route configuration for SageMaker handler types. This resolver handles both core SageMaker routes (/ping, /invocations) and LoRA-specific routes (/adapters, etc.). It serves as a unified entry point for all SageMaker routing configuration. Args: handler_type: The handler type identifier (e.g., 'ping', 'invoke', 'register_adapter', 'unregister_adapter') Returns: RouteConfig: The route configuration if the handler type has a route None: If the handler type doesn't have a route (e.g., transform-only handlers) pingz/pingGEThealth sagemakerzHealth check endpoint)pathmethodtagssummaryinvokez /invocationsPOST inferencezModel inference endpoint)rr )r s /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/model_hosting_container_standards/sagemaker/sagemaker_router.pyget_sagemaker_route_configr s| vK(+      ! !{+.      ! . ..ctjdttdg}tjdt |jd|S)aCreate a FastAPI router with all registered SageMaker handlers mounted. This convenience function creates an APIRouter and automatically mounts all registered SageMaker handlers using the unified route resolver. It provides a complete SageMaker-compatible routing solution out of the box. Supported Routes: - Core SageMaker endpoints: /ping (health check), /invocations (inference) - LoRA adapter management: /adapters (list/register), /adapters/{name} (unregister) - Any additional SageMaker-specific routes registered via decorators The router uses a unified route resolver that handles both core SageMaker routes and LoRA-specific routes, providing a single entry point for all SageMaker routing configuration. Alternative Usage: For more control over router configuration or to mount handlers to an existing router, use the generic mount_handlers function: ```python from model_hosting_container_standards.common.fastapi.routing import mount_handlers from model_hosting_container_standards.sagemaker.sagemaker_router import get_sagemaker_route_config # Mount to existing router mount_handlers(my_router, route_resolver=get_sagemaker_route_config) ``` Returns: APIRouter: Configured router with SageMaker handlers mounted and tagged with ["sagemaker"] for OpenAPI documentation Example: ```python from fastapi import FastAPI from model_hosting_container_standards.sagemaker import create_sagemaker_router app = FastAPI() sagemaker_router = create_sagemaker_router() app.include_router(sagemaker_router) ``` z5Creating SageMaker router with unified route resolverr)route_resolverrz+SageMaker router created successfully with z routes)rinforrlenroutes)routers rcreate_sagemaker_routerr"1siT KGHHH1]F  KQc&-6H6HQQQ MrN)typingrfastapircommon.fastapi.routingrrlogging_configr lora.routesr strrr"rrr*s@???????######/..... /S /Xk5J / / / /H5555555r