uPj1UdZ ddlZddlZddlmZddlmZddlmZddlZddl m Z m Z m Z m Z mZddlZddlmZmZddldd lmZdd lmZdd lmZdd lmZe Zeed <dZeed<dZeed<de fdZ de!de!fdZ"de!de!fdZ#de!de!fdZ$de%de!dzde!dzfdZ&de!de!de%fdZ'ejQdgd d!d"g#de!de d$e fd%Z)y)&z/ What is this? Logging Pass-Through Endpoints N) b64encode)Finalunquote) APIRouter HTTPExceptionRequestResponsestatus) SSRFError validate_url)*)user_api_key_auth)_safe_get_request_headers)_get_dynamic_logging_metadata)create_pass_through_routerouterdefault_vertex_configzhttps://cloud.langfuse.com_DEFAULT_LANGFUSE_HOSTrequestc|jt|jt|j |j t |jdS)N)methodurlheaderscookies query_params)rstrrrcopyrdictr)rs y/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/proxy/vertex_ai_endpoints/langfuse_endpoints.pycreate_request_copyr!#sE..7;;,W5::<??W112  valuereturnc2|} t|}||k(r|S|})Nr)r#previousdecodeds r _decode_to_convergencer(-s+H (# h N r"base_target_urlc|jds|jdsd|z} tj|}|jdvs |jst t j ddi|jrt t j ddit|S#t$r%}t t j dd|id}~wwxYw) Nzhttp://zhttps://errorInvalid Langfuse host:  status_codedetail)httphttpszInvalid Langfuse hostz*Langfuse host must not include credentials) startswithhttpxURL Exceptionrr HTTP_400_BAD_REQUESTschemehostuserinfor)r)base_urles r _normalize_langfuse_base_urlr<6s  & &y 1_5O5OPZ5[#o5 ))O4//x}}3345  33IJ  x=%  336qc:;   sB)) C2 CCendpointct|}td|Drttjddid|vs|j drttjddid|j dz}td|jdDrttjddi|S) Nc38K|]}t|dkyw) N)ord).0chars r z0_validate_langfuse_proxy_path..Ts 7d3t9r> 7sr+zInvalid Langfuse endpoint pathr-\z///c3$K|]}|dv yw)).z..N)rBsegments r rDz0_validate_langfuse_proxy_path..`s Jg7k ! Js)r(anyrr r6r2lstripsplit)r=decoded_endpoint endpoint_paths r _validate_langfuse_proxy_pathrPRs4X> 7&6 7733=>   #3#>#>t#D33=>  !1!8!8!==M J1D1DS1I JJ33=>   r"dynamic_host_supplieddynamic_langfuse_public_keydynamic_langfuse_secret_keyc|r%|r|sttjddi||fS|xs tjj d|xs tjj dfS)Nr+z@Dynamic Langfuse hosts must include dynamic Langfuse credentialsr-LANGFUSE_PUBLIC_KEY) secret_nameLANGFUSE_SECRET_KEY)rr r6litellmutils get_secretrQrRrSs r _get_langfuse_proxy_credentialsr\hs| *2M"77!cd +,GGG $bw}}'?'?La'?'b#bw}}'?'?La'?'b r"c`t|}tjt|}|j |}i}|r2t t ddr! tt|\}}||d<||fSt||fS#t$r%} ttjdd| id} ~ wwxYw)N)pathuser_url_validationTr+r,r-Host) rPr3r4r< copy_withgetattrrXr rr rr r6) r=r)rQrOr: updated_urlcustom_headers target_url host_headerr;s r _build_langfuse_proxy_targetrg|s 9BMii <_ MNH!+++?KN2G!N &23{3C&D #J "-v>)) { ^ ++ "77#:1#!>?  sA?? B- B((B-z/langfuse/{endpoint:path})GETPOSTPUTDELETEPATCHzLangfuse Pass-throughz pass-through)methodstagsfastapi_responsecJKddlm}|jjdxsd}|j dd}t j |}|jd}|jdd}t|d | d {}t|| }d } d } d } |F|j:|jjD]\} } | d k(r| } | dk(r| } | dk(s| } | d u}| xs"tjdtxst}t!|| | \}}t#|||\}}dt%|d|j'jdz}||d<t)|||t+|j,}||||d {}|S77 w)z Call Langfuse via LiteLLM proxy. Works with Langfuse SDK. [Docs](https://docs.litellm.ai/docs/pass_through/langfuse) r) proxy_config AuthorizationzBasic zutf-8:zBearer )rapi_keyN)user_api_key_dictrqlangfuse_public_keylangfuse_secret_key langfuse_host LANGFUSE_HOSTr[)r=r)rQascii)r=targetrdr)litellm.proxy.proxy_serverrqrgetreplacebase64 b64decodedecoderMrr callback_varsitemsosgetenvrr\rgrencoderrr)r=rrorqrv decoded_bytes decoded_strrwcallback_settings_objrRrSdynamic_langfuse_hostkvrQr)rxryretarget_headerslangfuse_combined_key endpoint_funcreceived_values r langfuse_proxy_routers8oo!!/28bGooh+G!++G4M&--g6K$Q'G%6wRYZaYbPc%dd@]+,A/3.2(,(-B-P-P-\)77==? *DAq))./+++./+o%()%  *$9#Dm?'3"J $,i;N:OqQdPe8f8m8m8o.p.w.w/$'rsD   GGH"BSN #u# <<####8CC,"%t"%t (,,, ,2 5 !> 2 II II  Ir"