h#@ddlmZmZmZmZmZddlmZGddZy))AnyDictListOptionalType)verbose_proxy_loggerceZdZdZgdZgdZddgZedee e e ffdZ ede e e fd e d e e e fdd fd Z ede e e fd ee de dd fdZede e e fded e d ee de de e e ff dZede e e fde e e ffdZede e e fde e e ffdZede e e fde e e ffdZede 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 /responsesreturnc |jS#t$r$ |jcYS#t$rYYywxYwwxYw)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 N)model_json_schemaAttributeErrorschema) model_classs j/var/www/Befach/backend/env/lib/python3.12/site-packages/litellm/proxy/common_utils/custom_openapi_spec.pyget_pydantic_schemaz%CustomOpenAPISpec.get_pydantic_schemasK 002 2  "))++!    s  ?.? ;?;?openapi_schema schema_name schema_defNcJd|vri|d<d|dvri|dd<||dd|<y)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)rrrs radd_schema_to_componentsz*CustomOpenAPISpec.add_schema_to_components5sJ ~ -+-N< ( N<8 868N< ( 3@J|$Y/ <paths schema_refc|D]9}||jdivsd|d|vs#dddd|iiid|d|dd<;y ) a* Add request body schema reference to specified paths. 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") rpostTzapplication/jsonrz$ref)requiredcontent requestBodyN)get)rrrpaths radd_request_body_to_pathsz+CustomOpenAPISpec.add_request_body_to_pathsHsoD~))'2666^T[E\]aEb;b $*$ & '-  Hw'-f5mDrroperation_namec v tj|}|Ltj|||tj||d|t j d|d|St j d| |S#t $r/}t j d|dt|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 Nz#/components/schemas/zSuccessfully added z schema to OpenAPI speczCould not get schema for zFailed to add z request schema: )r rrr%rdebug Exceptionstr)rrrrr&request_schemaes radd_request_schemaz$CustomOpenAPISpec.add_request_schema_s* c.BB;ON)!::>;Xfg";;"+K=9 %**-@ Md+ef %**-F{m+TU   c & &7GGXY\]^Y_X`'a b b  csA!B%B B8 $B33B8c ddlm}tj||dtjdS#t $r-}t jdt||cYd}~Sd}~wwxYw)aI 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)ProxyChatCompletionRequestr/zchat completionrrrrr&z-Failed to import ProxyChatCompletionRequest: N) litellm.proxy._typesr/r r-CHAT_COMPLETION_PATHS ImportErrorrr(r*)rr/r,s r"add_chat_completion_request_schemaz4CustomOpenAPISpec.add_chat_completion_request_schemasm " G$77-68'==0 8  " & &)VWZ[\W]V^'_ `! ! "-0 A&"A!A&!A&c ddlm}tj||dtjdS#t $r-}t jdt||cYd}~Sd}~wwxYw)a9 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)EmbeddingRequestr7 embeddingr0z#Failed to import EmbeddingRequest: N) litellm.types.embeddingr7r r-EMBEDDING_PATHSr3rr(r*)rr7r,s radd_embedding_request_schemaz.CustomOpenAPISpec.add_embedding_request_schemasj " @$77-,.'77* 8  " & &)LSQRVH'U V! ! "r5c ddlm}tj||dtjdS#t $r-}t jdt||cYd}~Sd}~wwxYw)aF 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)ResponsesAPIRequestParamsr=z responses APIr0z,Failed to import ResponsesAPIRequestParams: N) litellm.types.llms.openair=r r-RESPONSES_API_PATHSr3rr(r*)rr=r,s r add_responses_api_request_schemaz2CustomOpenAPISpec.add_responses_api_request_schemasm " K$77-57';;. 8  " & &)UVYZ[V\U]'^ _! ! "r5ctj|}tj|}tj|}|S)a 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 r4r;r@)rs radd_llm_api_request_schema_bodyz1CustomOpenAPISpec.add_llm_api_request_schema_bodys>+MMn]+GGW+KKN[r)__name__ __module__ __qualname____doc__r2r:r? staticmethodrrr*rrrrr%rr-r4r;r@rBrrrr r s  O  HT#s(^,D*Jc3hJcJ_cdgildl_mJrvJJ$$sCx.c`chl,,S#X,,,Cy ,  , c3h ,,\"4S>"dSVX[S[n""2"T#s(^"SRUX""2"c3h"DQTVYQYN""2S#X4PSUXPX>rr N) typingrrrrrlitellm._loggingrr rrrrJs221jjr