nebu 0.1.27__py3-none-any.whl → 0.1.30__py3-none-any.whl

nebu/processors/consumer.py

@@ -342,7 +342,9 @@ def process_message(message_id: str, message_data: Dict[str, str]) -> None:
             if content_type_name and content_type_name in local_namespace:
                 # Try to create the content type model first
                 try:
-                    content_model = local_namespace[content_type_name](**content)
+                    content_model = local_namespace[content_type_name].model_validate(
+                        content
+                    )
                     print(f"Content model: {content_model}")
                     input_obj = local_namespace["Message"](
                         kind=kind,
@@ -371,7 +373,7 @@ def process_message(message_id: str, message_data: Dict[str, str]) -> None:
                     )
             else:
                 # Just use the raw content
-                print(f"Using raw content")
+                print("Using raw content")
                 input_obj = local_namespace["Message"](
                     kind=kind,
                     id=msg_id,

nebu/processors/decorate.py

@@ -798,7 +798,6 @@ def processor(
 
         processor_instance = Processor(
             name=processor_name,
-            stream=processor_name,
             namespace=namespace,
             labels=labels,
             container=container_request,
@@ -812,6 +811,7 @@ def processor(
         print(
             f"[DEBUG Decorator] Processor instance '{processor_name}' created successfully."
         )
+        processor_instance.original_func = func
         return processor_instance
 
     return decorator

nebu/processors/processor.py

@@ -12,6 +12,7 @@ from nebu.processors.models import (
     V1Processors,
     V1ProcessorScaleRequest,
     V1Scale,
+    V1StreamData,
     V1UpdateProcessor,
 )
 
@@ -24,7 +25,6 @@ class Processor:
     def __init__(
         self,
         name: str,
-        stream: str,
         namespace: Optional[str] = None,
         labels: Optional[Dict[str, str]] = None,
         container: Optional[V1ContainerRequest] = None,
@@ -48,7 +48,6 @@ class Processor:
         self.name = name
         self.namespace = namespace
         self.labels = labels
-        self.stream = stream
         self.container = container
         self.schema_ = schema_
         self.common_schema = common_schema
@@ -123,7 +122,6 @@ class Processor:
             )
 
             update_processor = V1UpdateProcessor(
-                stream=stream,
                 container=container,
                 schema_=schema_,
                 common_schema=common_schema,
@@ -143,18 +141,23 @@ class Processor:
             patch_response.raise_for_status()
             print(f"Updated Processor {self.processor.metadata.name}")
 
-    def send(self, data: Dict[str, Any]) -> Dict[str, Any]:
+    def send(self, data: Dict[str, Any], wait: bool = False) -> Dict[str, Any]:
         """
         Send data to the processor.
         """
         if not self.processor or not self.processor.metadata.name:
             raise ValueError("Processor not found")
 
-        url = f"{self.processors_url}/{self.processor.metadata.namespace}/{self.processor.metadata.name}/send"
+        url = f"{self.processors_url}/{self.processor.metadata.namespace}/{self.processor.metadata.name}/messages"
 
-        response = requests.get(
+        stream_data = V1StreamData(
+            content=data,
+            wait=wait,
+        )
+
+        response = requests.post(
             url,
-            params=data,
+            json=stream_data.model_dump(exclude_none=True),
             headers={"Authorization": f"Bearer {self.api_key}"},
         )
         response.raise_for_status()
@@ -208,7 +211,6 @@ class Processor:
         out.namespace = namespace
 
         # Set specific fields from the processor
-        out.stream = processor_v1.stream
         out.container = processor_v1.container
         out.schema_ = processor_v1.schema_
         out.common_schema = processor_v1.common_schema

nebu/processors/remote.py

@@ -0,0 +1,47 @@
+from abc import ABC, abstractmethod
+from typing import Any, Dict, Optional, Type, TypeVar
+
+from pydantic import BaseModel
+
+from nebu.config import GlobalConfig
+from nebu.processors.models import V1ContainerRequest, V1Scale
+
+from .models import Message
+from .processor import Processor
+
+I = TypeVar("I", bound=BaseModel)
+O = TypeVar("O", bound=BaseModel)
+
+
+class RemoteProcessor(ABC, Processor):
+    def __init__(
+        self,
+        name: str,
+        namespace: Optional[str] = None,
+        labels: Optional[Dict[str, str]] = None,
+        container: Optional[V1ContainerRequest] = None,
+        schema_: Optional[Any] = None,
+        common_schema: Optional[str] = None,
+        min_replicas: Optional[int] = None,
+        max_replicas: Optional[int] = None,
+        scale_config: Optional[V1Scale] = None,
+        config: Optional[GlobalConfig] = None,
+        no_delete: bool = False,
+    ):
+        super().__init__(
+            name,
+            namespace,
+            labels,
+            container,
+            schema_,
+            common_schema,
+            min_replicas,
+            max_replicas,
+            scale_config,
+            config,
+            no_delete,
+        )
+
+    @abstractmethod
+    def process(self, message: Message[I]) -> Type[BaseModel]:
+        pass

{nebu-0.1.27.dist-info → nebu-0.1.30.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nebu
-Version: 0.1.27
+Version: 0.1.30
 Summary: A globally distributed container runtime
 Requires-Python: >=3.10.14
 Description-Content-Type: text/markdown
@@ -10,6 +10,7 @@ Requires-Dist: dill>=0.3.8
 Requires-Dist: openai>=1.68.2
 Requires-Dist: pillow>=11.1.0
 Requires-Dist: pydantic>=2.10.6
+Requires-Dist: pydantic-ai-slim[openai]>=0.0.55
 Requires-Dist: pysocks>=1.7.1
 Requires-Dist: pyyaml>=6.0.2
 Requires-Dist: redis[socks]>=5.0

nebu-0.1.30.dist-info/RECORD

@@ -0,0 +1,26 @@
+nebu/__init__.py,sha256=5sepbzdAdoA_8TIxws60S4ugFY1apQd_savzn20a4cY,465
+nebu/adapter.py,sha256=yWQCpAn2lJxXPSyQacmFpfVzAL3RrKtKgrSMws0BoT8,211
+nebu/auth.py,sha256=N_v6SPFD9HU_UoRDTaouH03g2Hmo9C-xxqInE1FweXE,1471
+nebu/cache.py,sha256=1aY1plIXWOPmUY6GGq_s_QDXzIi5UMuG34XYBA8PpW8,3803
+nebu/config.py,sha256=aZzQltkobtOLHFCGcIkpKoE3ITn3Z11Dp0E72w84TA0,5769
+nebu/data.py,sha256=kIH9-JJ1-iO7P2t28bku6Gn0Y5tgQszGeTW_rpmO03A,38725
+nebu/meta.py,sha256=CzFHMND9seuewzq9zNNx9WTr6JvrCBExe7BLqDSr7lM,745
+nebu/chatx/convert.py,sha256=4rjccr9bI0xmCAumKlxyFpVzW0up9Yj-gJYvDjH6C54,8798
+nebu/chatx/openai.py,sha256=LLSPvVGnauUViAXY7OifwoWlkUGZWfgxEjxR4mjSqZg,44961
+nebu/containers/container.py,sha256=yb7KaPTVXnEEAlrpdlUi4HNqF6P7z9bmwAILGlq6iqU,13502
+nebu/containers/decorator.py,sha256=uFtzlAXRHYZECJ-NPusY7oN9GXvdHrHDd_JNrIGr8aQ,3244
+nebu/containers/models.py,sha256=0j6NGy4yto-enRDh_4JH_ZTbHrLdSpuMOqNQPnIrwC4,6815
+nebu/containers/server.py,sha256=yFa2Y9PzBn59E1HftKiv0iapPonli2rbGAiU6r-wwe0,2513
+nebu/processors/consumer.py,sha256=becQxjQoerlaDdr2_wxzIux3wl8SVmo2LA93RZn0Jk8,19694
+nebu/processors/decorate.py,sha256=9mf25RawOX6_6WyQcRLBIHQC3oCDtofQZijJ13aQM9E,34576
+nebu/processors/default.py,sha256=W4slJenG59rvyTlJ7gRp58eFfXcNOTT2Hfi6zzJAobI,365
+nebu/processors/models.py,sha256=y40HoW-MEzDWB2dm_tsYlUy3Nf3s6eiLC0iGO9BoNog,3956
+nebu/processors/processor.py,sha256=EQ3-dBf432fSAQE2A_9ATX-cG5LkJ4fjVaOtlxCoXvc,9719
+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.30.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+nebu-0.1.30.dist-info/METADATA,sha256=Y_o6hvOdihVgFJMRQPIpALA2V6JfaXybtrlSaQyMNtI,1786
+nebu-0.1.30.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+nebu-0.1.30.dist-info/top_level.txt,sha256=uLIbEKJeGSHWOAJN5S0i5XBGwybALlF9bYoB1UhdEgQ,5
+nebu-0.1.30.dist-info/RECORD,,

nebu-0.1.27.dist-info/RECORD

@@ -1,22 +0,0 @@
-nebu/__init__.py,sha256=fdDvhSm-KU8I6PtL00ePEv2D3lP0NZ-O-MhtG_Sc5SM,359
-nebu/auth.py,sha256=rApCd-7_c3GpIb7gjCB79rR7SOcmkG7MmaTE6zMbvr0,1125
-nebu/config.py,sha256=aZzQltkobtOLHFCGcIkpKoE3ITn3Z11Dp0E72w84TA0,5769
-nebu/convert.py,sha256=x85CE1X4rABgAAx9rSEo9TVBq-i1-nm8QU_dYTijQxc,5512
-nebu/data.py,sha256=CzxEvESDdSApvuEO7NJUmW4yJ1vXtQQCXTjYDoOlcqk,40846
-nebu/meta.py,sha256=CzFHMND9seuewzq9zNNx9WTr6JvrCBExe7BLqDSr7lM,745
-nebu/containers/container.py,sha256=yb7KaPTVXnEEAlrpdlUi4HNqF6P7z9bmwAILGlq6iqU,13502
-nebu/containers/decorator.py,sha256=uFtzlAXRHYZECJ-NPusY7oN9GXvdHrHDd_JNrIGr8aQ,3244
-nebu/containers/models.py,sha256=0j6NGy4yto-enRDh_4JH_ZTbHrLdSpuMOqNQPnIrwC4,6815
-nebu/containers/server.py,sha256=yFa2Y9PzBn59E1HftKiv0iapPonli2rbGAiU6r-wwe0,2513
-nebu/processors/consumer.py,sha256=gbPyJE-odgmd3UbE2Y1tZ2x8yqFaaBtNzUKZh5dpKBA,19636
-nebu/processors/decorate.py,sha256=B2qK1B2Jm1SzO5IN9pWk32sS0z5WLqXPdJtP-y-iSO4,34563
-nebu/processors/default.py,sha256=W4slJenG59rvyTlJ7gRp58eFfXcNOTT2Hfi6zzJAobI,365
-nebu/processors/models.py,sha256=y40HoW-MEzDWB2dm_tsYlUy3Nf3s6eiLC0iGO9BoNog,3956
-nebu/processors/processor.py,sha256=oy2YdI-cy6qQWxrZhpZahJV46oWZlu_Im-jm811R_oo,9667
-nebu/redis/models.py,sha256=coPovAcVXnOU1Xh_fpJL4PO3QctgK9nBe5QYoqEcnxg,1230
-nebu/services/service.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nebu-0.1.27.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-nebu-0.1.27.dist-info/METADATA,sha256=V01d1IJl34syXgiaMGULDJ_IVydAdDzjVd_c8PXVIYU,1738
-nebu-0.1.27.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-nebu-0.1.27.dist-info/top_level.txt,sha256=uLIbEKJeGSHWOAJN5S0i5XBGwybALlF9bYoB1UhdEgQ,5
-nebu-0.1.27.dist-info/RECORD,,