aip-agents-binary 0.6.0__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_binary-0.6.0.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.6.0
+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

{aip_agents_binary-0.6.0.dist-info → aip_agents_binary-0.6.1.dist-info}/RECORD

@@ -34,8 +34,8 @@ aip_agents/agent/langflow_agent.py,sha256=3KjB4ZBAIAqPKK5pML3DuiQ_yixx2L4dTQSW2p
 aip_agents/agent/langflow_agent.pyi,sha256=HOzMWIkga8F_ZFrzT62JdX9vhh8JhQVg-ylrr2Tglyo,5800
 aip_agents/agent/langgraph_memory_enhancer_agent.py,sha256=1AvvqvFfeLf80Jw2W2Mm7TdpwO3o_OC7r-30M4EIWeU,17859
 aip_agents/agent/langgraph_memory_enhancer_agent.pyi,sha256=6n6cd3udUhSaVTxqulnjCCgR-eUROnBEW2lFkAQCBQ4,2644
-aip_agents/agent/langgraph_react_agent.py,sha256=oGUzQZu7XiXrbuLAEJote5sBGkjnKYYck352E5LhZdE,111126
-aip_agents/agent/langgraph_react_agent.pyi,sha256=bVzM5w98biRwV6GTzaIeKu_rwIQ8d-q7EugZk9N-0vQ,8487
+aip_agents/agent/langgraph_react_agent.py,sha256=dOsMAFhYpojMtLH6gJmpT8BBfhBv1ICSksCeaiOx0kA,119389
+aip_agents/agent/langgraph_react_agent.pyi,sha256=JYETZaQMmp4Ja0QN0fg6YZKBlH-GVQ25Ydb73tdeJUo,10645
 aip_agents/agent/system_instruction_context.py,sha256=6lQAJ6DJv1aF8wjalxdsNWqEDZmomHwnhnPikLCfzj4,1220
 aip_agents/agent/system_instruction_context.pyi,sha256=gpYUr_CjWU_qVFobX_sduKDQN_wwtRL9BdnYv0WZ0Mc,424
 aip_agents/agent/hitl/__init__.py,sha256=qyxPIfSJ08MrvkE1qBZxdiom4VxSJSGEkrjqUZY2kNg,850
@@ -209,6 +209,8 @@ aip_agents/examples/hello_world_multi_agent_langgraph_lm_invoker.py,sha256=B0I92
 aip_agents/examples/hello_world_multi_agent_langgraph_lm_invoker.pyi,sha256=OHkTLSqRAM7TOZEp6TH5dsPq9zbpi9UA_l0rDIrqFlM,257
 aip_agents/examples/hello_world_pii_logger.py,sha256=xH4PNGetuZv1TJj5Y1fqvm1Vto0pp1u5ju-ORccEFwo,605
 aip_agents/examples/hello_world_pii_logger.pyi,sha256=tZ0oOfd_gK97KHczzEh3PQfTt2eedh25IL7Q1PhVDWU,139
+aip_agents/examples/hello_world_ptc.py,sha256=crx641hItpnUmh1b09a0mJcKSygJQDmicFfcKQOcqIA,1513
+aip_agents/examples/hello_world_ptc.pyi,sha256=uGDk7WAt4iXbKJ4zKp-qd2IzBLWUAZ5ehPdLDhteVaw,236
 aip_agents/examples/hello_world_sentry.py,sha256=HRnIVTkaqcUGbVbVVCzB8HQv5nVqHbt_3aPiflrD31M,4109
 aip_agents/examples/hello_world_sentry.pyi,sha256=Z6t3ywrVlgBI9NFxbcbhRVAwt71Q3NsB_JvwZhcd0aQ,692
 aip_agents/examples/hello_world_step_limits.py,sha256=0f02AFrpfAertnjSHxQiU6uAyIWt0ovQ3aQ5QCJPSx4,10082
@@ -352,6 +354,48 @@ aip_agents/middleware/manager.py,sha256=cTSzJN-3ty2o7x7iS1xd__hdBsQiq2Rqb5bnLPvW
 aip_agents/middleware/manager.pyi,sha256=UbQ-5sehP7G0sokwxCSBe4q8JLub7acFrmr0P-z65vE,3661
 aip_agents/middleware/todolist.py,sha256=YxE1kcaP6Ov1WnfxM5z8cv5cNsJViG0NRPpbAGfpeMQ,10788
 aip_agents/middleware/todolist.pyi,sha256=nkRSDplMiQPS0qhNrMoDB1ksOtarEclaGErvIZH7QQ4,4320
+aip_agents/ptc/__init__.py,sha256=k7OsdTWjGB_LQ1YVZDR2B0Pa4NIwtwx7BBAGBn2bdaI,1458
+aip_agents/ptc/__init__.pyi,sha256=Gi0YKXYWGFzkwRbigQuwtSk45Y7QLO997uxIL6QfIO4,575
+aip_agents/ptc/doc_gen.py,sha256=CcSFar80f_C_r_YAZwzTVg49rk9krhA_qwKSvYiVDtE,3504
+aip_agents/ptc/doc_gen.pyi,sha256=i7hSGLtCzF-JXNqcazD13cewncTOy34_OMtwdLjYlhU,1306
+aip_agents/ptc/exceptions.py,sha256=nBt5QlW9BGSAQNd1smb4bmGCMPeREwnmsU68vya9IQo,944
+aip_agents/ptc/exceptions.pyi,sha256=jAjmWTiOqFosC0XhO5LxX8r8B2E5H0GAXQYknTm5HK4,686
+aip_agents/ptc/executor.py,sha256=0Qw4CgAgxd6YrZSqf566zVnUs-0Dy6ORVbq1ai8qx8E,5206
+aip_agents/ptc/executor.pyi,sha256=2kyD7Hv_QX-_sQ6NnqSmryZ_01mBy1iRll9fY67wfBE,3324
+aip_agents/ptc/naming.py,sha256=CY7whoWCynbx2K_NeWT7gPoFcm_5aqfum4l3Mnjqv0U,5185
+aip_agents/ptc/naming.pyi,sha256=yWNQmj45v5diLPflJbm1409G-JtIG4VBRl1bAGvX-Bg,2217
+aip_agents/ptc/payload.py,sha256=G5AV8G0BwaBdO85Z8u87jZLj1xY7vfA98LTI_mlDrXo,861
+aip_agents/ptc/payload.pyi,sha256=D6Pxb89T2Ln-_vmAaxz6HL3jDSBHz0zbmgzxAnBsJpk,646
+aip_agents/ptc/prompt_builder.py,sha256=FdW15QONAn3ml8G5GUQ-J-nlqTjOxQRn2XpPypIbHAU,18211
+aip_agents/ptc/prompt_builder.pyi,sha256=0OC46oVfp7yrcR-zNEVqt4_2A-pM9QtdokTX8rzjDNM,2289
+aip_agents/ptc/ptc_helper.py,sha256=KGPL2ZbtVKRwfbEmN4pcqFoGmjPiTPOCkNUljaPDFh8,455
+aip_agents/ptc/ptc_helper.pyi,sha256=Tw6zTztn-umW5b--gy4YI0T1EfbzQ6m0Rhf_0VVgm0o,78
+aip_agents/ptc/sandbox_bridge.py,sha256=xKR6si9z5AbmbkNK-QRIsutTgGpQUj5RaZm_KlgtPXk,1693
+aip_agents/ptc/sandbox_bridge.pyi,sha256=Gh3mF9pkZOWT6BIas-xd_8TXgvk0GfAVTJ3MougBG-I,1004
+aip_agents/ptc/template_utils.py,sha256=IzswGL7PfNy7JMAuVFb7TWKZeptISfHjlxNYwGtbLx4,931
+aip_agents/ptc/template_utils.pyi,sha256=V7XZTY1pvL5j0Kdct2Gu-s4r6HzszaprrtZffuN_RCU,451
+aip_agents/ptc/mcp/__init__.py,sha256=mYNGO_dAoVm6gF0TaTFpgcCUBR-uRgMBWMcs4MUZ9Ac,1163
+aip_agents/ptc/mcp/__init__.pyi,sha256=1Jhp9lKz4w-sxITUsgJumEz9Tz0BZXysdnP0fUAfESs,923
+aip_agents/ptc/mcp/sandbox_bridge.py,sha256=KHpdGR4JhAzjORl0qSQpbofpqVkgXLtpSB-7kC9sSFA,21086
+aip_agents/ptc/mcp/sandbox_bridge.pyi,sha256=aRdOeJBQNao5qYLIcM5EoDKnPYkArmGEvpBKgPpu3iw,2080
+aip_agents/ptc/mcp/templates/__init__.py,sha256=UXiWrFXTMnKozmU3JX_hZhN53J4P_mvcc2vzfA_RZI8,58
+aip_agents/ptc/mcp/templates/__init__.pyi,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+aip_agents/ptc/mcp/templates/mcp_client.py.template,sha256=ZVjaa5nQd0uSJlS_lb5Q6M42cKYZzD4xrNDGRSQ2ruI,7685
+aip_agents/ptc/templates/__init__.py,sha256=92KlBHjxR2_CSZVvkvDtZfRqKn2k3A8hjXQANHupZUY,57
+aip_agents/ptc/templates/__init__.pyi,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+aip_agents/ptc/templates/ptc_helper.py.template,sha256=x68OY0l_68DFlq-yBs8I2LqHom6dmUQfFSCI68yL0bw,4198
+aip_agents/sandbox/__init__.py,sha256=DLRxtsvymaKPZePPdpFbaISb9KiAkcHMODfasvh_qO8,1409
+aip_agents/sandbox/__init__.pyi,sha256=kAELy84zz3NL10pX2q20gcQLwHZB9rePxewMe5aRoQg,346
+aip_agents/sandbox/defaults.py,sha256=qvKzLeHeAnVed7DRXpIiPmNK1EgYv0rpWMF0KCTbJqw,241
+aip_agents/sandbox/defaults.pyi,sha256=zbhPsGixnyLiRtdo-9H4icOVFpCfZ2zf0LrFGAfCSng,66
+aip_agents/sandbox/e2b_runtime.py,sha256=t00YgwEn62iuBdWEw54DCUwbUvUL0vEeCljJy81SH4Q,9484
+aip_agents/sandbox/e2b_runtime.pyi,sha256=2Y0WoYwDsuKoyp4f1GhZrFnQi0lyP6L9AhCgJ9bEokA,2203
+aip_agents/sandbox/template_builder.py,sha256=6d-pAoXsuUHYGavvK10vB-KY07OKeTpd-3RMCN3VFTw,4479
+aip_agents/sandbox/template_builder.pyi,sha256=_e7g1wM7X2gGCnRevNUfVyifWdWu5ykTly9wG4QMkkc,1558
+aip_agents/sandbox/types.py,sha256=IFZinB6b8uv8W4e7WzVXo8JSh-aqdgfnUEtCJK82Njk,552
+aip_agents/sandbox/types.pyi,sha256=kERf60VMpbgFGnPnHPHr9vbg5vWzlU4G8yS7qfK45fA,380
+aip_agents/sandbox/validation.py,sha256=gr5M5HHWJnqxsT-2y51SaEn2LeDgQ98z_rmSBq7VSqI,1386
+aip_agents/sandbox/validation.pyi,sha256=T0-E-i5gH-nkLDe6gOuLFYEQ2zEAHIgMAO1k6VjK-KE,607
 aip_agents/schema/__init__.py,sha256=Pxqd6kvSJkKHpmkBjy3N9ncxK7aEyYFf60STfNuHgkQ,1767
 aip_agents/schema/__init__.pyi,sha256=ATPF_cX0XxXw1cDxsaTXOS6b3wjWGZyW9GxM6jKXGQ8,1917
 aip_agents/schema/a2a.py,sha256=00MZ4aTHGhd4j6kjUDzULhMqUpyOrCupVyRfK1aTuY4,1424
@@ -390,10 +434,12 @@ aip_agents/storage/providers/memory.py,sha256=Zd7QwYacEfaHLCpjvV3XaHSRulhfS464U0
 aip_agents/storage/providers/memory.pyi,sha256=NpsqKzboQGWtxtFV1ch_C3KT0KHOLyac6TtBgkLRgjI,2188
 aip_agents/storage/providers/object_storage.py,sha256=RlhXdaL1YTH9oqf7crvftsicHRxJlaLO3UgLPRCWlfg,6627
 aip_agents/storage/providers/object_storage.pyi,sha256=4pvDaNqhDleOm3ijMe_Yk3meBn2UCpLP9bKAH1l6C7w,2976
-aip_agents/tools/__init__.py,sha256=-4NJEX914Zhvl7ivBy6ztWowtzpaR4r4ammXFydjWGw,1988
-aip_agents/tools/__init__.pyi,sha256=b4plRUwyRypopaZfKMFSj0kTGb0paYL1FPm0EsWRPVw,897
+aip_agents/tools/__init__.py,sha256=2EMXjbvj2oKKjZmLkNSb56nSb_fOXAKrx-rN5DSeK6M,2157
+aip_agents/tools/__init__.pyi,sha256=QRPRqIMlmiRq9uOxKwcO2C-mmgnJm-X0yQ1W9oPLCp4,1037
 aip_agents/tools/constants.py,sha256=xNf9_pv_Rf4TCODr7BYLO05Kw5cD55MWiYfDyLWPJ3Y,5900
 aip_agents/tools/constants.pyi,sha256=ftkzuxDMnVUqVeqm4W848olK-VSlRH4oSv1p66sZnHk,3690
+aip_agents/tools/execute_ptc_code.py,sha256=ZaOPbuhSSwPSdgff_Zw4pIZ1iDnGF1WxJx8_hVVFpiU,11713
+aip_agents/tools/execute_ptc_code.pyi,sha256=ch-yVYTz2NVeCLNnHL0UeVQ-lLOTa-JkFmMSGrcFBxI,4133
 aip_agents/tools/gl_connector_tools.py,sha256=1VrXUKRiYO21v4JWOaoo1XNaIKSm_0KmXp3AMx_UL_8,4055
 aip_agents/tools/gl_connector_tools.pyi,sha256=wPK8hjFEZvQtRxBwX2MEebw6d4bXzn7nWxvcWPgYJ-c,1405
 aip_agents/tools/memory_search_tool.py,sha256=ClXW6BEjZNHb6NHe44KxbwZ6GSGo7v7AUzaV99LMouc,679
@@ -560,7 +606,7 @@ aip_agents/utils/pii/pii_helper.py,sha256=8QGVC9lb7dic_vSfLDUaDvqm45BUbYyPeQTVva
 aip_agents/utils/pii/pii_helper.pyi,sha256=wEgOasJxwKObtQ9Cb1UsiyIaqq2JUYmdRwOZR9PnIjA,3017
 aip_agents/utils/pii/uuid_deanonymizer_mapping.py,sha256=X9zeX1bhb3rlCc8P5QnbHCILx2AIhGmZwjsjh_2G4ZQ,7543
 aip_agents/utils/pii/uuid_deanonymizer_mapping.pyi,sha256=6H1xRV2Nr0LpP5K6fbz2uCobehTpM2626v8kiOd9W9Y,3157
-aip_agents_binary-0.6.0.dist-info/METADATA,sha256=g7imubdoedmX0qX3mj2-NuuyLQ8XCG6pqOK_CmhavH4,23055
-aip_agents_binary-0.6.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-aip_agents_binary-0.6.0.dist-info/top_level.txt,sha256=PEz8vcwC1bH4UrkhF0LkIYCNfXGWZUHdSklbvkBe25E,11
-aip_agents_binary-0.6.0.dist-info/RECORD,,
+aip_agents_binary-0.6.1.dist-info/METADATA,sha256=YP2lsKKsd4vrjc3HJ1cr3srb7v-ere4-wZdg7MY_rQM,23055
+aip_agents_binary-0.6.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+aip_agents_binary-0.6.1.dist-info/top_level.txt,sha256=PEz8vcwC1bH4UrkhF0LkIYCNfXGWZUHdSklbvkBe25E,11
+aip_agents_binary-0.6.1.dist-info/RECORD,,