swarmauri_tool_containernewsession 0.7.5.dev1__tar.gz → 0.9.0.dev2__tar.gz

{swarmauri_tool_containernewsession-0.7.5.dev1 → swarmauri_tool_containernewsession-0.9.0.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: swarmauri_tool_containernewsession
-Version: 0.7.5.dev1
+Version: 0.9.0.dev2
 Summary: Create a new Docker or Kubernetes session.
 License: Apache-2.0
 Author: Jacob Stewart
@@ -47,7 +47,8 @@ pip install swarmauri_tool_containernewsession
 ```python
 from swarmauri_tool_containernewsession import ContainerNewSessionTool
 
-result = ContainerNewSessionTool()(container_name="my-container", image="ubuntu:latest")
+tool = ContainerNewSessionTool(container_name="my-container", image="ubuntu:latest")
+result = tool()
 print(result["stdout"])
 ```
 

{swarmauri_tool_containernewsession-0.7.5.dev1 → swarmauri_tool_containernewsession-0.9.0.dev2}/README.md

@@ -30,6 +30,7 @@ pip install swarmauri_tool_containernewsession
 ```python
 from swarmauri_tool_containernewsession import ContainerNewSessionTool
 
-result = ContainerNewSessionTool()(container_name="my-container", image="ubuntu:latest")
+tool = ContainerNewSessionTool(container_name="my-container", image="ubuntu:latest")
+result = tool()
 print(result["stdout"])
 ```

{swarmauri_tool_containernewsession-0.7.5.dev1 → swarmauri_tool_containernewsession-0.9.0.dev2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "swarmauri_tool_containernewsession"
-version = "0.7.5.dev1"
+version = "0.9.0.dev2"
 description = "Create a new Docker or Kubernetes session."
 license = "Apache-2.0"
 readme = "README.md"

{swarmauri_tool_containernewsession-0.7.5.dev1 → swarmauri_tool_containernewsession-0.9.0.dev2}/swarmauri_tool_containernewsession/ContainerNewSessionTool.py

@@ -17,32 +17,43 @@ class ContainerNewSessionTool(ToolBase):
 
     driver: Literal["docker", "kubernetes"] = "docker"
 
+    container_name: str | None = Field(
+        default=None, description="Name for the container or pod"
+    )
+    image: str | None = Field(default=None, description="Container image to launch")
+
     parameters: List[Parameter] = Field(
         default_factory=lambda: [
             Parameter(
                 name="container_name",
                 input_type="string",
                 description="Name for the container or pod",
-                required=True,
+                required=False,
             ),
             Parameter(
                 name="image",
                 input_type="string",
                 description="Container image to launch",
-                required=True,
+                required=False,
             ),
         ]
     )
 
-    def __call__(self, container_name: str, image: str) -> dict:
+    def __call__(
+        self, container_name: str | None = None, image: str | None = None
+    ) -> dict:
+        name = container_name or self.container_name
+        img = image or self.image
+        if name is None or img is None:
+            raise ValueError("container_name and image must be provided")
         if self.driver == "docker":
             cmd = [
                 "docker",
                 "run",
                 "-d",
                 "--name",
-                container_name,
-                image,
+                name,
+                img,
                 "sleep",
                 "infinity",
             ]
@@ -50,9 +61,9 @@ class ContainerNewSessionTool(ToolBase):
             cmd = [
                 "kubectl",
                 "run",
-                container_name,
+                name,
                 "--image",
-                image,
+                img,
                 "-i",
                 "--restart=Never",
                 "--command",
@@ -63,4 +74,4 @@ class ContainerNewSessionTool(ToolBase):
         result = subprocess.run(cmd, capture_output=True, text=True)
         if result.returncode != 0:
             raise RuntimeError(result.stderr.strip())
-        return {"session_name": container_name, "stdout": result.stdout.strip()}
+        return {"session_name": name, "stdout": result.stdout.strip()}