[pipeline-manager] Disable stream compression for streaming endpoints

Karakatiza666 · Karakatiza666 · commit a50460c773f8 · 2025-11-20T11:14:08.000Z
Update tests that involve support_bundle

Signed-off-by: Karakatiza666 &lt;bulakh.96@gmail.com&gt;
diff --git a/.devcontainer/Dockerfile b/.devcontainer/Dockerfile
@@ -37,6 +37,7 @@ RUN apt-get update --fix-missing && \
 RUN apt-get install -y \
    gcc clang libclang-dev python3-pip python3-plumbum \
    hub numactl openjdk-21-jre-headless maven
+RUN curl -LsSf https://astral.sh/uv/install.sh | sh
 
 ## Install nodejs (optional)
 RUN mkdir -p /etc/apt/keyrings
diff --git a/crates/pipeline-manager/src/runner/interaction.rs b/crates/pipeline-manager/src/runner/interaction.rs
@@ -540,6 +540,8 @@ impl RunnerInteraction {
         for header in response.headers().into_iter() {
             builder.append_header(header);
         }
+        // Disable compression to avoid gzip frame buffering that causes clients to block
+        builder.insert_header(actix_http::ContentEncoding::Identity);
         Ok(builder.streaming(response))
     }
 
@@ -612,6 +614,8 @@ impl RunnerInteraction {
         for header in response.headers().into_iter() {
             builder.append_header(header);
         }
+        // Disable compression to avoid gzip frame buffering that causes clients to block
+        builder.insert_header(actix_http::ContentEncoding::Identity);
         Ok(builder.streaming(response))
     }
 
diff --git a/python/feldera/pipeline.py b/python/feldera/pipeline.py
@@ -1337,6 +1337,7 @@ def support_bundle(
         stats: bool = True,
         pipeline_config: bool = True,
         system_config: bool = True,
+        dataflow_graph: bool = True,
     ) -> bytes:
         """
         Generate a support bundle containing diagnostic information from this pipeline.
@@ -1354,6 +1355,7 @@ def support_bundle(
         :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)
+        :param dataflow_graph: Whether to collect dataflow graph (default: True)
         :return: The support bundle as bytes (ZIP archive)
         :raises FelderaAPIError: If the pipeline does not exist or if there's an error
         """
@@ -1374,6 +1376,8 @@ def support_bundle(
             params["pipeline_config"] = "false"
         if not system_config:
             params["system_config"] = "false"
+        if not dataflow_graph:
+            params["dataflow_graph"] = "false"
 
         support_bundle_bytes = self.client.get_pipeline_support_bundle(
             self.name, params=params
diff --git a/python/tests/platform/test_shared_pipeline.py b/python/tests/platform/test_shared_pipeline.py
@@ -653,7 +653,7 @@ def test_support_bundle(self):
             with zipfile.ZipFile(io.BytesIO(support_bundle_bytes), "r") as zip_file:
                 # Check that the ZIP file contains some files
                 file_list = zip_file.namelist()
-                assert any("json_circuit_profile.zip" in item for item in file_list)
+                assert any("circuit_profile.json" in item for item in file_list)
                 assert len(file_list) > 0
         except zipfile.BadZipFile:
             self.fail("Support bundle is not a valid ZIP file")
@@ -688,6 +688,7 @@ def test_support_bundle_with_selectors(self):
             stats=False,
             pipeline_config=False,
             system_config=False,
+            dataflow_graph=False,
         )
 
         assert isinstance(support_bundle_bytes, bytes)

Original file line number	Diff line number	Diff line change
`@@ -540,6 +540,8 @@ impl RunnerInteraction {`
`540`	`540`	`for header in response.headers().into_iter() {`
`541`	`541`	`builder.append_header(header);`
`542`	`542`	`}`
	`543`	`+ // Disable compression to avoid gzip frame buffering that causes clients to block`
	`544`	`+ builder.insert_header(actix_http::ContentEncoding::Identity);`
`543`	`545`	`Ok(builder.streaming(response))`
`544`	`546`	`}`
`545`	`547`
`@@ -612,6 +614,8 @@ impl RunnerInteraction {`
`612`	`614`	`for header in response.headers().into_iter() {`
`613`	`615`	`builder.append_header(header);`
`614`	`616`	`}`
	`617`	`+ // Disable compression to avoid gzip frame buffering that causes clients to block`
	`618`	`+ builder.insert_header(actix_http::ContentEncoding::Identity);`
`615`	`619`	`Ok(builder.streaming(response))`
`616`	`620`	`}`
`617`	`621`