nebu 0.1.10__py3-none-any.whl → 0.1.12__py3-none-any.whl

nebu/processors/decorate.py

@@ -167,14 +167,56 @@ def processor(
 
         # Get the source code of the function
         try:
-            function_source = inspect.getsource(func)
+            raw_function_source = inspect.getsource(func)
+            print(
+                f"[DEBUG Decorator] Raw source for {func.__name__}:\n{raw_function_source}"
+            )
+
             # Clean up the indentation
-            function_source = textwrap.dedent(function_source)
-        except (IOError, TypeError):
-            raise ValueError(
-                f"Could not retrieve source code for function {func.__name__}"
+            dedented_function_source = textwrap.dedent(raw_function_source)
+            print(
+                f"[DEBUG Decorator] Dedented source for {func.__name__}:\n{dedented_function_source}"
             )
 
+            # Find the start of the function definition ('def')
+            # Skip lines starting with '@' or empty lines until 'def' is found
+            lines = dedented_function_source.splitlines()
+            func_def_index = -1
+            for i, line in enumerate(lines):
+                stripped_line = line.strip()
+                if stripped_line.startswith("def "):
+                    func_def_index = i
+                    break
+                elif stripped_line.startswith("@") or not stripped_line:
+                    # Skip decorator lines and empty lines before 'def'
+                    continue
+                else:
+                    # Found something unexpected before 'def'
+                    raise ValueError(
+                        f"Unexpected content found before 'def' in source for {func.__name__}. Cannot reliably strip decorators."
+                    )
+
+            if func_def_index != -1:
+                # Keep lines from the 'def' line onwards
+                function_source = "\n".join(
+                    lines[func_def_index:]
+                )  # Use \n for env var
+            else:
+                # If 'def' wasn't found (shouldn't happen with valid function source)
+                raise ValueError(
+                    f"Could not find function definition 'def' in source for {func.__name__}"
+                )
+
+            print(
+                f"[DEBUG Decorator] Processed function source for {func.__name__}:\n{function_source}"
+            )
+
+        except (IOError, TypeError) as e:
+            print(f"[DEBUG Decorator] Error getting source for {func.__name__}: {e}")
+            raise ValueError(
+                f"Could not retrieve source code for function {func.__name__}: {e}"
+            ) from e
+
         # Get source code for the models
         input_model_source = None
         output_model_source = None
@@ -195,7 +237,11 @@ def processor(
         output_model_source = get_type_source(return_type)
 
         # Add function source code to environment variables
+        print(
+            f"[DEBUG Decorator] Setting FUNCTION_SOURCE: {function_source[:100]}..."
+        )  # Print first 100 chars
         all_env.append(V1EnvVar(key="FUNCTION_SOURCE", value=function_source))
+        print(f"[DEBUG Decorator] Setting FUNCTION_NAME: {func.__name__}")
         all_env.append(V1EnvVar(key="FUNCTION_NAME", value=func.__name__))
 
         # Add model source codes

nebu/processors/models.py

@@ -94,7 +94,7 @@ class V1StreamData(BaseModel):
     wait: Optional[bool] = None
 
 
-class V1StreamMessage(Generic[T], BaseModel):
+class V1StreamMessage(BaseModel, Generic[T]):
     kind: str = "StreamMessage"
     id: str
     content: Optional[T] = None

{nebu-0.1.10.dist-info → nebu-0.1.12.dist-info}/METADATA

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

{nebu-0.1.10.dist-info → nebu-0.1.12.dist-info}/RECORD

@@ -7,14 +7,14 @@ nebu/containers/decorator.py,sha256=qiM7hbHne9MhSp1gDgX5z5bimsXr_YPjTIZoe09dwr4,
 nebu/containers/models.py,sha256=0j6NGy4yto-enRDh_4JH_ZTbHrLdSpuMOqNQPnIrwC4,6815
 nebu/containers/server.py,sha256=yFa2Y9PzBn59E1HftKiv0iapPonli2rbGAiU6r-wwe0,2513
 nebu/processors/consumer.py,sha256=rFqd6gg2OYgXi3gf11GFpuaOOzuK1TYaPO-t_leSR8Y,15097
-nebu/processors/decorate.py,sha256=8jemT7QvY-2aAJAOiYzjClja-Zso0QgWXZH377uIW4I,13126
+nebu/processors/decorate.py,sha256=m5zDs7wxHLS0drgTwXUQTiUEFml1qoYDuVY8VMsgHQs,15218
 nebu/processors/default.py,sha256=W4slJenG59rvyTlJ7gRp58eFfXcNOTT2Hfi6zzJAobI,365
-nebu/processors/models.py,sha256=JpX5GouqdIdCQWUUNzQh68OsD57_Po2aQreDQ9VMRec,3654
+nebu/processors/models.py,sha256=GvnI8UJrQSjHo2snP07cPfisCH90cEGTY-PZV5_AtXI,3654
 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.10.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-nebu-0.1.10.dist-info/METADATA,sha256=apL1G3eCfv0BncDZL5Sqq4mc2e4yeA2sTmbMgACQOlU,1588
-nebu-0.1.10.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-nebu-0.1.10.dist-info/top_level.txt,sha256=uLIbEKJeGSHWOAJN5S0i5XBGwybALlF9bYoB1UhdEgQ,5
-nebu-0.1.10.dist-info/RECORD,,
+nebu-0.1.12.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+nebu-0.1.12.dist-info/METADATA,sha256=QEN1FBqo3AsqbtLdcOnu7wqiYf6Las8cI1dRrL8V7oY,1588
+nebu-0.1.12.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+nebu-0.1.12.dist-info/top_level.txt,sha256=uLIbEKJeGSHWOAJN5S0i5XBGwybALlF9bYoB1UhdEgQ,5
+nebu-0.1.12.dist-info/RECORD,,