鎒j9.dZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlZddlZddlmZmZmZmZmZddlmZddlmZddlmZddlZddl m!Z!ddl"mZddl#m$Z$dd l%m&Z&dd l%m'Z'dd l(m)Z*dd l(m+Z+dd l,m-Z-ddl,m.Z/e&j0Z0e&j1Z1e&j2Z2dZ3dZ4ej5j6e!j7j6ej5j8e!j7j8ej5j9e!j7j9iZ:dej5dzde!j7j5fdZ;ej<j=e!j>ej<j?e!j@iZAdej<dzde!j<fdZBdejCde!jCfdZDdejEdzde!jEfdZFdejGdzde!jGdzfdZHdejIdzde!jIdzfdZJdeKejLdeKe!jMfdZN d6ded efd!e/jOdzde!jPfd"ZQe jRd#ZSd$eTdeTfd%ZUd&ejVdeWeTzdzfd'ZXGd(d)e$jYZZd*ej[de!j\j]fd+Z^deTfd,Z_d-Z`d.eaeTeTfdzdeTfd/ZbebZcd0ejdde!jdfd1Zed2eTejfzdzde!jfdzfd3ZgGd4d5e$jhZidS)7z0Local connection for the Google Antigravity SDK.N)Any AsyncIteratorCallableSequencecast)types) json_format)override)localharness_pb2) connection)event_processor)local_connection_config) hook_runner)policy) schema_utils tool_runnermodereturncP|tvr t|StjjSN)_SESSION_CONTINUATION_MODE_MAPr HarnessConfig%SESSION_CONTINUATION_MODE_UNSPECIFIED)rs cC:\Projects\DSCons\.venv\Lib\site-packages\google\antigravity\connections\local\local_connection.py"to_proto_session_continuation_moderGs' +++ )$ //  ' MMbehaviorcJ||tvr t|StjSr)_AGENT_BEHAVIOR_MAPr AGENT_BEHAVIOR_AUTONOMOUS)r s rto_proto_agent_behaviorr$Ys)h*=== x ((  33r model_typec|tjjkr tjS|tjjkr tjStjSr)r ModelTypeTEXTr MODEL_TYPE_TEXTIMAGEMODEL_TYPE_IMAGEMODEL_TYPE_UNSPECIFIED)r%s rto_proto_model_typer-as?5?'''  ++5?(((  ,,  00roptionsc"tj}|ry|jrt|jj|_|jrMt |jdrt|jj|_nt|j|_|S)zHConverts SDK GeminiModelOptions to localharness protobuf representation.value)r GeminiModelOptionsthinking_levelstrr0 service_tierhasattr)r.protos rbuild_gemini_options_protor7ks  - / /% 7? !7!=>>e7 %w / /7 !5!;<< !566 ,rconfigc|sdStj}|jrF|jd}|r)|jtjdi||jrF|jd}|r)|jtjdi||ds|dsdS|S)z4Builds a RetryConfig proto from a RetryConfig model.NT exclude_none api_retrymodel_output_retry) r RetryConfigr< model_dumpCopyFromModelAPIRetryConfigr=ModelOutputRetryConfigHasField)r8r6api_data output_datas rbuild_retry_config_protorG{s   4  & ( (% ***==H o  . : : : : +66D6IIK ''  1 @ @K @ @ .. % %nn122 4 ,rc`|sdS|d}|sdStjdi|S)z6Builds a BudgetConfig proto from a BudgetConfig model.NTr:r>)r@r BudgetConfig)r8datas rbuild_budget_config_protorKsK  4     - -$  4  & . . . ..rmodelsc`g}|D]'}tj|jpdd|jD}t |jtjrtj|jjpd|jjpi|jj pd}|jj rQ|jj dr1|j t|jj |j |nt |jtjrtj|jjpd|jjpi|jjpd|jjpd|jj pd}|jj rQ|jj dr1|j t|jj |j |n$t'dt)|j||)|S) z)r-).0ts r z&build_models_proto..s!777!"1%%777r)namer)base_url http_headersapi_keyTr:)rTrUprojectlocationrVzUnrecognized endpoint type: )r ModelConfigrSr isinstanceendpointGeminiAPIEndpointrTrUrVr.r@rAr7gemini_api_endpointVertexEndpointrWrXvertex_endpoint ValueErrortypeappend)rLprotosmr6api_endpoint_protovertex_endpoint_protos rbuild_models_protorgsJ & ##a  ( V\r77qw777   E!*e566J+=:&,"z.4"*$*    2 = =!>!!  "++ &qz'9 : :    (();<<<< AJ 4 5 5J.=:&,"z.4"*$*:&,"*$*     2 = =!>!!  %.. &qz'9 : :    $$%:;;;; Hd1:6F6FHH I II MM% -rfn.rc t|tjrVtjt |dd|jpdtjtj |j S|}t |ddpt|j }|||jvr||}t!|ds|fd}||_ t#|dt dd t#|dt%jn7#t(t*f$r#t#|dt diYnwxYw|}t,j|d }|jr|jd }n|jr|j}nd id}tj|j|jpdtjtj |S)aConverts a Python callable to a localharness Tool proto. Uses google.genai.types.FunctionDeclaration for schema extraction. If a ``tool_runner`` is provided, the runner's ``get_public_callable`` is used to strip injectable parameters (e.g. ``ToolContext``) from the schema so the model never sees them. Args: fn: The Python callable to convert. tool_runner: Optional ToolRunner that owns schema-hiding logic. Returns: A localharness_pb2.Tool proto. __name__rN)rS descriptionparameters_json_schemaNc|i|Srr>)argskwargsorig_fns rwrappedz'callable_to_tool_proto..wrappeds Wd %f % %%r__doc__ __signature____annotations__ GEMINI_API)callable api_optionTr:object)ra properties)rZt_runnerToolWithSchemar Toolgetattrrrjsondumpsrnormalize_schema input_schemararjtoolsget_public_callabler5setattrinspect signaturer` TypeError genai_typesFunctionDeclarationfrom_callable_with_api_option parametersr@rlrSrk)rhr target_fn tool_namerqdeclrrps @rcallable_to_tool_protors@$H+,,  RR ( (J$"#z  )"/ : :     )b*d++@tBxx/@)k.?!?!?// ::I J ' 'G&&&&&!G GYD A ABBB g(9'(B(BCCCC  " $gg7H"&M&MI  ( F F G  $ _6+++>>JJ "6,JJ""55J   9"(b!Z  ' 3 3   s8#D1EEz%[\x00-\x08\x0b\x0c\x0e-\x1f\x7f-\x9f]textcp|sdStd|}|sdS|S)zHStrips null bytes and replaces dangerous control characters with spaces.rN )_CONTROL_CHAR_REsubstrip)r sanitizeds r_sanitize_promptrsA  2""3--)    3 rect|dd}|t|dr|jSt|dd}|t|dr|jS||dSt|ddS)zMSafely retrieves the WebSocket close code across websockets library versions.rcvdNcodesenti)r}r5r)rrrs r_get_ws_close_coder"s~ FD ! !$ '$// 9 FD ! !$ '$// 9 \dl 4 FD ! !!rceZdZdZ d0dejedzdedej dzde j dzde e jdzdeeefdzd ejdzd eed fdzd e jdzd eee jfdzfdZedefdZede e jfdZeedejdzfdZedefdZede jfdZedeee jffdZede jfdZde j dzdeddfdZ!de"e#j$fdZ%d1dZ&d1dZ'd1dZ(d1dZ)d1dZ*d e+j,ddfd!Z-d"eddfd#Z.ede/j0efd$Z1ede/j2fd%Z3ededzfd&Z4e4j5d'edzddfd(Z4d)e+j6ddfd*Z7d+e j8deeeffd,Z9d-e+j:ddfd.Z;d-e+j:ddfd/Z|jjS)z;Returns True if the connection is idle and ready for input.)ris_idleis_setrs rrzLocalConnection.is_idle^s ? " ) ) + ++rc|jS)zAReturns the pre-existing session steps restored during handshake.)rrs r_initial_historyz LocalConnection._initial_historycs   !!rc|jS)z4Returns the debug configuration for this connection.rrs rrzLocalConnection.debug_configh   rc|jjpdS)z3Returns the conversation identifier, if one exists.rNrmain_trajectory_idrs rconversation_idzLocalConnection.conversation_idns ? - 33rc|jjS)z6Returns total cumulative token usage from the backend.)rcumulative_usagers rrz LocalConnection.cumulative_usagess ? ++rc|jjS)z?Returns per-trajectory cumulative token usage from the backend.)rtrajectory_usagesrs rrz!LocalConnection.trajectory_usagesxs ? ,,rc|jjS)zAReturns the stop reason of the most recent turn from the backend.)r_last_turn_stop_reasonrs rrz&LocalConnection._last_turn_stop_reason}s ? 11rpromptrocvKd|_|j|dg}nAt|tjjrt|ttfs|}n|g}tj d|D}tj |}| |d{VdS)zSends a prompt to the agent. Args: prompt: The user prompt or content to send. **kwargs: Strategy-specific options. FNrNc,g|]}t|Sr>)to_proto_input_content)rPcs rrRz(LocalConnection.send..s!???Q%a((???r)parts) user_input) rrreset_for_turnrZrabcrr3bytesr UserInput InputEventr)rrro content_list user_input_pbevents rsendzLocalConnection.sends#DO""$$$ ~Tll FKO4 5 5je ??llXl$.??,???M  '= A A AE   ' ''''''''''rcK|jrtdd|_ |jrB|jjr$|jrtj d|_dS |jrB|jjr$|jrtj d|_dS|jj d{V}t|tr||tj ur|$|jrtj d|_dSt|tr|ttj|}|WV|jtjjkre|jtjjkrKt-|dd}|dvrtj|jpdt3jd ||jp#d|_wxYw) z/Receives steps as they complete from the agent.zFConcurrent receive_steps() calls are not supported on this connection.TFN http_coder)iiizSystem error occurred.zSystem step error (HTTP %s): %s)r RuntimeErrorrr step_queueemptyrrAntigravityCancelledErrorgetrZ Exceptionr IDLE_SENTINELrLocalConnectionStepstatus StepStatusERRORsource StepSourceSYSTEMr}AntigravityConnectionErrorerrorloggingwarning)rstep_objrs r receive_stepszLocalConnection.receive_stepss<     D,! $/4::<<  ! 2/11 1P!dM$ < DO6<<>>   # 4133 3 D!dA37799999999 h * * . 4 4 4     # 4133 3 ,!d+ h * * .;XFF Ou/5 5 55#3#:::h Q77) / ) )2:":  O19hn   E$L!d    s ?G*AG4A&G$B7G G$c:K|jjd{V|jjsV |jjn#t j$rYdSwxYw|jjTdSdS)z)Blocks until the connection becomes idle.N)rrwaitrr get_nowaitr QueueEmptyrs r wait_for_idlezLocalConnection.wait_for_idles / ! & & ( ((((((((o(..00 "--////   o(..00sA%%A87A8czfd}tj|dd}||_dS)zStarts a background daemon thread that drains the harness stderr. Args: stderr_stream: The binary stderr stream from the harness process. c D]Z}|dd}j|t jd|[dS#t $rYdSwxYw)Nutf-8replace)errorszharness stderr: %s)decoderstriprrbrinfor`)raw_lineliner stderr_streams r_drainz4LocalConnection._start_stderr_reader.._drains % 3 3H;;BBDD$   # #D ) ) ) ,+T 2 2 2 2 3 3     sAA"" A0/A0Tzharness-stderr)targetdaemonrSN) threadingThreadstartr)rr r rQs`` r_start_stderr_readerz$LocalConnection._start_stderr_readersW       t:JKKKAGGIIIDrc`Kd|_d}|jry|jjrm |t jdd{V|jjd{Vn#t$r }|}Yd}~nd}~wwxYw |j d{V|j  |j d{Vn#tj$rYnwxYw tj|jdd{Vn#tj$rYnwxYw|jr*|jjr|jj|jr |jt*n#t,j$r|j |jdnI#t,j$r7|j|jdYnwxYwYnwxYw||dS#||wxYw)z5Tears down the harness connection in a careful order.TN)session_end_requestg?)timeout)rron_session_end_hooksrr rrsession_end_donerrcancel_background_tasksrcancelrCancelledErrorwait_forrclose TimeoutErrorrstdin_PROCESS_WAIT_TIMEOUT_SECONDS subprocessTimeoutExpired terminatekill)r hook_errorrs r disconnectzLocalConnection.disconnectsDJ T.C$$  'D A A A         o.335555555555  ! O 3 3 5 55555555       #      tx~~//===========  !      $4=.$ !!###  * * -  %B  C C C C( * * * - ! ! # # # * M  q  ) ) ) )* * * * M   M  q  ) ) ) ) ) *  *    sAA33 B =BB  8H' CH'C&#H'%C&&H'*3DH'D0-H'/D00;H', F H' (H6GHAHHHHH'HH''H-cxKd|_tjd}||d{VdS)zCancels the current turn.T) halt_requestN)rr rrrrs rrzLocalConnection.cancel"sM!D  'T : : :E   ' ''''''''''rcK |j23d{V}tjd|tj}t j|||j|d{Ve6n#tj $r}t|}|j rtjd|nod |jpd}d|d|}tj||jjt%j|d{VYd}~nid}~wt($rY}tjd||jjt%jd |d{VYd}~nd}~wwxYw|jjt,jd{VdS#|jjt,jd{VwxYw) zAReads OutputEvents from the WebSocket and delegates to processor.NzRAW WS MSG: %sz,WebSocket closed (code %s); normal shutdown. z(no stderr output)z3Harness process exited unexpectedly (WS close code z). Harness stderr: zError in reader loop: %szError in reader loop: )rrrr OutputEventr Parser process_event websocketsConnectionClosedrrjoinrrrputrrr exceptionr CLOSE_SENTINEL)rraw_msgrr close_code stderr_tail error_msgs rrzLocalConnection._ws_reader_loop(sK83333333 %w/// ,..'5)))o++E2222222222 $8  &   %a((j    CZPPPPii 233K7K  0* 0 0"- 0 0   i   o(,,  ,Y 7 7           2A666 O & * *  *+GA+G+G H H   O & * *?+I J JJJJJJJJJJDO & * *?+I J JJJJJJJJJsNA3A0A A31F<3FBD"F<" F/AF>F<FF<<1G-rcnK|jtj|d{VdS)z0Helper to send an InputEvent over the WebSocket.N)rrr MessageToJsonr)s rrz!LocalConnection._send_input_eventIs= (-- 1%88 9 9999999999rcontentcjKtj|}||d{VdS)z%Sends a trigger message to the agent.)automated_triggerN)r rr)rr;rs rsend_trigger_notificationz)LocalConnection.send_trigger_notificationMsE  '' B B BE   ' ''''''''''rc|jjSr)rrrs r _step_queuezLocalConnection._step_queueSs ? %%rc|jjSr)rrrs r_is_idlezLocalConnection._is_idleWs ? ""rc|jjSrrrs r_main_trajectory_idz#LocalConnection._main_trajectory_id[s ? --rvalc||j_dSrr)rrEs rrDz#LocalConnection._main_trajectory_id_s),DO&&&r tool_callcJK|j|d{VdS)z-Handles tool execution and hook interception.N)rhandle_tool_call)rrGs r_handle_tool_callz!LocalConnection._handle_tool_callcs6 / * *9 5 5555555555rresultc6|j|S)z5Converts a ToolResult to a dictionary representation.)rtool_result_to_dict)rrKs r_tool_result_to_dictz$LocalConnection._tool_result_to_dictis ? . .v 6 66r step_updatecJK|j|d{VdS)z+Handles question requests from the harness.N)rhandle_question_requestrrOs r_handle_question_requestz(LocalConnection._handle_question_requestms6 / 1 1+ > >>>>>>>>>>rcJK|j|d{VdS)z4Handles tool confirmation requests from the harness.N)r handle_tool_confirmation_requestrRs r!_handle_tool_confirmation_requestz1LocalConnection._handle_tool_confirmation_requestss6 / : :; G GGGGGGGGGGr)NNNNNNNNrN)=rj __module__ __qualname__rrr!Popenrrrz ToolRunnerh_runner HookRunnerrrStepdictr3r DebugConfig UsageMetadatarpropertyboolrrr rrrr StopReasonrContentrrr rrrrr&rrr rrr>rQueuer@EventrBrDsetterToolCallrJ ToolResultrN StepUpdaterSrVr>rrrr/s11 150459#'48>B26IM*8*8&-*8 *8&- *8 &- *8   +d2 *8 S>D *8*T1*8sO34t;*8(4/*8"&c5+>&>!?$!F*8*8*8*8X ,t,,, 8, " 4""" 8"  J2T9 8 8 4s444 84 , 3,,, 8, -c5+>&>!?--- 8- 2e&6222 82(u}t3(s(t((((26!_896!6!6!6!p(0000d(((( KKKKB:+;+F:4::::(S(T((((  &7=-&&& 8& # ### 8# .3:... 8.-S4Z-D----6'06 6666 7)97d38n7777?)4? ???? H)4H HHHHHHrrr;ct|tr-tjt |St|t jrEtj|j}tj|St|t j t j t j t j f}|rQtj |j|j|j}tj|St#dt%|)z3Converts dynamic prompt fragments into proto Parts.rrS) slash_command) mime_typerJrk)mediaz!Unsupported prompt content type: )rZr3r rPartrr SlashCommandrSImageDocumentAudioVideoMediarprJrkrra)r;sc_pbis_semantic_mediamedia_pbs rrrzs%K  % * *0@0I0I * J JJ+,,@  & 3 3 \ 4  E  % * * * ? ??   U^U[%+F;)//# \'0H  % * * * : ::Ed7mmEEFFFrcx tjdS#tjj$rYdSwxYw)z2Returns the version of the Google Antigravity SDK.google-antigravityz 0.0.0-dev) importlibmetadataversionPackageNotFoundErrorr>rr_get_sdk_versionrsG   % %&: ; ;;   0 ;;s !99ANTIGRAVITY_HARNESS_PATHrc|rt|vr |tStjtx}r|S tjd}|jr|jD]}t| dd}| dratj t| }tj |r|cSn'#tjjt t"f$rYnwxYw t$jdkrdnd}ttjd|}tj |r|Sn#t,t"t.f$rYnwxYwt1jd x}r|St5d td ) z-Returns the default localharness binary path.r}\/)z#google/antigravity/bin/localharnessz'google/antigravity/bin/localharness.exewin32zbin/localharness.exezbin/localharnesszgoogle.antigravity localharnessz[Could not find default localharness binary. Please specify binary_path explicitly, set the z environment variable, or ensure it is in your PATH. Note: If you are running from the root of the repository, the local source tree might shadow your pip-installed package and prevent resource discovery.)_HARNESS_PATH_ENV_VARosenvironrr~r distributionfilesr3rendswithpathabspathlocateexistsrr`AttributeErrorsysplatform resourcesjoinpath ImportErrorKeyErrorshutilwhichr)r harness_pathdistfnormalized_path binary_pathsuffixrs r!_get_default_binary_path_externalrs & "c ) ) $ %%Z^^$9:::\     * *+? @ @D z z!a&&..s33  # #%    AHHJJ88+ W^^K ( (       1:~ N   D  <7 " "   !!"677@@HHK w~~k""  ~x 0   D \. ) ))T K   s,CD D !D0/D04A3F))GG server_cfgc t|tjrYtj|j|jpg|jpg|jpdtj |j |j |j piSt|tj rStj|j|jpg|jpg|jpdtj|j|jpiSt#dt%|)z7Converts an McpServerConfig to a McpServerConfig proto.r)commandrnr)rS enabled_toolsdisabled_toolstimeout_secondsstdio)urlheaders)rSrrrhttpzUnknown McpServerConfig type: )rZrMcpStdioServerr McpServerConfigrSrrrMcpStdioTransportrrnrMcpStreamableHttpServerMcpHttpTransportrrr`ra)rs r_to_mcp_server_protors   E011  + _ .4"!06B"27a0&$"      *e;<<   + _ .4"!06B"27a  .&,"       FD4D4DFFGGGr instructionsc|sdSt|tr)tjtj|g}t j}t|tjrR|j t jt j |j gnt|tjrpt j }|j r |j |_|jD](}|j|j|j)|j ||S)aConverts string or SystemInstructions model to localharness SystemInstructions proto. String instructions and TemplatedSystemInstructions map to AppendedSystemInstructions (appending to default agent system instructions). CustomSystemInstructions map to CustomSystemInstructions (replacing default system instructions). Args: instructions: System instructions as string or SystemInstructions object. Returns: The localharness SystemInstructions proto, or None if instructions is empty. Nr;sectionsrm)part)titler;)rZr3rTemplatedSystemInstructionsSystemInstructionSectionr SystemInstructionsCustomSystemInstructionscustomrArrrAppendedSystemInstructionsidentitycustom_identityrappended_sectionsaddrr;appended)rr6rsecs rto_system_instructions_protorsZ  4 c""40FFFGL  - / /% e<==& L1 9>>%*?   , ABB&:<LocalConnectionStrategy._to_subagent_system_instructions_protorrcB|||}d}d}g}|?t|ddotjj|v}t|dd}|jpg}n|stjj|v}t j||}|||_d}|"t|ddpt|dd}||j nd} |+|j $tt|j d znd } t j |t jtjj|v t jtjj|v t jtjj|v| | t jtjj|v t jtjj|v t jtjj|v t jtjj|v t jtjj|v t jtjj|v t j tjj!|v t j"tjj#|v  S)NrFrTmax_subagent_depth)enabledallowed_subagentsrun_command_config run_commandir)r)renable_daemon_commandsmax_timeout_ms) rfinduser_questionsr file_edit view_file write_to_file grep_searchlist_dirgenerate_image search_webread_url_content)$rr}rrSTART_SUBAGENTrr SubagentsConfigmax_nesting_depthenable_daemonsrintroundHarnessSideToolsFindToolConfig FIND_FILEUserQuestionsConfig ASK_QUESTIONRunCommandToolConfig RUN_COMMANDFileEditToolConfig EDIT_FILEViewFileToolConfig VIEW_FILEWriteToFileToolConfig CREATE_FILEGrepSearchToolConfig SEARCH_DIRListDirToolConfigLIST_DIRGenerateImageToolConfigGENERATE_IMAGESearchWebToolConfig SEARCH_WEBReadUrlContentToolConfigREAD_URL_CONTENT) rrr active_toolssubagent_enabled max_depthrsubagents_proto run_cmd_cfg enable_daemon timeout_mss r_to_harness_side_tools_protoz4LocalConnectionStrategy._to_harness_side_tools_protos --c{-KKLI  &8$??   +| ;#3T::i/52 K+:lJ&6 +O*3o'K C!5t<< }dAAk'2&= ""5  "{'B'N E+-4 5 5666   ,!  ,&0L@   (;&3|C   %9&2lB#0%   #5&0L@   #5&0L@   '<&2lB   %9&1\A   "3&/<?   (?&5E   $7&1\A   *B&7<G   E% % % %rall_tool_protosc g}|jD]}}|jp0tjtj}g}|jpgD]}t|trK|}||vr| ||9| tj |bt|r6t||j}|||j<| |t!d|jd|| tj|j|j||j||d|t-|j|S) zCResolves and builds CustomAgent configuration protos for subagents.rrnrzInvalid tool type in subagent 'z' tools list: Tr)rSrkrharness_side_toolsragent_behavior)r capabilitiesrrrrrrZr3rbr r|rvrrrSr` CustomAgentrkrrr4r$r8) rr5custom_agents_protossubagentr9resolved_subagent_toolstoolrSr6s r_build_custom_subagents_protosz6LocalConnectionStrategy._build_custom_subagents_protoss O&&*e.H*4466///l!#.&B  $ dC  $ _ $ $ # * *?4+@ A A A A # * *+;+@d+K+K+K L L L L d^^ (4;LMMM%(-/%* % ! ( ( / / / /    !!  &="."&"M"M.##"&!B!BD"C"",4-  rci}|jrA|jjD]"}t||j}|||j<#g}|j|jD]}t |trI||vr|||7|tj |`t|r5t||j}|||j<||n|jrt|j D]x}|jpgD]l}t |tr|-t|r0t||j}|jmyfd|D}||j} g} |jrt'|j} d|jD} ||j} d|jpgD} |}||}tj|| |jpdt9|j| | |jpg| |jjpd|jj pd|j!pJttEj#d d z d z $%| ||tM|jj' }|j(r0tS|j(}|r|j*+||j,r0t[|j,}|r|j.+||j/r;taj1|j/\}|_2|j3+||S) z>Translates Pydantic config objects into a HarnessConfig proto.rNrnc"g|] \}}|v | Sr>r>)rPrSr6subagent_tool_namess rrRzALocalConnectionStrategy._build_harness_config..Ds3dE , , ,  , , ,rc g|]N}tjtjtj|OS)) directory)filesystem_workspace)r WorkspaceFilesystemWorkspacepathlibPathas_posix)rPps rrRzALocalConnectionStrategy._build_harness_config..Qsh  "!1!E!,q//2244"""   rc,g|]}t|Sr>)r)rPss rrRzALocalConnectionStrategy._build_harness_config..^s.$%QrrNr~z.gemini antigravity)rr cascade_idrrLrrr7compaction_thresholdfinish_tool_schema_jsonrr enabled_hookscustom_subagentsr8)4rrvaluesrrSrrZr3rbr r|rvrrritemsrrrrgrr4rr_get_enabled_hooksr?rrrrrrQrRrrHrI expanduserresolver$r8rrGrrArrKrrr_to_policy_config_protor policy_config)rr5rhr6root_tool_protosr>sarQsubagent_protosystem_instructions_proto models_protosworkspace_protosr7mcp_server_protosrSr;harness_config retry_proto budget_protor[rBs @r_build_harness_configz-LocalConnectionStrategy._build_harness_config#sdO ,!'..00,,"&rt7HIII&+ ## {+ ) )$ dC  ) _ $ $  # #OD$9 : : : :  # #$4$9t$D$D$D E E E E d^^ )(4;LMMM%(-/%* %  ! !% ( ( ( )   EE99"R 9 9A 3   9  # #A & & & &{{ 93t0N # #N$7 8 8 8 9,2244 !% B B !!!M |7(66m ! :: !)-):)@b++--M>>OO%35(.B"D  +# # #'-2-  % : ?a  % = C'  \#   *] : Z\\ WYY  &#-.  % 4  7N> :,T-?@@k :#,,[999 <.t/BCCl <$--l;;; ~;060N .11-mT-"++M::: rc g}|js|S|jjtjf|jjtjf|jjtjf|jjtj f|jj tj f|jj tj f|jjtjf|jjtjfg}|D]\}}|r|||S)zAReturns a list of proto enum IDs for registered hook collections.)ron_session_start_hooksr LIFECYCLE_HOOK_ON_SESSION_STARTrLIFECYCLE_HOOK_ON_SESSION_ENDpre_turn_hooksLIFECYCLE_HOOK_PRE_TURNpost_turn_hooksLIFECYCLE_HOOK_POST_TURNpre_tool_call_decide_hooksLIFECYCLE_HOOK_PRE_TOOLpost_tool_call_hooksLIFECYCLE_HOOK_POST_TOOLon_tool_error_hooksLIFECYCLE_HOOK_ON_TOOL_ERRORon_compaction_hooksLIFECYCLE_HOOK_ON_COMPACTIONrb)rrS hook_mapping hooks_list hook_types rrWz*LocalConnectionStrategy._get_enabled_hookss M      4  <   2  :   ,  4   -  5   8  4   2  5   1  9   1  9 ;!LD".(( I (Y''' rc|jS)z2Returns the debug configuration for this strategy.rrs rrz$LocalConnectionStrategy.debug_configrrc\t|dr|jtd|jS)z#Returns the established Connection.rNz5Connection not established. Use as a context manager.)r5rrrs rconnectzLocalConnectionStrategy.connects< 4 ' '4+;+C  A    rc|jsdS|jD]t}|jrO |j$#t$r'}t jt ||d}~wwxYwt jd|jddS)z7Validates that all required configurations are present.NzModel 'z#' must have an endpoint configured.)rr[validate_endpointr`rAntigravityValidationErrorr3rS)rrdrs r_validate_connectionz,LocalConnectionStrategy._validate_connections <  f \      @ * & & ( ( ( ( @ @ @0Q88a ? @. Aaf A A A      s5 A&"A!!A&portrVrcKttD]|}dD]W}d|d|d} tj|d|idd{V}||fccS#ttjf$r }|} Yd}~Pd}~wwxYwt jdd |zzd{V}|d } |j ,|j d } td |d td| | )z@Attempts to connect to the local harness WebSocket with backoff.) localhostz 127.0.0.1zws://:rzx-goog-api-keyN)additional_headersmax_sizeg?rNrz"Failed to connect to WebSocket at z after z attempts. Stderr: ) range_MAX_WEBSOCKET_CONNECT_RETRIESr/r|OSErrorWebSocketExceptionrsleepr$stderrreadrr) rrrVrattempthostws_urlrrlast_exceptionrs r_connect_websocketz*LocalConnectionStrategy._connect_websockets788 . .,  $'''''' '"2G!<         " V     67   ......  M#G, - --------- LLNNN F~!~""$$++G44f  HV H H * H H?E H H   s"AA0$A++A0c $ K||}t}tjd|t jt jt j }|j r#d|j Dni}tj |j pd||}|j itj|nd}t!j|jgt jt jt j|}|}|jJ|jJ|jJ|jt3jdt7||z|j|jd } | s>|jd } t?d | t3j d| d } tj!} | "|j| |#| j$| j%|d{V\} } tj&| }| 'tQj)|d{V| *d{V}g}d}i}tW|tXtZfrtj.}tQj/|||j0}d|j1D}|2drtgj4|j5}|j6D]?}|j7r6|2dr!tgj4|j8||j7<@nh#tr$r[}|:|jd } t?d|d| |d}~wwxYwtw|| |j<|j=||j |j>|j?pd|| |_@|j@A|jdS)zStarts the backend.python)languagerlanguage_versionr os_versioncNi|]"\}}t|t|#Sr>)r3)rPkvs r z6LocalConnectionStrategy.__aenter__.. s*666DAqQQ666rrN)storage_directory client_inforN)rstdoutrrz.AsW   "  / 9 9)%4     rrusagez%Failed to initialize conversation at z . Stderr: ) rrrrrrrrrr)Brrfrr ClientInforpython_versionsystemlowerreleaserrV InputConfigrrrr!rZrPIPESerializeToStringrrrwritestructpacklenflushrrrunpack OutputConfigParseFromStringrrrVInitializeConversationEventrr r:recvrZr3rr,r- initialize_conversation_responsehistoryrDr parse_usage_metadatartrajectory_usage trajectory_idrrr$rrrrrrr)rrc sdk_versionclient_info_protoenv_map input_config merged_envr serializedraw_len stderr_outputlength output_configrr init_event raw_init_resprrrinit_resp_event init_respentryrs r __aenter__z"LocalConnectionStrategy.__aenter__s//11N"$$K(3!022 ?   " " $ $#%% ;?)K66DIOO$5$56666 $/..B% L /3i.C*BJ*'**J o  G//11J = $ $ $ > % % % > % % % M D#j//::ZGHHH Mn!!!$$G n))++227;;m  G G G  ]4 ) )! ,F$133M!!'."5"5f"="=>>> ..M17JB$#?j GGK-j99 : ::::::::GGIIoooooom*,om"$ MC< 0 0*688-999#D   &/%6       0 1 1 )>(  -/  E   U^^G%<%< 4U[AA &e&9 :  llnnnn))++227;;m  &      ' %%' I'3;t#";   D ))'.99999sD3O P5AP00P5cKt|dr/|jr*|jd{Vd|_dSdSdS)z2Tears down the backend and releases all resources.rN)r5rr&)rexc_typeexc_valexc_tbs r __aexit__z!LocalConnectionStrategy.__aexit__gskt]##(8   ' ' ) ))))))))dr)FrW)rrrr*s44 ! """"047777%%%% 1504/3'+CG=A$(HL!%)!%<@#'5948/315-19=+P&P&P&&-P&&- P& 5$%, P& I$ P&!99D@P&!3d:P&TzP&"'!>!EP&d P&s)d"P&$JP&E12T9P& S>D P& e*+d2!P&"*T1#P&$%,%P&&'$.'P&(V]#d*)P&*hsCx(3. /$ 6+P&P&P&P&j   #e&@ @4 G 5  ,6%22T96*T16666 6%22T96*T16666 MM  #e&@ @4 GMM( MMMM^- C!1!667-   ()- - - - ^q%5%Cqqqqf+$s)++++Z  J2T9 8 8z,     #.8.>u.E S#X6i:i:i:i:Vrrr)jrrrrimportlib.metadatar~importlib.resourcesrr~rrrHrrerrr!rrtypingrrrrr google.genairrgoogle.protobufr typing_extensionsr r/google.antigravity.protor google.antigravitygoogle.antigravity.connectionsr $google.antigravity.connections.localr rgoogle.antigravity.hooksrr\rgoogle.antigravity.toolsrrrzrrr4r rrRESUMErCREATE_OR_RESUME CREATE_ONLYrr AgentBehavior AUTONOMOUSr# INTERACTIVEAGENT_BEHAVIOR_INTERACTIVEr"r$r'r-r1r7r?rGrIrKrrrYrgr[r|rcompilerr3rr0rrrrContentPrimitiverrrrrrr_rrrrrrConnectionStrategyrr>rrrs76    ??????????????------''''''&&&&&&555555$$$$$$555555@@@@@@HHHHHH<<<<<<++++++111111<<<<<<%9-  /!'!" !(*:*H*O !2&7 !-&2 "N  '$ .N#;NNNN "2 #3 4!D(4#4444111111   % , (        $!D(6 /   % /"T) / / / /) " #)  &'))))\/3AAcA$t+AAAAAN2:FGG33 "*5 "#)d:J " " " "HHHHHHHHHHj+HHHHHHV G  #G$GGGG6#384S>D+@8S8888v=H%H%HHHH>)0047)(4/))))XA A A A A j;A A A A A r