uPj d dl mZmZmZ d dlZd dlmZ d dlmZ d dl m
Z
d dl m Z d dl
mZ dd lmZ er
d dlZd d
lmZ eZneZ G d d e Zy)
)
TYPE_CHECKINGAnyFinalN)BaseLLMException OpenAIErrorget_secret_str)AllMessageValues)
ModelResponse )OpenAIGPTConfig)Loggingc v e Zd ZdZdedefdZed dedz dedz fd Zed dedz dedz fd Z ed dedz dedz fd
Z
dedz dedz de edz edz f fd Z fd Z
d ded
ej dedededee dededddedz dedz defdZdededeej. z defdZ xZS ) ClarifaiConfigz
Configuration class for Clarifai chat completions.
Since Clarifai is OpenAI-compatible, we extend OpenAIGPTConfig.
modelreturnc
g dS )zE
Get the supported OpenAI params for the given model
)
max_tokensmax_completion_tokensresponse_formatstream temperaturetop_p tool_choicetoolspresence_penaltyfrequency_penaltystream_options )selfr s n/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/clarifai/chat/transformation.pyget_supported_openai_paramsz*ClarifaiConfig.get_supported_openai_params s
Napi_keyc | xs t d S )NCLARIFAI_API_KEYr )r% s r" get_api_keyzClarifaiConfig.get_api_key1 s <.);<[bccr$ raw_responsemodel_response logging_obj request_datar= r> r? encodingztiktoken.Encoding | None json_modec |j ||
|j d|i |j } t di | }|j
d|z |_ |S # t $ r+}
t |j
d|
|j |
d}
~
ww xY w)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_codemessager@ Nz clarifai/r ) post_calltextjson Exceptionr rN r@ r r )r! r rB rC rD rE r= r> r? rF r% rG completion_responseeresponses r" transform_responsez!ClarifaiConfig.transform_responseP s & * / / 2L A
)5):):)< (>*=> >> % (5 0HN ( 4 4=aSA$,,
s A B
&BB
error_messagerN r@ c t ||| S )z
Get the appropriate error class for Clarifai errors.
Since Clarifai is OpenAI-compatible, we use OpenAI error handling.
rM r )r! rX rN r@ s r" get_error_classz ClarifaiConfig.get_error_classz s
#!
r$ r: )NN)__name__
__module__ __qualname____doc__strlistr# staticmethodr( r, r4 tupler8 r< httpxResponser LiteLLMLoggingObjdictr boolrW intHeadersr rZ
__classcell__)rA s @r" r r s
$ =S4Z =3: = = GsTz GS4Z G G cDj C$J
)*
) t
)
sTz3:% &
)d #!%(( nn( & (
' (
( '(( ( ( -( t( $;(
(T
S
s
TTYTaTaMa
fv
r$ r )typingr r r rc )litellm.llms.base_llm.chat.transformationr litellm.llms.openai.common_utilsr litellm.secret_managers.mainr
litellm.types.llms.openair litellm.types.utilsr openai.chat.gpt_transformationr tiktoken*litellm.litellm_core_utils.litellm_loggingr _LiteLLMLoggingObjre r r r$ r" ru sG , , F 8 7 . =X*j
_ j
r$