aip-agents-binary 0.6.0__py3-none-macosx_13_0_arm64.whl → 0.6.2__py3-none-macosx_13_0_arm64.whl

aip_agents/tools/execute_ptc_code.pyi

@@ -0,0 +1,90 @@
+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 pydantic import BaseModel
+from typing import Any
+
+logger: Incomplete
+
+class PTCCodeInput(BaseModel):
+    """Input schema for PTCCodeTool."""
+    code: str
+
+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
+    args_schema: type[BaseModel]
+    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.
+        """
+    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.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aip-agents-binary
-Version: 0.6.0
+Version: 0.6.2
 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.2.dist-info}/RECORD

@@ -34,8 +34,8 @@ aip_agents/agent/langflow_agent.py,sha256=ZoJGOS8GcgGJ9Xfu4upzk3Nc1XVkpS8GSpcyKh
 aip_agents/agent/langflow_agent.pyi,sha256=z89Y7JifrsHi4Vn6LE_s0zbo77nLhkDKI-eRcxmnOr0,5667
 aip_agents/agent/langgraph_memory_enhancer_agent.py,sha256=EqH_olZ6Ue-SyQMrvYfuLkiCQAAZRMWx8izB19IYSlI,17426
 aip_agents/agent/langgraph_memory_enhancer_agent.pyi,sha256=e5sz1hbyOLw0eHo6fKf3HsCpyTe3yGqEidn9EOGb3QQ,2595
-aip_agents/agent/langgraph_react_agent.py,sha256=C1JSXFemfCdpv9CxAy1F62YqFa2uCidfPmXOWkudTR8,108486
-aip_agents/agent/langgraph_react_agent.pyi,sha256=z9qvkaDZZO0oSqX-2O_cacuXEtMSGY8bbcI4PAN6gUg,8356
+aip_agents/agent/langgraph_react_agent.py,sha256=oKmrKm1qalpSpp4bMpg81fIbKa8Lx3QvA4VDIBqS8iI,116557
+aip_agents/agent/langgraph_react_agent.pyi,sha256=15rZxqRKwjooqiwKUZoq2BJt6V32GhKi2IP_Xxl888E,10475
 aip_agents/agent/system_instruction_context.py,sha256=xTWdpKVJsWcR1uI0alCLnUsbP5sh_VWHxfF2zoi7NY0,1186
 aip_agents/agent/system_instruction_context.pyi,sha256=mdSg5sApS9G7-zr3d9BunJqwe2GB6dDD3DOCDfRrpkU,411
 aip_agents/agent/hitl/__init__.py,sha256=IyE9kWVPO-uI2_HowNm_aAxr9jyto0ZM0sZKVtpUpzo,826
@@ -209,6 +209,8 @@ aip_agents/examples/hello_world_multi_agent_langgraph_lm_invoker.py,sha256=sppLD
 aip_agents/examples/hello_world_multi_agent_langgraph_lm_invoker.pyi,sha256=zFNns9i-sPY8aBy4HaVS6tXDWTsU7f-ApLD8gDtb1uk,252
 aip_agents/examples/hello_world_pii_logger.py,sha256=pm9LK-doZwQL-VdJYFHziuvwH562c8wwAwmKZeAWQS0,584
 aip_agents/examples/hello_world_pii_logger.pyi,sha256=5gOVrvhQV2DxIwr7ocL1-oTUXsO8XSyl5WN70c7rl_w,134
+aip_agents/examples/hello_world_ptc.py,sha256=A3aadYO3H41UgKZoHqrThXz2_hOFGX8-E4XwFnRi7P4,1410
+aip_agents/examples/hello_world_ptc.pyi,sha256=Dbo4myHbndrL6PyA_mgf3HTIYqLudTaF4h_nsdqOyKs,231
 aip_agents/examples/hello_world_sentry.py,sha256=Gccr6WCevSnMFZYT5M6uJ_22EvLmAMUE8Et_H510_kM,3976
 aip_agents/examples/hello_world_sentry.pyi,sha256=k0wFj46QqXEjBU0CEERmiEf2bovVFkupzuzQzeD6h00,671
 aip_agents/examples/hello_world_step_limits.py,sha256=hRV9XERlq2qW_yZ4kHAe1hTgukrSZjThIrsY9NuTdJg,9809
@@ -352,6 +354,48 @@ aip_agents/middleware/manager.py,sha256=dlFhWC8fmKzbL6KIioeBv6cO8TSNfwmSYbciTxGN
 aip_agents/middleware/manager.pyi,sha256=a-OJeUVP6pIVvG9p8nyIC3oCOWfPVqH1EGykUeePXAs,3577
 aip_agents/middleware/todolist.py,sha256=BwElJRwn_hG-EC6YJOiaCG3zbAab03e0HkIbHv4LSik,10514
 aip_agents/middleware/todolist.pyi,sha256=AvRUtHdchxbP9A3ikJwF1EmQqJ-tbzy12eSXpQ-ZDiU,4195
+aip_agents/ptc/__init__.py,sha256=khwauiDZ7ZozQaPmnkszDcwEwnBkfuhZ69_obuQVXWY,1410
+aip_agents/ptc/__init__.pyi,sha256=oWHcEv30flFGfDSNHoOCbR_W5hLeyPPLSZcf02Qok9o,565
+aip_agents/ptc/doc_gen.py,sha256=ym6psNvZ_8iKB3AGrb1TTFoQiXXLdP2zKp6c4SPf4JA,3382
+aip_agents/ptc/doc_gen.pyi,sha256=fJ92pbil2mMPv_9AHctVqFNZVE7P8njisLrn9zSSmdA,1266
+aip_agents/ptc/exceptions.py,sha256=iNO9Xmz5P34K6FLnxgiGxA9bjPceVf6rUqhrugc5Xq8,905
+aip_agents/ptc/exceptions.pyi,sha256=VmypiFm4ZWv5KIYwxUMN9GWOmDMMzkdYdVSoKxfqLTw,664
+aip_agents/ptc/executor.py,sha256=I9YqEZx1BN7QhU00TIOmIGQ5xY3-kxWlU0zNxp4SgeM,5063
+aip_agents/ptc/executor.pyi,sha256=h9kAq8Un7ShSQxi-_JXFIXyCQz3pDCwx0zs5h56B3rQ,3251
+aip_agents/ptc/naming.py,sha256=Cb0Q3SoBNPLGrmWG6Fn2KBzKoJIuSXzy-A_hUw0kWCg,5001
+aip_agents/ptc/naming.pyi,sha256=TgIbV0lLStbtjTKadnGlgDVjH90L0IOHXd5W10YUpvg,2141
+aip_agents/ptc/payload.py,sha256=DPEeunaKGk3iWxdpE-1YVD_iuHQPhWqIvcL2TaYPfig,835
+aip_agents/ptc/payload.pyi,sha256=qPgwyDY14tAsx-r_nt3ACZAfgBmJiAPdH7MXBJJVqjk,631
+aip_agents/ptc/prompt_builder.py,sha256=3Xu0Fi3j6yhxRWJAuST8yjYTRfTbYUEBdz53dXWwjMw,17640
+aip_agents/ptc/prompt_builder.pyi,sha256=b0njp0fb0LmlqCJs59xCv-yc2hGzaadoXVQWTdfrKhU,2234
+aip_agents/ptc/ptc_helper.py,sha256=DARdrCwEfuaKOGdVYE1YuudXCbIvkv6rSgcGYK7S_z8,439
+aip_agents/ptc/ptc_helper.pyi,sha256=FOgjwJR_J15JQjjTNU9oTG-s6d7ocks364ijMuVTHT0,77
+aip_agents/ptc/sandbox_bridge.py,sha256=DeLA1REOcaOfsHHGB_446VNRo9UfrgNExxx4eOWU-OM,1635
+aip_agents/ptc/sandbox_bridge.pyi,sha256=RqZmnNDMcvFJREWC3QeBcgSvZDYeYf1vhG5dNe8zAmw,979
+aip_agents/ptc/template_utils.py,sha256=wHwnMNnf_lz-dj8oWnVXwCp7szw4_Nk4kh4FLcRxlfM,898
+aip_agents/ptc/template_utils.pyi,sha256=hNpgwyalUMezalSJry9FeQsFfjwuCxl_dZHBimkShF0,438
+aip_agents/ptc/mcp/__init__.py,sha256=DlT25Zs_SDstW8vAFxomKlSF4QCpe_k6Sa42fN22ikg,1118
+aip_agents/ptc/mcp/__init__.pyi,sha256=nKRY7DrlqgmDThlBh9uDQ5LCchfWteaPX0Er37-AtMI,916
+aip_agents/ptc/mcp/sandbox_bridge.py,sha256=HzzmEJmkkNb6ZNVUIq0gHBSD-ZXOZnp1n_PyYGAHk58,20418
+aip_agents/ptc/mcp/sandbox_bridge.pyi,sha256=dzQ5gzkF1_h0kZxlKt9nY6R2HvOn9q_TZ50T4V7w0gQ,2033
+aip_agents/ptc/mcp/templates/__init__.py,sha256=gUaLjHPncH1Js6dHiNRSpbAwngKY83gQcRCARKl42DU,57
+aip_agents/ptc/mcp/templates/__init__.pyi,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+aip_agents/ptc/mcp/templates/mcp_client.py.template,sha256=Ql6Wtj31oz8Kb6rUJP2_RnaTScXJS94ujgyg0MuZPVc,7446
+aip_agents/ptc/templates/__init__.py,sha256=T_jAb0bAhq1en0feNp4kNDkyFCU67GuzCBeoGtSOgD0,56
+aip_agents/ptc/templates/__init__.pyi,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+aip_agents/ptc/templates/ptc_helper.py.template,sha256=_heBHy5XtLWYGONtof7qstyEbeHMtHDwCYhxL4-BNDM,4064
+aip_agents/sandbox/__init__.py,sha256=1zA88_E_WaOm0KaDMsaznWm9qEQvqPF8Ain4w4LFFQo,1366
+aip_agents/sandbox/__init__.pyi,sha256=l-anSofX18C7Mk0PReCTDZ-tIv8PGg9w0YdNnppve2k,341
+aip_agents/sandbox/defaults.py,sha256=rKvzBl494N1Qx3-GZ7yBM07xUVSZo675OCVfROlDcUI,232
+aip_agents/sandbox/defaults.pyi,sha256=TV4aNiyAwsERFBa6cnYmUC2jyBVxccDR1v5wW6M6YJU,64
+aip_agents/sandbox/e2b_runtime.py,sha256=_Fht2K9rB-RGuX0lulQedw4DLxdOkG288g_ALEWBQaM,9217
+aip_agents/sandbox/e2b_runtime.pyi,sha256=sQvrc-hg9CV4KU7D9vJdaLjVY7knT_S-BnszwLB4Sno,2152
+aip_agents/sandbox/template_builder.py,sha256=COZr33dVodBag4Kz3o7qHqyyykQGErHbL4DZw3Hn7Zs,4348
+aip_agents/sandbox/template_builder.pyi,sha256=DFOrf1IVkRBcrI4fPm-_CrO_n4JQM4xMfX2z2APIm68,1522
+aip_agents/sandbox/types.py,sha256=D4f3k8bfJwRYM2re0e-5Dxfg58FLIFYY3YE_z1Wqgh8,528
+aip_agents/sandbox/types.pyi,sha256=P-BGzoiSJA8nQPQNluowNyrFKt5UnLBxeeOL5pamiRU,366
+aip_agents/sandbox/validation.py,sha256=UqBv6TTTmiIv6oCymEHDRlo0QCi45BQ6ceW9Cs7_TM0,1336
+aip_agents/sandbox/validation.pyi,sha256=rCgd4JJkHECFFo2Oa0LtPaBK0wjvIqOZeV9-aFU9tC4,587
 aip_agents/schema/__init__.py,sha256=eQdptrXX7Rxo0xAJ1tXfoPhIaFoVQTDvrRAJAF3SjSM,1698
 aip_agents/schema/__init__.pyi,sha256=kho3g8pKCp1FBZ5SwyK3Vom5rSEgHJ4rYv81QIX3bGg,1908
 aip_agents/schema/a2a.py,sha256=04WVxpFjVuiQc2-w3m3TLO3h3NuApVuFrzJQq5casU4,1368
@@ -390,10 +434,12 @@ aip_agents/storage/providers/memory.py,sha256=81E9yT_oRkH7tMVr5azZv1oXIaeTzTAtTu
 aip_agents/storage/providers/memory.pyi,sha256=r0meeCsjKyE9FHJqCysfQ5ZH_FWRRv2SWg-NOHCUD-g,2109
 aip_agents/storage/providers/object_storage.py,sha256=Q268Sn8Y09gS-ilP6fe4CR0YIOvdbunh3V6SmuQsAVs,6413
 aip_agents/storage/providers/object_storage.pyi,sha256=nBIKJjUAWkIKhYhBFmFcypjoOieIYOghy55f62x7AX8,2878
-aip_agents/tools/__init__.py,sha256=J0wDdcKre3Qc_z1hhzKfPiaa1m2OPOXyHjl6uwOkApU,1935
-aip_agents/tools/__init__.pyi,sha256=w1gRWL5iRdLtuBNXBhmD87uzfvIRrXwq3uZ1meku8js,888
+aip_agents/tools/__init__.py,sha256=Kvd185p8zZ2xpNqimxURJUA6joPXP-3MIrr8K3Lf9Rw,2102
+aip_agents/tools/__init__.pyi,sha256=K2j6MjKo5aaMAbWkknLwyRbSu14LxJyRMoi88DbHROk,1027
 aip_agents/tools/constants.py,sha256=AabnuPQG_mc2sVdr9jV23_6bFempAsxQv3kdCR_ztLA,5723
 aip_agents/tools/constants.pyi,sha256=kFY8dKgRqHKGvPqG3QUyuEc8C5yRaDj7DYQDH88K2T0,3552
+aip_agents/tools/execute_ptc_code.py,sha256=mN8-G50voxVqWH9r3uJGQOtgPtxsHaTOnJrJuojQLIM,11396
+aip_agents/tools/execute_ptc_code.pyi,sha256=nDlpxV-kcKuNmtghahjXAtjWvtNv6D38x8-VTNCKYjU,4089
 aip_agents/tools/gl_connector_tools.py,sha256=bxl_3VQYZDv3lFn6Y3kDVVRFwH4cntOLz3f74YzDcic,3936
 aip_agents/tools/gl_connector_tools.pyi,sha256=2ATn_MW_FRg5Uv7dLI_ToBOtlgTSfj0zgDQpN1N-cJs,1366
 aip_agents/tools/memory_search_tool.py,sha256=gqTTb_Fao_Hl-SavfaFsqPDhnFQUjzIQUkzizSD2L0A,653
@@ -560,7 +606,7 @@ aip_agents/utils/pii/pii_helper.py,sha256=g0yRzakfA2AA6vjUNLWHqFlcxyLql6MXQ90NN3
 aip_agents/utils/pii/pii_helper.pyi,sha256=dulZs150ikbAL3Bw2YLcz3_g4DsGmL3lciwf8mKxEjI,2939
 aip_agents/utils/pii/uuid_deanonymizer_mapping.py,sha256=Gks8l8t0cuS9pzoQnrpiK1CaLmWYksjOnTeiHh3_7EE,7348
 aip_agents/utils/pii/uuid_deanonymizer_mapping.pyi,sha256=gnWfD1rWZh_tloJjgKiZ6f6iNUuBaHpKqCSiP0d-9bs,3084
-aip_agents_binary-0.6.0.dist-info/METADATA,sha256=PC6WXiABFDyUIszE8akBBT0YZyQGLbVllkvhdnTLd7g,22371
-aip_agents_binary-0.6.0.dist-info/WHEEL,sha256=KxCTaSkoYs_EnWvWxmau4HAvN-_rCRYV_bfRc_41A9k,106
-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.2.dist-info/METADATA,sha256=pwc5Qenz-QKrlBcmtfP-CwkSLbEp0Wi-LjqBMmtqFRY,22371
+aip_agents_binary-0.6.2.dist-info/WHEEL,sha256=KxCTaSkoYs_EnWvWxmau4HAvN-_rCRYV_bfRc_41A9k,106
+aip_agents_binary-0.6.2.dist-info/top_level.txt,sha256=PEz8vcwC1bH4UrkhF0LkIYCNfXGWZUHdSklbvkBe25E,11
+aip_agents_binary-0.6.2.dist-info/RECORD,,