From 14bf3c125583534ffc3e57ee0f7ec07658ef03ab Mon Sep 17 00:00:00 2001
From: "databricks-ci-ghec-1[bot]"
 <184311507+databricks-ci-ghec-1[bot]@users.noreply.github.com>
Date: Fri, 17 Jul 2026 05:22:22 +0000
Subject: [PATCH] Update SDK to a6998209ce3202b5a4c5dae41c4bf5cc905d35cb

---
 .codegen/_openapi_sha                         |  2 +-
 NEXT_CHANGELOG.md                             |  1 +
 .../sdk/service/ml/StreamSource.java          | 42 +++++++++++++++++--
 3 files changed, 41 insertions(+), 4 deletions(-)

diff --git a/.codegen/_openapi_sha b/.codegen/_openapi_sha
index 3fa3c1d2d..5ae26219d 100644
--- a/.codegen/_openapi_sha
+++ b/.codegen/_openapi_sha
@@ -1 +1 @@
-e0647ba6804a08bc727c2c58a0f4cd9774da1313
\ No newline at end of file
+a6998209ce3202b5a4c5dae41c4bf5cc905d35cb
\ No newline at end of file
diff --git a/NEXT_CHANGELOG.md b/NEXT_CHANGELOG.md
index 5361e4c66..cdd3b0ba1 100644
--- a/NEXT_CHANGELOG.md
+++ b/NEXT_CHANGELOG.md
@@ -15,3 +15,4 @@
 ### Internal Changes
 
 ### API Changes
+* Add `dataframeSchema` and `transformationSql` fields for `com.databricks.sdk.service.ml.StreamSource`.
\ No newline at end of file
diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/StreamSource.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/StreamSource.java
index 536ef9735..e177bab4d 100644
--- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/StreamSource.java
+++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/StreamSource.java
@@ -10,6 +10,13 @@
 /** A Stream entity used as a data source for a feature. */
 @Generated
 public class StreamSource {
+  /**
+   * Schema of the resulting dataframe after transformations, in Spark StructType JSON format (from
+   * df.schema.json()). Any subsequent functions operate against this dataframe.
+   */
+  @JsonProperty("dataframe_schema")
+  private String dataframeSchema;
+
   /** The filter condition applied to the source data before aggregation. */
   @JsonProperty("filter_condition")
   private String filterCondition;
@@ -18,6 +25,22 @@ public class StreamSource {
   @JsonProperty("full_name")
   private String fullName;
 
+  /**
+   * The pipeline runs these SQL statements immediately after conversion into the schema specified
+   * on the Stream object.
+   */
+  @JsonProperty("transformation_sql")
+  private String transformationSql;
+
+  public StreamSource setDataframeSchema(String dataframeSchema) {
+    this.dataframeSchema = dataframeSchema;
+    return this;
+  }
+
+  public String getDataframeSchema() {
+    return dataframeSchema;
+  }
+
   public StreamSource setFilterCondition(String filterCondition) {
     this.filterCondition = filterCondition;
     return this;
@@ -36,25 +59,38 @@ public String getFullName() {
     return fullName;
   }
 
+  public StreamSource setTransformationSql(String transformationSql) {
+    this.transformationSql = transformationSql;
+    return this;
+  }
+
+  public String getTransformationSql() {
+    return transformationSql;
+  }
+
   @Override
   public boolean equals(Object o) {
     if (this == o) return true;
     if (o == null || getClass() != o.getClass()) return false;
     StreamSource that = (StreamSource) o;
-    return Objects.equals(filterCondition, that.filterCondition)
-        && Objects.equals(fullName, that.fullName);
+    return Objects.equals(dataframeSchema, that.dataframeSchema)
+        && Objects.equals(filterCondition, that.filterCondition)
+        && Objects.equals(fullName, that.fullName)
+        && Objects.equals(transformationSql, that.transformationSql);
   }
 
   @Override
   public int hashCode() {
-    return Objects.hash(filterCondition, fullName);
+    return Objects.hash(dataframeSchema, filterCondition, fullName, transformationSql);
   }
 
   @Override
   public String toString() {
     return new ToStringer(StreamSource.class)
+        .add("dataframeSchema", dataframeSchema)
         .add("filterCondition", filterCondition)
         .add("fullName", fullName)
+        .add("transformationSql", transformationSql)
         .toString();
   }
 }