Skip to content

Commit 9261558

Browse files
stainless-app[bot]stainless-app[bot]
committed
chore: update example values in tests and docs (#61)
1 parent 5a3f954 commit 9261558

File tree

6 files changed

+262
-54
lines changed

6 files changed

+262
-54
lines changed

‎README.md

+9-3
Original file line numberDiff line numberDiff line change
@@ -79,13 +79,19 @@ import java.util.List;
7979
InferencePipelineDataStreamParams params = InferencePipelineDataStreamParams.builder()
8080
.inferencePipelineId("182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e")
8181
.config(InferencePipelineDataStreamParams.Config.ofLlmData(InferencePipelineDataStreamParams.Config.LlmData.builder()
82-
.outputColumnName("output")
83-
.costColumnName("cost")
8482
.inputVariableNames(List.of("user_query"))
83+
.outputColumnName("output")
8584
.numOfTokenColumnName("tokens")
85+
.costColumnName("cost")
8686
.timestampColumnName("timestamp")
8787
.build()))
88-
.row(List.of(InferencePipelineDataStreamParams.Row.builder().build()))
88+
.row(List.of(InferencePipelineDataStreamParams.Row.builder()
89+
.putAdditionalProperty("user_query", JsonValue.from("what is the meaning of life?"))
90+
.putAdditionalProperty("output", JsonValue.from("42"))
91+
.putAdditionalProperty("tokens", JsonValue.from(7))
92+
.putAdditionalProperty("cost", JsonValue.from(0.02))
93+
.putAdditionalProperty("timestamp", JsonValue.from(1610000000))
94+
.build()))
8995
.build();
9096
InferencePipelineDataStreamResponse response = client.inferencePipelines().data().stream(params);
9197
```

‎openlayer-java-core/src/test/kotlin/com/openlayer/api/models/InferencePipelineDataStreamParamsTest.kt

+72-12
Original file line numberDiff line numberDiff line change
@@ -21,10 +21,10 @@ class InferencePipelineDataStreamParamsTest {
2121
.costColumnName("cost")
2222
.groundTruthColumnName("ground_truth")
2323
.inferenceIdColumnName("id")
24-
.inputVariableNames(listOf("string"))
24+
.inputVariableNames(listOf("user_query"))
2525
.latencyColumnName("latency")
2626
.metadata(JsonValue.from(mapOf<String, Any>()))
27-
.numOfTokenColumnName("num_tokens")
27+
.numOfTokenColumnName("tokens")
2828
.prompt(
2929
listOf(
3030
InferencePipelineDataStreamParams.Config.LlmData.Prompt.builder()
@@ -38,7 +38,17 @@ class InferencePipelineDataStreamParamsTest {
3838
.build()
3939
)
4040
)
41-
.rows(listOf(InferencePipelineDataStreamParams.Row.builder().build()))
41+
.rows(
42+
listOf(
43+
InferencePipelineDataStreamParams.Row.builder()
44+
.putAdditionalProperty("user_query", JsonValue.from("bar"))
45+
.putAdditionalProperty("output", JsonValue.from("bar"))
46+
.putAdditionalProperty("tokens", JsonValue.from("bar"))
47+
.putAdditionalProperty("cost", JsonValue.from("bar"))
48+
.putAdditionalProperty("timestamp", JsonValue.from("bar"))
49+
.build()
50+
)
51+
)
4252
.build()
4353
}
4454

@@ -55,10 +65,10 @@ class InferencePipelineDataStreamParamsTest {
5565
.costColumnName("cost")
5666
.groundTruthColumnName("ground_truth")
5767
.inferenceIdColumnName("id")
58-
.inputVariableNames(listOf("string"))
68+
.inputVariableNames(listOf("user_query"))
5969
.latencyColumnName("latency")
6070
.metadata(JsonValue.from(mapOf<String, Any>()))
61-
.numOfTokenColumnName("num_tokens")
71+
.numOfTokenColumnName("tokens")
6272
.prompt(
6373
listOf(
6474
InferencePipelineDataStreamParams.Config.LlmData.Prompt
@@ -73,7 +83,17 @@ class InferencePipelineDataStreamParamsTest {
7383
.build()
7484
)
7585
)
76-
.rows(listOf(InferencePipelineDataStreamParams.Row.builder().build()))
86+
.rows(
87+
listOf(
88+
InferencePipelineDataStreamParams.Row.builder()
89+
.putAdditionalProperty("user_query", JsonValue.from("bar"))
90+
.putAdditionalProperty("output", JsonValue.from("bar"))
91+
.putAdditionalProperty("tokens", JsonValue.from("bar"))
92+
.putAdditionalProperty("cost", JsonValue.from("bar"))
93+
.putAdditionalProperty("timestamp", JsonValue.from("bar"))
94+
.build()
95+
)
96+
)
7797
.build()
7898
val body = params.getBody()
7999
assertThat(body).isNotNull
@@ -86,10 +106,10 @@ class InferencePipelineDataStreamParamsTest {
86106
.costColumnName("cost")
87107
.groundTruthColumnName("ground_truth")
88108
.inferenceIdColumnName("id")
89-
.inputVariableNames(listOf("string"))
109+
.inputVariableNames(listOf("user_query"))
90110
.latencyColumnName("latency")
91111
.metadata(JsonValue.from(mapOf<String, Any>()))
92-
.numOfTokenColumnName("num_tokens")
112+
.numOfTokenColumnName("tokens")
93113
.prompt(
94114
listOf(
95115
InferencePipelineDataStreamParams.Config.LlmData.Prompt.builder()
@@ -104,7 +124,17 @@ class InferencePipelineDataStreamParamsTest {
104124
)
105125
)
106126
assertThat(body.rows())
107-
.isEqualTo(listOf(InferencePipelineDataStreamParams.Row.builder().build()))
127+
.isEqualTo(
128+
listOf(
129+
InferencePipelineDataStreamParams.Row.builder()
130+
.putAdditionalProperty("user_query", JsonValue.from("bar"))
131+
.putAdditionalProperty("output", JsonValue.from("bar"))
132+
.putAdditionalProperty("tokens", JsonValue.from("bar"))
133+
.putAdditionalProperty("cost", JsonValue.from("bar"))
134+
.putAdditionalProperty("timestamp", JsonValue.from("bar"))
135+
.build()
136+
)
137+
)
108138
}
109139

110140
@Test
@@ -119,7 +149,17 @@ class InferencePipelineDataStreamParamsTest {
119149
.build()
120150
)
121151
)
122-
.rows(listOf(InferencePipelineDataStreamParams.Row.builder().build()))
152+
.rows(
153+
listOf(
154+
InferencePipelineDataStreamParams.Row.builder()
155+
.putAdditionalProperty("user_query", JsonValue.from("bar"))
156+
.putAdditionalProperty("output", JsonValue.from("bar"))
157+
.putAdditionalProperty("tokens", JsonValue.from("bar"))
158+
.putAdditionalProperty("cost", JsonValue.from("bar"))
159+
.putAdditionalProperty("timestamp", JsonValue.from("bar"))
160+
.build()
161+
)
162+
)
123163
.build()
124164
val body = params.getBody()
125165
assertThat(body).isNotNull
@@ -132,7 +172,17 @@ class InferencePipelineDataStreamParamsTest {
132172
)
133173
)
134174
assertThat(body.rows())
135-
.isEqualTo(listOf(InferencePipelineDataStreamParams.Row.builder().build()))
175+
.isEqualTo(
176+
listOf(
177+
InferencePipelineDataStreamParams.Row.builder()
178+
.putAdditionalProperty("user_query", JsonValue.from("bar"))
179+
.putAdditionalProperty("output", JsonValue.from("bar"))
180+
.putAdditionalProperty("tokens", JsonValue.from("bar"))
181+
.putAdditionalProperty("cost", JsonValue.from("bar"))
182+
.putAdditionalProperty("timestamp", JsonValue.from("bar"))
183+
.build()
184+
)
185+
)
136186
}
137187

138188
@Test
@@ -147,7 +197,17 @@ class InferencePipelineDataStreamParamsTest {
147197
.build()
148198
)
149199
)
150-
.rows(listOf(InferencePipelineDataStreamParams.Row.builder().build()))
200+
.rows(
201+
listOf(
202+
InferencePipelineDataStreamParams.Row.builder()
203+
.putAdditionalProperty("user_query", JsonValue.from("bar"))
204+
.putAdditionalProperty("output", JsonValue.from("bar"))
205+
.putAdditionalProperty("tokens", JsonValue.from("bar"))
206+
.putAdditionalProperty("cost", JsonValue.from("bar"))
207+
.putAdditionalProperty("timestamp", JsonValue.from("bar"))
208+
.build()
209+
)
210+
)
151211
.build()
152212
assertThat(params).isNotNull
153213
// path param "inferencePipelineId"

‎openlayer-java-core/src/test/kotlin/com/openlayer/api/models/ProjectCommitCreateParamsTest.kt

+12
Original file line numberDiff line numberDiff line change
@@ -93,8 +93,12 @@ class ProjectCommitCreateParamsTest {
9393
ProjectCommitCreateParams.Commit.builder()
9494
.id("3fa85f64-5717-4562-b3fc-2c963f66afa6")
9595
.authorId("589ece63-49a2-41b4-98e1-10547761d4b0")
96+
.fileSize(1024L)
9697
.message("Updated the prompt.")
98+
.mlModelId("3fa85f64-5717-4562-b3fc-2c963f66afa6")
9799
.storageUri("s3://...")
100+
.trainingDatasetId("3fa85f64-5717-4562-b3fc-2c963f66afa6")
101+
.validationDatasetId("3fa85f64-5717-4562-b3fc-2c963f66afa6")
98102
.build()
99103
)
100104
.storageUri("s3://...")
@@ -106,8 +110,12 @@ class ProjectCommitCreateParamsTest {
106110
ProjectCommitCreateParams.Commit.builder()
107111
.id("3fa85f64-5717-4562-b3fc-2c963f66afa6")
108112
.authorId("589ece63-49a2-41b4-98e1-10547761d4b0")
113+
.fileSize(1024L)
109114
.message("Updated the prompt.")
115+
.mlModelId("3fa85f64-5717-4562-b3fc-2c963f66afa6")
110116
.storageUri("s3://...")
117+
.trainingDatasetId("3fa85f64-5717-4562-b3fc-2c963f66afa6")
118+
.validationDatasetId("3fa85f64-5717-4562-b3fc-2c963f66afa6")
111119
.build()
112120
)
113121
assertThat(body.storageUri()).isEqualTo("s3://...")
@@ -122,8 +130,12 @@ class ProjectCommitCreateParamsTest {
122130
ProjectCommitCreateParams.Commit.builder()
123131
.id("3fa85f64-5717-4562-b3fc-2c963f66afa6")
124132
.authorId("589ece63-49a2-41b4-98e1-10547761d4b0")
133+
.fileSize(1024L)
125134
.message("Updated the prompt.")
135+
.mlModelId("3fa85f64-5717-4562-b3fc-2c963f66afa6")
126136
.storageUri("s3://...")
137+
.trainingDatasetId("3fa85f64-5717-4562-b3fc-2c963f66afa6")
138+
.validationDatasetId("3fa85f64-5717-4562-b3fc-2c963f66afa6")
127139
.build()
128140
)
129141
.storageUri("s3://...")

0 commit comments

Comments
 (0)