uPj)UdZddlmZmZddlZddlmZddlmZddl m Z m Z ddl m Z mZmZmZdZeed <ehd Zeed <Gd d eZy)ah Google AI Studio Agents API configuration. Proxies the Gemini v1beta Agents API: POST /v1beta/agents create GET /v1beta/agents list GET /v1beta/agents/{name} get DELETE /v1beta/agents/{name} delete GET /v1beta/agents/{name}/versions list versions )AnyFinalN)verbose_logger)BaseAgentsAPIConfig) GeminiErrorGeminiModelInfo)AgentCreateResponseAgentDeleteResultAgentListResponseAgentVersionsResponse) base_agent instructionsbase_environment_GEMINI_AGENT_BODY_KEYS>api_keyagent_idapi_base is_public agent_name guardrails make_publiccost_per_querymax_iterationsagent_card_paramscustom_llm_providerinput_cost_per_tokenoutput_cost_per_tokenmax_budget_per_sessionprovider_agent_response"require_trace_id_on_calls_by_agent"require_trace_id_on_calls_to_agent_LITELLM_INTERNAL_KEYSc eZdZdZedefdZdedzdefdZdeded e e jzde fd Z dedzd e eefdefd Zd e eefd e eefde eeffd Zde j"ddfdZded e eefde eeffdZde j"dedefdZdedzd e eefdeee eefffdZde j"defdZdededzd e eefdeee eefffdZde j"dedefdZdededzd e eefdefdZde j"dedefdZdededzd e eefdeee eefffdZde j"dede fdZ!y)GeminiAgentsConfigz Configuration for the Google AI Studio (Gemini) native Agents API. Authentication uses x-goog-api-key, resolved from (in order): 1. litellm_params["api_key"] 2. GOOGLE_API_KEY env var 3. GEMINI_API_KEY env var returncy)Nv1beta)selfs n/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/gemini/agents/transformation.py api_versionzGeminiAgentsConfig.api_version@srNcJtj|d|jS)N/)r get_api_baser+)r)rs r* _base_urlzGeminiAgentsConfig._base_urlDs&!..x894;K;K:LMMr, error_message status_codeheadersc0t||t|S)Nmessager2r3)rdict)r)r1r2r3s r*get_error_classz"GeminiAgentsConfig.get_error_classKs !#M  r,litellm_paramsc*|j|dS)N/agentsr0)r)rr9s r*get_complete_urlz#GeminiAgentsConfig.get_complete_urlWs ..*+733r,ct|}d|d<|jd}|jdr |s tdtj|}|s td||d<|S)Nzapplication/jsonz Content-TyperrzWhen overriding api_base for Gemini agents, you must also supply an explicit api_key. Falling back to GOOGLE_API_KEY / GEMINI_API_KEY env vars with a custom api_base is refused to prevent leaking the shared provider key to arbitrary hosts.zRGoogle API key is required. Set GOOGLE_API_KEY or GEMINI_API_KEY, or pass api_key.zx-goog-api-key)r7get ValueErrorr get_api_key)r)r3r9explicit_api_keyrs r*validate_environmentz'GeminiAgentsConfig.validate_environment^s w-"4"0"4"4Y"?   j )2BQ  )445EFqr r$+ !r, raw_responsecd|jcxkrdks7nt|j|jt|jy)Ni,r5)r2rtextr7r3)r)rDs r*_raise_for_statusz$GeminiAgentsConfig._raise_for_statusxsG|//5#5$))(44\112 6r,namecd|i}tD]}|j|}||||<tjd||S)NrIz"GeminiAgentsConfig create body: %s)rr?rdebug)r)rIr9bodykeyvalues r*transform_create_requestz+GeminiAgentsConfig.transform_create_requestsS (.tn* "C"&&s+E !S  " A4H r,c<|j| |j}|j d||j d|dtjd|tdi|S#t$r'tjd|j d|i}YwwxYw)z Gemini returns: {"id": "my-agent", "base_agent": "waverunner", "system_instruction": "...", "base_environment": {...}} z9GeminiAgentsConfig: non-JSON create response (status=%d).idrIz&GeminiAgentsConfig create response: %sr() rHjson Exceptionrwarningr2 setdefaultrKr r)rDrIdatas r*transform_create_responsez,GeminiAgentsConfig.transform_create_responses |, #/#4#4#6D d# T +EtL"*T**  " "K(( $F; F{r,c|j| |j}tjd|t |j dg|j dS#t$ri}YOwxYw)Nz$GeminiAgentsConfig list response: %sagents nextPageToken)rcnext_page_token)rHrRrSrrKr r?)r)rDrWs r*transform_list_responsez*GeminiAgentsConfig.transform_list_responsesu |, $$&D CTJ 88Hb) HH_5   D sA%% A32A3c6|j|d|}|ifSN/agents/r<)r)rIrr9r_s r*transform_get_requestz(GeminiAgentsConfig.transform_get_requests' x01$@Bwr,c|j| |j}|jd||jd|dt j d|t di|S#t$rd|i}YWwxYw)uDSame shape as create response — Gemini returns "id" as identifier.rQrIz#GeminiAgentsConfig get response: %sr()rHrRrSrUrrKr rVs r*transform_get_responsez)GeminiAgentsConfig.transform_get_responses |, $$&D d# T +BDI"*T**  $ |, >  $ $  !dD99r,c|j|d|d}i}|jdr|d|d<|jdr|d|d<||fS)Nriz /versionsrZr[r\r]r^)r)rIrr9r_r`s r*transform_list_versions_requestz2GeminiAgentsConfig.transform_list_versions_requestsi x01$yI(*   k *!/ !F; F{r,c|j| |j}tjd||t |j dg|j dS#t$ri}YPwxYw)z Gemini returns: {"agentVersions": [{"agent": "waverunner", "name": "agents/.../versions/uuid", ...}]} z6GeminiAgentsConfig list_versions response for '%s': %s agentVersionsrd)agent_versionsre)rHrRrSrrKr r?rVs r* transform_list_versions_responsez3GeminiAgentsConfig.transform_list_versions_responsesy |, $$&D UW[]ab$88OR8 HH_5   D sA&& A43A4)"__name__ __module__ __qualname____doc__propertystrr+r0intr7httpxHeadersrSr8rr=rCResponserHrOr rXtuplerar rfrjrlrnr rqrsr rwr(r,r*r$r$6sSN#*NN       %    4*4S#X4 4c3hS#X c3h 4enn  S#X  c3h +nn++  +: * S#X  sDcN" #   nn     &*S#X  sDcN" # +nn++  +(;;*;S#X ; ; :nn : :  :$  * S#X  sDcN" #  nn     r,r$)r{typingrrrlitellm._loggingr+litellm.llms.base_llm.agents.transformationr litellm.llms.gemini.common_utilsrrlitellm.types.agentsr r r r r__annotations__ frozensetr"r$r(r,r*rs[  +KI"TS!*!.o ,o r,