nebu 0.1.0__tar.gz

nebu-0.1.0/LICENSE

@@ -0,0 +1,201 @@
+                                 Apache License
+                           Version 2.0, January 2004
+                        http://www.apache.org/licenses/
+
+   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+   1. Definitions.
+
+      "License" shall mean the terms and conditions for use, reproduction,
+      and distribution as defined by Sections 1 through 9 of this document.
+
+      "Licensor" shall mean the copyright owner or entity authorized by
+      the copyright owner that is granting the License.
+
+      "Legal Entity" shall mean the union of the acting entity and all
+      other entities that control, are controlled by, or are under common
+      control with that entity. For the purposes of this definition,
+      "control" means (i) the power, direct or indirect, to cause the
+      direction or management of such entity, whether by contract or
+      otherwise, or (ii) ownership of fifty percent (50%) or more of the
+      outstanding shares, or (iii) beneficial ownership of such entity.
+
+      "You" (or "Your") shall mean an individual or Legal Entity
+      exercising permissions granted by this License.
+
+      "Source" form shall mean the preferred form for making modifications,
+      including but not limited to software source code, documentation
+      source, and configuration files.
+
+      "Object" form shall mean any form resulting from mechanical
+      transformation or translation of a Source form, including but
+      not limited to compiled object code, generated documentation,
+      and conversions to other media types.
+
+      "Work" shall mean the work of authorship, whether in Source or
+      Object form, made available under the License, as indicated by a
+      copyright notice that is included in or attached to the work
+      (an example is provided in the Appendix below).
+
+      "Derivative Works" shall mean any work, whether in Source or Object
+      form, that is based on (or derived from) the Work and for which the
+      editorial revisions, annotations, elaborations, or other modifications
+      represent, as a whole, an original work of authorship. For the purposes
+      of this License, Derivative Works shall not include works that remain
+      separable from, or merely link (or bind by name) to the interfaces of,
+      the Work and Derivative Works thereof.
+
+      "Contribution" shall mean any work of authorship, including
+      the original version of the Work and any modifications or additions
+      to that Work or Derivative Works thereof, that is intentionally
+      submitted to Licensor for inclusion in the Work by the copyright owner
+      or by an individual or Legal Entity authorized to submit on behalf of
+      the copyright owner. For the purposes of this definition, "submitted"
+      means any form of electronic, verbal, or written communication sent
+      to the Licensor or its representatives, including but not limited to
+      communication on electronic mailing lists, source code control systems,
+      and issue tracking systems that are managed by, or on behalf of, the
+      Licensor for the purpose of discussing and improving the Work, but
+      excluding communication that is conspicuously marked or otherwise
+      designated in writing by the copyright owner as "Not a Contribution."
+
+      "Contributor" shall mean Licensor and any individual or Legal Entity
+      on behalf of whom a Contribution has been received by Licensor and
+      subsequently incorporated within the Work.
+
+   2. Grant of Copyright License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      copyright license to reproduce, prepare Derivative Works of,
+      publicly display, publicly perform, sublicense, and distribute the
+      Work and such Derivative Works in Source or Object form.
+
+   3. Grant of Patent License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      (except as stated in this section) patent license to make, have made,
+      use, offer to sell, sell, import, and otherwise transfer the Work,
+      where such license applies only to those patent claims licensable
+      by such Contributor that are necessarily infringed by their
+      Contribution(s) alone or by combination of their Contribution(s)
+      with the Work to which such Contribution(s) was submitted. If You
+      institute patent litigation against any entity (including a
+      cross-claim or counterclaim in a lawsuit) alleging that the Work
+      or a Contribution incorporated within the Work constitutes direct
+      or contributory patent infringement, then any patent licenses
+      granted to You under this License for that Work shall terminate
+      as of the date such litigation is filed.
+
+   4. Redistribution. You may reproduce and distribute copies of the
+      Work or Derivative Works thereof in any medium, with or without
+      modifications, and in Source or Object form, provided that You
+      meet the following conditions:
+
+      (a) You must give any other recipients of the Work or
+          Derivative Works a copy of this License; and
+
+      (b) You must cause any modified files to carry prominent notices
+          stating that You changed the files; and
+
+      (c) You must retain, in the Source form of any Derivative Works
+          that You distribute, all copyright, patent, trademark, and
+          attribution notices from the Source form of the Work,
+          excluding those notices that do not pertain to any part of
+          the Derivative Works; and
+
+      (d) If the Work includes a "NOTICE" text file as part of its
+          distribution, then any Derivative Works that You distribute must
+          include a readable copy of the attribution notices contained
+          within such NOTICE file, excluding those notices that do not
+          pertain to any part of the Derivative Works, in at least one
+          of the following places: within a NOTICE text file distributed
+          as part of the Derivative Works; within the Source form or
+          documentation, if provided along with the Derivative Works; or,
+          within a display generated by the Derivative Works, if and
+          wherever such third-party notices normally appear. The contents
+          of the NOTICE file are for informational purposes only and
+          do not modify the License. You may add Your own attribution
+          notices within Derivative Works that You distribute, alongside
+          or as an addendum to the NOTICE text from the Work, provided
+          that such additional attribution notices cannot be construed
+          as modifying the License.
+
+      You may add Your own copyright statement to Your modifications and
+      may provide additional or different license terms and conditions
+      for use, reproduction, or distribution of Your modifications, or
+      for any such Derivative Works as a whole, provided Your use,
+      reproduction, and distribution of the Work otherwise complies with
+      the conditions stated in this License.
+
+   5. Submission of Contributions. Unless You explicitly state otherwise,
+      any Contribution intentionally submitted for inclusion in the Work
+      by You to the Licensor shall be under the terms and conditions of
+      this License, without any additional terms or conditions.
+      Notwithstanding the above, nothing herein shall supersede or modify
+      the terms of any separate license agreement you may have executed
+      with Licensor regarding such Contributions.
+
+   6. Trademarks. This License does not grant permission to use the trade
+      names, trademarks, service marks, or product names of the Licensor,
+      except as required for reasonable and customary use in describing the
+      origin of the Work and reproducing the content of the NOTICE file.
+
+   7. Disclaimer of Warranty. Unless required by applicable law or
+      agreed to in writing, Licensor provides the Work (and each
+      Contributor provides its Contributions) on an "AS IS" BASIS,
+      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+      implied, including, without limitation, any warranties or conditions
+      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+      PARTICULAR PURPOSE. You are solely responsible for determining the
+      appropriateness of using or redistributing the Work and assume any
+      risks associated with Your exercise of permissions under this License.
+
+   8. Limitation of Liability. In no event and under no legal theory,
+      whether in tort (including negligence), contract, or otherwise,
+      unless required by applicable law (such as deliberate and grossly
+      negligent acts) or agreed to in writing, shall any Contributor be
+      liable to You for damages, including any direct, indirect, special,
+      incidental, or consequential damages of any character arising as a
+      result of this License or out of the use or inability to use the
+      Work (including but not limited to damages for loss of goodwill,
+      work stoppage, computer failure or malfunction, or any and all
+      other commercial damages or losses), even if such Contributor
+      has been advised of the possibility of such damages.
+
+   9. Accepting Warranty or Additional Liability. While redistributing
+      the Work or Derivative Works thereof, You may choose to offer,
+      and charge a fee for, acceptance of support, warranty, indemnity,
+      or other liability obligations and/or rights consistent with this
+      License. However, in accepting such obligations, You may act only
+      on Your own behalf and on Your sole responsibility, not on behalf
+      of any other Contributor, and only if You agree to indemnify,
+      defend, and hold each Contributor harmless for any liability
+      incurred by, or claims asserted against, such Contributor by reason
+      of your accepting any such warranty or additional liability.
+
+   END OF TERMS AND CONDITIONS
+
+   APPENDIX: How to apply the Apache License to your work.
+
+      To apply the Apache License to your work, attach the following
+      boilerplate notice, with the fields enclosed by brackets "[]"
+      replaced with your own identifying information. (Don't include
+      the brackets!)  The text should be enclosed in the appropriate
+      comment syntax for the file format. We also recommend that a
+      file or class name and description of purpose be included on the
+      same "printed page" as the copyright notice for easier
+      identification within third-party archives.
+
+   Copyright [yyyy] [name of copyright owner]
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.

nebu-0.1.0/PKG-INFO

@@ -0,0 +1,58 @@
+Metadata-Version: 2.4
+Name: nebu
+Version: 0.1.0
+Summary: A globally distributed container runtime
+Requires-Python: >=3.10.14
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: openai>=1.68.2
+Requires-Dist: pydantic>=2.10.6
+Requires-Dist: pyyaml>=6.0.2
+Requires-Dist: requests>=2.32.3
+Dynamic: license-file
+
+# nebulous-py
+A python library for the [Nebulous runtime](https://github.com/agentsea/nebulous)
+
+## Installation
+
+```bash
+pip install nebu
+```
+
+## Usage
+
+```python
+from nebu import Container, V1EnvVar, V1ResourceMeta
+
+container = Container(
+    metadata=V1ResourceMeta(
+        name="pytorch-example",
+        namespace="test",
+    ),
+    image="pytorch/pytorch:latest",
+    platform="runpod",
+    env=[V1EnvVar(name="MY_ENV_VAR", value="my-value")],
+    command="nvidia-smi",
+    accelerators=["1:A100_SXM"],
+    proxy_port=8080,
+)
+
+while container.status.status.lower() != "running":
+    print(f"Container '{container.metadata.name}' is not running, it is '{container.status.status}', waiting...")
+    time.sleep(1)
+
+print(f"Container '{container.metadata.name}' is running")
+
+print(f"You can access the container at {container.status.tailnet_url}")
+```
+
+## Contributing
+
+Please open an issue or a PR to contribute to the project.
+
+## Development
+
+```bash
+make test
+```

nebu-0.1.0/README.md

@@ -0,0 +1,45 @@
+# nebulous-py
+A python library for the [Nebulous runtime](https://github.com/agentsea/nebulous)
+
+## Installation
+
+```bash
+pip install nebu
+```
+
+## Usage
+
+```python
+from nebu import Container, V1EnvVar, V1ResourceMeta
+
+container = Container(
+    metadata=V1ResourceMeta(
+        name="pytorch-example",
+        namespace="test",
+    ),
+    image="pytorch/pytorch:latest",
+    platform="runpod",
+    env=[V1EnvVar(name="MY_ENV_VAR", value="my-value")],
+    command="nvidia-smi",
+    accelerators=["1:A100_SXM"],
+    proxy_port=8080,
+)
+
+while container.status.status.lower() != "running":
+    print(f"Container '{container.metadata.name}' is not running, it is '{container.status.status}', waiting...")
+    time.sleep(1)
+
+print(f"Container '{container.metadata.name}' is running")
+
+print(f"You can access the container at {container.status.tailnet_url}")
+```
+
+## Contributing
+
+Please open an issue or a PR to contribute to the project.
+
+## Development
+
+```bash
+make test
+```

nebu-0.1.0/pyproject.toml

@@ -0,0 +1,24 @@
+[project]
+name = "nebu"
+version = "0.1.0"
+description = "A globally distributed container runtime"
+readme = "README.md"
+requires-python = ">=3.10.14"
+dependencies = [
+    "openai>=1.68.2",
+    "pydantic>=2.10.6",
+    "pyyaml>=6.0.2",
+    "requests>=2.32.3",
+]
+
+[dependency-groups]
+dev = [
+    "pytest>=8.3.5",
+]
+
+[tool.pyright]
+reportUnknownParameterType = false
+reportMissingTypeArgument = false
+reportUnknownMemberType = false
+reportUnknownVariableType = false
+reportUnknownArgumentType = false

nebu-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

nebu-0.1.0/src/nebu/__init__.py

@@ -0,0 +1,9 @@
+# type: ignore
+# pylint: disable=all
+# ruff: noqa: F401
+# ruff: noqa: F403
+
+from .config import *
+from .containers.container import Container
+from .containers.models import *
+from .meta import *

nebu-0.1.0/src/nebu/config.py

@@ -0,0 +1,135 @@
+from __future__ import annotations
+
+import os
+from dataclasses import dataclass, field
+from typing import List, Optional
+
+import yaml
+
+
+@dataclass
+class ServerConfig:
+    """
+    Python equivalent of the Rust ServerConfig struct.
+    """
+
+    name: Optional[str] = None
+    api_key: Optional[str] = None
+    server: Optional[str] = None
+    auth_server: Optional[str] = None
+
+
+@dataclass
+class GlobalConfig:
+    """
+    Python equivalent of the Rust GlobalConfig struct.
+    Manages multiple ServerConfig entries and a current_server pointer.
+    """
+
+    servers: List[ServerConfig] = field(default_factory=list)
+    current_server: Optional[str] = None
+
+    @classmethod
+    def read(cls) -> GlobalConfig:
+        """
+        Read the config from ~/.agentsea/nebu.yaml, or create a default if it doesn’t exist.
+        Then ensure that we either find or create a matching server from environment variables,
+        and set that as the `current_server` if relevant (mimicking the Rust logic).
+        """
+        path = _get_config_file_path()
+        path_exists = os.path.exists(path)
+
+        # Load from disk or create a default
+        if path_exists:
+            with open(path, "r") as yaml_file:
+                data = yaml.safe_load(yaml_file) or {}
+            # Convert each server entry into a ServerConfig
+            servers_data = data.get("servers", [])
+            servers = [ServerConfig(**srv) for srv in servers_data]
+            current_server = data.get("current_server")
+            config = cls(servers=servers, current_server=current_server)
+        else:
+            config = cls()  # default
+
+        # Collect environment variables (no fallback defaults here)
+        env_api_key = os.environ.get("NEBU_API_KEY") or os.environ.get(
+            "AGENTSEA_API_KEY"
+        )
+        env_server = os.environ.get("NEBU_SERVER") or os.environ.get("AGENTSEA_SERVER")
+        env_auth_server = os.environ.get("NEBU_AUTH_SERVER") or os.environ.get(
+            "AGENTSEA_AUTH_SERVER"
+        )
+
+        # Only proceed if all three environment variables are present
+        if env_api_key and env_server and env_auth_server:
+            # Find a matching server
+            found_server = None
+            for srv in config.servers:
+                if (
+                    srv.api_key == env_api_key
+                    and srv.server == env_server
+                    and srv.auth_server == env_auth_server
+                ):
+                    found_server = srv
+                    break
+
+            server_name = "env-based-server"
+            if found_server:
+                # Ensure it has a name, so we can set current_server to it
+                if found_server.name is None:
+                    found_server.name = server_name
+                # Use that server’s name as current
+                config.current_server = found_server.name
+            else:
+                # Create a new server entry
+                new_server = ServerConfig(
+                    name=server_name,
+                    api_key=env_api_key,
+                    server=env_server,
+                    auth_server=env_auth_server,
+                )
+                config.servers.append(new_server)
+                config.current_server = server_name
+
+        # Write if the file didn't already exist
+        if not path_exists:
+            config.write()
+
+        return config
+
+    def write(self) -> None:
+        """
+        Write the current GlobalConfig to disk as YAML.
+        """
+        path = _get_config_file_path()
+        # Create parent directories if they don't exist
+        os.makedirs(os.path.dirname(path), exist_ok=True)
+
+        # Convert our dataclass-based objects into simple dictionaries
+        data = {
+            "servers": [srv.__dict__ for srv in self.servers],
+            "current_server": self.current_server,
+        }
+
+        with open(path, "w") as yaml_file:
+            yaml.dump(data, yaml_file)
+            yaml_file.flush()
+
+    def get_current_server_config(self) -> Optional[ServerConfig]:
+        """
+        Get the server config for the current_server name, or None if unset/missing.
+        """
+        if self.current_server:
+            for srv in self.servers:
+                if srv.name == self.current_server:
+                    return srv
+        return None
+
+
+def _get_config_file_path() -> str:
+    """
+    Return the path to ~/.agentsea/nebu.yaml
+    """
+    home = os.path.expanduser("~")
+    config_dir = os.path.join(home, ".agentsea")
+    return os.path.join(config_dir, "nebu.yaml")

nebu-0.1.0/src/nebu/containers/container.py

@@ -0,0 +1,241 @@
+from typing import List, Optional
+
+import requests
+
+from nebu.config import GlobalConfig  # or wherever your GlobalConfig is defined
+from nebu.containers.models import (
+    DEFAULT_RESTART_POLICY,
+    V1AuthzConfig,
+    V1Container,
+    V1ContainerRequest,
+    V1ContainerResources,
+    V1Containers,
+    V1EnvVar,
+    V1Meter,
+    V1PortRequest,
+    V1ResourceMetaRequest,
+    V1SSHKey,
+    V1VolumePath,
+)
+
+
+class Container:
+    def __init__(
+        self,
+        name: str,
+        namespace: str = "default",
+        platform: Optional[str] = None,
+        image: str = "",
+        env: Optional[List[V1EnvVar]] = None,
+        command: Optional[str] = None,
+        volumes: Optional[List[V1VolumePath]] = None,
+        accelerators: Optional[List[str]] = None,
+        resources: Optional[V1ContainerResources] = None,
+        meters: Optional[List[V1Meter]] = None,
+        restart: str = DEFAULT_RESTART_POLICY,
+        queue: Optional[str] = None,
+        timeout: Optional[str] = None,
+        ssh_keys: Optional[List[V1SSHKey]] = None,
+        ports: Optional[List[V1PortRequest]] = None,
+        proxy_port: Optional[int] = None,
+        authz: Optional[V1AuthzConfig] = None,
+        config: Optional[GlobalConfig] = None,
+    ):
+        # Fallback to a default config if none is provided
+        config = config or GlobalConfig.read()
+        current_server = config.get_current_server_config()
+        if not current_server:
+            raise ValueError("No current server config found")
+        self.api_key = current_server.api_key
+        self.nebu_host = current_server.server
+
+        # print(f"nebu_host: {self.nebu_host}")
+        # print(f"api_key: {self.api_key}")
+
+        # Construct the containers base URL
+        self.containers_url = f"{self.nebu_host}/v1/containers"
+
+        # Attempt to find an existing container
+        response = requests.get(
+            self.containers_url,
+            headers={"Authorization": f"Bearer {self.api_key}"},
+        )
+        response.raise_for_status()
+
+        meta_request = V1ResourceMetaRequest(
+            name=name,
+            namespace=namespace,
+        )
+
+        containers = V1Containers.model_validate(response.json())
+        print(f"containers: {containers}")
+        existing = next(
+            (
+                c
+                for c in containers.containers
+                if c.metadata.name == name and c.metadata.namespace == namespace
+            ),
+            None,
+        )
+
+        print(f"existing: {existing}")
+
+        if not existing:
+            # If there's no existing container, create one:
+            if not image:
+                raise ValueError("An 'image' is required to create a new container.")
+
+            create_request = V1ContainerRequest(
+                kind="Container",
+                platform=platform,
+                metadata=meta_request,
+                image=image,
+                env=env,
+                command=command,
+                volumes=volumes,
+                accelerators=accelerators,
+                resources=resources,
+                meters=meters,
+                restart=restart,
+                queue=queue,
+                timeout=timeout,
+                ssh_keys=ssh_keys,
+                ports=ports,
+                proxy_port=proxy_port,
+                authz=authz,
+            )
+            create_response = requests.post(
+                self.containers_url,
+                json=create_request.model_dump(),
+                headers={"Authorization": f"Bearer {self.api_key}"},
+            )
+            create_response.raise_for_status()
+            self.container = V1Container.model_validate(create_response.json())
+            print(f"Created container {self.container.metadata.name}")
+        else:
+            # If container is found, check if anything has changed
+            # Gather the updated fields from the function arguments
+            updated_image = image or existing.image
+            updated_env = env if env is not None else existing.env
+            updated_command = command if command is not None else existing.command
+            updated_volumes = volumes if volumes is not None else existing.volumes
+            updated_accelerators = (
+                accelerators if accelerators is not None else existing.accelerators
+            )
+            updated_resources = (
+                resources if resources is not None else existing.resources
+            )
+            updated_meters = meters if meters is not None else existing.meters
+            updated_restart = restart if restart else existing.restart
+            updated_queue = queue if queue else existing.queue
+            updated_timeout = timeout if timeout else existing.timeout
+            updated_proxy_port = proxy_port if proxy_port else existing.proxy_port
+            updated_authz = authz if authz else existing.authz
+
+            # Determine if fields differ. You can adapt these checks as needed
+            # (for example, deep comparison for complex field structures).
+            fields_changed = (
+                existing.image != updated_image
+                or existing.env != updated_env
+                or existing.command != updated_command
+                or existing.volumes != updated_volumes
+                or existing.accelerators != updated_accelerators
+                or existing.resources != updated_resources
+                or existing.meters != updated_meters
+                or existing.restart != updated_restart
+                or existing.queue != updated_queue
+                or existing.timeout != updated_timeout
+                or existing.proxy_port != updated_proxy_port
+                or existing.authz != updated_authz
+            )
+
+            if not fields_changed:
+                # Nothing changed—do nothing
+                print(f"No changes detected for container {existing.metadata.name}.")
+                self.container = existing
+                return
+
+            print(
+                f"Detected changes for container {existing.metadata.name}, deleting and recreating."
+            )
+
+            # Construct the URL to delete the existing container
+            delete_url = (
+                f"{self.containers_url}/{existing.metadata.namespace}/{existing.metadata.name}"
+                if existing.metadata.namespace
+                else f"{self.containers_url}/{existing.metadata.name}"
+            )
+
+            # Delete the existing container
+            delete_response = requests.delete(
+                delete_url,
+                headers={"Authorization": f"Bearer {self.api_key}"},
+            )
+            delete_response.raise_for_status()
+            print(f"Deleted container {existing.metadata.name}")
+
+            # Now recreate the container using the updated parameters
+            create_request = V1ContainerRequest(
+                kind="Container",
+                platform=platform,
+                metadata=meta_request,
+                image=updated_image,
+                env=updated_env,
+                command=updated_command,
+                volumes=updated_volumes,
+                accelerators=updated_accelerators,
+                resources=updated_resources,
+                meters=updated_meters,
+                restart=updated_restart,
+                queue=updated_queue,
+                timeout=updated_timeout,
+                ssh_keys=ssh_keys,
+                ports=ports,
+                proxy_port=updated_proxy_port,
+                authz=updated_authz,
+            )
+            create_response = requests.post(
+                self.containers_url,
+                json=create_request.model_dump(),
+                headers={"Authorization": f"Bearer {self.api_key}"},
+            )
+            create_response.raise_for_status()
+            self.container = V1Container.model_validate(create_response.json())
+            print(f"Recreated container {self.container.metadata.name}")
+
+        # Save constructor params to `self` for reference, like you do in ReplayBuffer.
+        self.kind = "Container"
+        self.namespace = namespace
+        self.name = name
+        self.platform = platform
+        self.metadata = meta_request
+        self.image = image
+        self.env = env
+        self.command = command
+        self.volumes = volumes
+        self.accelerators = accelerators
+        self.resources = resources
+        self.meters = meters
+        self.restart = restart
+        self.queue = queue
+        self.timeout = timeout
+        self.ssh_keys = ssh_keys
+
+    @classmethod
+    def from_request(cls, request: V1ContainerRequest) -> V1Container:
+        return V1Container(**request.model_dump())
+
+    def delete(self) -> None:
+        """
+        Deletes the container by making a DELETE request to /v1/containers/:namespace/:name.
+        """
+        # Construct the url using instance attributes
+        delete_url = f"{self.containers_url}/{self.namespace}/{self.name}"
+
+        # Perform the deletion
+        response = requests.delete(
+            delete_url,
+            headers={"Authorization": f"Bearer {self.api_key}"},
+        )
+        response.raise_for_status()
+        print(f"Deleted container {self.name} in namespace {self.namespace}")

nebu-0.1.0/src/nebu/containers/models.py

@@ -0,0 +1,187 @@
+from enum import Enum
+from typing import Dict, List, Optional
+
+from pydantic import BaseModel, Field
+
+from nebu.meta import V1ResourceMeta, V1ResourceMetaRequest
+
+
+# 1) If you're still using V1ErrorResponse, no change needed here.
+class V1ErrorResponse(BaseModel):
+    response_type: str = Field(default="ErrorResponse", alias="type")
+    request_id: str
+    error: str
+    traceback: Optional[str] = None
+
+
+class V1Meter(BaseModel):
+    cost: Optional[float] = None
+    costp: Optional[float] = None
+    currency: str
+    unit: str
+    metric: str
+    json_path: Optional[str] = None
+
+
+class V1EnvVar(BaseModel):
+    key: str
+    value: Optional[str] = None
+    secret_name: Optional[str] = None
+
+
+class V1ContainerResources(BaseModel):
+    min_cpu: Optional[float] = None
+    min_memory: Optional[float] = None
+    max_cpu: Optional[float] = None
+    max_memory: Optional[float] = None
+
+
+class V1SSHKey(BaseModel):
+    public_key: Optional[str] = None
+    public_key_secret: Optional[str] = None
+    copy_local: Optional[bool] = None
+
+
+DEFAULT_RESTART_POLICY = "Never"
+
+
+class V1VolumeDriver(str, Enum):
+    RCLONE_SYNC = "RCLONE_SYNC"
+    RCLONE_BISYNC = "RCLONE_BISYNC"
+    RCLONE_MOUNT = "RCLONE_MOUNT"
+    RCLONE_COPY = "RCLONE_COPY"
+
+
+class V1VolumePath(BaseModel):
+    source: str
+    dest: str
+    resync: bool = False
+    continuous: bool = False
+    driver: V1VolumeDriver = V1VolumeDriver.RCLONE_SYNC
+
+
+class V1VolumeConfig(BaseModel):
+    paths: List[V1VolumePath]
+    cache_dir: str = "/nebu/cache"
+
+
+class V1ContainerStatus(BaseModel):
+    status: Optional[str] = None
+    message: Optional[str] = None
+    accelerator: Optional[str] = None
+    public_ip: Optional[str] = None
+    cost_per_hr: Optional[float] = None
+
+
+class V1AuthzSecretRef(BaseModel):
+    name: Optional[str] = None
+    key: Optional[str] = None
+
+
+class V1AuthzJwt(BaseModel):
+    secret_ref: Optional[V1AuthzSecretRef] = None
+
+
+class V1AuthzPathMatch(BaseModel):
+    path: Optional[str] = None
+    pattern: Optional[str] = None
+
+
+class V1AuthzFieldMatch(BaseModel):
+    json_path: Optional[str] = None
+    pattern: Optional[str] = None
+
+
+class V1AuthzRuleMatch(BaseModel):
+    roles: Optional[List[str]] = None
+
+
+class V1AuthzRule(BaseModel):
+    name: str
+    rule_match: Optional[V1AuthzRuleMatch] = Field(default=None, alias="match")
+    allow: bool
+    field_match: Optional[List[V1AuthzFieldMatch]] = None
+    path_match: Optional[List[V1AuthzPathMatch]] = None
+
+
+class V1AuthzConfig(BaseModel):
+    enabled: bool = False
+    default_action: str = "deny"
+    auth_type: str = "jwt"
+    jwt: Optional[V1AuthzJwt] = None
+    rules: Optional[List[V1AuthzRule]] = None
+
+
+class V1PortRequest(BaseModel):
+    port: int
+    protocol: Optional[str] = None
+    public: Optional[bool] = None
+
+
+class V1Port(BaseModel):
+    port: int
+    protocol: Optional[str] = None
+    public_ip: Optional[str] = None
+
+
+class V1ContainerRequest(BaseModel):
+    kind: str = Field(default="Container")
+    platform: Optional[str] = None
+    metadata: Optional[V1ResourceMetaRequest] = None
+    image: str
+    env: Optional[List[V1EnvVar]] = None
+    command: Optional[str] = None
+    volumes: Optional[List[V1VolumePath]] = None
+    accelerators: Optional[List[str]] = None
+    resources: Optional[V1ContainerResources] = None
+    meters: Optional[List[V1Meter]] = None
+    restart: str = Field(default=DEFAULT_RESTART_POLICY)
+    queue: Optional[str] = None
+    timeout: Optional[str] = None
+    ssh_keys: Optional[List[V1SSHKey]] = None
+    ports: Optional[List[V1PortRequest]] = None
+    proxy_port: Optional[int] = None
+    authz: Optional[V1AuthzConfig] = None
+
+
+class V1Container(BaseModel):
+    kind: str = Field(default="Container")
+    platform: str
+    metadata: V1ResourceMeta
+    image: str
+    env: Optional[List[V1EnvVar]] = None
+    command: Optional[str] = None
+    volumes: Optional[List[V1VolumePath]] = None
+    accelerators: Optional[List[str]] = None
+    meters: Optional[List[V1Meter]] = None
+    restart: str = Field(default=DEFAULT_RESTART_POLICY)
+    queue: Optional[str] = None
+    timeout: Optional[str] = None
+    resources: Optional[V1ContainerResources] = None
+    status: Optional[V1ContainerStatus] = None
+    ssh_keys: Optional[List[V1SSHKey]] = None
+    ports: Optional[List[V1Port]] = None
+    proxy_port: Optional[int] = None
+    authz: Optional[V1AuthzConfig] = None
+
+
+class V1UpdateContainer(BaseModel):
+    image: Optional[str] = None
+    env: Optional[List[V1EnvVar]] = None
+    command: Optional[str] = None
+    volumes: Optional[List[V1VolumePath]] = None
+    accelerators: Optional[List[str]] = None
+    labels: Optional[Dict[str, str]] = None
+    cpu_request: Optional[str] = None
+    memory_request: Optional[str] = None
+    platform: Optional[str] = None
+    meters: Optional[List[V1Meter]] = None
+    restart: Optional[str] = None
+    queue: Optional[str] = None
+    timeout: Optional[str] = None
+    resources: Optional[V1ContainerResources] = None
+    proxy_port: Optional[int] = None
+
+
+class V1Containers(BaseModel):
+    containers: List[V1Container]

nebu-0.1.0/src/nebu/meta.py

@@ -0,0 +1,24 @@
+from typing import Dict, Optional
+
+from pydantic import BaseModel
+
+
+# Match Rust "V1ResourceMetaRequest" struct
+class V1ResourceMetaRequest(BaseModel):
+    name: Optional[str] = None
+    namespace: Optional[str] = None
+    labels: Optional[Dict[str, str]] = None
+    owner: Optional[str] = None
+    owner_ref: Optional[str] = None
+
+
+class V1ResourceMeta(BaseModel):
+    name: str
+    namespace: str
+    id: str
+    owner: str
+    created_at: int
+    updated_at: int
+    created_by: str
+    owner_ref: Optional[str] = None
+    labels: Optional[Dict[str, str]] = None

nebu-0.1.0/src/nebu/processors/models.py

@@ -0,0 +1,67 @@
+from typing import Any, Optional
+
+from pydantic import BaseModel, Field
+
+from nebu.containers.models import V1Container
+from nebu.meta import V1ResourceMeta, V1ResourceMetaRequest
+
+# If these are in another module, import them as:
+# from .containers import V1Container, V1ResourceMeta, V1ResourceMetaRequest
+# For demonstration, simply assume they're available in scope:
+# class V1Container(BaseModel): ...
+# class V1ResourceMeta(BaseModel): ...
+# class V1ResourceMetaRequest(BaseModel): ...
+
+
+class V1ProcessorStatus(BaseModel):
+    status: Optional[str] = None
+    message: Optional[str] = None
+    pressure: Optional[int] = None
+
+
+class V1ScaleUp(BaseModel):
+    above_pressure: Optional[int] = None
+    duration: Optional[str] = None
+
+
+class V1ScaleDown(BaseModel):
+    below_pressure: Optional[int] = None
+    duration: Optional[str] = None
+
+
+class V1ScaleZero(BaseModel):
+    duration: Optional[str] = None
+
+
+class V1Scale(BaseModel):
+    up: Optional[V1ScaleUp] = None
+    down: Optional[V1ScaleDown] = None
+    zero: Optional[V1ScaleZero] = None
+
+
+DEFAULT_PROCESSOR_KIND = "Processor"
+
+
+class V1Processor(BaseModel):
+    kind: str = Field(default=DEFAULT_PROCESSOR_KIND)
+    metadata: V1ResourceMeta
+    container: Optional["V1Container"] = None
+    stream: Optional[str] = None
+    schema_: Optional[Any] = None  # Or Dict[str, Any], if you know the schema format
+    common_schema: Optional[str] = None
+    min_replicas: Optional[int] = None
+    max_replicas: Optional[int] = None
+    scale: Optional[V1Scale] = None
+    status: Optional[V1ProcessorStatus] = None
+
+
+class V1ProcessorRequest(BaseModel):
+    kind: str = Field(default=DEFAULT_PROCESSOR_KIND)
+    metadata: V1ResourceMetaRequest
+    container: Optional["V1Container"] = None
+    stream: Optional[str] = None
+    schema_: Optional[Any] = None
+    common_schema: Optional[str] = None
+    min_replicas: Optional[int] = None
+    max_replicas: Optional[int] = None
+    scale: Optional[V1Scale] = None

nebu-0.1.0/src/nebu/redis/models.py

@@ -0,0 +1,45 @@
+from typing import Optional
+
+from openai.types.chat.completion_create_params import CompletionCreateParamsBase
+from openai.types.responses import Response
+from pydantic import BaseModel
+
+
+class V1StreamMessage(BaseModel):
+    kind: str = "V1StreamMessage"
+    id: str
+    content: dict  # type: ignore
+    created_at: int
+    return_stream: Optional[str] = None
+    user_id: Optional[str] = None
+    organizations: Optional[dict] = None  # type: ignore
+    handle: Optional[str] = None
+    adapter: Optional[str] = None
+
+
+class V1StreamResponseMessage(BaseModel):
+    kind: str = "V1StreamResponseMessage"
+    id: str
+    content: dict  # type: ignore
+    created_at: int
+    user_id: Optional[str] = None
+
+
+class V1OpenAIStreamMessage(BaseModel):
+    kind: str = "V1OpenAIStreamMessage"
+    id: str
+    content: CompletionCreateParamsBase
+    created_at: int
+    return_stream: Optional[str] = None
+    user_id: Optional[str] = None
+    organizations: Optional[dict] = None  # type: ignore
+    handle: Optional[str] = None
+    adapter: Optional[str] = None
+
+
+class V1OpenAIStreamResponse(BaseModel):
+    kind: str = "V1OpenAIStreamResponse"
+    id: str
+    content: Response
+    created_at: int
+    user_id: Optional[str] = None

nebu-0.1.0/src/nebu.egg-info/PKG-INFO

@@ -0,0 +1,58 @@
+Metadata-Version: 2.4
+Name: nebu
+Version: 0.1.0
+Summary: A globally distributed container runtime
+Requires-Python: >=3.10.14
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: openai>=1.68.2
+Requires-Dist: pydantic>=2.10.6
+Requires-Dist: pyyaml>=6.0.2
+Requires-Dist: requests>=2.32.3
+Dynamic: license-file
+
+# nebulous-py
+A python library for the [Nebulous runtime](https://github.com/agentsea/nebulous)
+
+## Installation
+
+```bash
+pip install nebu
+```
+
+## Usage
+
+```python
+from nebu import Container, V1EnvVar, V1ResourceMeta
+
+container = Container(
+    metadata=V1ResourceMeta(
+        name="pytorch-example",
+        namespace="test",
+    ),
+    image="pytorch/pytorch:latest",
+    platform="runpod",
+    env=[V1EnvVar(name="MY_ENV_VAR", value="my-value")],
+    command="nvidia-smi",
+    accelerators=["1:A100_SXM"],
+    proxy_port=8080,
+)
+
+while container.status.status.lower() != "running":
+    print(f"Container '{container.metadata.name}' is not running, it is '{container.status.status}', waiting...")
+    time.sleep(1)
+
+print(f"Container '{container.metadata.name}' is running")
+
+print(f"You can access the container at {container.status.tailnet_url}")
+```
+
+## Contributing
+
+Please open an issue or a PR to contribute to the project.
+
+## Development
+
+```bash
+make test
+```

nebu-0.1.0/src/nebu.egg-info/SOURCES.txt

@@ -0,0 +1,18 @@
+LICENSE
+README.md
+pyproject.toml
+src/nebu/__init__.py
+src/nebu/config.py
+src/nebu/meta.py
+src/nebu.egg-info/PKG-INFO
+src/nebu.egg-info/SOURCES.txt
+src/nebu.egg-info/dependency_links.txt
+src/nebu.egg-info/requires.txt
+src/nebu.egg-info/top_level.txt
+src/nebu/containers/container.py
+src/nebu/containers/models.py
+src/nebu/processors/models.py
+src/nebu/processors/processor.py
+src/nebu/redis/models.py
+src/nebu/services/service.py
+tests/test_containers.py

nebu-0.1.0/src/nebu.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

nebu-0.1.0/src/nebu.egg-info/requires.txt

@@ -0,0 +1,4 @@
+openai>=1.68.2
+pydantic>=2.10.6
+pyyaml>=6.0.2
+requests>=2.32.3

nebu-0.1.0/src/nebu.egg-info/top_level.txt

@@ -0,0 +1 @@
+nebu

nebu-0.1.0/tests/test_containers.py

@@ -0,0 +1,84 @@
+import random
+import string
+
+import pytest
+
+from nebu.config import GlobalConfig
+from nebu.containers.container import Container
+
+
+@pytest.fixture
+def random_container_name() -> str:
+    """Generate a random container name to avoid collisions."""
+    suffix = "".join(random.choices(string.ascii_lowercase + string.digits, k=6))
+    return f"test-container-{suffix}"
+
+
+@pytest.fixture
+def global_config() -> GlobalConfig:
+    """
+    Set your real API key and server here or ensure GlobalConfig.read()
+    is retrieving valid values for these.
+    """
+    return GlobalConfig.read()
+
+
+def test_container_integration(global_config: GlobalConfig, random_container_name: str):
+    """
+    Single end-to-end test that:
+      1) Creates a container.
+      2) Creates it again with no changes (should detect no differences).
+      3) Updates a field (should trigger a recreate).
+      4) Deletes the container.
+    """
+
+    # 1) Create container
+    container_instance = Container(
+        name=random_container_name,
+        image="pytorch/pytorch:latest",
+        command="nvidia-smi",
+        platform="runpod",
+        accelerators=["1:L4"],
+        config=global_config,
+        restart="Never",
+    )
+    assert container_instance.container is not None, "Container object should be set."
+    assert container_instance.container.metadata.name == random_container_name
+    print(f"Created container {random_container_name}")
+
+    # 2) Attempt to create again with no changes
+    same_params_instance = Container(
+        name=random_container_name,
+        image="pytorch/pytorch:latest",
+        command="nvidia-smi",
+        platform="runpod",
+        accelerators=["1:L4"],
+        config=global_config,
+        restart="Never",
+    )
+    assert (
+        same_params_instance.container is not None
+    ), "Container object should remain set for no-change scenario."
+    # If your code prints something like "No changes detected..." this is good.
+    print(f"No changes detected for container {random_container_name}")
+
+    # 3) Update the image to trigger a recreate
+    updated_container_instance = Container(
+        name=random_container_name,
+        image="pytorch/pytorch:latest",
+        command="ls -la /",
+        platform="runpod",
+        accelerators=["1:L4"],
+        config=global_config,
+        restart="Never",
+    )
+    assert (
+        updated_container_instance.container is not None
+    ), "Updated container should be set."
+    assert (
+        updated_container_instance.container.command == "ls -la /"
+    ), "Container image should have been updated."
+    print(f"Updated container image to python:3.10-slim for {random_container_name}")
+
+    # 4) Delete the container
+    updated_container_instance.delete()  # Hypothetical method you plan to add