aip-agents-binary 0.6.1__py3-none-any.whl → 0.6.2__py3-none-any.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_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