|
9 | 9 | import com.openlayer.api.core.JsonString; |
10 | 10 | import com.openlayer.api.models.InferencePipelineDataStreamParams; |
11 | 11 | import com.openlayer.api.models.InferencePipelineDataStreamResponse; |
12 | | - |
13 | | -import java.util.List; |
| 12 | +import java.util.Arrays; |
14 | 13 |
|
15 | 14 | public class StreamData { |
16 | 15 | public static void main(String[] args) { |
@@ -42,20 +41,20 @@ public static void main(String[] args) { |
42 | 41 | // Create Inference Pipeline Data Stream Parameters |
43 | 42 | InferencePipelineDataStreamParams params = InferencePipelineDataStreamParams.builder() |
44 | 43 | .inferencePipelineId(inferencePipelineId) |
45 | | - .rows(List.of(row)) |
46 | | - .config(InferencePipelineDataStreamParams.Config |
47 | | - .ofLlmData(InferencePipelineDataStreamParams.Config.LlmData.builder() |
| 44 | + .rows(Arrays.asList(row)) |
| 45 | + .config(InferencePipelineDataStreamParams.Config.ofLlmData( |
| 46 | + InferencePipelineDataStreamParams.Config.LlmData.builder() |
48 | 47 | .outputColumnName("output") |
49 | 48 | .costColumnName("cost") |
50 | | - .inputVariableNames(List.of("user_query")) |
| 49 | + .inputVariableNames(Arrays.asList("user_query")) |
51 | 50 | .numOfTokenColumnName("tokens") |
52 | 51 | .timestampColumnName("timestamp") |
53 | 52 | .build())) |
54 | 53 | .build(); |
55 | 54 |
|
56 | 55 | // Execute the request |
57 | | - InferencePipelineDataStreamResponse inferencePipelineDataStreamResponse = client.inferencePipelines().data() |
58 | | - .stream(params); |
| 56 | + InferencePipelineDataStreamResponse inferencePipelineDataStreamResponse = |
| 57 | + client.inferencePipelines().data().stream(params); |
59 | 58 |
|
60 | 59 | // Print the response |
61 | 60 | System.out.println(inferencePipelineDataStreamResponse); |
|
0 commit comments