鎒j2\dZddlZddlmZmZmZddlmZddlm Z dZ GddZ dS) aStateful conversation session for the Google Antigravity SDK. Conversation is the Layer 2 session API. It wraps a Connection with: - Step history accumulation (with compaction index tracking) - A chat() convenience method (send + collect in one call) - State introspection (idle, turn count, last response) Layer 1 (Agent) delegates to Conversation; power users can use Conversation directly with any ConnectionStrategy. N)Any AsyncIteratorSequence)types) connectioni'cdeZdZdZedddejdedee j dzfdZ e e jdejd edfd Zd e jdzd ed dfd Zd ee j fdZd ee je jzfdZd edzfdZ d&d e jdzd ed e jfdZed ee j fdZed efdZ ed efdZ!ed eefdZ"d'dZ#d'dZ$ed ejfdZ%ed e&fdZ'ed efdZ(ed e j)fdZ*ed e+ee j)ffdZ,ed e j)dzfdZ-ed e j)dzfdZ.ed e j/fdZ0d'd Z1d'd!Z2d(d#e3d e&fd$Z4d'd%Z5dS)) ConversationzStateful session wrapping a single conversation with the agent. Accumulates step history, tracks turn start indices and compaction indices, and provides convenience methods for common send/receive patterns. N)max_history_sizehistoryconnr r cV||_g|_g|_g|_||_d|_|pgD]e}|j||jtj j kr/|jt|jdz f| dS)a2Initializes the conversation with a connection and empty history. Args: conn: The established connection to the agent backend. max_history_size: Maximum number of steps to retain in history. When exceeded, the oldest steps are discarded. Set to 0 to disable the limit. N) _connection_steps_turn_start_indices_compaction_indices_max_history_size_turn_start_usageappendtyperStepType COMPACTIONlen_enforce_max_history)selfr r r steps ZC:\Projects\DSCons\.venv\Lib\site-packages\google\antigravity\conversation\conversation.py__init__zConversation.__init__+sD$&DK*,D*,D-D9=D 2>> k en/ / /  ''DK(8(81(<===strategyreturncK|4d{V|}|||jWVdddd{VdS#1d{VswxYwYdS)zCreates a new conversation. Args: strategy: The connection strategy to use to interact with an agent. Yields: A new Conversation instance. N)r )connect_initial_history)clsr r s rcreatezConversation.createFs55555555     d Cd3 4 4 44444555555555555555555555555555555s*A AApromptkwargscK|jjdurN |23d{V}6n/#t$r"|jd{VYnwxYw|jt|j|jj |_ |jj |fi|d{VdS)aSends a message to the agent. If a turn is still in progress, drains all remaining steps into history before sending the new message. When another coroutine is already iterating ``receive_steps()`` (and therefore recording steps into history), falls back to ``wait_for_idle()``. Args: prompt: The user message to send. **kwargs: Strategy-specific options. FN) ris_idle receive_steps RuntimeError wait_for_idlerrrrcumulative_usage model_copyrsend)rr'r(_s rr0zConversation.send\s  5((/))++       ! ,+ /// ,,........... /  ##C $4$4555!->IIKKD $   1 1& 1 1111111111s0.0)AAcFK|j23d{V}|j||jt jjkr/|jt|jdz | |WV6dS)aReceives steps as they complete, blocks until execution is idle. Steps are recorded in history as they arrive. The iterator exits once the execution turn is complete. Returns: An async iterator of Step objects. Yields: Steps as they complete. Nr) rr+rrrrrrrrrrrs rr+zConversation.receive_stepszs&4466d k en/ / /  ''DK(8(81(<=== !!! jjjjj 766sB cKt}|23d{V}|jtjjk}|jtjjk}|rV|rT|j r#tj |j |j WV|j r#tj |j |j WV|jr@|jD]8}|j |j|vr&|j||j|WV96dS)zReceives and yields real-time semantic chunks for the current turn. Returns: An async iterator of Thought, Text, or ToolCall events. Yields: Thought, Text, or ToolCall events in real-time. N) step_indextext)setr+sourcer StepSourceMODELtarget StepTargetUSERthinking_deltaThoughtr5 content_deltaText tool_callsidadd)r seen_tool_idsris_modelis_target_usercalls rreceive_chunkszConversation.receive_chunkssN"eeM((**d 0 66h{e&6&;;n  Pn P    t/B         P doDz5Conversation._enforce_max_history...""" (]] h, ]]rc&g|] }|k|z Srdrdres rriz5Conversation._enforce_max_history.. rjr)rrrrr)rrhs @rrz!Conversation._enforce_max_historys  #dk"2"2T5K"K"KT[!!D$::hK *dk""""+"""d """"+"""d  "K"Krc|jS)uReturns the underlying Connection transport. Intended for advanced use cases that need direct transport access. Prefer Conversation or Agent methods for normal interaction — bypassing the Conversation layer skips history tracking and hook dispatch. )rrUs rrzConversation.connections   rc|jjS)z=Returns True if the conversation is idle and ready for input.)rr*rUs rr*zConversation.is_idles   ##rc|jjS)z3Returns the conversation identifier, if one exists.)rconversation_idrUs rrozConversation.conversation_id!s   ++rc>|jjS)z@Returns cumulative token usage across all turns in this session.)rr.r/rUs r total_usagezConversation.total_usage&s   , 7 7 9 99rcRd|jjDS)zReturns a dictionary mapping trajectory ID to cumulative token usage. The main agent's trajectory ID matches conversation_id (or 'main'). Subagents each have a unique trajectory ID. c>i|]\}}||Srd)r/)rfkvs r z2Conversation.trajectory_usages..2s3   "a1<<>>   r)rtrajectory_usagesitemsrUs rrwzConversation.trajectory_usages+s5  &*&6&H&N&N&P&P   rcZ|jdS|jj|jz }|jdkrdS|S)z7Returns token usage from the most recent turn, or None.Nr)rrr.total_token_count)rdiffs rlast_turn_usagezConversation.last_turn_usage6s= % T   ,t/E ED "" T Krc|jS)z#Internal alias for last_turn_usage.)r|rUs r_last_turn_usagezConversation._last_turn_usage@s   rc|jjS)z0Returns the stop reason of the most recent turn.)r_last_turn_stop_reasonrUs rrz#Conversation._last_turn_stop_reasonEs   22rcHK|jd{VdS)z%Cancels the current turn in progress.N)rcancelrUs rrzConversation.cancelNs3   ! ! # ##########rcHK|jd{VdS)zBBlocks until the conversation is idle and ready for the next turn.N)rr-rUs rr-zConversation.wait_for_idleRs3   ( ( * **********rr@timeoutcFK|j|d{VS)zBlocks until the conversation wakes up or the timeout is reached. Args: timeout: Maximum seconds to wait. Returns: True if the conversation woke up, False on timeout. N)rwait_for_wakeup)rrs rrzConversation.wait_for_wakeupVs1!11':: : : : : : ::rcHK|jd{VdS)zBCloses the connection transport and releases background resources.N)r disconnectrUs rrzConversation.disconnectas3   % % ' ''''''''''r)N)r!N)r)6__name__ __module__ __qualname____doc___DEFAULT_MAX_HISTORY_SIZEconnection_lib ConnectionintrrStepr classmethod contextlibasynccontextmanagerConnectionStrategyrr&Contentrr0r+ StreamChunkToolCallrIrNrQrRpropertyrTr strrZr\r^rarrboolr*ro UsageMetadatarqdictrwr|r~ StopReasonrrr-floatrrrdrrr r $s38-1     %    #d*     6 !515^$555"!;5(2md"22 2222<=#<&$U&78$$$$L #*    ,0HHMD(H;>H HHHH, tEJ' 8 S 8 )#))) 8) *$s)*** 8* " " " "" .3 8 $t$$$ 8$ ,s,,, 8, :5.::: 8: c5+>&>!? 8 u2T9 8   3d :    8  3e&6333 83$$$$++++ ; ;5 ;T ; ; ; ;((((((rr ) rrtypingrrrgoogle.antigravityrgoogle.antigravity.connectionsrrrr rdrrrs  //////////$$$$$$GGGGGG#((((((((((r