)`i8, ddlmZddlmZmZmZddlmZmZddl m Z ddl m Z ddl mZd ed efd Zed GddZ ddedeeedeeegeefd dfdZ dd edeeegeefd efdZ ddeded ed dfdZ ddeded ed eefdZdeded dfdZgdZdS)) dataclass)CallableListOptional) APIRouterFastAPI)APIRoute)logger)handler_registryprefixreturncp|sdS|d}|r|dsd|}|S)aNormalize a URL prefix to ensure consistent path handling. Args: prefix: The URL prefix to normalize Returns: Normalized prefix with leading slash and no trailing slash, or empty string if input is empty/None Examples: normalize_prefix("api/v1/") -> "/api/v1" normalize_prefix("/api/v1") -> "/api/v1" normalize_prefix("api/v1") -> "/api/v1" normalize_prefix("") -> "" normalize_prefix("/") -> "" /)rstrip startswith)rs /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/model_hosting_container_standards/common/fastapi/routing.pynormalize_prefixr sS" r]]3  Ff'',,V MT)frozencheZdZUdZeed<eed<dZeeeed<dZ eeed<dS) RouteConfigaConfiguration for a handler route. This immutable dataclass defines all the metadata needed to mount a handler as a FastAPI route. The frozen=True ensures that route configurations cannot be accidentally modified after creation. Attributes: path: The URL path for the route (e.g., "/adapters", "/health") method: The HTTP method (e.g., "POST", "GET", "DELETE") tags: Optional list of tags for documentation grouping summary: Optional short summary for documentation pathmethodNtagssummary) __name__ __module__ __qualname____doc__str__annotations__rrrrrrrr)s]   III KKK $D(49 $$$!GXc]!!!!!rrNrouter handler_namesroute_resolverc p|tjddS||ntj}tjdt |dtjd||D]}tj|}|s ||}|rW||j ||j g|j |j tjd|j d|j d |ntjd |d #t$r$}tjd |d |Yd}~d}~wwxYwdS) aMount handlers to a FastAPI router with their configured routes. This function iterates through registered handlers and mounts them to the provided router using route configurations obtained from the route_resolver function. This design allows different handler types to provide their own routing logic without modifying this generic mounting code. The mounting process: 1. Determine which handlers to mount (all registered, or a specific subset) 2. For each handler type, get the handler function from the registry 3. Use the route_resolver to get the route configuration 4. Mount the handler to the router if a configuration exists Args: router: FastAPI APIRouter instance to mount handlers to handler_names: Optional list of specific handler types to mount. If None, attempts to mount all registered handlers. Useful when you only want to expose certain handlers. route_resolver: Function that maps handler_name (str) -> RouteConfig. This function encapsulates the routing logic for a specific handler category (e.g., LoRA handlers, monitoring handlers). If None, a warning is logged and no routes are mounted. Notes: - Handlers without route configurations are skipped (logged at DEBUG level) - ValueError exceptions from the route_resolver are caught and logged - Successfully mounted handlers are logged at INFO level NznNo route_resolver provided to mount_handlers. No routes will be mounted. This is likely a configuration error.z Mounting z handlers to routerzHandlers to mount: )methodsrrzMounted handler:  z -> z Skipping z - no default route configuredz - no route mapping available: )r warningr list_handlersinfolendebug get_handler add_api_routerrrr ValueError)r&r'r(handlers_to_mount handler_namehandler route_configes rmount_handlersr9>sD O    '2 8H8V8X8X KGC 122GGGHHH L:'8::;;;*WW ".|<<   W)>,77L W$$ %)01%*(0 % ) (;))l>O))&)) UUUUVVV W W W LU\UURSUU V V V V V V V V W;WWsA RouteConfig. See mount_handlers() for details. **router_kwargs: Additional keyword arguments passed to APIRouter constructor. Common options include: - tags: List of tags for all routes - dependencies: List of FastAPI dependencies Returns: APIRouter: A configured FastAPI router with handlers mounted zCreating router with prefix='z', tags=rrz1Router instance created, now mounting handlers...)r(zRouter created with  routesr%)r r.getrr0r9r/routes)rr( router_kwargsr&s r create_routerr?s0 KWWW 8I8I&RT8U8UWW  6 6f 6 6 6F LDEEE6.9999 KBs6='9'9BBBCCC Mrappct|}t}|jD]c}t|trL|r ||jn|j}t t|j}| ||fd|sdS|j j}d}tt|dz ddD]o} || }t|trP|j} t t|j} | | f|vr"tjd| d| || =|dz }p|dkr!d|_tjd|ddSdS) zRemove conflicting routes from app before including router. Args: app: The FastAPI application router: The router to be included prefix: URL prefix that will be applied to router routes NrzRemoving conflicting route: r+zRemoved z conflicting routes)rsetr= isinstancer rtuplesortedr*addr&ranger/r r.openapi_schema) r@r&rincoming_routesroute router_pathrouter_methods app_routes removed_counti route_path route_methodss rremove_conflicting_routesrTsf % %FeeO?? eX & & ?5;KV1UZ111K"6%-#8#899N   n = > > > "JM 3z??Q&B / / # #1  eX & & #J!&"7"788MM*o== W=WW:WWXXXqM" q! A}AAABBBBBrc t|}t}|jjD]R}t |t r;||jtt|j fSg}|jD]}t |t rj|r ||jn|j}tt|j }||f|vr/d |}| |d||S)a=Check for route conflicts between existing app routes and router routes. Args: app: The FastAPI application to check router: The router to be included prefix: URL prefix that will be applied to router routes Returns: List of conflict descriptions (empty if no conflicts) z, r+) rrDr&r=rEr rHrrFrGr*joinappend) r@r&rexisting_pathsrL conflictsrMrN methods_strs rcheck_route_conflictsr[sf % %FUUN"KK eX & & K    E&2G2G,H,HI J J JIAA eX & & A5;KV1UZ111K"6%-#8#899N^,>>"ii77   K!?!?+!?!?@@@ rctjdt||}|r=d|}tjd|t ||ntjd||td|j D}tjd|ddS) aSafely include a router in a FastAPI app with automatic conflict resolution. This function automatically replaces any conflicting routes and warns about conflicts in the logs. This ensures predictable behavior where the most recently included router takes precedence. Args: app: The FastAPI application router: The router to include Example: # Simple inclusion with automatic conflict resolution safe_include_router(app, my_router) z(Including router with conflict detectionz - zNRoute conflicts detected. The following existing routes will be replaced: - zNo route conflicts detectedc<g|]}t|t|Sr%)rEr ).0rs r z'safe_include_router...s'KKKQ:a3J3JKqKKKrz"Successfully included router with r;N) r r.r[rVr,rTr0include_routerr/r=)r@r&rY conflict_list route_counts rsafe_include_routerrd s$ K:;;;&c622I4 i00  m^k m m    "#v.... 2333vKK&-KKKLLK KI[IIIJJJJJr)rrr9r?rTr[rd)NN)rN)r) dataclassesrtypingrrrfastapirrfastapi.routingr logging_configr r6r r#rrr9r?rTr[rd__all__r%rrrks!!!!!!++++++++++&&&&&&&&$$$$$$$$$$$$&&&&&&SS< $"""""""",*.GKQWQW QWDI&QWXseXk-B&BCDQW QWQWQWQWjGK&& &XseXk-B&BCD& &&&&T46)C)C )C#)C-0)C )C)C)C)CZ46!! !#!-0! #Y!!!!H$K $K $K $K$K$K$KN   r