uPjo UddlmZddlmZddlmZddlmZddlm Z Gdde Z d e d e d zfd Z d Zeed<de d e fdZGddZy ))Final)Headersget_secret_str)AllMessageValues)BaseLLMExceptionc eZdZy)FireworksAIExceptionN)__name__ __module__ __qualname__k/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/fireworks_ai/common_utils.pyr r srr litellm_paramsreturnNc|}dD]"}|j|}|st|cS|jd}t|tr|jd}|r t|Sy)a9 Session id to send as `x-session-affinity`, or None when the caller gave none. Deliberately does not fall back to `litellm_trace_id`: that is generated per request (`str(uuid.uuid4())` when absent), so using it pins every request to a different Fireworks node and prompt caching never hits. )litellm_session_id session_idmetadatarN)getstr isinstancedict)rparamskeyvaluers rget_fireworks_session_idrsj#F3 3 u: jj,H(D! \* u:  rzFW-'AZURE_FOUNDRY_FIREWORKS_MODEL_ID_PREFIXmodelc|jd}|jdtfsd|vr|S|jdrd|S|jdrd|Sd|S) Nz fireworks_ai/z accounts/#)zrouters/zmodels/zaccounts/fireworks/z-fastzaccounts/fireworks/routers/zaccounts/fireworks/models/) removeprefix startswithr endswith)r!strippeds rresolve_fireworks_resource_namer('sx((9HK)PQRVY]eVe23$XJ//!,XJ77 'z 22rceZdZdZdededeezdefdZ dedzdedzfd Z dded ed e e d ed ededzdedzdefdZ ded edefdZy)FireworksAIMixinzD Common Base Config functions across Fireworks AI Endpoints error_message status_codeheadersrct|||S)N)r,messager-)r )selfr+r,r-s rget_error_classz FireworksAIMixin.get_error_class7s##!  rapi_keyNcr|xs2tdxs%tdxstdxs td}|S)NFIREWORKS_API_KEYFIREWORKS_AI_API_KEYFIREWORKSAI_API_KEYFIREWORKS_AI_TOKENr)r0r2dynamic_api_keys r _get_api_keyzFireworksAIMixin._get_api_key>sL!(" . / 445 434 423  rr!messagesoptional_paramsrapi_basec|j|}| tddd|i|}td|Drinddi} |ji|| |S)NzFIREWORKS_API_KEY is not set AuthorizationzBearer c3BK|]}|jdk(yw)z content-typeNlower.0rs r z8FireworksAIMixin.validate_environment..WsLciik^3Lz Content-Typezapplication/json)r9 ValueErrorany_add_session_affinity_header) r0r-r!r:r;rr2r< auth_headerscontent_type_headers rvalidate_environmentz%FireworksAIMixin.validate_environmentGs{##G, ?;< <.''0COwO L|LLBSacuRv 001XL1XDW1XZhiircXtd|Dr|St|}|s|Si|d|iS)Nc3BK|]}|jdk(yw)x-session-affinityNr@rBs rrDz@FireworksAIMixin._add_session_affinity_header..\sFssyy{22FrErN)rGr)r0r-rrs rrHz-FireworksAIMixin._add_session_affinity_header[s: FgF FN4^D N<'</<rXsb76; + TcDj*27'633333/=/=r