鎒jdZddlmZddlZddlZddlmZddlmZddl m Z ddl m ZddlmZdd lmZdd lmZdd lmZd gZGd d ZdS)z Layer 1 API for Antigravity SDK.)SequenceN)cast)types) connection) conversation) hook_runner) tool_context) tool_runner)trigger_runnerAgentceZdZdZdejfdZd dZdZde j de j fdZ e defd Ze dejfd Ze ded zfd Zd S)r z0High-level Agent API for simplified interaction.configc|d|_|jjr.tt|jj|jj_d|_d|_d|_ d|_ d|_ t|j |_t|j|_t#j|_dS)zWInitializes the Agent. Args: config: Declarative agent configuration. T)deepN) model_copy_configresponse_schemarstr capabilitiesfinish_tool_schema_json _strategy _conversation _tool_runner _hook_runner_trigger_runnerlisthooks_pending_hookstriggers_pending_triggers contextlibAsyncExitStack _exit_stack)selfrs FC:\Projects\DSCons\.venv\Lib\site-packages\google\antigravity\agent.py__init__zAgent.__init__%s $$$$//DL |#:> t|+;;dl7DNDDDD v|,,D!&/22D!022DreturncKtjd tj|_|jD]}|j||jt|j j }|j j }ttj}|jt|j}nO|j/ttjt|jz }nttj}t%||z }t%|j j}t%|jj}|s|r|s|st+dt-j|j |_|j |j|j|_tjd|jt<j |jd{V|_!|j"rtjd|jtGj$t|j"|j%j&d{V|_'|j"|jr3tQj)|j%} |j*| |S#tV$r5tj,d |j-d{VwxYw) zPStarts the agent session. Returns: The started Agent instance. zStarting Agent sessionNzWrite tools or MCP servers are enabled without a safety policy. Add policies=[policy.allow_all()] to approve all tool calls, or policies=[policy.deny_all(), policy.allow('tool_name')] to selectively allow specific tools.)tools)r rz0Starting connection and creating conversation...zStarting triggers...)rrz-Failed to start Agent session, cleaning up...).logginginfor HookRunnerrr register_hookclearrrpoliciesrsetr BuiltinTools read_only enabled_toolsdisabled_toolsbool mcp_serverspre_tool_call_decide_hooks ValueErrorr ToolRunner_get_all_custom_toolsrcreate_strategyrr#enter_async_contextconversation_lib Conversationcreaterr r TriggerRunnerrrrr ToolContext set_context Exception exceptionaclose) r$hookactive_policiescfgread_only_tools active_toolshas_write_toolshas_mcp_servershas_tool_decide_hookctxs r% __aenter__zAgent.__aenter__>sb  L)***G %022d%..$ ''---- !!!T\233o L %cE.88::;;o  &3,--   )5-..S5G1H1HH 5-.. \O;< >""d  ' +,,,%)%5%I%I  (d455,7   & &        $$&&&  +&t'899 %%c*** k     GHHH   # # % %%%%%%%%  s K&K???L>crKtjd|j|||d{VS)aStops the agent session. Args: exc_type: The exception type, if any. exc_val: The exception value, if any. exc_tb: The traceback, if any. Returns: True if the exception was suppressed, False or None otherwise. zStopping Agent sessionN)r+r,r# __aexit__)r$exc_typeexc_valexc_tbs r%rRzAgent.__aexit__sF L)***!++HgvFF F F F F F FFr'promptcnK|)t|tr&|std|t|trBt|ts-|rt d|Drtd||j|d{VS)aZSends a prompt and returns the final response. Args: prompt: The user prompt or content to send. Returns: The final response from the agent. Raises: ValueError: If prompt is None, an empty or whitespace-only string, or an empty sequence / sequence containing only empty or whitespace strings. Nz1chat() requires a non-empty message string. Got: c3jK|].}t|to| V/dS)N) isinstancerstrip).0ps r% zAgent.chat..s;HHA:a%%7aggii-HHHHHHr'z0chat() requires non-empty message content. Got: )rYrrZr9rallrchat)r$rVs r%r_z Agent.chats~*VS11~&,,..~  Hf H H   68$$ 63''   HHHHHHH   GV G G  "''// / / / / / //r'c|jduS)z+Whether the agent session has been started.N)rr$s r% is_startedzAgent.is_starteds  T ))r'c<|jstd|jS)a4Returns the active Conversation session. Use this for advanced session introspection: history, turn count, compaction indices, usage, or direct send/receive_steps control. For most use cases, prefer chat() instead. Raises: RuntimeError: If the agent session has not been started. z7Agent session not started. Use 'async with Agent(...)'.)r RuntimeErrorras r%rzAgent.conversations/    C    r'Nc0|jsdS|jjpdS)a$Returns the conversation identifier assigned by the runtime. Available after the session has started and at least one message has been exchanged. Pass this value back via SessionConfig.conversation_id to resume from a saved session. Returns None before the session starts. N)rconversation_idras r%rfzAgent.conversation_ids%   T   - 55r')r(r )__name__ __module__ __qualname____doc__connection_module AgentConfigr&rPrRrContent ChatResponser_propertyr6rbr>r?rrrfr'r%r r "s883.:33332N N N N ` G G G0u}01C0000< *$*** 8* ,9 8   6sTz 6 6 6 8 6 6 6r')rjcollections.abcrr!r+typingrgoogle.antigravityrgoogle.antigravity.connectionsrrkgoogle.antigravity.conversationrr>google.antigravity.hooksrgoogle.antigravity.toolsr r google.antigravity.triggersr __all__r rpr'r%rzs '&$$$$$$$$$$$$JJJJJJLLLLLL000000111111000000666666 )x6x6x6x6x6x6x6x6x6x6r'