nebu 0.1.53__tar.gz → 0.1.56__tar.gz

{nebu-0.1.53/src/nebu.egg-info → nebu-0.1.56}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nebu
-Version: 0.1.53
+Version: 0.1.56
 Summary: A globally distributed container runtime
 Requires-Python: >=3.10.14
 Description-Content-Type: text/markdown

{nebu-0.1.53 → nebu-0.1.56}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "nebu"
-version = "0.1.53"
+version = "0.1.56"
 description = "A globally distributed container runtime"
 readme = "README.md"
 requires-python = ">=3.10.14"

{nebu-0.1.53 → nebu-0.1.56}/src/nebu/config.py

@@ -52,12 +52,17 @@ class GlobalConfig:
             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_api_key = (
+            os.environ.get("NEBU_API_KEY")
+            or os.environ.get("AGENTSEA_API_KEY")
+            or os.environ.get("ORIGN_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"
+        env_server = os.environ.get("NEBU_SERVER")
+        env_auth_server = (
+            os.environ.get("NEBU_AUTH_SERVER")
+            or os.environ.get("AGENTSEA_AUTH_SERVER")
+            or os.environ.get("ORIGN_AUTH_SERVER")
+            or os.environ.get("AGENTSEA_AUTH_URL")
         )
 
         # Only proceed if all three environment variables are present
@@ -158,7 +163,9 @@ class ContainerConfig:
         Load configuration from environment variables.
         """
         return cls(
-            api_key=os.environ.get("NEBU_API_KEY"),
+            api_key=os.environ.get("NEBU_API_KEY")
+            or os.environ.get("AGENTSEA_API_KEY")
+            or os.environ.get("ORIGN_API_KEY"),
             server=os.environ.get("NEBU_SERVER"),
             namespace=os.environ.get("NEBU_NAMESPACE"),
             name=os.environ.get("NEBU_NAME"),

{nebu-0.1.53 → nebu-0.1.56}/src/nebu/processors/decorate.py

@@ -396,13 +396,13 @@ def processor(
 ):
     def decorator(
         func: Callable[[Any], Any],
-    ) -> Processor | Callable[[Any], Any]:
+    ) -> Processor:
         # --- Prevent Recursion Guard ---
         if os.environ.get(_NEBU_INSIDE_CONSUMER_ENV_VAR) == "1":
             print(
                 f"[DEBUG Decorator] Guard triggered for '{func.__name__}'. Returning original function."
             )
-            return func
+            return func  # type: ignore
         # --- End Guard ---
 
         print(

{nebu-0.1.53 → nebu-0.1.56}/src/nebu/processors/processor.py

@@ -1,6 +1,6 @@
 import json
 import threading
-from typing import Any, Dict, List, Optional
+from typing import Any, Dict, Generic, List, Optional, TypeVar
 
 import requests
 from pydantic import BaseModel
@@ -19,6 +19,8 @@ from nebu.processors.models import (
     V1UpdateProcessor,
 )
 
+InputType = TypeVar("InputType", bound=BaseModel)
+
 
 def _fetch_and_print_logs(log_url: str, api_key: str, processor_name: str):
     """Helper function to fetch logs in a separate thread."""
@@ -78,7 +80,7 @@ def _fetch_and_print_logs(log_url: str, api_key: str, processor_name: str):
         )
 
 
-class Processor:
+class Processor(Generic[InputType]):
     """
     A class for managing Processor instances.
     """
@@ -203,14 +205,18 @@ class Processor:
             patch_response.raise_for_status()
             print(f"Updated Processor {self.processor.metadata.name}")
 
-    def __call__(self, data: BaseModel, wait: bool = False) -> Dict[str, Any]:
+    def __call__(self, data: InputType, wait: bool = False) -> Dict[str, Any]:
         """
         Allows the Processor instance to be called like a function, sending data.
         """
         return self.send(data=data, wait=wait)
 
     def send(
-        self, data: BaseModel, wait: bool = False, logs: bool = False
+        self,
+        data: InputType,
+        wait: bool = False,
+        logs: bool = False,
+        api_key: Optional[str] = None,
     ) -> Dict[str, Any]:
         """
         Send data to the processor and optionally stream logs in the background.
@@ -225,6 +231,9 @@ class Processor:
         processor_name = self.processor.metadata.name
         processor_namespace = self.processor.metadata.namespace
 
+        if not api_key:
+            api_key = self.api_key
+
         # --- Send Data ---
         messages_url = (
             f"{self.processors_url}/{processor_namespace}/{processor_name}/messages"
@@ -236,7 +245,7 @@ class Processor:
         response = requests.post(
             messages_url,
             json=stream_data.model_dump(mode="json", exclude_none=True),
-            headers={"Authorization": f"Bearer {self.api_key}"},
+            headers={"Authorization": f"Bearer {api_key}"},
         )
         response.raise_for_status()
         send_response_json = response.json()

{nebu-0.1.53 → nebu-0.1.56/src/nebu.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nebu
-Version: 0.1.53
+Version: 0.1.56
 Summary: A globally distributed container runtime
 Requires-Python: >=3.10.14
 Description-Content-Type: text/markdown