鎒j8VdZddlmZddlZddlZddlZddlZddlmZm Z m Z m Z m Z ddl Z ddlmZddlmZddlmZddlmZGd d eje jZGd d ejZGd dejZGdde jZedS)uBase interfaces for connections in the Google Antigravity SDK. A Connection is the SDK's public interface for interacting with an agent backend, regardless of where the agent runs. Layer 2 APIs (Conversation, AgentConfig) depend ONLY on this interface — never on transport details. A ConnectionStrategy knows how to establish a Connection for a specific backend type and how to tear it down. ) annotationsN)Any AsyncIteratorCallableMappingSequence)types)hooks)policy)triggersceZdZUdZejdZdZded<ej dZ d ed <ej e Z d ed <ej e Z d ed<ej e Zded<ej e Zded<ej e Zded<ej e Zded<dZded<dZded<dZded<dZded<dZded<ej e Zded<ej e Zd ed!<dZd"ed#<dZd$ed%<dZd&ed'<ejd#d()edBd-Zejd.)dCd0Z ejdedDd1Z!ejd.)dCd2Z"ejdd3Z#ejdd()ed4Z$dd5d6dEfd;Z%dFd=Z&e'j(dGdAZ)xZ*S)H AgentConfigzAbstract base class for agent configuration. Each ConnectionStrategy defines a concrete subclass with the config fields it needs. Agent introspects the config type to auto-dispatch to the correct strategy factory. Tarbitrary_types_allowedNz%str | types.SystemInstructions | Nonesystem_instructionscdtjtjS)N) enabled_tools)r CapabilitiesConfig BuiltinTools read_onlyWC:\Projects\DSCons\.venv\Lib\site-packages\google\antigravity\connections\connection.pyzAgentConfig.5s*e6*4466r)default_factoryztypes.CapabilitiesConfig capabilitieszlist[Callable[..., Any] | str]toolszlist[policy.Policy]policieszlist[hooks_mod.Hook]r zlist[triggers_mod.Trigger]r zlist[types.McpServerConfig] mcp_serversz list[str] workspaces str | Noneconversation_idz$types.SessionContinuationMode | Nonesession_continuation_modesave_dir app_data_dirz6dict[str, Any] | type[pydantic.BaseModel] | str | Noneresponse_schema skills_pathszlist[types.SubagentConfig] subagentsDebugConfig | None debug_configztypes.RetryConfig | None retry_configztypes.BudgetConfig | None budget_configbefore)modevrreturnct|tr|rtndSt|tr tdi|S|S)Nr) isinstancebool DebugConfigdictclsr/s r_validate_debug_configz"AgentConfig._validate_debug_configNsT!T* )[]]]T)!T   1   Hrafter 'AgentConfig'cz|j3t|jtr|j|SN)r*r2r4 apply_loggingselfs r_apply_debug_settingsz!AgentConfig._apply_debug_settingsWs> $ ;**$ %%''' Krc|rYt|dkrtdt|tjd|std||S)N z9conversation_id must be at least 32 characters long, got z^[a-zA-Z0-9-]+$z-conversation_id must match [a-zA-Z0-9-], got )len ValueErrorrematchr6s r_validate_conversation_idz%AgentConfig._validate_conversation_id_sy  Q" PA P P   X(! , ,  AA A A    Hrcf|jtjjkr|jst d|S)NzJconversation_id must be specified when session_continuation_mode is RESUME)r#r SessionContinuationModeRESUMEr"rDr>s r_validate_session_continuationz*AgentConfig._validate_session_continuationmsD &%*G*NNN$ O  0   Krc|dSt|tr> tj||S#tj$r}t d|d}~wwxYwt|t rtj|St|tr@t|tj r&tj| St dt|j d)Nz)response_schema string is not valid JSON.z$Unsupported response_schema format: z?. Expected a JSON string, dict, or pydantic.BaseModel subclass.)r2strjsonloadsJSONDecodeErrorrDr5dumpstype issubclasspydantic BaseModelmodel_json_schema__name__)r7r/excs r_validate_schemazAgentConfig._validate_schemaysy T!SOO 1   !OOODEE3NO!T Z]]!T/z!X-?@@/ Z++-- . ..  HtAww/? H H H  s1AAAc|gSt|tttfrt|tt fr|g}gfd|S)Nct|tttfr.t|tt fs|D] }|dS|dSr<)r2listtuplerrMbytesappend)itemsub_item flat_policiesflattens rrcz/AgentConfig._validate_policies..flattens D41 2 2#: e <<#  H '(       T"""""r)r2r\r]rrMr^)r7r/rbrcs @@r_validate_policieszAgentConfig._validate_policiess y i a$x0 1 1Z C<66 #aM###### GAJJJ rFupdatedeeprfMapping[str, Any] | Nonergr3ct||}|rdt|j|_t|j|_t|j|_t|j|_|S)Nre)super model_copyr\rr r r)r?rfrgcopied __class__s rrkzAgentConfig.model_copysrWW  vD  9 9F ,$*%%fl$*%%flT]++foT]++fo Mrlist[Callable[..., Any]]c .g}i}|jpgD]t}t|rct|ddpt|j}||vr |||krt d|dZ|||<||u|jpgD]}|jpgD]|}t|rkt|ddpt|j}||vr(|||krt d|d|jdb|||<||}|S)zFReturns all callable custom tools across the main agent and subagents.rWNzDuplicate custom tool name 'z4' detected across agent and subagent configurations.z&' detected across agent and subagent 'z' configurations.) rcallablegetattrrRrWrDr_r(name)r?r seen_namestrrsubtools r_get_all_custom_toolsz!AgentConfig._get_all_custom_toolss&(E02J Z 2   ! q*d++?tAww/? :     " "0t000 # *T  ,,q///~#  )/r  $ D>> z400GDJJ4G$ Z  $4''@@@$'H@@@( $Jt  LL     Lr tool_runner hook_runner'ConnectionStrategy'cdS)azCreates the ConnectionStrategy for this config. The Agent calls this after setting up ToolRunner, HookRunner, and policies. The strategy receives the fully-wired runners. Args: tool_runner: The fully-wired ToolRunner. hook_runner: The fully-wired HookRunner. Returns: A ConnectionStrategy instance configured with the specified runners. Nr)r?rxrys rcreate_strategyzAgentConfig.create_strategys ,Cr)r/rr0r))r0r:)r/r!r0r!)rfrhrgr3r0r:)r0rn)rxrryrr0rz)+rW __module__ __qualname____doc__rT ConfigDict model_configr__annotations__Fieldrr\rrr r rr r"r#r$r%r&r'r(r*r+r,field_validator classmethodr8model_validatorr@rGrKrYrdrkrwabcabstractmethodr| __classcell__)rms@rrr)s%$TBBB,?CCCCC+98>,,,, +9(.*N*N*N%NNNN"0(."F"F"F(FFFF .t D D D%DDDD)7)M)M)M(MMMM-;X^...+)(.>>>*>>>> $/$$$$DHHHHH(!,!!!!LP/PPPP*HN4@@@,@@@@*8(.*N*N*N)NNNN%),)))),0,////-1-11118N:::   ;;: 8)))*)8-..    ;/.  8)))   *) 8-../.$8JX666;76,37U">rrcJeZdZdZed&dZed'dZed(dZed)d Zed*d Z ed+dZ ed,dZ e j d-dZe j d.dZd/dZd/dZd/dZd0d1dZd2d"Ze j d3d$Zd%S)4 ConnectionzA live session with an agent backend. This is the common contract that all connection types implement. Layer 2 APIs depend only on this interface. r0r3cdS)z;Returns True if the connection is idle and ready for input.Trr>s ris_idlezConnection.is_idle  4rSequence[types.Step]cgS)zAReturns the pre-existing session steps restored during handshake.rr>s r_initial_historyzConnection._initial_historys  IrrMcdS)z>Returns the conversation identifier, or empty string if unset.rr>s rr"zConnection.conversation_ids  2rr)cdS)zIReturns the debug configuration for this connection, or None if disabled.Nrr>s rr*zConnection.debug_configrrtypes.UsageMetadatac(tjS)zReturns total cumulative token usage from the backend. Subclasses override to provide live usage data. Default returns an empty UsageMetadata instance (all fields None, indicating no tracking). )r UsageMetadatar>s rcumulative_usagezConnection.cumulative_usages    rdict[str, types.UsageMetadata]ciS)zReturns per-trajectory cumulative token usage from the backend. Subclasses override to provide live usage data. Default returns an empty dictionary (no tracking). rr>s rtrajectory_usageszConnection.trajectory_usagess  Irtypes.StopReasonc$tjjS)zReturns the stop reason of the most recent turn. Subclasses override to provide live stop reason data. Default returns UNSPECIFIED. )r StopReason UNSPECIFIEDr>s r_last_turn_stop_reasonz!Connection._last_turn_stop_reasons   ''rprompttypes.Content | NonekwargsrNonec KdS)zSends a prompt to the agent. Args: prompt: The user message to send. **kwargs: Strategy-specific options (model overrides, media, etc.). Nr)r?rrs rsendzConnection.send"s CrAsyncIterator[types.Step]cdS)zReceives steps as they complete from the agent. Yields Step objects representing agent actions. The exact fields populated depend on the backend, but all steps conform to the Step model. Yields: Step objects as they occur. Nrr>s r receive_stepszConnection.receive_steps,s Crc KdS)z/Disconnects the session and releases resources.Nrr>s r disconnectzConnection.disconnect8 Drc KdS)z%Cancels the current turn in progress.Nrr>s rcancelzConnection.cancel<rrc KdS)z)Blocks until the connection becomes idle.Nrr>s r wait_for_idlezConnection.wait_for_idle@rrr@timeoutfloatc KdS)zBlocks until the connection wakes up or the timeout is reached. Args: timeout: Maximum seconds to wait. Returns: True if the connection woke up, False on timeout. Fr)r?rs rwait_for_wakeupzConnection.wait_for_wakeupDs  5rresultslist[types.ToolResult]c KdS)zSends tool execution results back to the agent. Each connection strategy serializes the results into the backend wire format. Args: results: A list of ToolResult objects. Nr)r?rs r_send_tool_resultszConnection._send_tool_resultsOs  Drcontentc KdS)zaSends a trigger message to the agent. Args: content: The trigger message content. Nr)r?rs rsend_trigger_notificationz$Connection.send_trigger_notificationZs CrN)r0r3)r0r)r0rMr0r))r0r)r0r)r0r)rrrrr0r)r0rr0r)r)rrr0r3)rrr0r)rrMr0r)rWr}r~rpropertyrrr"r*rrrrrrrrrrrrrrrrrrs   8  8  8  8 !!! 8!  8 ((( 8(                          rrceZdZdZejd dZejd dZejd dZe d d Z d S)ConnectionStrategyaStrategy for establishing a Connection to an agent backend. Each backend type (local, Interactions API, cloud agent) provides its own ConnectionStrategy implementation that handles process management, transport setup, authentication, and health checking. r0rcdS)zReturns the established Connection. Returns: The active Connection object. Raises: RuntimeError: If the connection has not been established. Nrr>s rconnectzConnectionStrategy.connectls Crrc KdS)z0Starts the backend and prepares for connections.Nrr>s r __aenter__zConnectionStrategy.__aenter__{s Crc KdS)zTears down the backend and releases all resources. Args: exc_type: The exception type, if any. exc_val: The exception value, if any. exc_tb: The traceback, if any. Nr)r?exc_typeexc_valexc_tbs r __aexit__zConnectionStrategy.__aexit__s Crr)cdS)zGReturns the debug configuration for this strategy, or None if disabled.Nrr>s rr*zConnectionStrategy.debug_configrrN)r0rrr) rWr}r~rrrrrrrr*rrrrrds      8rrcdeZdZUdZejdZdZded<e j Z ded<d d Z d S) r4zConfiguration for client-side and server-side debugging and observability. When instantiated with default parameters (`DebugConfig()`), all debug features and detailed logging (`logging.DEBUG`) are enabled by default. Trr3enable_server_side_tracingzint | str | None logging_levelr0rcp|j.tjd|jdSdS)z8Applies the configured logging level to the SDK loggers.Nzgoogle.antigravity)rlogging getLoggersetLevelr>s rr=zDebugConfig.apply_loggings< % ,--66t7IJJJJJ&%rNr) rWr}r~rrTrrrrrDEBUGrr=rrrr4r4sx %$TBBB,&*))))$+M-1111KKKKKKrr4)r __future__rrrNrrEtypingrrrrrrTgoogle.antigravityr google.antigravity.hooksr hooks_modr google.antigravity.triggersr triggers_modABCrUrrrr4 model_rebuildrrrrs#"""""  BBBBBBBBBBBBBB$$$$$$777777++++++@@@@@@@@@@@#'8-@@@Fuuuuuuuup********ZKKKKK($KKK( r