feldera 0.123.0__tar.gz → 0.124.0__tar.gz

{feldera-0.123.0 → feldera-0.124.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: feldera
-Version: 0.123.0
+Version: 0.124.0
 Summary: The feldera python client
 Author-email: Feldera Team <dev@feldera.com>
 License: MIT

{feldera-0.123.0 → feldera-0.124.0}/feldera/pipeline.py

@@ -1061,7 +1061,18 @@ pipeline '{self.name}' to sync checkpoint '{uuid}'"""
             errors.append(derr)
         return errors
 
-    def support_bundle(self, output_path: Optional[str] = None) -> bytes:
+    def support_bundle(
+        self,
+        output_path: Optional[str] = None,
+        *,
+        circuit_profile: bool = True,
+        heap_profile: bool = True,
+        metrics: bool = True,
+        logs: bool = True,
+        stats: bool = True,
+        pipeline_config: bool = True,
+        system_config: bool = True,
+    ) -> bytes:
         """
         Generate a support bundle containing diagnostic information from this pipeline.
 
@@ -1071,11 +1082,37 @@ pipeline '{self.name}' to sync checkpoint '{uuid}'"""
 
         :param output_path: Optional path to save the support bundle file. If None,
             the support bundle is only returned as bytes.
+        :param circuit_profile: Whether to collect circuit profile data (default: True)
+        :param heap_profile: Whether to collect heap profile data (default: True)
+        :param metrics: Whether to collect metrics data (default: True)
+        :param logs: Whether to collect logs data (default: True)
+        :param stats: Whether to collect stats data (default: True)
+        :param pipeline_config: Whether to collect pipeline configuration data (default: True)
+        :param system_config: Whether to collect system configuration data (default: True)
         :return: The support bundle as bytes (ZIP archive)
         :raises FelderaAPIError: If the pipeline does not exist or if there's an error
         """
 
-        support_bundle_bytes = self.client.get_pipeline_support_bundle(self.name)
+        # Build query parameters
+        params = {}
+        if not circuit_profile:
+            params["circuit_profile"] = "false"
+        if not heap_profile:
+            params["heap_profile"] = "false"
+        if not metrics:
+            params["metrics"] = "false"
+        if not logs:
+            params["logs"] = "false"
+        if not stats:
+            params["stats"] = "false"
+        if not pipeline_config:
+            params["pipeline_config"] = "false"
+        if not system_config:
+            params["system_config"] = "false"
+
+        support_bundle_bytes = self.client.get_pipeline_support_bundle(
+            self.name, params=params
+        )
 
         if output_path is not None:
             path = pathlib.Path(output_path)

{feldera-0.123.0 → feldera-0.124.0}/feldera/rest/feldera_client.py

@@ -138,7 +138,17 @@ class FelderaClient:
                             err_msg += f"Code snippet:\n{sql_error['snippet']}"
                         raise RuntimeError(err_msg)
 
-                raise RuntimeError(f"The program failed to compile: {status}")
+                error_message = f"The program failed to compile: {status}\n"
+
+                rust_error = p.program_error.get("rust_compilation")
+                if rust_error is not None:
+                    error_message += f"Rust Error: {rust_error}\n"
+
+                system_error = p.program_error.get("system_error")
+                if system_error is not None:
+                    error_message += f"System Error: {system_error}"
+
+                raise RuntimeError(error_message)
 
             logging.debug("still compiling %s, waiting for 100 more milliseconds", name)
             time.sleep(0.1)
@@ -952,7 +962,9 @@ Reason: The pipeline is in a STOPPED state due to the following error:
 
         return FelderaConfig(resp)
 
-    def get_pipeline_support_bundle(self, pipeline_name: str) -> bytes:
+    def get_pipeline_support_bundle(
+        self, pipeline_name: str, params: Optional[Dict[str, Any]] = None
+    ) -> bytes:
         """
         Generate a support bundle containing diagnostic information from a pipeline.
 
@@ -961,12 +973,14 @@ Reason: The pipeline is in a STOPPED state due to the following error:
         and packages them into a single ZIP file for support purposes.
 
         :param pipeline_name: The name of the pipeline
+        :param params: Optional query parameters to control data collection
         :return: The support bundle as bytes (ZIP file)
         :raises FelderaAPIError: If the pipeline does not exist or if there's an error
         """
 
         resp = self.http.get(
             path=f"/pipelines/{pipeline_name}/support_bundle",
+            params=params,
             stream=True,
         )
 

{feldera-0.123.0 → feldera-0.124.0}/feldera.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: feldera
-Version: 0.123.0
+Version: 0.124.0
 Summary: The feldera python client
 Author-email: Feldera Team <dev@feldera.com>
 License: MIT

{feldera-0.123.0 → feldera-0.124.0}/pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "setuptools.build_meta"
 name = "feldera"
 readme = "README.md"
 description = "The feldera python client"
-version = "0.123.0"
+version = "0.124.0"
 license = { text = "MIT" }
 requires-python = ">=3.10"
 authors = [

{feldera-0.123.0 → feldera-0.124.0}/tests/test_shared_pipeline.py

@@ -555,7 +555,7 @@ class TestPipeline(SharedTestPipeline):
     def test_support_bundle(self):
         self.pipeline.start()
 
-        # Test getting support bundle as bytes
+        # Test getting support bundle as bytes (all collections enabled)
         support_bundle_bytes = self.pipeline.support_bundle()
         assert isinstance(support_bundle_bytes, bytes)
         assert len(support_bundle_bytes) > 0
@@ -587,6 +587,50 @@ class TestPipeline(SharedTestPipeline):
             if os.path.exists(temp_path):
                 os.unlink(temp_path)
 
+    def test_support_bundle_with_selectors(self):
+        self.pipeline.start()
+
+        # Test with all collections disabled
+        support_bundle_bytes = self.pipeline.support_bundle(
+            circuit_profile=False,
+            heap_profile=False,
+            metrics=False,
+            logs=False,
+            stats=False,
+            pipeline_config=False,
+            system_config=False,
+        )
+
+        assert isinstance(support_bundle_bytes, bytes)
+        assert len(support_bundle_bytes) > 0
+
+        # Verify it's a valid ZIP file
+        with zipfile.ZipFile(io.BytesIO(support_bundle_bytes), "r") as zip_file:
+            file_list = zip_file.namelist()
+            # Should only contain manifest.txt when all collections are disabled
+            assert len(file_list) == 1
+            assert "manifest.txt" in file_list
+
+        # Test with partial collections enabled
+        support_bundle_bytes = self.pipeline.support_bundle(
+            circuit_profile=True,
+            heap_profile=False,
+            metrics=True,
+            logs=False,
+            stats=True,
+            pipeline_config=False,
+            system_config=True,
+        )
+
+        assert isinstance(support_bundle_bytes, bytes)
+        assert len(support_bundle_bytes) > 0
+
+        # Verify it's a valid ZIP file
+        with zipfile.ZipFile(io.BytesIO(support_bundle_bytes), "r") as zip_file:
+            file_list = zip_file.namelist()
+            assert len(file_list) >= 2
+            assert "manifest.txt" in file_list
+
 
 if __name__ == "__main__":
     unittest.main()