uPj- UddlmZddlmZddlZddlmZmZddlm Z m Z ddl m Z dd l mZd d lmZeZeed <Gd de Zy))Callable)FinalN)AsyncHTTPHandler HTTPHandler)CustomStreamingDecoder ModelResponse)OpenAILikeChatHandler)_get_api_params)IBMWatsonXChatConfigwatsonx_chat_transformationc!eZdZfdZdidddddddd dedededzded ed ed ed edzd edededzde e jzdzde e zdzdedzdedzdef fdZxZS)WatsonXChatHandlerc $t|di|y)N)super__init__)selfkwargs __class__s f/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/watsonx/chat/handler.pyrzWatsonXChatHandler.__init__s "6"NF) acompletionlitellm_paramsheaders logger_fntimeoutclientcustom_endpointstreaming_decoder fake_streammodelmessagesapi_basecustom_llm_providercustom_prompt_dictmodel_response print_verboseapi_keyoptional_paramsrrrr r!r"r#clt| |}tj|xsi||| | | }tj||}| j |tj || || | | j dd}t|!|j dxsd|||||||| | | | | ||||d | S) N)paramsr$)rr$r%r,r+r)r$ api_paramsstreamF)r&r+r$r,rr0model_idT)r$r%r&r'r(r)r*encodingr+ logging_objr,rrrrrr r!r") r rvalidate_environment_prepare_payloadupdateget_complete_urlgetr completion)rr$r%r&r'r(r)r*r3r+r4r,rrrrrr r!r"r#r/watsonx_auth_payloadrs rr:zWatsonXChatHandler.completions 0,?%P .BBMr+) C 'B&R&R!'S'  34/??+)"&&x7 @ w!&**:6<" 31)'#+#) /'"  r)__name__ __module__ __qualname__rstrlistdictrrfloathttpxTimeoutrrboolrr: __classcell__)rs@rrrs#"!#048<'+;?!-I I  I * I ! I !I &I  I tI I I !I $&-%I &..5'I ()I *2D8+I ,-I I rr)collections.abcrtypingrrC&litellm.llms.custom_httpx.http_handlerrrlitellm.types.utilsrropenai_like.chat.handlerr common_utilsr transformationrr__annotations__rrrrrOs:$ PE=*0%9%;U;M .M r