i"ddlmZmZmZmZmZmZddlZddlm Z ddl m Z ddl m Z ddlmZddlmZdd lmZer dd lmZeZneZGd d eZy) ) TYPE_CHECKINGAnyListOptionalTupleUnionNget_secret_str) ModelResponse)AllMessageValues OpenAIError)BaseLLMException)OpenAIGPTConfig)LoggingczeZdZdZdedefdZeddeedeefdZ eddeedeefdZ eddeedeefd Z deedeede eeeeffd Z fd Z dded ej d edededeededededeedeedefdZdededeeej6fdefdZxZS)ClarifaiConfigz Configuration class for Clarifai chat completions. Since Clarifai is OpenAI-compatible, we extend OpenAIGPTConfig. modelreturnc gdS)zE Get the supported OpenAI params for the given model ) max_tokensmax_completion_tokensresponse_formatstream temperaturetop_p tool_choicetoolspresence_penaltyfrequency_penaltystream_options)selfrs z/Users/manta/Documents/Projects/TheRoad-I1/.venv/lib/python3.12/site-packages/litellm/llms/clarifai/chat/transformation.pyget_supported_openai_paramsz*ClarifaiConfig.get_supported_openai_paramss   api_keyc |xs tdS)NCLARIFAI_API_KEYr )r(s r% get_api_keyzClarifaiConfig.get_api_key/s<.);<s/JJ!M^4F%GM2((r'cT|j|xs|}t| |||||SN)r7supertransform_request)r$rmessagesoptional_paramslitellm_paramsheaders __class__s r%r?z ClarifaiConfig.transform_requestJs7##E*3ew( 8_ng  r' raw_responsemodel_response logging_obj request_datar@rArBencoding json_modec ,|j|| |jd|i |j} tdi| }|j d|z|_ |S#t$r4} t |j dt | |j| d} ~ wwxYw)z Transform the Clarifai response to a standard ModelResponse. Since Clarifai is OpenAI-compatible, we use OpenAI response transformation. complete_input_dict)inputr(original_responseadditional_argsz#Failed to parse Clarifai response:  status_codemessagerCNz clarifai/r#) post_calltextjson ExceptionrrQstrrCr r)r$rrErFrGrHr@rArBrIr(rJcompletion_responseeresponses r%transform_responsez!ClarifaiConfig.transform_responseRs& *//2LA   "."3"3"5 !7#67 >> %(50HN (44=c!fXF$,,   sA B/BB error_messagerQrCct|||S)z Get the appropriate error class for Clarifai errors. Since Clarifai is OpenAI-compatible, we use OpenAI error handling. rPr )r$r\rQrCs r%get_error_classzClarifaiConfig.get_error_class|s#!  r'r=)NN)__name__ __module__ __qualname____doc__rWlistr& staticmethodrr+r/r7rr;r?httpxResponser LiteLLMLoggingObjdictrr rboolr[intrHeadersrr^ __classcell__)rDs@r%rrs    $=Xc]=hsm==Gx}G GGhsmx} )3- )# ) x}hsm+ , ) &"&$(((nn(& ( ' (  ('(((((#(D>( (T     /2  =B4CV=W    r'r)typingrrrrrrrelitellm.secret_managers.mainr litellm.types.utilsr litellm.types.llms.openair litellm.llms.openai.common_utilsr)litellm.llms.base_llm.chat.transformationropenai.chat.gpt_transformationr*litellm.litellm_core_utils.litellm_loggingr_LiteLLMLoggingObjrgrr#r'r%rvsDCC 7-9F=X*p _p r'