tPj<DddlmZmZddlmZmZddlmZGddZy))MappingSequence)AnyFinal)verbose_proxy_loggercdeZdZdZgdZgdZddgZedee e fdzfdZ ed e e e fd e d ee e fddfd Zed e e e fd ee de ddfdZed e e e fdee ee e ffddfdZede de fdZede e e fde e e ffdZede e e fde e e ffdZed e e e fded e d ee de de e e ff dZed e e e fde e e ffdZed e e e fde e e ffdZed e e e fde e e ffdZed e e e fde e e ffdZy)CustomOpenAPISpecz Handler for customizing OpenAPI specifications with Pydantic models for documentation purposes without runtime validation. )z/v1/chat/completionsz/chat/completionsz!/engines/{model}/chat/completionsz,/openai/deployments/{model}/chat/completions)z/v1/embeddingsz /embeddingsz/engines/{model}/embeddingsz&/openai/deployments/{model}/embeddingsz /v1/responsesz /responsesreturnNc |jS#t$r$ |jcYS#t$rYYywxYwt$r!}t j d||Yd}~yd}~wwxYw)z Get JSON schema from a Pydantic model, handling both v1 and v2 APIs. Args: model_class: Pydantic model class Returns: JSON schema dict or None if failed Nz$Failed to generate schema for %s: %s)model_json_schemaAttributeErrorschema Exceptionrdebug) model_classes s/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/proxy/common_utils/custom_openapi_spec.pyget_pydantic_schemaz%CustomOpenAPISpec.get_pydantic_schemaso 002 2  "))++!    ! & &'M{\] ^  s0 A'.A' ;A'; A'A""A'openapi_schema schema_name schema_defcdd|vri|d<d|dvri|dd<tj|||iy)a Add a schema definition to the OpenAPI components/schemas section. Args: openapi_schema: The OpenAPI schema dict to modify schema_name: Name for the schema component schema_def: The schema definition componentsschemasN)r _move_defs_to_components)rrrs radd_schema_to_componentsz*CustomOpenAPISpec.add_schema_to_components8sJ ~ -+-N< ( N<8 868N< ( 3 22>KQ[C\]paths schema_refc|D]l}||jdivsd|d|vs$|jdd}|jdijdij|i}|jdi}|jdg}d |vrtj||d d |id }|j D]E\} } tj | } tj | } | d k(r d ddg| d<| |d| <Gddd|iid|d|dd<d|d|dvs-|d|dd} | D cgc]} | jddk(s| }} ||d|dd<oycc} w)a Add request body with expanded form fields for better Swagger UI display. This keeps the request body but expands it to show individual fields in the UI. Args: openapi_schema: The OpenAPI schema dict to modify paths: List of paths to update schema_ref: Reference to the schema component (e.g., "#/components/schemas/ModelName") rpost/rr propertiesrequired$defsobject)typer%r$messagesuserzHello, how are you?)rolecontentexampleTzapplication/jsonr)r%r, requestBody parametersinpathN)getsplitr ritems_expand_field_definition_rewrite_defs_refs)rrrr1r actual_schemaschema_propertiesrequired_fieldsexpanded_schema field_name field_defexpanded_fieldexisting_paramsparamfiltered_paramss radd_request_body_to_pathsz+CustomOpenAPISpec.add_request_body_to_pathsMs- ZD~))'2666^T[E\]aEb;b(..s3B7 . 2 2< D H HTV W [ [\gik l $1$5$5lB$G!"/"3"3J"Cm+%>>~}]dOef % /"$#.?-D-D-F O)J %6%O%OPY%ZN&7%I%I.%YN"Z/>DQf5g4hy1@NOL1*= O!% 2X4OPHw'-f5mD  >'#:4#@#HH&4W&=d&CF&KL&YO:I&gUYYW[_`fMfu&gO&gJYN7+D1&9,G[- ZX'hs E4E4defsc|syd|vri|d<d|dvri|dd<|jD]C\}}tj|}||dd|<d|vs+tj||dEy)a& Move $defs from Pydantic v2 schema to OpenAPI components/schemas. This makes the definitions resolvable in Swagger/OpenAPI viewers. Args: openapi_schema: The OpenAPI schema dict to modify defs: The $defs dictionary from Pydantic schema Nrrr&)r4r r6r)rrBdef_name def_schema rewritten_defs rrz*CustomOpenAPISpec._move_defs_to_componentss  ~ -+-N< ( N<8 868N< ( 3%)JJL ` Hj-@@LM@MN< ( 3H =*$!::>:V]K^_ `rrct|tr{i}|jD]d\}}|dk(rR RMSs Cr<cxd|vrd|diSd|vr(|d}|D]}|jddk7s|cSddiSddiS)z Extract a simple schema from a Pydantic field definition for parameter display. Args: field_def: Pydantic field definition Returns: Simplified schema for OpenAPI parameter r(anyOfnullstring)r2)r<any_ofoptions r_extract_field_schemaz'CustomOpenAPISpec._extract_field_schemaso Y If-. . i %g.F  "::f%/!M "H% %!!rc"|jS)aD Expand a Pydantic field definition for inline use in OpenAPI schema. This creates a full field definition that Swagger UI can render as individual form fields. Args: field_def: Pydantic field definition Returns: Expanded field definition for OpenAPI schema )copy)r<s rr5z*CustomOpenAPISpec._expand_field_definitions~~rroperation_namecR tj|}|Itj|||tj||d|t j d||St j d| |S#t $r"}t j d||Yd}~|Sd}~wwxYw)a Generic method to add a request schema to OpenAPI specification. Args: openapi_schema: The OpenAPI schema dict to modify model_class: The Pydantic model class to get schema from schema_name: Name for the schema component paths: List of paths to add the request body to operation_name: Name of the operation for logging (e.g., "chat completion", "embedding") Returns: Modified OpenAPI schema NrIz,Successfully added %s schema to OpenAPI speczCould not get schema for %sz#Failed to add %s request schema: %s)r rrrArrr)rrrrr]request_schemars radd_request_schemaz$CustomOpenAPISpec.add_request_schemas* a$5$I$I+$VN)!::>;Xfg";;"E-B;-+P%**+Y[fg %**+H+V   a & &'Ln^_ ` `  asAA;"A;; B&B!!B&c ddlm}tj||dtjdS#t $r"}t jd||cYd}~Sd}~wwxYw)a5 Add ProxyChatCompletionRequest schema to chat completion endpoints for documentation. This shows the request body in Swagger without runtime validation. Args: openapi_schema: The OpenAPI schema dict to modify Returns: Modified OpenAPI schema r)ProxyChatCompletionRequestrbzchat completionrrrrr]z/Failed to import ProxyChatCompletionRequest: %sN)litellm.proxy._typesrbr r`CHAT_COMPLETION_PATHS ImportErrorrr)rrbrs r"add_chat_completion_request_schemaz4CustomOpenAPISpec.add_chat_completion_request_schemasa " G$77-68'==0 8  " & &'XZ[ \! ! "-0 AAAAc ddlm}tj||dtjdS#t $r"}t jd||cYd}~Sd}~wwxYw)a% Add EmbeddingRequest schema to embedding endpoints for documentation. This shows the request body in Swagger without runtime validation. Args: openapi_schema: The OpenAPI schema dict to modify Returns: Modified OpenAPI schema r)EmbeddingRequestrj embeddingrcz%Failed to import EmbeddingRequest: %sN)litellm.types.embeddingrjr r`EMBEDDING_PATHSrfrr)rrjrs radd_embedding_request_schemaz.CustomOpenAPISpec.add_embedding_request_schema7sa " @$77-,.'77* 8  " & &'NPQ R! ! "rhc ddlm}tj||dtjdS#t $r"}t jd||cYd}~Sd}~wwxYw)a2 Add ResponsesAPIRequestParams schema to responses API endpoints for documentation. This shows the request body in Swagger without runtime validation. Args: openapi_schema: The OpenAPI schema dict to modify Returns: Modified OpenAPI schema r)ResponsesAPIRequestParamsrpz responses APIrcz.Failed to import ResponsesAPIRequestParams: %sN)litellm.types.llms.openairpr r`RESPONSES_API_PATHSrfrr)rrprs r add_responses_api_request_schemaz2CustomOpenAPISpec.add_responses_api_request_schemaQsa " K$77-57';;. 8  " & &'WYZ [! ! "rhctj|}tj|}tj|}|S)z Add LLM API request schema bodies to OpenAPI specification for documentation. Args: openapi_schema: The base OpenAPI schema Returns: OpenAPI schema with added request body schemas )r rgrnrs)rs radd_llm_api_request_schema_bodyz1CustomOpenAPISpec.add_llm_api_request_schema_bodyms>+MMn]+GGW+KKN[r)__name__ __module__ __qualname____doc__rermrr staticmethodrrLr'rrKrrrrArr6rZr5r(r`rgrnrsrurrr r s O+L9GCK,@4,G4^S#X^58^FMcSYkFZ^ ^^(7Z$sCx.7ZRU7Zdg7Zlp7Z7Zr`c3h`wsT[\_ad\dTeOeGf`ko``833<"c3h"DcN""6  Df,=  $sF{BS    *S&[)***} *  * c6k  **X"S&[)" c6k ""6"T#v+5F"4PSU[P[K\""2"S&[)" c6k ""6S&[) c6k rr N) collections.abcrrtypingrrlitellm._loggingrr r{rrrs-1||r