uPjg UdZddlZddlZddlZddlmZmZddlmZddl m Z ddl m Z m Z ddlmZmZe r ddlmZdd lmZe ej,d z Ze ed <hd Ze ed <d%dZdeeefdzfdZdeeefddfdZGddeZeddGddZ dddddedzfdZ!ddddde"ededzfdZ#de fd Z$ee$fd!e d"ege fde%fd#Z&e'd$k(rejPe&yy)&ak Per-feature OpenAPI snapshot for lazy-loaded routers. The committed JSON is generated by `python -m litellm.proxy._lazy_openapi_snapshot` and consumed at runtime so /openapi.json can show full route info for unloaded features without importing them. check-ui-api-types.yml (mirrored locally by `make check`) regenerates this file and fails when the committed copy differs, then rebuilds schema.d.ts from app.openapi() with the snapshot injected. After changing any lazily loaded route or this generator, rerun the module and commit the JSON, then run `npm run gen:api` in ui/litellm-dashboard and commit schema.d.ts. N)CallableMapping) dataclass)Path) TYPE_CHECKINGFinal)ReadOnly TypedDict)FastAPI) LazyFeaturez_lazy_openapi_snapshot.json SNAPSHOT_FILE>getputheadpostpatchtracedeleteoptionsHTTP_METHOD_SUFFIXESreturnc|D]}tt|ddxsg}t|dks t|dds9|j|j}t j dd|}|d|dj|_y)zGFastAPI derives route IDs from a set of methods; make snapshots stable.methodsN path_formatz\W_r) sortedgetattrlennamerresublower unique_id)routesrouter operation_ids i/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/proxy/_lazy_openapi_snapshot.py!_stabilize_multi_method_route_idsr))sA 48>B? w<1 GE=$$G **e&7&7%89 vveS,7 )N!GAJ,<,<,>+?@Actjsy tj5}tj|cdddS#1swYyxYw#tj t f$rYywxYwN)r existsopenjsonloadJSONDecodeErrorOSError)fs r( load_snapshotr46s`    !    ! Q99Q<    ' *s.AA  A AAAA21A2pathsc|jD]}t|ts|Dchc] }|tvs |}}|s/|j D]w\}}|tvst|ts|j d}t|t sA|D]2}d|}|j|s|dt| d|z|d<wyycc}w)a\Make FastAPI-generated operation IDs stable for multi-method routes. FastAPI derives the default operation ID suffix from the first item in the route's methods set. For routes registered with several HTTP methods, that set iteration order can vary between processes, which makes the snapshot drift even when no routes changed. operationIdrN) values isinstancedictritemsrstrendswithr)r5path_opsmethodr operationr'suffix suffix_tokens r(_normalize_operation_idsrC@sLLN(D) (0SfF>R4R6SS !)!1  FI11It9T$==7LlC0! !"6(| ((6/;P/QVWX^W_T`/`Im,   Ts C C cVeZdZUeeeeeeffed<eeeeeeffed<y)SnapshotFragmentr5 componentsN)__name__ __module__ __qualname__r rr<object__annotations__r*r(rErE_s> GCf!556 77gc6k&:!:;<$))Bse2>?yys+2 B;A<6B<Bused_operation_idsc ddlm}ddlm}|jDcgc]!}|j t |dds |#}}|syt|||j|j|}|jdi}t||jD]} | jD]}\} } t| ts| jd} t| t r6t"D]-} | j%d | s| dt'|  | z| d<n|j(g| d <|||}|d |jd ijd iid Scc}w)Nr) get_openapi)#ensure_unique_openapi_operation_idspath)titleversionr%r5r7rtagsschemasrF)r5rF)fastapi.openapi.utilsrdlitellm.proxy.proxy_serverrer%matchesrr)rhrirrCr8r;r9r:r<rr=rr )rTrUrbrdrer feat_routesfullr5r>r?opr'rAuniques r(_feature_fragmentrtusV1N%(ZZX4<<6SU@V3W!XKX %k2CIIs{{;WD88GR(EU#LLN )"..* )JFB"d#!vvm4 lC0"6"'001VH>0<^F |0Lv0UB}-!"#ii[6  ) )8>PQF &**\2">"B"B9b"QR %Ys !E"E"cddlm}ddlmt fd|D}t }|Dcic]}t ||x} |j| }}t||Scc}w)Nr) LAZY_FEATURES)rTc3@K|]}t|xywr,)ra).0rUrTr s r( z$generate_snapshot..s%pDFWX[]aFb>bd=o4ps)rQrR) litellm.proxy._lazy_featuresrvrmrTrSsetrtr rP)rvrRrbrUfragmentrQrTr s @@r(generate_snapshotr}sr:.pMppG*-%" )#t5GH HH U 8I Iw ?? s#A$ snapshot_filegeneratec|}|jrStjjdt |jddj |jdy|j tj|jdddztjjd t |jd |dy ) Nzerror: zQ feature(s) failed to import, so their fragments would vanish from the snapshot: z, rWrT)indent sort_keyszwrote z feature fragments to r) rRr]r^r_rjoin write_textr/dumpsrQstdout)r~rresults r(mainrsJF ~~ c&..)*+6>>232 7 TZZ(8(8dSVZZ[JJvc&"2"2344J=/Y[\] r*__main__)rN))__doc__r/r!r]collections.abcrr dataclassesrpathlibrtypingrrtyping_extensionsr r fastapir rzr __file__parentr rKrr)r:r<r4rCrErPrar{rtr}intrrGexitrLr*r(rsa  -!'18H~,,/LL uL e  AtCI-DdO>=y=  $d#$ 9McDj9MsSVx\los\s: @> @ -Wh    ^AS8T  mp   z CHHTVr*