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

aip_agents/examples/hello_world_ptc.py

@@ -7,8 +7,6 @@ Required environment variables:
 
 import asyncio
 
-from langchain_openai import ChatOpenAI
-
 from aip_agents.agent import LangGraphReactAgent
 from aip_agents.ptc import PromptConfig, PTCSandboxConfig
 
@@ -24,7 +22,7 @@ async def main() -> None:
     agent = LangGraphReactAgent(
         name="ptc_hello_world",
         instruction=instruction,
-        model=ChatOpenAI(model="gpt-5.2"),
+        model="openai/gpt-5.2",
         ptc_config=PTCSandboxConfig(enabled=True, sandbox_timeout=180.0, prompt=PromptConfig(mode="index")),
     )
     agent.add_mcp_server(

aip_agents/tools/execute_ptc_code.py

@@ -18,6 +18,7 @@ from langchain_core.callbacks import (
     CallbackManagerForToolRun,
 )
 from langchain_core.tools import BaseTool
+from pydantic import BaseModel, Field
 
 from aip_agents.ptc.naming import sanitize_function_name
 from aip_agents.tools.tool_config_injector import TOOL_CONFIGS_KEY
@@ -31,6 +32,21 @@ if TYPE_CHECKING:
 logger = get_logger(__name__)
 
 
+class PTCCodeInput(BaseModel):
+    """Input schema for PTCCodeTool."""
+
+    code: str = Field(
+        ...,
+        description=(
+            "Python code to execute. Import MCP tools from the generated `tools` package, "
+            "for example: `from tools.yfinance import get_stock_history`. "
+            "The code runs in a sandboxed environment with access to all configured MCP tools. "
+            "Use print() to output results. The tool returns JSON with keys: "
+            "ok, stdout, stderr, exit_code."
+        ),
+    )
+
+
 def _merge_config_layer(
     merged: dict[str, dict[str, Any]],
     source: dict[str, Any],
@@ -123,6 +139,9 @@ class PTCCodeTool(BaseTool):
         "This tool is useful for chaining multiple MCP tool calls with local data processing."
     )
 
+    # Input schema for LangChain tool invocation
+    args_schema: type[BaseModel] = PTCCodeInput
+
     # Internal attributes (not exposed to LLM)
     _ptc_executor: "PTCSandboxExecutor" = None  # type: ignore[assignment]
     _ptc_runtime: "E2BSandboxRuntime" = None  # type: ignore[assignment]
@@ -149,22 +168,6 @@ class PTCCodeTool(BaseTool):
         object.__setattr__(self, "_ptc_runtime", runtime)
         object.__setattr__(self, "_agent_tool_configs", agent_tool_configs)
 
-    @property
-    def args(self) -> dict[str, Any]:
-        """Return the argument schema for the tool."""
-        return {
-            "code": {
-                "type": "string",
-                "description": (
-                    "Python code to execute. Import MCP tools from the generated `tools` package, "
-                    "for example: `from tools.yfinance import get_stock_history`. "
-                    "The code runs in a sandboxed environment with access to all configured MCP tools. "
-                    "Use print() to output results. The tool returns JSON with keys: "
-                    "ok, stdout, stderr, exit_code."
-                ),
-            }
-        }
-
     def _run(
         self,
         code: str,

aip_agents/tools/execute_ptc_code.pyi

@@ -6,10 +6,15 @@ from aip_agents.sandbox.e2b_runtime import E2BSandboxRuntime as E2BSandboxRuntim
 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.
 
@@ -42,6 +47,7 @@ class PTCCodeTool(BaseTool):
     """
     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.
 
@@ -51,9 +57,6 @@ class PTCCodeTool(BaseTool):
             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."""
 

{aip_agents_binary-0.6.1.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.1
+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.1.dist-info → aip_agents_binary-0.6.2.dist-info}/RECORD

@@ -209,7 +209,7 @@ 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=EkSHUfRiGWLhUBcwpjWTRzAILqYzEPpys0pXmndWjk0,1462
+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
@@ -438,8 +438,8 @@ aip_agents/tools/__init__.py,sha256=Kvd185p8zZ2xpNqimxURJUA6joPXP-3MIrr8K3Lf9Rw,
 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=h1-bnW4u5HoDFZGQbROitREE4MGblvlwKqHXSNL3XUM,11408
-aip_agents/tools/execute_ptc_code.pyi,sha256=4wXrAPjUYoXWODxg90expdvIetTlULvFZwGvKbdbJrE,4046
+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
@@ -606,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.1.dist-info/METADATA,sha256=SHkg6ELXvsKcMmpLPVaJSxPYoSHbpMqUmhouUUFqAVg,22371
-aip_agents_binary-0.6.1.dist-info/WHEEL,sha256=KxCTaSkoYs_EnWvWxmau4HAvN-_rCRYV_bfRc_41A9k,106
-aip_agents_binary-0.6.1.dist-info/top_level.txt,sha256=PEz8vcwC1bH4UrkhF0LkIYCNfXGWZUHdSklbvkBe25E,11
-aip_agents_binary-0.6.1.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,,