߲jI UdZddlmZddlZddlZddlZddlmZmZddl m Z ejdZ dZ iZded <d Zdd Zgd Zd ddgdfdddgdfddedfddedfgZddZd d!dZddZd"dZy)#aKimi Code and Z.AI endpoint auto-detection, LM Studio base-URL normalization. Re-exported from ``hermes_cli/auth.py`` (patch targets unchanged); origin helpers are imported lazily per function so ``hermes_cli.auth.`` patches still intercept and no cycle forms. ) annotationsN)DictOptional)httpxzhermes_cli.authi,zDict[str, float]_zai_probe_failed_untilzhttps://api.kimi.com/codingc@|r|S|r|jdrtS|S)zIKimi base URL from the key prefix; an explicit KIMI_BASE_URL always wins.zsk-kimi-) startswithKIMI_CODE_BASE_URL)api_key default_url env_overrides =/home/ubuntu/.hermes/hermes-agent/hermes_cli/auth_zai_kimi.py_resolve_kimi_base_urlrs'7%%j1!! )zglm-5.3z glm-5.3-flashzglm-5.2zglm-5.1z glm-5v-turbozglm-4.7globalzhttps://api.z.ai/api/paas/v4zglm-5Globalcnz$https://open.bigmodel.cn/api/paas/v4Chinaz coding-globalz#https://api.z.ai/api/coding/paas/v4zGlobal (Coding Plan)z coding-cnz+https://open.bigmodel.cn/api/coding/paas/v4zChina (Coding Plan)c x|\}}}}|D]} tj|dd|dd|ddddd gd | }|jd k(r!tj d |||||||dcStj d|||jy#t $r"} tj d||| Yd} ~ d} ~ wwxYw)zWProbe one Z.AI endpoint, trying its candidate models in order; None when none succeeds.z/chat/completionszBearer zapplication/json) Authorizationz Content-TypeFuserping)rolecontent)modelstream max_tokensmessages)headersjsontimeoutz(Z.AI endpoint probe: %s (%s) model=%s OK)idbase_urlrlabelz,Z.AI endpoint probe: %s model=%s returned %sz+Z.AI endpoint probe: %s model=%s failed: %sN)rpost status_codeloggerdebug Exception) r endpointr"ep_idr% probe_modelsr&rrespexcs r_probe_single_zai_endpointr11s+3(E8\5 [ [::*-.,3G9*=Oab$Q^dqwUxTyz D 3& GPXZ_`#ETYZZ LLGPUW[WgWg h [  [ LLFuVY Z Z [sAB)"B B9B44B9c " ddlm}m}|tt} tDcic]}|j t ||||d }}|jDcic]\}}|| c}} id fd } ||D]>} | j} | | || <| d} | *| c|jdS| d|jdScc}wcc}}w#t$rYTwxYw#|jdwxYw) z]Probe z.ai endpoints in parallel; first working one in ZAI_ENDPOINTS priority order, or None.r)ThreadPoolExecutor as_completed) max_workerscrtD]-}|r|djsy|dvs%|dcSy)Nr) ZAI_ENDPOINTSdone) require_doneepby_idresultss r _first_readyz)detect_zai_endpoint.._first_readyOsL$ *be (9(9(;a5G#"2a5>)  * rT)r9F)wait)r9boolreturnOptional[Dict[str, str]]) concurrent.futuresr3r4lenr7submitr1itemsresultr+shutdown)r r"r3r4poolr:futuresfr-r=futurerFwinnerr;r<s @@rdetect_zai_endpointrMEsC #m*< =D"bop\^4;;97BPRTUVRWWpp*1--/:ha:-/ #7+ F %/5GGFO,"t4F!  5 ! / 5 !5q:$   5 !sXC:#C  C:" C%/C:C+" C:.C: C: C:+ C74C:6C77C::Dc2ddlm}m}m}m}m}m}|r|S|s|Stj|jjdd} ||dxsi} | jd} t| trD| jdr3| jdd | k(rtjd | d| dSt j| d t#j"kDr|S||} | r| jds:tjd |t#j"t$zt | <|S| d| jd d | jdd | jdd | d} |5|}||dxsi}| |d<||d|d||dddtj+d| d| d| dS#1swY+xYw#t&$r }tj)d|Yd}~Rd}~wwxYw)zZ.AI base URL by probing endpoints; an explicit GLM_BASE_URL always wins. The detected endpoint is cached in provider state (auth.json) keyed on a hash of the API key so subsequent starts skip the probe. r)_auth_store_lock_load_auth_store_load_provider_state_save_auth_store_store_provider_staterMNzaidetected_endpointr%key_hashzZ.AI: using cached endpoint %sgz.Z.AI: probe failed, falling back to default %sr$rr&)r% endpoint_idrr&rWF) set_activezHZ.AI: could not persist detected endpoint (%s); will re-probe next startz$Z.AI: auto-detected endpoint %s (%s))hermes_cli.authrOrPrQrRrSrMhashlibsha256encode hexdigestget isinstancedictr)r*rtime_ZAI_PROBE_FAILURE_TTL_SECONDSr+warninginfo)r r r rOrPrQrRrSrMrWstatecacheddetectedrV auth_storestate_under_lockr0s r_resolve_zai_base_urlrlhs WW ~~gnn./99;CR@H !1!3U ; ArE YY* +F&$FJJz$:vzz*VX?Y]e?e 5vj7IJj!!""8S1DIIK?#7+H j1 E{S,0IIK:X,X)Z(dB9Ogr*X\\'25N h   ))+J3JFL" 4E 0 1 !*e5ERW X Z (  ) KK68I8T^K_` J  ) ) hacfgghs0G- .G!7G-!G*&G-- H6HHct|xsdjjd}dD]3}|j|s|dt | jd}n|xsddzS)aReturn the OpenAI-compatible LM Studio runtime base URL. LM Studio's native management API lives under ``/api/v1`` while its OpenAI-compatible chat endpoint lives under ``/v1``; users paste either form, so normalize before the SDK appends ``/chat/completions``. rX/)z/api/v1z/api/v1Nzhttp://127.0.0.1:1234ro)strstriprstripendswithrC)r%rootsuffixs r$_normalize_lmstudio_runtime_base_urlrvst x~2  $ $ & - -c 2D, == 3v;,'..s3D   ++u 44r)r rpr rpr rpr@rp)r rpr,tupler"floatr@rA)g @)r rpr"rxr@rA)r%rpr@rp)__doc__ __future__rloggingr\rctypingrrhermes_cli.auth_constantsr getLoggerr)rdr__annotations__r r_ZAI_CODING_PROBE_MODELSr7r1rMrlrvrrrs # !+   , -"%,.).3i4gY(S VXnoCE]_tu  ( "F0 f 5r