鎒j HdZddlmZddlmZGddejZdS)u:Conversation-aware context for custom tools. ToolContext is the Layer 2 counterpart to TriggerContext. It wraps a Conversation and exposes conversation capabilities — identity and a per-conversation key-value store — to tools that opt in by declaring a ``ToolContext``-typed parameter. Tools that do not declare a ``ToolContext`` parameter are unaffected. The ToolRunner handles detection at registration time and injection at execution time; schema generation automatically hides the ``ToolContext`` parameter from the model. Example:: from google.antigravity.tools.tool_context import ToolContext def my_tool(query: str, ctx: ToolContext) -> str: \"\"\"Searches and records the query in conversation state.\"\"\" ctx.set_state("last_query", query) return f"Searching for {query}..." ) conversation)statec^eZdZdZdejddffd ZedefdZ dfd Z xZ S) ToolContexta/Conversation-aware context injected into tools that request it. Modeled after ``TriggerContext``, this handle wraps a ``Conversation`` and provides a curated set of conversation capabilities. One ``ToolContext`` is created per session and shared across all tools. Per-conversation state (``get_state`` / ``set_state``) persists for the lifetime of the ``ToolContext`` instance (i.e., the session). State access methods (``get_state``, ``set_state``, ``update_state``) are thread-safe and protected by a reentrant lock (``threading.RLock``). rreturnNcZtd||_dS)zaInitializes the ToolContext. Args: conversation: The active conversation session. N)parent)super__init__ _conversation)selfr __class__s SC:\Projects\DSCons\.venv\Lib\site-packages\google\antigravity\tools\tool_context.pyr zToolContext.__init__6s. GGD!!!%Dc|jjS)z$Returns the conversation identifier.)r conversation_id)r s rrzToolContext.conversation_idBs   --rcHt|S)z8Acquires the context lock for use in a `with` statement.)r __enter__)r rs rrzToolContext.__enter__Gs GG Kr)rr) __name__ __module__ __qualname____doc__conversation_module Conversationr propertystrrr __classcell__)rs@rrr)s   &'4 &  & & & & & & .s... 8.rrN) rgoogle.antigravity.conversationrrgoogle.antigravity.utilsr state_module StateStorerrrr#sm,POOOOO::::::!!!!!,)!!!!!r