feat: improvement: add a working example for data streaming

Author: Stainless Bot

examples/rest-api/StreamData.java

@@ -0,0 +1,63 @@
+/*
+ * This source file was generated by the Gradle 'init' task
+ */
+package org.example;
+
+import com.openlayer.api.client.OpenlayerClient;
+import com.openlayer.api.client.okhttp.OpenlayerOkHttpClient;
+import com.openlayer.api.core.JsonNumber;
+import com.openlayer.api.core.JsonString;
+import com.openlayer.api.models.InferencePipelineDataStreamParams;
+import com.openlayer.api.models.InferencePipelineDataStreamResponse;
+
+import java.util.List;
+
+public class StreamData {
+    public static void main(String[] args) {
+        // Configure the client using environment variables
+        OpenlayerClient client = OpenlayerOkHttpClient.fromEnv();
+
+        // Or you can configure the client with additional properties
+        /*
+         * OpenlayerClient client = OpenlayerOkHttpClient.builder()
+         * .fromEnv()
+         * // Additional properties can be set here
+         * .build();
+         */
+
+        // Replace with your inference pipeline id
+        String inferencePipelineId = "your-inference-pipeline-id";
+
+        // Let's say we want to stream the following row, which represents a model
+        // prediction:
+        // Define a row with the relevant fields
+        InferencePipelineDataStreamParams.Row row = InferencePipelineDataStreamParams.Row.builder()
+                .putAdditionalProperty("user_query", JsonString.of("what's the meaning of life?"))
+                .putAdditionalProperty("output", JsonString.of("42"))
+                .putAdditionalProperty("tokens", JsonNumber.of(7))
+                .putAdditionalProperty("cost", JsonNumber.of(0.02))
+                .putAdditionalProperty("timestamp", JsonNumber.of(1620000000))
+                .build();
+
+        // Create Inference Pipeline Data Stream Parameters
+        InferencePipelineDataStreamParams params = InferencePipelineDataStreamParams.builder()
+                .inferencePipelineId(inferencePipelineId)
+                .rows(List.of(row))
+                .config(InferencePipelineDataStreamParams.Config
+                        .ofLlmData(InferencePipelineDataStreamParams.Config.LlmData.builder()
+                                .outputColumnName("output")
+                                .costColumnName("cost")
+                                .inputVariableNames(List.of("user_query"))
+                                .numOfTokenColumnName("tokens")
+                                .timestampColumnName("timestamp")
+                                .build()))
+                .build();
+
+        // Execute the request
+        InferencePipelineDataStreamResponse inferencePipelineDataStreamResponse = client.inferencePipelines().data()
+                .stream(params);
+
+        // Print the response
+        System.out.println(inferencePipelineDataStreamResponse);
+    }
+}