uPjUddlZddlmZddlmZmZddlZddlmZddl m Z m Z m Z ddl mZdd lmZerdd lmZeZeed <iddfd ed ededzde dededefdZ dd edede dedzdedzdddefdZy)N)Callable) TYPE_CHECKINGFinal)_get_httpx_client)EmbeddingResponse ModelResponseUsage)OobaboogaError)OobaboogaConfig)Loggingoobabooga_configmodelmessagesapi_basemodel_response print_verboseoptional_paramslitellm_paramsc tj|i|||| } |jdr|}n|r|}n tdd|}|dz}tj |||| | }|j ||d|i t }|j|| tj|d |vr|d nd }d |vr|d d ur|jStj||||||||| | S)Napi_keyheadersrrrrzhttp://zhttps://DAPI Base not set. Set one via completion(..,api_base='your-api-url') status_codemessagez/v1/chat/completions)rrrrrcomplete_input_dictinputradditional_argsstreamF)rdatar%T) r raw_responser logging_objr request_datarrrencoding) rvalidate_environment startswithr transform_requestpre_callrpostjsondumps iter_linestransform_response)rrrrrr*rr(rrcustom_prompt_dict logger_fndefault_max_tokens_to_samplercompletion_urlr&clientresponses j/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/oobabooga/chat/oobabooga.py completionr;sP&::'% ;G /0 !Z   E#&<r[s $'DAA)+W)+%+!%G G G DjG " G  G G G dD D D&D4Z D Dj D % DDr<