nebu 0.1.53__py3-none-any.whl → 0.1.56__py3-none-any.whl

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/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/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.dist-info → nebu-0.1.56.dist-info}/METADATA

@@ -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.dist-info → nebu-0.1.56.dist-info}/RECORD

@@ -1,7 +1,7 @@
 nebu/__init__.py,sha256=gJ49VnAs4GC_bUV90j-SS-iiYWESuCf4uuE7mx4T78k,385
 nebu/auth.py,sha256=N_v6SPFD9HU_UoRDTaouH03g2Hmo9C-xxqInE1FweXE,1471
 nebu/cache.py,sha256=jmluqvWnE9N8uNq6nppXSxEJK7DKWaB79GicaGg9KmY,4718
-nebu/config.py,sha256=aZzQltkobtOLHFCGcIkpKoE3ITn3Z11Dp0E72w84TA0,5769
+nebu/config.py,sha256=ZHLVKASiMHesWdRPBJOEfVgdsdQuLS9nieaWt7tkAbQ,6006
 nebu/data.py,sha256=yAQ5EXmdwsQfaFC30DB8IrzKNS53NhUkPrwKyM5nbxA,39782
 nebu/meta.py,sha256=CzFHMND9seuewzq9zNNx9WTr6JvrCBExe7BLqDSr7lM,745
 nebu/builders/builder.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -11,15 +11,15 @@ nebu/containers/decorator.py,sha256=uFtzlAXRHYZECJ-NPusY7oN9GXvdHrHDd_JNrIGr8aQ,
 nebu/containers/models.py,sha256=0j6NGy4yto-enRDh_4JH_ZTbHrLdSpuMOqNQPnIrwC4,6815
 nebu/containers/server.py,sha256=yFa2Y9PzBn59E1HftKiv0iapPonli2rbGAiU6r-wwe0,2513
 nebu/processors/consumer.py,sha256=q4cFKPEfMf1xIs0Y3CjUDCR35QVjr0F_hSOZUKUYc_U,33667
-nebu/processors/decorate.py,sha256=XZqCkcLKYev3CO1wJUZXKV0Q8otQUdLTxogBSVUYKcg,53182
+nebu/processors/decorate.py,sha256=f4y0BL2W07XMMVzm0GPwh7GfJ-QRioSjBofJSa7hQtg,53175
 nebu/processors/default.py,sha256=W4slJenG59rvyTlJ7gRp58eFfXcNOTT2Hfi6zzJAobI,365
 nebu/processors/models.py,sha256=y40HoW-MEzDWB2dm_tsYlUy3Nf3s6eiLC0iGO9BoNog,3956
-nebu/processors/processor.py,sha256=cptZEN9ZGcaoFNreaw3BkwV0qKHvjP9b4nNxlQjFT3s,15405
+nebu/processors/processor.py,sha256=1iFR5JEOYrUpljSJtRzNKFpmUyZU-KWb_dJQDE9IEPs,15613
 nebu/processors/remote.py,sha256=TeAIPGEMqnDIb7H1iett26IEZrBlcbPB_-DSm6jcH1E,1285
 nebu/redis/models.py,sha256=coPovAcVXnOU1Xh_fpJL4PO3QctgK9nBe5QYoqEcnxg,1230
 nebu/services/service.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nebu-0.1.53.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-nebu-0.1.53.dist-info/METADATA,sha256=GKhvMkM4FEpn8Ln1wUwcuBgtBTTes2D9l0fNmdHiawY,1731
-nebu-0.1.53.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
-nebu-0.1.53.dist-info/top_level.txt,sha256=uLIbEKJeGSHWOAJN5S0i5XBGwybALlF9bYoB1UhdEgQ,5
-nebu-0.1.53.dist-info/RECORD,,
+nebu-0.1.56.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+nebu-0.1.56.dist-info/METADATA,sha256=k1ZNw6tIK0LLhliNmgcqNjRKz_BQZyuRkY6SfjIt5PY,1731
+nebu-0.1.56.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
+nebu-0.1.56.dist-info/top_level.txt,sha256=uLIbEKJeGSHWOAJN5S0i5XBGwybALlF9bYoB1UhdEgQ,5
+nebu-0.1.56.dist-info/RECORD,,