types: fix openapi spec of StartFromCheckpoint

abhizer · abhizer · commit a81bc780a604 · 2025-07-22T07:49:22.000Z
* py: set `runtime_version` in `PipelineBuilder`

Signed-off-by: Abhinav Gyawali &lt;22275402+abhizer@users.noreply.github.com&gt;

[ci] apply automatic fixes

Signed-off-by: feldera-bot &lt;feldera-bot@feldera.com&gt;
diff --git a/Cargo.toml b/Cargo.toml
@@ -238,7 +238,7 @@ tracing-subscriber = "0.3.18"
 typedmap = "0.3.0"
 url = "2.5.0"
 urlencoding = "2.1.3"
-utoipa = "4.2"
+utoipa = { version =  "4.2", features = ["uuid"] }
 utoipa-swagger-ui = { version = "7.1", features = ["vendored"] }
 uuid = { version = "1.17.0", features = ["serde"] }
 wiremock = "0.6"
diff --git a/crates/feldera-types/src/config.rs b/crates/feldera-types/src/config.rs
@@ -272,12 +272,42 @@ pub enum StorageCompression {
     Snappy,
 }
 
-#[derive(Debug, Clone, Eq, PartialEq, ToSchema)]
+#[derive(Debug, Clone, Eq, PartialEq)]
 pub enum StartFromCheckpoint {
     Latest,
     Uuid(uuid::Uuid),
 }
 
+impl ToSchema<'_> for StartFromCheckpoint {
+    fn schema() -> (
+        &'static str,
+        utoipa::openapi::RefOr<utoipa::openapi::schema::Schema>,
+    ) {
+        (
+            "StartFromCheckpoint",
+            utoipa::openapi::RefOr::T(Schema::OneOf(
+                OneOfBuilder::new()
+                    .item(
+                        ObjectBuilder::new()
+                            .schema_type(SchemaType::String)
+                            .enum_values(Some(["latest"].into_iter()))
+                            .build(),
+                    )
+                    .item(
+                        ObjectBuilder::new()
+                            .schema_type(SchemaType::String)
+                            .format(Some(utoipa::openapi::SchemaFormat::KnownFormat(
+                                utoipa::openapi::KnownFormat::Uuid,
+                            )))
+                            .build(),
+                    )
+                    .nullable(true)
+                    .build(),
+            )),
+        )
+    }
+}
+
 impl<'de> Deserialize<'de> for StartFromCheckpoint {
     fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
     where
diff --git a/openapi.json b/openapi.json
@@ -7238,22 +7238,15 @@
           {
             "type": "string",
             "enum": [
-              "Latest"
+              "latest"
             ]
           },
           {
-            "type": "object",
-            "required": [
-              "Uuid"
-            ],
-            "properties": {
-              "Uuid": {
-                "type": "string",
-                "format": "uuid"
-              }
-            }
+            "type": "string",
+            "format": "uuid"
           }
-        ]
+        ],
+        "nullable": true
       },
       "StorageBackendConfig": {
         "oneOf": [
diff --git a/python/feldera/pipeline_builder.py b/python/feldera/pipeline_builder.py
@@ -1,3 +1,6 @@
+import os
+from typing import Optional
+
 from feldera.rest.feldera_client import FelderaClient
 from feldera.rest.pipeline import Pipeline as InnerPipeline
 from feldera.pipeline import Pipeline
@@ -32,6 +35,7 @@ def __init__(
         description: str = "",
         compilation_profile: CompilationProfile = CompilationProfile.OPTIMIZED,
         runtime_config: RuntimeConfig = RuntimeConfig.default(),
+        runtime_version: Optional[str] = None,
     ):
         self.client: FelderaClient = client
         self.name: str | None = name
@@ -41,6 +45,9 @@ def __init__(
         self.udf_toml: str = udf_toml
         self.compilation_profile: CompilationProfile = compilation_profile
         self.runtime_config: RuntimeConfig = runtime_config
+        self.runtime_version: Optional[str] = runtime_version or os.environ.get(
+            "FELDERA_RUNTIME_VERSION"
+        )
 
     def create(self) -> Pipeline:
         """
@@ -67,6 +74,7 @@ def create(self) -> Pipeline:
             udf_toml=self.udf_toml,
             program_config={
                 "profile": self.compilation_profile.value,
+                "runtime_version": self.runtime_version,
             },
             runtime_config=self.runtime_config.to_dict(),
         )
diff --git a/python/feldera/runtime_config.py b/python/feldera/runtime_config.py
@@ -1,4 +1,3 @@
-import os
 from typing import Optional, Any, Mapping
 from feldera.enums import FaultToleranceModel
 
@@ -79,7 +78,6 @@ def __init__(
         clock_resolution_usecs: Optional[int] = None,
         provisioning_timeout_secs: Optional[int] = None,
         resources: Optional[Resources] = None,
-        runtime_version: Optional[str] = None,
         fault_tolerance_model: Optional[FaultToleranceModel] = None,
         checkpoint_interval_secs: Optional[int] = None,
     ):
@@ -91,9 +89,6 @@ def __init__(
         self.min_batch_size_records = min_batch_size_records
         self.clock_resolution_usecs = clock_resolution_usecs
         self.provisioning_timeout_secs = provisioning_timeout_secs
-        self.runtime_version = runtime_version or os.environ.get(
-            "FELDERA_RUNTIME_VERSION"
-        )
         if fault_tolerance_model is not None:
             self.fault_tolerance = {
                 "model": str(fault_tolerance_model),
