uPjUddlZddlZddlmZmZmZddlZddlmZm Z er ddl m Z e Z neZ ddlmZddlmZdd lmZmZej*eZeed <d Zeed <d edefdZGddeZy)N) TYPE_CHECKINGAnyFinal)AllMessageValuesChatCompletionRequest)Logging)BaseLLMException)OpenAIGPTConfig)HuggingFaceError!_fetch_inference_provider_mappingloggerzhttps://router.huggingface.coBASE_URL model_urlreturnc|jd}|jdr|dz }|jds|dz }|S)N/z/v1z/chat/completions/v1/chat/completions)rstripendswith)rs q/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/huggingface/chat/transformation.py_build_chat_completion_urlrsO  %I% ((    1 2++ ceZdZdZ ddededeededededzd edzd efd Zd ed e dee jzd e fdZ dededzd edzfdZ dd edzdedzdededededzd efdZdedeedededed ef dZy)HuggingFaceChatConfigzQ Reference: https://huggingface.co/docs/huggingface_hub/guides/inference Nheadersmodelmessagesoptional_paramslitellm_paramsapi_keyapi_baserc.ddi}|d||d<i||}|S)Nz content-typezapplication/jsonzBearer Authorization) selfrrrr r!r"r#default_headerss rvalidate_environmentz*HuggingFaceChatConfig.validate_environment.s> ."   18 /BOO ,0W00r error_message status_codect|||S)N)r+messager)r )r'r*r+rs rget_error_classz%HuggingFaceChatConfig.get_error_classBsKX_``rbase_urlc|jdr|}|S|-tjdxstjdd}|S)z Get the API base for the Huggingface API. Do not add the chat/embedding/rerank extension here. Let the handler do this. zhttp://zhttps:// HF_API_BASEHUGGINGFACE_API_BASE) startswithosgetenv)r'rr/s r get_base_urlz"HuggingFaceChatConfig.get_base_urlEsL   3 4H yy/X299=SUW3XHrstreamc||}t|}ntjdstjdr?ttjdxsttjd}np|j dr|}t|}nQd}|j dd\}} d| vr6|} | dk(r | d|d } n| d k(r| d } n| d k(r| d } n| d } t d| }|jd}|S)zr Get the complete URL for the API call. For provider-specific routing through huggingface r2r3r1z1https://router.huggingface.co/v1/chat/completionsrz hf-inferencez/models/rnovitaz/v3/openai/chat/completionsz fireworks-aiz/inference/v1/chat/completions)rr6r7strr5splitrr) r'r#r"rr r!r9 complete_url first_part remainingproviderroutes rget_complete_urlz&HuggingFaceChatConfig.get_complete_urlQs  #L5lCL YY} %3I)Jryy78bC J`@arr _transform_messages) r'rrr r!rr@rA mapped_modelrBmodel_idprovider_mappings rtransform_requestz'HuggingFaceChatConfig.transform_request{s#   j )-`EH`P_`a a O + NNP Q    t 4 % C 3 I ) (H'H@J //&$XJ.MhZX # 09 )Y6`bjlt,L9L++X\+R)c xcSbcddr)NN)N)__name__ __module__ __qualname____doc__rKr=listrr)inthttpxHeadersr r.r8boolrDrRr&rrrr)sl##'(     t* (aSasaTTYTaTaMaafva # t d &#(*(t( (  (  (t ( (T!e!e'(!e !e  !e  !e !err)loggingr6typingrrrrYlitellm.types.llms.openairr*litellm.litellm_core_utils.litellm_loggingrLiteLLMLoggingObj LoggingClass)litellm.llms.base_llm.chat.transformationr openai.chat.gpt_transformationr common_utilsr r getLoggerrSr__annotations__rr=rrr&rrrgst ,, MW$LLF=N!!!(++1%1 # # seOser