nebu 0.1.114__py3-none-any.whl → 0.1.115__py3-none-any.whl

nebu/processors/decorate.py

@@ -1194,6 +1194,52 @@ def processor(
 
         # --- Final Setup ---
         logger.debug("Decorator: Preparing final Processor object...")
+
+        # Determine ResolvedInputType for Processor Generic
+        ResolvedInputType: type[BaseModel] = (
+            BaseModel  # Default to BaseModel to satisfy generic bound
+        )
+        if is_stream_message:
+            if (
+                content_type
+                and isinstance(content_type, type)
+                and issubclass(content_type, BaseModel)
+            ):
+                ResolvedInputType = content_type
+            else:
+                logger.warning(
+                    f"Decorator: Message type hint found, but ContentType '{content_type!s}' is not a valid Pydantic Model. Defaulting InputType to BaseModel."
+                )
+                # ResolvedInputType remains BaseModel (default)
+        elif (
+            param_type
+            and isinstance(param_type, type)
+            and issubclass(param_type, BaseModel)
+        ):
+            ResolvedInputType = param_type  # Function takes the data model directly
+        else:
+            logger.warning(
+                f"Decorator: Parameter type '{param_type!s}' is not a valid Pydantic Model. Defaulting InputType to BaseModel."
+            )
+            # ResolvedInputType remains BaseModel (default)
+
+        ResolvedOutputType: type[BaseModel] = BaseModel  # Default to BaseModel
+        if (
+            return_type
+            and isinstance(return_type, type)
+            and issubclass(return_type, BaseModel)
+        ):
+            ResolvedOutputType = return_type
+        elif return_type is not None:  # It was something, but not a BaseModel subclass
+            logger.warning(
+                f"Decorator: Return type '{return_type!s}' is not a valid Pydantic Model. Defaulting OutputType to BaseModel."
+            )
+        # Else (return_type is None), ResolvedOutputType remains BaseModel
+
+        logger.debug(
+            f"Decorator: Resolved Generic Types for Processor: InputType={ResolvedInputType.__name__}, OutputType={ResolvedOutputType.__name__}"
+        )
+
         metadata = V1ResourceMetaRequest(
             name=processor_name, namespace=effective_namespace, labels=labels
         )
@@ -1225,7 +1271,7 @@ def processor(
         final_command = "\n".join(all_commands)
 
         logger.debug(
-            f"Decorator: Final container command:\n-------\n{final_command}\n-------"
+            f"Decorator: Final container command:\\n-------\\n{final_command}\\n-------"
         )
 
         container_request = V1ContainerRequest(
@@ -1259,21 +1305,28 @@ def processor(
             else:
                 logger.debug(f"  {env_var.key}: {value_str}")
 
-        processor_instance = Processor(
+        # Create the generically typed Processor instance
+        _processor_instance = Processor[ResolvedInputType, ResolvedOutputType](
             name=processor_name,
             namespace=effective_namespace,
             labels=labels,
             container=container_request,
-            schema_=None,  # Schema info might be derived differently now if needed
+            input_model_cls=ResolvedInputType,
+            output_model_cls=ResolvedOutputType,
             common_schema=None,
             min_replicas=min_replicas,
             max_replicas=max_replicas,
             scale_config=scale,
+            config=effective_config,
+            api_key=api_key,
             no_delete=no_delete,
             wait_for_healthy=wait_for_healthy,
         )
+        # Type hint for the variable. The instance itself IS correctly typed with specific models.
+        processor_instance: Processor[BaseModel, BaseModel] = _processor_instance
+
         logger.debug(
-            f"Decorator: Processor instance '{processor_name}' created successfully."
+            f"Decorator: Processor instance '{processor_name}' created successfully with generic types."
         )
         # Store original func for potential local invocation/testing? Keep for now.
         # TODO: Add original_func to Processor model definition if this is desired

nebu/processors/processor.py

@@ -2,7 +2,17 @@ import json
 import threading
 import time
 import uuid
-from typing import Any, Dict, Generic, List, Optional, TypeVar, cast, get_args
+from typing import (
+    Any,
+    Dict,
+    Generic,
+    List,
+    Optional,
+    TypeVar,
+    cast,
+    get_args,
+    get_origin,
+)
 
 import requests
 from pydantic import BaseModel
@@ -101,7 +111,8 @@ class Processor(Generic[InputType, OutputType]):
         namespace: Optional[str] = None,
         labels: Optional[Dict[str, str]] = None,
         container: Optional[V1ContainerRequest] = None,
-        schema_: Optional[Any] = None,
+        input_model_cls: Optional[type[BaseModel]] = None,
+        output_model_cls: Optional[type[BaseModel]] = None,
         common_schema: Optional[str] = None,
         min_replicas: Optional[int] = None,
         max_replicas: Optional[int] = None,
@@ -124,7 +135,8 @@ class Processor(Generic[InputType, OutputType]):
         self.namespace = namespace
         self.labels = labels
         self.container = container
-        self.schema_ = schema_
+        self.input_model_cls = input_model_cls
+        self.output_model_cls = output_model_cls
         self.common_schema = common_schema
         self.min_replicas = min_replicas
         self.max_replicas = max_replicas
@@ -132,37 +144,26 @@ class Processor(Generic[InputType, OutputType]):
         self.processors_url = f"{self.orign_host}/v1/processors"
         self._log_thread: Optional[threading.Thread] = None
 
-        # Attempt to infer OutputType if schema_ is not provided
-        print(">>> self.schema_: ", self.schema_)
-        print("self.__dict__: ", self.__dict__)
-        if self.schema_ is None and hasattr(self, "__orig_class__"):
+        # Infer OutputType Pydantic class if output_model_cls is not provided
+        if self.output_model_cls is None and hasattr(self, "__orig_class__"):
             type_args = get_args(self.__orig_class__)  # type: ignore
-            print(">>> type_args: ", type_args)
             if len(type_args) == 2:
                 output_type_candidate = type_args[1]
-                print(">>> output_type_candidate: ", output_type_candidate)
-                # Check if it looks like a Pydantic model class
                 if isinstance(output_type_candidate, type) and issubclass(
                     output_type_candidate, BaseModel
                 ):
-                    print(">>> output_type_candidate is a Pydantic model class")
                     logger.debug(
-                        f"Inferred OutputType {output_type_candidate.__name__} from generic arguments."
+                        f"Inferred output_model_cls {output_type_candidate.__name__} from generic arguments."
                     )
-                    self.schema_ = output_type_candidate
+                    self.output_model_cls = output_type_candidate
                 else:
-                    print(">>> output_type_candidate is not a Pydantic model class")
                     logger.debug(
                         f"Second generic argument {output_type_candidate} is not a Pydantic BaseModel. "
-                        "Cannot infer OutputType."
+                        "Cannot infer output_model_cls."
                     )
             else:
-                print(
-                    "Could not infer OutputType from generic arguments: wrong number of type args found "
-                    f"(expected 2, got {len(type_args) if type_args else 0})."
-                )
                 logger.debug(
-                    "Could not infer OutputType from generic arguments: wrong number of type args found "
+                    "Could not infer output_model_cls from generic arguments: wrong number of type args found "
                     f"(expected 2, got {len(type_args) if type_args else 0})."
                 )
 
@@ -201,7 +202,6 @@ class Processor(Generic[InputType, OutputType]):
             processor_request = V1ProcessorRequest(
                 metadata=metadata,
                 container=container,
-                schema_=schema_,
                 common_schema=common_schema,
                 min_replicas=min_replicas,
                 max_replicas=max_replicas,
@@ -226,7 +226,6 @@ class Processor(Generic[InputType, OutputType]):
 
             update_processor = V1UpdateProcessor(
                 container=container,
-                schema_=schema_,
                 common_schema=common_schema,
                 min_replicas=min_replicas,
                 max_replicas=max_replicas,
@@ -341,38 +340,37 @@ class Processor(Generic[InputType, OutputType]):
 
         # Attempt to parse into OutputType if conditions are met
         print(f">>> wait: {wait}")
-        print(f">>> self.schema_: {self.schema_}")
-        print(">>> type(self.schema_): ", type(self.schema_))
-        print(f">>> isinstance(self.schema_, type): {isinstance(self.schema_, type)}")
+        print(f">>> self.output_model_cls: {self.output_model_cls}")
+        print(">>> type(self.output_model_cls): ", type(self.output_model_cls))
+        print(
+            f">>> isinstance(self.output_model_cls, type): {isinstance(self.output_model_cls, type)}"
+        )
         print(f">>> isinstance(raw_content, dict): {isinstance(raw_content, dict)}")
         if (
             wait
-            and self.schema_
-            and isinstance(self.schema_, type)
-            and issubclass(self.schema_, BaseModel)  # type: ignore
+            and self.output_model_cls
+            and isinstance(self.output_model_cls, type)
+            and issubclass(self.output_model_cls, BaseModel)  # type: ignore
             and isinstance(raw_content, dict)
-        ):  # Check if raw_content is a dict
+        ):
             print(f">>> raw_content: {raw_content}")
             try:
-                # self.schema_ is assumed to be the Pydantic model class for OutputType
-                # Parse raw_content instead of the full response
-                parsed_model = self.schema_.model_validate(raw_content)
+                parsed_model = self.output_model_cls.model_validate(raw_content)
                 print(f">>> parsed_model: {parsed_model}")
-                # Cast to OutputType to satisfy the linter with generics
                 parsed_output: OutputType = cast(OutputType, parsed_model)
                 print(f">>> parsed_output: {parsed_output}")
                 return parsed_output
-            except (
-                Exception
-            ) as e:  # Consider pydantic.ValidationError for more specific handling
+            except Exception as e:
                 print(f">>> error: {e}")
-                schema_name = getattr(self.schema_, "__name__", str(self.schema_))
+                model_name = getattr(
+                    self.output_model_cls, "__name__", str(self.output_model_cls)
+                )
                 logger.error(
-                    f"Processor {processor_name}: Failed to parse 'content' field into output type {schema_name}. "
+                    f"Processor {processor_name}: Failed to parse 'content' field into output type {model_name}. "
                     f"Error: {e}. Returning raw JSON response."
                 )
-                # Fallback to returning the raw JSON response
                 return raw_content
+        # Fallback logic using self.schema_ has been removed.
 
         return raw_content
 
@@ -401,6 +399,8 @@ class Processor(Generic[InputType, OutputType]):
         namespace: Optional[str] = None,
         config: Optional[GlobalConfig] = None,
         api_key: Optional[str] = None,
+        input_model_cls: Optional[type[BaseModel]] = None,
+        output_model_cls: Optional[type[BaseModel]] = None,
     ):
         """
         Get a Processor from the remote server.
@@ -412,27 +412,60 @@ class Processor(Generic[InputType, OutputType]):
             raise ValueError("Processor not found")
         processor_v1 = processors[0]
 
-        out = cls.__new__(cls)
+        # Try to infer Input/Output model classes if Processor.load is called as generic
+        # e.g., MyProcessor = Processor[MyInput, MyOutput]; MyProcessor.load(...)
+        loaded_input_model_cls: Optional[type[BaseModel]] = None
+        loaded_output_model_cls: Optional[type[BaseModel]] = None
+
+        # __orig_bases__ usually contains the generic version of the class if it was parameterized.
+        # We look for Processor[...] in the bases.
+        if hasattr(cls, "__orig_bases__"):
+            for base in cls.__orig_bases__:  # type: ignore
+                if get_origin(base) is Processor:
+                    type_args = get_args(base)
+                    if len(type_args) == 2:
+                        input_arg, output_arg = type_args
+                        if isinstance(input_arg, type) and issubclass(
+                            input_arg, BaseModel
+                        ):
+                            loaded_input_model_cls = input_arg
+                        if isinstance(output_arg, type) and issubclass(
+                            output_arg, BaseModel
+                        ):
+                            loaded_output_model_cls = output_arg
+                    break  # Found Processor generic base
+
+        # Determine final model classes, prioritizing overrides
+        final_input_model_cls = (
+            input_model_cls if input_model_cls is not None else loaded_input_model_cls
+        )
+        final_output_model_cls = (
+            output_model_cls
+            if output_model_cls is not None
+            else loaded_output_model_cls
+        )
+
+        out = cls.__new__(cls)  # type: ignore
+        # If generic types were successfully inferred or overridden, pass them to init
+        # Otherwise, they will be None, and __init__ might try __orig_class__ if called on instance
+        out.__init__(  # type: ignore
+            name=processor_v1.metadata.name,  # Use name from fetched metadata
+            namespace=processor_v1.metadata.namespace,  # Use namespace from fetched metadata
+            labels=processor_v1.metadata.labels,  # Use labels from fetched metadata
+            container=processor_v1.container,
+            input_model_cls=final_input_model_cls,  # Use determined input model
+            output_model_cls=final_output_model_cls,  # Use determined output model
+            common_schema=processor_v1.common_schema,
+            min_replicas=processor_v1.min_replicas,
+            max_replicas=processor_v1.max_replicas,
+            scale_config=processor_v1.scale,
+            config=config,  # Pass original config
+            api_key=api_key,  # Pass original api_key
+        )
+        # The __init__ call above handles most setup. We store the fetched processor data.
         out.processor = processor_v1
-        out.config = config or GlobalConfig.read()
-        if not out.config:
-            raise ValueError("No config found")
-        out.current_server = out.config.get_current_server_config()
-        if not out.current_server:
-            raise ValueError("No server config found")
-        out.api_key = api_key or out.current_server.api_key
-        out.orign_host = out.current_server.server
-        out.processors_url = f"{out.orign_host}/v1/processors"
-        out.name = name
-        out.namespace = namespace
-
-        # Set specific fields from the processor
-        out.container = processor_v1.container
-        out.schema_ = processor_v1.schema_
-        out.common_schema = processor_v1.common_schema
-        out.min_replicas = processor_v1.min_replicas
-        out.max_replicas = processor_v1.max_replicas
-        out.scale_config = processor_v1.scale
+        # self.schema_ was removed, so no assignment for it here from processor_v1.schema_
+        # out.common_schema = processor_v1.common_schema # This is now set in __init__
 
         return out
 
@@ -549,7 +582,7 @@ class Processor(Generic[InputType, OutputType]):
 
                 # Send health check and wait for response
                 response = self.send(
-                    data=health_check_data,  # type: ignore
+                    data=health_check_data,  # type: ignore[arg-type]
                     wait=True,
                     timeout=30.0,  # Short timeout for individual health check
                 )

{nebu-0.1.114.dist-info → nebu-0.1.115.dist-info}/METADATA

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

{nebu-0.1.114.dist-info → nebu-0.1.115.dist-info}/RECORD

@@ -15,14 +15,14 @@ nebu/namespaces/models.py,sha256=EqUOpzhVBhvJw2P92ONDUbIgC31M9jMmcaG5vyOrsWg,497
 nebu/namespaces/namespace.py,sha256=oeZyGqsIGIrppyjif1ZONsdTmqRgd9oSLFE1BChXTTE,5247
 nebu/processors/consumer.py,sha256=9WapzBTPuXRunH-vjPerTlGZy__hn_d4m13l1ajebY8,62732
 nebu/processors/consumer_process_worker.py,sha256=h--eNFKaLbUayxn88mB8oGGdrU2liE1dnwm_TPlewX8,36960
-nebu/processors/decorate.py,sha256=aT6TT8QOOpl8HKo8kYdwmX6785_l9kSfoyUSPp4tkuY,58726
+nebu/processors/decorate.py,sha256=wnos4w7htBUSOnNVyhY7NQgUuz9d2_-Kre5H99UiFpw,61089
 nebu/processors/default.py,sha256=cy4ETMdbdRGkrvbYec1o60h7mGDlGN5JsuUph0ENtDU,364
 nebu/processors/models.py,sha256=g4B1t6Rgoy-NUEHBLeQc0EENzHXLDlWSio8Muv7cTDU,4093
-nebu/processors/processor.py,sha256=Gh-RQ2EBvO2Ab91CJOKmzHjZiWG6TKvftTS6_Ogyctk,23104
+nebu/processors/processor.py,sha256=fMXzodTe2qWnoCBry9nLsHLMEI6FxOOmcsR9DbuY_W0,24574
 nebu/redis/models.py,sha256=coPovAcVXnOU1Xh_fpJL4PO3QctgK9nBe5QYoqEcnxg,1230
 nebu/services/service.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nebu-0.1.114.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-nebu-0.1.114.dist-info/METADATA,sha256=6jG9Q52pjeBsLFyU5QwaZ7nSeL0bfGGYA9qaiIQvvTs,1798
-nebu-0.1.114.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-nebu-0.1.114.dist-info/top_level.txt,sha256=uLIbEKJeGSHWOAJN5S0i5XBGwybALlF9bYoB1UhdEgQ,5
-nebu-0.1.114.dist-info/RECORD,,
+nebu-0.1.115.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+nebu-0.1.115.dist-info/METADATA,sha256=tud697lMA-FHCANFaBZKT20hTCyIxRMMQT3ZNdc0tQk,1798
+nebu-0.1.115.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+nebu-0.1.115.dist-info/top_level.txt,sha256=uLIbEKJeGSHWOAJN5S0i5XBGwybALlF9bYoB1UhdEgQ,5
+nebu-0.1.115.dist-info/RECORD,,