aip-agents-binary 0.5.25b9__py3-none-any.whl → 0.6.1__py3-none-any.whl

aip_agents/tools/execute_ptc_code.pyi

@@ -0,0 +1,87 @@
+from _typeshed import Incomplete
+from aip_agents.mcp.client.base_mcp_client import BaseMCPClient as BaseMCPClient
+from aip_agents.ptc.executor import PTCSandboxConfig as PTCSandboxConfig, PTCSandboxExecutor as PTCSandboxExecutor
+from aip_agents.ptc.naming import sanitize_function_name as sanitize_function_name
+from aip_agents.sandbox.e2b_runtime import E2BSandboxRuntime as E2BSandboxRuntime
+from aip_agents.tools.tool_config_injector import TOOL_CONFIGS_KEY as TOOL_CONFIGS_KEY
+from aip_agents.utils.logger import get_logger as get_logger
+from langchain_core.tools import BaseTool
+from typing import Any
+
+logger: Incomplete
+
+def merge_tool_configs(agent_configs: dict[str, Any] | None, runtime_configs: dict[str, Any] | None) -> dict[str, dict[str, Any]]:
+    '''Merge agent-level and runtime tool configs with sanitized keys.
+
+    Merges tool configurations from two sources:
+    1. Agent-level defaults (from agent.tool_configs)
+    2. Runtime overrides (from RunnableConfig.metadata["tool_configs"])
+
+    Both sources support two formats (matching LangGraphReactAgent behavior):
+    - Direct per-tool keys: {"time_tool": {"timezone": "UTC"}}
+    - Nested structure: {"tool_configs": {"time_tool": {"timezone": "UTC"}}}
+
+    The nested "tool_configs" key has higher precedence than direct keys.
+    Tool names are sanitized to match sandbox expectations (e.g., "Time Tool" -> "time_tool").
+
+    Args:
+        agent_configs: Agent-level tool configs (may be None or contain nested dicts)
+        runtime_configs: Runtime overrides from metadata (may be None)
+
+    Returns:
+        Merged dict with sanitized tool names as keys and config dicts as values.
+        Only includes entries that are dicts (non-dict values are agent-wide defaults).
+    '''
+
+class PTCCodeTool(BaseTool):
+    """Tool for executing Python code with MCP tool access in a sandbox.
+
+    This tool uses BaseTool to properly access runtime config via run_manager.metadata.
+    The config parameter is NOT exposed to the LLM schema - it's extracted from
+    the callback manager during execution.
+    """
+    name: str
+    description: str
+    def __init__(self, executor: PTCSandboxExecutor, runtime: E2BSandboxRuntime, agent_tool_configs: dict[str, Any] | None = None, **kwargs: Any) -> None:
+        """Initialize the PTC code tool.
+
+        Args:
+            executor: The PTC sandbox executor.
+            runtime: The E2B sandbox runtime.
+            agent_tool_configs: Optional agent-level tool configs.
+            **kwargs: Additional keyword arguments passed to BaseTool.
+        """
+    @property
+    def args(self) -> dict[str, Any]:
+        """Return the argument schema for the tool."""
+    async def cleanup(self) -> None:
+        """Clean up the sandbox runtime."""
+
+def create_execute_ptc_code_tool(mcp_client: BaseMCPClient | None, config: PTCSandboxConfig | None = None, agent_tool_configs: dict[str, Any] | None = None) -> PTCCodeTool:
+    '''Create a tool that executes Python code with MCP tool access.
+
+    The code runs inside an E2B sandbox with access to generated MCP tool modules.
+    This tool is designed for LLM-generated code that needs to call multiple tools
+    programmatically in a single execution.
+
+    Args:
+        mcp_client: The MCP client with configured servers.
+        config: Optional sandbox executor configuration.
+        agent_tool_configs: Optional agent-level tool configs (from agent.tool_configs).
+            These are merged with runtime overrides from RunnableConfig.metadata.
+
+    Returns:
+        PTCCodeTool configured for PTC code execution.
+
+    Example:
+        ```python
+        from aip_agents.mcp.client import LangchainMCPClient
+        from aip_agents.tools.execute_ptc_code import create_execute_ptc_code_tool
+
+        mcp_client = LangchainMCPClient()
+        await mcp_client.add_server("yfinance", {...})
+
+        tool = create_execute_ptc_code_tool(mcp_client)
+        result = await tool.ainvoke({"code": "from tools.yfinance import get_stock\\\\nprint(get_stock(\'AAPL\'))"})
+        ```
+    '''

aip_agents/utils/langgraph/tool_managers/delegation_tool_manager.py

@@ -134,7 +134,19 @@ class DelegationToolManager(BaseLangGraphToolManager):
             Returns:
                 The result from the delegated agent, including artifacts if any.
             """
-            writer: StreamWriter = get_stream_writer()
+            try:
+                writer: StreamWriter = get_stream_writer()
+            except Exception as exc:
+                logger.warning(
+                    "DelegationToolManager: Stream writer unavailable; delegation streaming disabled.",
+                    extra={"error": str(exc), "error_type": type(exc).__name__},
+                )
+
+                def _noop_writer(_: Any) -> None:
+                    """No-op writer for non-graph execution contexts."""
+                    return None
+
+                writer = _noop_writer
 
             try:
                 # Check delegation depth limit before executing
@@ -295,6 +307,19 @@ class DelegationToolManager(BaseLangGraphToolManager):
             cfg_conf = cfg.get("configurable") if isinstance(cfg, dict) else None
             if isinstance(cfg_conf, dict):
                 parent_step_id = cfg_conf.get("parent_step_id")
+                if not parent_step_id:
+                    metadata = cfg.get("metadata") or {}
+                    tool_call_id = (
+                        cfg_conf.get("tool_call_id")
+                        or metadata.get("tool_call_id")
+                        or metadata.get("id")
+                        or (metadata.get("tool_call") or {}).get("id")
+                    )
+                    if tool_call_id and self.parent_agent is not None:
+                        thread_key = getattr(self.parent_agent, "thread_id_key", "thread_id")
+                        thread_id = cfg_conf.get(thread_key)
+                        parent_map = self.parent_agent._tool_parent_map_by_thread.get(str(thread_id), {})
+                        parent_step_id = parent_map.get(str(tool_call_id))
             _DELEGATION_PARENT_STEP_ID_CVAR.set(parent_step_id)
         except Exception:
             _DELEGATION_PARENT_STEP_ID_CVAR.set(None)

aip_agents/utils/langgraph/tool_output_management.py

@@ -471,6 +471,86 @@ class ToolOutputManager:
         """
         return self._generate_json_summary(thread_id, max_entries)
 
+    def get_latest_reference(self, thread_id: str) -> str | None:
+        """Return the most recent tool output reference for a thread.
+
+        Args:
+            thread_id: Thread ID to retrieve the latest output reference for.
+
+        Returns:
+            Latest tool output reference string or None when unavailable.
+        """
+        try:
+            summary = json.loads(self.generate_summary(thread_id, max_entries=1))
+        except Exception as exc:
+            logger.debug("Failed to parse tool output summary: %s", exc)
+            return None
+
+        if not summary:
+            return None
+        latest = summary[0].get("reference")
+        return latest if isinstance(latest, str) and latest else None
+
+    def has_reference(self, value: Any) -> bool:
+        """Check whether a value contains a tool output reference.
+
+        Args:
+            value: Value to inspect for tool output references.
+
+        Returns:
+            True if any tool output reference is present.
+        """
+        if isinstance(value, str):
+            return value.startswith(TOOL_OUTPUT_REFERENCE_PREFIX)
+        if isinstance(value, dict):
+            return any(self.has_reference(item) for item in value.values())
+        if isinstance(value, list):
+            return any(self.has_reference(item) for item in value)
+        return False
+
+    def should_replace_with_reference(self, value: Any) -> bool:
+        """Check whether a tool argument value should use a tool output reference.
+
+        Args:
+            value: Value to evaluate for replacement.
+
+        Returns:
+            True if the value should be replaced with a reference.
+        """
+        if isinstance(value, dict | list | tuple):
+            return True
+        if isinstance(value, str):
+            return len(value) > DATA_PREVIEW_TRUNCATION_LENGTH
+        return False
+
+    def rewrite_args_with_latest_reference(self, args: dict[str, Any], thread_id: str) -> dict[str, Any]:
+        """Rewrite tool args to use the latest tool output reference when appropriate.
+
+        Args:
+            args: Tool arguments to rewrite.
+            thread_id: Thread ID used for resolving stored outputs.
+
+        Returns:
+            Updated args dictionary with references substituted when needed.
+        """
+        if not self.has_outputs(thread_id):
+            return args
+        if self.has_reference(args):
+            return args
+
+        latest_reference = self.get_latest_reference(thread_id)
+        if not latest_reference:
+            return args
+
+        updated_args = dict(args)
+        replaced_any = False
+        for key, value in args.items():
+            if self.should_replace_with_reference(value):
+                updated_args[key] = latest_reference
+                replaced_any = True
+
+        return updated_args if replaced_any else args
+
     def _generate_json_summary(self, thread_id: str, max_entries: int) -> str:
         """Generate simplified JSON summary optimized for LLM prompts.
 

aip_agents/utils/langgraph/tool_output_management.pyi

@@ -232,6 +232,43 @@ class ToolOutputManager:
             A JSON string containing structured data about tool outputs. Always returns
             valid JSON, even when no outputs are stored (empty entries list).
         """
+    def get_latest_reference(self, thread_id: str) -> str | None:
+        """Return the most recent tool output reference for a thread.
+
+        Args:
+            thread_id: Thread ID to retrieve the latest output reference for.
+
+        Returns:
+            Latest tool output reference string or None when unavailable.
+        """
+    def has_reference(self, value: Any) -> bool:
+        """Check whether a value contains a tool output reference.
+
+        Args:
+            value: Value to inspect for tool output references.
+
+        Returns:
+            True if any tool output reference is present.
+        """
+    def should_replace_with_reference(self, value: Any) -> bool:
+        """Check whether a tool argument value should use a tool output reference.
+
+        Args:
+            value: Value to evaluate for replacement.
+
+        Returns:
+            True if the value should be replaced with a reference.
+        """
+    def rewrite_args_with_latest_reference(self, args: dict[str, Any], thread_id: str) -> dict[str, Any]:
+        """Rewrite tool args to use the latest tool output reference when appropriate.
+
+        Args:
+            args: Tool arguments to rewrite.
+            thread_id: Thread ID used for resolving stored outputs.
+
+        Returns:
+            Updated args dictionary with references substituted when needed.
+        """
     def clear_all(self) -> None:
         """Clear all stored outputs from both metadata and storage.
 

{aip_agents_binary-0.5.25b9.dist-info → aip_agents_binary-0.6.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aip-agents-binary
-Version: 0.5.25b9
+Version: 0.6.1
 Summary: A library for managing agents in Gen AI applications.
 Author-email: Raymond Christopher <raymond.christopher@gdplabs.id>
 Requires-Python: <3.13,>=3.11
@@ -21,22 +21,12 @@ Requires-Dist: langchain<0.4.0,>=0.3.0
 Requires-Dist: langchain-openai<0.4.0,>=0.3.17
 Requires-Dist: langchain-mcp-adapters<0.1.0,>=0.0.10
 Requires-Dist: langchain-experimental<0.4.0,>=0.3.4
-Requires-Dist: langgraph<0.3.0,>=0.2.16
+Requires-Dist: langgraph<0.7.0,>=0.6.0
 Requires-Dist: minio<8.0.0,>=7.2.20
 Requires-Dist: pydantic<3.0.0,>=2.11.7
 Requires-Dist: python-dotenv<2.0.0,>=1.1.0
 Requires-Dist: requests<3.0.0,>=2.32.4
 Requires-Dist: uvicorn<0.35.0,>=0.34.3
-Provides-Extra: dev
-Requires-Dist: coverage<8.0.0,>=7.4.4; extra == "dev"
-Requires-Dist: mypy<2.0.0,>=1.15.0; extra == "dev"
-Requires-Dist: pre-commit<4.0.0,>=3.7.0; extra == "dev"
-Requires-Dist: pytest<9.0.0,>=8.1.1; extra == "dev"
-Requires-Dist: pytest-asyncio<0.24.0,>=0.23.6; extra == "dev"
-Requires-Dist: pytest-cov<6.0.0,>=5.0.0; extra == "dev"
-Requires-Dist: pytest-xdist>=3.8.0; extra == "dev"
-Requires-Dist: ruff<0.7.0,>=0.6.7; extra == "dev"
-Requires-Dist: pillow<12.0.0,>=11.3.0; extra == "dev"
 Provides-Extra: memory
 Requires-Dist: gllm-memory-binary[mem0ai]<0.2.0,>=0.1.1; extra == "memory"
 Provides-Extra: privacy
@@ -55,6 +45,17 @@ Requires-Dist: unidecode<2.0.0,>=1.3.0; extra == "local"
 Requires-Dist: gllm-docproc-binary[docx,pdf,xlsx]<0.8.0,>=0.7.21; extra == "local"
 Requires-Dist: gllm-multimodal-binary==0.2.0.post1; extra == "local"
 Requires-Dist: bosa-connectors-binary<0.4.0,>=0.3.1; extra == "local"
+Provides-Extra: dev
+Requires-Dist: coverage<8.0.0,>=7.4.4; extra == "dev"
+Requires-Dist: mypy<2.0.0,>=1.15.0; extra == "dev"
+Requires-Dist: nest-asyncio<2.0.0,>=1.6.0; extra == "dev"
+Requires-Dist: pre-commit<4.0.0,>=3.7.0; extra == "dev"
+Requires-Dist: pytest<9.0.0,>=8.1.1; extra == "dev"
+Requires-Dist: pytest-asyncio<0.24.0,>=0.23.6; extra == "dev"
+Requires-Dist: pytest-cov<6.0.0,>=5.0.0; extra == "dev"
+Requires-Dist: pytest-xdist>=3.8.0; extra == "dev"
+Requires-Dist: ruff<0.7.0,>=0.6.7; extra == "dev"
+Requires-Dist: pillow<12.0.0,>=11.3.0; extra == "dev"
 
 # AIP Agents
 
@@ -73,13 +74,13 @@ You can use `aip-agents` directly for local execution, or let `glaip-sdk` manage
 ### Prerequisites
 - Python 3.11 - 3.12 - [Install here](https://www.python.org/downloads/)
 - Pip (if using Pip) - [Install here](https://pip.pypa.io/en/stable/installation/)
-- uv (if using uv) - [Install here](https://docs.astral.sh/uv/)
+- Poetry 1.8.1+ (if using Poetry) - [Install here](https://python-poetry.org/docs/#installation)
 - Git (if using Git) - [Install here](https://git-scm.com/downloads)
 - For git installation:
   - Access to the [GDP Labs SDK repository](https://github.com/GDP-ADMIN/glaip-sdk)
 
 ### 1. Installation from the GDP Labs registry
-This package is published to the internal GDP Labs registry. Ensure your uv/pip config includes the registry:
+This package is published to the internal GDP Labs registry. Ensure your pip/Poetry config includes the registry:
 `https://glsdk.gdplabs.id/gen-ai-internal/simple/`.
 
 #### Using pip
@@ -87,15 +88,15 @@ This package is published to the internal GDP Labs registry. Ensure your uv/pip
 pip install aip-agents
 ```
 
-#### Using uv
+#### Using Poetry
 ```bash
-uv pip install aip-agents
+poetry add aip-agents
 ```
 
 ### 2. Development Installation (Git)
 For development purposes, you can install directly from the Git repository:
 ```bash
-uv pip install "git+ssh://git@github.com/GDP-ADMIN/glaip-sdk.git#subdirectory=python/aip-agents"
+poetry add "git+ssh://git@github.com/GDP-ADMIN/glaip-sdk.git#subdirectory=python/aip-agents"
 ```
 
 ### 3. Recommended: install via glaip-sdk for local mode
@@ -112,19 +113,21 @@ pip install "aip-agents[privacy]"
 
 ## Managing Dependencies
 1. Go to the `aip-agents` module root, e.g. `cd python/aip-agents`.
-2. Run `uv sync --extra dev` to install the `aip-agents` requirements.
-3. Run `uv lock` if you change any dependency versions in `pyproject.toml`.
+2. Run `poetry shell` to create a virtual environment.
+3. Run `poetry install` to install the `aip-agents` requirements (Poetry will generate a local lock file for you if needed; the repository ignores it).
+4. Run `poetry update` if you change any dependency versions in `pyproject.toml`.
 
 ## Contributing
 Please refer to this [Python Style Guide](https://docs.google.com/document/d/1uRggCrHnVfDPBnG641FyQBwUwLoFw0kTzNqRm92vUwM/edit?usp=sharing)
 to get information about code style, documentation standard, and SCA that you need to use when contributing to this project
 
 1. Activate `pre-commit` hooks using `pre-commit install`
-2. Run `uv sync --extra dev` to install the `aip-agents` requirements.
-3. Run `which python` to get the path to be referenced at Visual Studio Code interpreter path (`Ctrl`+`Shift`+`P` or `Cmd`+`Shift`+`P`)
-4. Try running the unit test to see if it's working:
+2. Run `poetry shell` to create a virtual environment.
+3. Run `poetry install` to install the `aip-agents` requirements (this will also create a local lock file that stays local).
+4. Run `which python` to get the path to be referenced at Visual Studio Code interpreter path (`Ctrl`+`Shift`+`P` or `Cmd`+`Shift`+`P`)
+5. Try running the unit test to see if it's working:
 ```bash
-uv run pytest -s tests/unit_tests/
+poetry run pytest -s tests/unit_tests/
 ```
 
 ## Hello World Examples
@@ -225,57 +228,57 @@ For STDIO, SSE, and HTTP transports using local servers, open a terminal in the
 - For STDIO:
 
 ```bash
-uv run python aip_agents/examples/mcp_servers/mcp_server_stdio.py
+poetry run python aip_agents/examples/mcp_servers/mcp_server_stdio.py
 ```
 
 - For SSE:
 
 ```bash
-uv run python aip_agents/examples/mcp_servers/mcp_server_sse.py
+poetry run python aip_agents/examples/mcp_servers/mcp_server_sse.py
 ```
 
 - For HTTP:
 
 ```bash
-uv run python aip_agents/examples/mcp_servers/mcp_server_http.py
+poetry run python aip_agents/examples/mcp_servers/mcp_server_http.py
 ```
 
 Note: Start the appropriate server before running the client examples for that transport.
 
 ### Running Examples
 
-All examples are run from the library root using `uv run python aip_agents/examples/<file>.py`. Examples support OpenAI for LangGraph/LangChain and Google ADK where specified.
+All examples are run from the library root using `poetry run python aip_agents/examples/<file>.py`. Examples support OpenAI for LangGraph/LangChain and Google ADK where specified.
 
 #### LangChain Examples
 
 ##### STDIO Transport
 - Non-Streaming:
 ```bash
-uv run python aip_agents/examples/hello_world_langchain_mcp_stdio.py
+poetry run python aip_agents/examples/hello_world_langchain_mcp_stdio.py
 ```
 - Streaming:
 ```bash
-uv run python aip_agents/examples/hello_world_langchain_mcp_stdio_stream.py
+poetry run python aip_agents/examples/hello_world_langchain_mcp_stdio_stream.py
 ```
 
 ##### SSE Transport
 - Non-Streaming:
 ```bash
-uv run python aip_agents/examples/hello_world_langchain_mcp_sse.py
+poetry run python aip_agents/examples/hello_world_langchain_mcp_sse.py
 ```
 - Streaming:
 ```bash
-uv run python aip_agents/examples/hello_world_langchain_mcp_sse_stream.py
+poetry run python aip_agents/examples/hello_world_langchain_mcp_sse_stream.py
 ```
 
 ##### HTTP Transport
 - Non-Streaming:
 ```bash
-uv run python aip_agents/examples/hello_world_langchain_mcp_http.py
+poetry run python aip_agents/examples/hello_world_langchain_mcp_http.py
 ```
 - Streaming:
 ```bash
-uv run python aip_agents/examples/hello_world_langchain_mcp_http_stream.py
+poetry run python aip_agents/examples/hello_world_langchain_mcp_http_stream.py
 ```
 
 #### Google ADK Examples
@@ -283,31 +286,31 @@ uv run python aip_agents/examples/hello_world_langchain_mcp_http_stream.py
 ##### STDIO Transport
 - Non-Streaming:
 ```bash
-uv run python aip_agents/examples/hello_world_google_adk_mcp_stdio.py
+poetry run python aip_agents/examples/hello_world_google_adk_mcp_stdio.py
 ```
 - Streaming:
 ```bash
-uv run python aip_agents/examples/hello_world_google_adk_mcp_stdio_stream.py
+poetry run python aip_agents/examples/hello_world_google_adk_mcp_stdio_stream.py
 ```
 
 ##### SSE Transport
 - Non-Streaming:
 ```bash
-uv run python aip_agents/examples/hello_world_google_adk_mcp_sse.py
+poetry run python aip_agents/examples/hello_world_google_adk_mcp_sse.py
 ```
 - Streaming:
 ```bash
-uv run python aip_agents/examples/hello_world_google_adk_mcp_sse_stream.py
+poetry run python aip_agents/examples/hello_world_google_adk_mcp_sse_stream.py
 ```
 
 ##### HTTP Transport
 - Non-Streaming:
 ```bash
-uv run python aip_agents/examples/hello_world_google_adk_mcp_http.py
+poetry run python aip_agents/examples/hello_world_google_adk_mcp_http.py
 ```
 - Streaming:
 ```bash
-uv run python aip_agents/examples/hello_world_google_adk_mcp_http_stream.py
+poetry run python aip_agents/examples/hello_world_google_adk_mcp_http_stream.py
 ```
 
 #### LangGraph Examples (OpenAI)
@@ -315,31 +318,31 @@ uv run python aip_agents/examples/hello_world_google_adk_mcp_http_stream.py
 ##### STDIO Transport
 - Non-Streaming:
 ```bash
-uv run python aip_agents/examples/hello_world_langgraph_mcp_stdio.py
+poetry run python aip_agents/examples/hello_world_langgraph_mcp_stdio.py
 ```
 - Streaming:
 ```bash
-uv run python aip_agents/examples/hello_world_langgraph_mcp_stdio_stream.py
+poetry run python aip_agents/examples/hello_world_langgraph_mcp_stdio_stream.py
 ```
 
 ##### SSE Transport
 - Non-Streaming:
 ```bash
-uv run python aip_agents/examples/hello_world_langgraph_mcp_sse.py
+poetry run python aip_agents/examples/hello_world_langgraph_mcp_sse.py
 ```
 - Streaming:
 ```bash
-uv run python aip_agents/examples/hello_world_langgraph_mcp_sse_stream.py
+poetry run python aip_agents/examples/hello_world_langgraph_mcp_sse_stream.py
 ```
 
 ##### HTTP Transport
 - Non-Streaming:
 ```bash
-uv run python aip_agents/examples/hello_world_langgraph_mcp_http.py
+poetry run python aip_agents/examples/hello_world_langgraph_mcp_http.py
 ```
 - Streaming:
 ```bash
-uv run python aip_agents/examples/hello_world_langgraph_mcp_http_stream.py
+poetry run python aip_agents/examples/hello_world_langgraph_mcp_http_stream.py
 ```
 
 ### Multi-Server Example
@@ -355,13 +358,13 @@ npx @playwright/mcp@latest --headless --port 8931
 2. In another terminal, start the Name Generator SSE server:
 
 ```bash
-uv run python aip_agents/examples/mcp_servers/mcp_name.py
+poetry run python aip_agents/examples/mcp_servers/mcp_name.py
 ```
 
 3. Run the multi-server client example:
 
 ```bash
-uv run python aip_agents/examples/hello_world_langchain_mcp_multi_server.py
+poetry run python aip_agents/examples/hello_world_langchain_mcp_multi_server.py
 ```
 **3. Running Individual A2A Examples:**
 
@@ -477,7 +480,7 @@ The library supports Mem0 as a memory backend for long-term conversation recall.
 Use the coordinator example with memory enabled:
 
 ```bash
-uv run python aip_agents/examples/hello_world_a2a_mem0_coordinator_server.py
+poetry run python aip_agents/examples/hello_world_a2a_mem0_coordinator_server.py
 ```
 
 In client: