yjQ dZddlmZddlmZddlmZGddeZeddd d d dd ZeedS)uGoogle Vertex AI provider profile. vertex: Gemini models via Google Cloud's OpenAI-compatible endpoint. Auth is OAuth2 — short-lived access tokens minted from a service-account JSON or Application Default Credentials (ADC), NOT a static API key. Token resolution and refresh live in ``agent/vertex_adapter.py``; runtime_provider.py calls it to obtain a fresh ``(token, base_url)`` pair, then hands the token to the standard OpenAI client as ``api_key``. Because the wire format is the OpenAI-compatible chat/completions surface, no message translation is needed — the only Gemini-specific concern is the ``thinking_config`` reasoning hook, which is emitted here exactly as the ``gemini`` provider does for its OpenAI-compat subpath (``extra_body.google.thinking_config``). ``auth_type="vertex"`` marks this as an OAuth-token provider (resolved specially, like bedrock's ``aws_sdk``) so it is never treated as an api_key provider that would mistake a credentials-file path for a key. )Any)register_provider)ProviderProfilec eZdZdZdddedzdedeeeffdZdddd d edzd edzd ede edzfd Z dS) VertexProfileuHVertex AI — reuse Gemini's thinking_config translation for extra_body.N) session_idrcontextreturnc ddlm}m}|dpd}|d}|||}|siS||}|siSddd|iiiS) zEmit ``extra_body.google.thinking_config`` for the OpenAI-compat Vertex surface, mirroring the ``gemini`` provider's behavior. r)_build_gemini_thinking_config"_snake_case_gemini_thinking_configmodelreasoning_config extra_bodygooglethinking_config)!agent.transports.chat_completionsr r get) selfrr r r rrraw_thinking_configrs M/home/ragecks/.hermes/hermes-agent/plugins/model-providers/vertex/__init__.pybuild_extra_bodyzVertexProfile.build_extra_bodys          G$$*";;'9::;;ECSTT" I<<=PQQ Ix*;_)MNOOg @)api_keybase_urltimeoutrrrcdS)zVertex's OpenAI-compat endpoint has no ``/models`` listing route; model discovery is not available. The setup wizard ships a curated list. N)rrrrs r fetch_modelszVertexProfile.fetch_models4s tr) __name__ __module__ __qualname____doc__strrdictrfloatlistr rrrrrsRR+/PPP 4ZP;>P c3hPPPP4##   t *    cT       rrvertex)z google-vertexz vertex-aiz gcp-vertexchat_completionsrz!https://aiplatform.googleapis.comzgoogle/gemini-3.6-flash)namealiasesapi_modeenv_varsr auth_typedefault_aux_modelN) r$typingr providersrproviders.baserrr)rrrr4s&''''''******$$$$$O$$$N   8   0/   &r