diff --git a/.codegen/_openapi_sha b/.codegen/_openapi_sha index 8cd956362..35bb18e53 100644 --- a/.codegen/_openapi_sha +++ b/.codegen/_openapi_sha @@ -1 +1 @@ -06a18b97d7996d6cd8dd88bfdb0f2c2792739e46 \ No newline at end of file +pr:1008805 \ No newline at end of file diff --git a/NEXT_CHANGELOG.md b/NEXT_CHANGELOG.md index 395a1a32c..e4b66bc5c 100644 --- a/NEXT_CHANGELOG.md +++ b/NEXT_CHANGELOG.md @@ -11,3 +11,4 @@ ### Internal Changes ### API Changes +* Added `outputs` field to `QueryEndpointResponseObject` to support feature serving endpoints responses. diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/core/DatabricksConfig.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/core/DatabricksConfig.java index fa89f5041..3feb86a3e 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/core/DatabricksConfig.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/core/DatabricksConfig.java @@ -374,13 +374,17 @@ public DatabricksConfig setAzureUseMsi(boolean azureUseMsi) { return this; } - /** @deprecated Use {@link #getAzureUseMsi()} instead. */ + /** + * @deprecated Use {@link #getAzureUseMsi()} instead. + */ @Deprecated() public boolean getAzureUseMSI() { return azureUseMsi; } - /** @deprecated Use {@link #setAzureUseMsi(boolean)} instead. */ + /** + * @deprecated Use {@link #setAzureUseMsi(boolean)} instead. + */ @Deprecated public DatabricksConfig setAzureUseMSI(boolean azureUseMsi) { this.azureUseMsi = azureUseMsi; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/QueryEndpointResponse.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/QueryEndpointResponse.java index cc5be3bc5..88b18ad59 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/QueryEndpointResponse.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/QueryEndpointResponse.java @@ -54,6 +54,10 @@ public class QueryEndpointResponse { @JsonProperty("object") private QueryEndpointResponseObject object; + /** The outputs returned by the feature serving endpoint. */ + @JsonProperty("outputs") + private Collection outputs; + /** The predictions returned by the serving endpoint. */ @JsonProperty("predictions") private Collection predictions; @@ -127,6 +131,15 @@ public QueryEndpointResponseObject getObject() { return object; } + public QueryEndpointResponse setOutputs(Collection outputs) { + this.outputs = outputs; + return this; + } + + public Collection getOutputs() { + return outputs; + } + public QueryEndpointResponse setPredictions(Collection predictions) { this.predictions = predictions; return this; @@ -165,6 +178,7 @@ public boolean equals(Object o) { && Objects.equals(id, that.id) && Objects.equals(model, that.model) && Objects.equals(object, that.object) + && Objects.equals(outputs, that.outputs) && Objects.equals(predictions, that.predictions) && Objects.equals(servedModelName, that.servedModelName) && Objects.equals(usage, that.usage); @@ -173,7 +187,7 @@ public boolean equals(Object o) { @Override public int hashCode() { return Objects.hash( - choices, created, data, id, model, object, predictions, servedModelName, usage); + choices, created, data, id, model, object, outputs, predictions, servedModelName, usage); } @Override @@ -185,6 +199,7 @@ public String toString() { .add("id", id) .add("model", model) .add("object", object) + .add("outputs", outputs) .add("predictions", predictions) .add("servedModelName", servedModelName) .add("usage", usage)