Skip to content

Commit 9aa09b1

Browse files
stainless-app[bot]stainless-app[bot]
committed
fix: add missing fields
1 parent 9f8de64 commit 9aa09b1

File tree

15 files changed

+213
-0
lines changed

15 files changed

+213
-0
lines changed

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

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,7 @@ internal class CommitRetrieveResponseTest {
3939
.projectId("3fa85f64-5717-4562-b3fc-2c963f66afa6")
4040
.status(CommitRetrieveResponse.Status.COMPLETED)
4141
.statusMessage("Commit successfully processed.")
42+
.storageUri("s3://...")
4243
.totalGoalCount(6L)
4344
.trainingDatasetId("3fa85f64-5717-4562-b3fc-2c963f66afa6")
4445
.validationDatasetId("3fa85f64-5717-4562-b3fc-2c963f66afa6")
@@ -85,6 +86,7 @@ internal class CommitRetrieveResponseTest {
8586
.isEqualTo(CommitRetrieveResponse.Status.COMPLETED)
8687
assertThat(commitRetrieveResponse.statusMessage())
8788
.contains("Commit successfully processed.")
89+
assertThat(commitRetrieveResponse.storageUri()).isEqualTo("s3://...")
8890
assertThat(commitRetrieveResponse.totalGoalCount()).isEqualTo(6L)
8991
assertThat(commitRetrieveResponse.trainingDatasetId())
9092
.contains("3fa85f64-5717-4562-b3fc-2c963f66afa6")
@@ -132,6 +134,7 @@ internal class CommitRetrieveResponseTest {
132134
.projectId("3fa85f64-5717-4562-b3fc-2c963f66afa6")
133135
.status(CommitRetrieveResponse.Status.COMPLETED)
134136
.statusMessage("Commit successfully processed.")
137+
.storageUri("s3://...")
135138
.totalGoalCount(6L)
136139
.trainingDatasetId("3fa85f64-5717-4562-b3fc-2c963f66afa6")
137140
.validationDatasetId("3fa85f64-5717-4562-b3fc-2c963f66afa6")

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

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,16 @@ internal class InferencePipelineRetrieveResponseTest {
4444
.BIGQUERY
4545
)
4646
.bigqueryConnectionId("182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e")
47+
.config(
48+
InferencePipelineRetrieveResponse.DataBackend.BigQueryDataBackend.Config
49+
.builder()
50+
.groundTruthColumnName("ground_truth")
51+
.humanFeedbackColumnName("human_feedback")
52+
.inferenceIdColumnName("id")
53+
.latencyColumnName("latency")
54+
.timestampColumnName("timestamp")
55+
.build()
56+
)
4757
.datasetId("my-dataset")
4858
.projectId("my-project")
4959
.tableId("my-table")
@@ -111,6 +121,7 @@ internal class InferencePipelineRetrieveResponseTest {
111121
.projectCount(0L)
112122
.slug("openlayer")
113123
.status(InferencePipelineRetrieveResponse.Workspace.Status.ACTIVE)
124+
.inviteCode("inviteCode")
114125
.addMonthlyUsage(
115126
InferencePipelineRetrieveResponse.Workspace.MonthlyUsage.builder()
116127
.executionTimeMs(0L)
@@ -167,6 +178,16 @@ internal class InferencePipelineRetrieveResponseTest {
167178
.BIGQUERY
168179
)
169180
.bigqueryConnectionId("182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e")
181+
.config(
182+
InferencePipelineRetrieveResponse.DataBackend.BigQueryDataBackend.Config
183+
.builder()
184+
.groundTruthColumnName("ground_truth")
185+
.humanFeedbackColumnName("human_feedback")
186+
.inferenceIdColumnName("id")
187+
.latencyColumnName("latency")
188+
.timestampColumnName("timestamp")
189+
.build()
190+
)
170191
.datasetId("my-dataset")
171192
.projectId("my-project")
172193
.tableId("my-table")
@@ -238,6 +259,7 @@ internal class InferencePipelineRetrieveResponseTest {
238259
.projectCount(0L)
239260
.slug("openlayer")
240261
.status(InferencePipelineRetrieveResponse.Workspace.Status.ACTIVE)
262+
.inviteCode("inviteCode")
241263
.addMonthlyUsage(
242264
InferencePipelineRetrieveResponse.Workspace.MonthlyUsage.builder()
243265
.executionTimeMs(0L)
@@ -287,6 +309,16 @@ internal class InferencePipelineRetrieveResponseTest {
287309
.BIGQUERY
288310
)
289311
.bigqueryConnectionId("182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e")
312+
.config(
313+
InferencePipelineRetrieveResponse.DataBackend.BigQueryDataBackend.Config
314+
.builder()
315+
.groundTruthColumnName("ground_truth")
316+
.humanFeedbackColumnName("human_feedback")
317+
.inferenceIdColumnName("id")
318+
.latencyColumnName("latency")
319+
.timestampColumnName("timestamp")
320+
.build()
321+
)
290322
.datasetId("my-dataset")
291323
.projectId("my-project")
292324
.tableId("my-table")
@@ -354,6 +386,7 @@ internal class InferencePipelineRetrieveResponseTest {
354386
.projectCount(0L)
355387
.slug("openlayer")
356388
.status(InferencePipelineRetrieveResponse.Workspace.Status.ACTIVE)
389+
.inviteCode("inviteCode")
357390
.addMonthlyUsage(
358391
InferencePipelineRetrieveResponse.Workspace.MonthlyUsage.builder()
359392
.executionTimeMs(0L)

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

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,16 @@ internal class InferencePipelineUpdateResponseTest {
4444
.BIGQUERY
4545
)
4646
.bigqueryConnectionId("182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e")
47+
.config(
48+
InferencePipelineUpdateResponse.DataBackend.BigQueryDataBackend.Config
49+
.builder()
50+
.groundTruthColumnName("ground_truth")
51+
.humanFeedbackColumnName("human_feedback")
52+
.inferenceIdColumnName("id")
53+
.latencyColumnName("latency")
54+
.timestampColumnName("timestamp")
55+
.build()
56+
)
4757
.datasetId("my-dataset")
4858
.projectId("my-project")
4959
.tableId("my-table")
@@ -111,6 +121,7 @@ internal class InferencePipelineUpdateResponseTest {
111121
.projectCount(0L)
112122
.slug("openlayer")
113123
.status(InferencePipelineUpdateResponse.Workspace.Status.ACTIVE)
124+
.inviteCode("inviteCode")
114125
.addMonthlyUsage(
115126
InferencePipelineUpdateResponse.Workspace.MonthlyUsage.builder()
116127
.executionTimeMs(0L)
@@ -167,6 +178,16 @@ internal class InferencePipelineUpdateResponseTest {
167178
.BIGQUERY
168179
)
169180
.bigqueryConnectionId("182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e")
181+
.config(
182+
InferencePipelineUpdateResponse.DataBackend.BigQueryDataBackend.Config
183+
.builder()
184+
.groundTruthColumnName("ground_truth")
185+
.humanFeedbackColumnName("human_feedback")
186+
.inferenceIdColumnName("id")
187+
.latencyColumnName("latency")
188+
.timestampColumnName("timestamp")
189+
.build()
190+
)
170191
.datasetId("my-dataset")
171192
.projectId("my-project")
172193
.tableId("my-table")
@@ -238,6 +259,7 @@ internal class InferencePipelineUpdateResponseTest {
238259
.projectCount(0L)
239260
.slug("openlayer")
240261
.status(InferencePipelineUpdateResponse.Workspace.Status.ACTIVE)
262+
.inviteCode("inviteCode")
241263
.addMonthlyUsage(
242264
InferencePipelineUpdateResponse.Workspace.MonthlyUsage.builder()
243265
.executionTimeMs(0L)
@@ -287,6 +309,16 @@ internal class InferencePipelineUpdateResponseTest {
287309
.BIGQUERY
288310
)
289311
.bigqueryConnectionId("182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e")
312+
.config(
313+
InferencePipelineUpdateResponse.DataBackend.BigQueryDataBackend.Config
314+
.builder()
315+
.groundTruthColumnName("ground_truth")
316+
.humanFeedbackColumnName("human_feedback")
317+
.inferenceIdColumnName("id")
318+
.latencyColumnName("latency")
319+
.timestampColumnName("timestamp")
320+
.build()
321+
)
290322
.datasetId("my-dataset")
291323
.projectId("my-project")
292324
.tableId("my-table")
@@ -354,6 +386,7 @@ internal class InferencePipelineUpdateResponseTest {
354386
.projectCount(0L)
355387
.slug("openlayer")
356388
.status(InferencePipelineUpdateResponse.Workspace.Status.ACTIVE)
389+
.inviteCode("inviteCode")
357390
.addMonthlyUsage(
358391
InferencePipelineUpdateResponse.Workspace.MonthlyUsage.builder()
359392
.executionTimeMs(0L)

‎openlayer-java-core/src/test/kotlin/com/openlayer/api/models/inferencepipelines/rows/RowUpdateParamsTest.kt‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@ internal class RowUpdateParamsTest {
1919
RowUpdateParams.Config.builder()
2020
.groundTruthColumnName("ground_truth")
2121
.humanFeedbackColumnName("human_feedback")
22+
.inferenceIdColumnName("id")
2223
.latencyColumnName("latency")
2324
.timestampColumnName("timestamp")
2425
.build()
@@ -51,6 +52,7 @@ internal class RowUpdateParamsTest {
5152
RowUpdateParams.Config.builder()
5253
.groundTruthColumnName("ground_truth")
5354
.humanFeedbackColumnName("human_feedback")
55+
.inferenceIdColumnName("id")
5456
.latencyColumnName("latency")
5557
.timestampColumnName("timestamp")
5658
.build()
@@ -89,6 +91,7 @@ internal class RowUpdateParamsTest {
8991
RowUpdateParams.Config.builder()
9092
.groundTruthColumnName("ground_truth")
9193
.humanFeedbackColumnName("human_feedback")
94+
.inferenceIdColumnName("id")
9295
.latencyColumnName("latency")
9396
.timestampColumnName("timestamp")
9497
.build()
@@ -103,6 +106,7 @@ internal class RowUpdateParamsTest {
103106
RowUpdateParams.Config.builder()
104107
.groundTruthColumnName("ground_truth")
105108
.humanFeedbackColumnName("human_feedback")
109+
.inferenceIdColumnName("id")
106110
.latencyColumnName("latency")
107111
.timestampColumnName("timestamp")
108112
.build()

‎openlayer-java-core/src/test/kotlin/com/openlayer/api/models/projects/commits/CommitCreateResponseTest.kt‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,7 @@ internal class CommitCreateResponseTest {
3939
.projectId("3fa85f64-5717-4562-b3fc-2c963f66afa6")
4040
.status(CommitCreateResponse.Status.COMPLETED)
4141
.statusMessage("Commit successfully processed.")
42+
.storageUri("s3://...")
4243
.totalGoalCount(6L)
4344
.trainingDatasetId("3fa85f64-5717-4562-b3fc-2c963f66afa6")
4445
.validationDatasetId("3fa85f64-5717-4562-b3fc-2c963f66afa6")
@@ -83,6 +84,7 @@ internal class CommitCreateResponseTest {
8384
.isEqualTo("3fa85f64-5717-4562-b3fc-2c963f66afa6")
8485
assertThat(commitCreateResponse.status()).isEqualTo(CommitCreateResponse.Status.COMPLETED)
8586
assertThat(commitCreateResponse.statusMessage()).contains("Commit successfully processed.")
87+
assertThat(commitCreateResponse.storageUri()).isEqualTo("s3://...")
8688
assertThat(commitCreateResponse.totalGoalCount()).isEqualTo(6L)
8789
assertThat(commitCreateResponse.trainingDatasetId())
8890
.contains("3fa85f64-5717-4562-b3fc-2c963f66afa6")
@@ -130,6 +132,7 @@ internal class CommitCreateResponseTest {
130132
.projectId("3fa85f64-5717-4562-b3fc-2c963f66afa6")
131133
.status(CommitCreateResponse.Status.COMPLETED)
132134
.statusMessage("Commit successfully processed.")
135+
.storageUri("s3://...")
133136
.totalGoalCount(6L)
134137
.trainingDatasetId("3fa85f64-5717-4562-b3fc-2c963f66afa6")
135138
.validationDatasetId("3fa85f64-5717-4562-b3fc-2c963f66afa6")

‎openlayer-java-core/src/test/kotlin/com/openlayer/api/models/projects/commits/CommitListResponseTest.kt‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,7 @@ internal class CommitListResponseTest {
4141
.projectId("3fa85f64-5717-4562-b3fc-2c963f66afa6")
4242
.status(CommitListResponse.Item.Status.COMPLETED)
4343
.statusMessage("Commit successfully processed.")
44+
.storageUri("s3://...")
4445
.totalGoalCount(6L)
4546
.trainingDatasetId("3fa85f64-5717-4562-b3fc-2c963f66afa6")
4647
.validationDatasetId("3fa85f64-5717-4562-b3fc-2c963f66afa6")
@@ -85,6 +86,7 @@ internal class CommitListResponseTest {
8586
.projectId("3fa85f64-5717-4562-b3fc-2c963f66afa6")
8687
.status(CommitListResponse.Item.Status.COMPLETED)
8788
.statusMessage("Commit successfully processed.")
89+
.storageUri("s3://...")
8890
.totalGoalCount(6L)
8991
.trainingDatasetId("3fa85f64-5717-4562-b3fc-2c963f66afa6")
9092
.validationDatasetId("3fa85f64-5717-4562-b3fc-2c963f66afa6")
@@ -133,6 +135,7 @@ internal class CommitListResponseTest {
133135
.projectId("3fa85f64-5717-4562-b3fc-2c963f66afa6")
134136
.status(CommitListResponse.Item.Status.COMPLETED)
135137
.statusMessage("Commit successfully processed.")
138+
.storageUri("s3://...")
136139
.totalGoalCount(6L)
137140
.trainingDatasetId("3fa85f64-5717-4562-b3fc-2c963f66afa6")
138141
.validationDatasetId("3fa85f64-5717-4562-b3fc-2c963f66afa6")

‎openlayer-java-core/src/test/kotlin/com/openlayer/api/models/projects/inferencepipelines/InferencePipelineCreateParamsTest.kt‎

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,16 @@ internal class InferencePipelineCreateParamsTest {
4141
.BIGQUERY
4242
)
4343
.bigqueryConnectionId("182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e")
44+
.config(
45+
InferencePipelineCreateParams.DataBackend.BigQueryDataBackend.Config
46+
.builder()
47+
.groundTruthColumnName("ground_truth")
48+
.humanFeedbackColumnName("human_feedback")
49+
.inferenceIdColumnName("id")
50+
.latencyColumnName("latency")
51+
.timestampColumnName("timestamp")
52+
.build()
53+
)
4454
.datasetId("my-dataset")
4555
.bodyProjectId("my-project")
4656
.tableId("my-table")
@@ -107,6 +117,7 @@ internal class InferencePipelineCreateParamsTest {
107117
.projectCount(0L)
108118
.slug("openlayer")
109119
.status(InferencePipelineCreateParams.Workspace.Status.ACTIVE)
120+
.inviteCode("inviteCode")
110121
.addMonthlyUsage(
111122
InferencePipelineCreateParams.Workspace.MonthlyUsage.builder()
112123
.executionTimeMs(0L)
@@ -189,6 +200,16 @@ internal class InferencePipelineCreateParamsTest {
189200
.BIGQUERY
190201
)
191202
.bigqueryConnectionId("182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e")
203+
.config(
204+
InferencePipelineCreateParams.DataBackend.BigQueryDataBackend.Config
205+
.builder()
206+
.groundTruthColumnName("ground_truth")
207+
.humanFeedbackColumnName("human_feedback")
208+
.inferenceIdColumnName("id")
209+
.latencyColumnName("latency")
210+
.timestampColumnName("timestamp")
211+
.build()
212+
)
192213
.datasetId("my-dataset")
193214
.bodyProjectId("my-project")
194215
.tableId("my-table")
@@ -256,6 +277,7 @@ internal class InferencePipelineCreateParamsTest {
256277
.projectCount(0L)
257278
.slug("openlayer")
258279
.status(InferencePipelineCreateParams.Workspace.Status.ACTIVE)
280+
.inviteCode("inviteCode")
259281
.addMonthlyUsage(
260282
InferencePipelineCreateParams.Workspace.MonthlyUsage.builder()
261283
.executionTimeMs(0L)
@@ -307,6 +329,16 @@ internal class InferencePipelineCreateParamsTest {
307329
.BIGQUERY
308330
)
309331
.bigqueryConnectionId("182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e")
332+
.config(
333+
InferencePipelineCreateParams.DataBackend.BigQueryDataBackend.Config
334+
.builder()
335+
.groundTruthColumnName("ground_truth")
336+
.humanFeedbackColumnName("human_feedback")
337+
.inferenceIdColumnName("id")
338+
.latencyColumnName("latency")
339+
.timestampColumnName("timestamp")
340+
.build()
341+
)
310342
.datasetId("my-dataset")
311343
.bodyProjectId("my-project")
312344
.tableId("my-table")
@@ -377,6 +409,7 @@ internal class InferencePipelineCreateParamsTest {
377409
.projectCount(0L)
378410
.slug("openlayer")
379411
.status(InferencePipelineCreateParams.Workspace.Status.ACTIVE)
412+
.inviteCode("inviteCode")
380413
.addMonthlyUsage(
381414
InferencePipelineCreateParams.Workspace.MonthlyUsage.builder()
382415
.executionTimeMs(0L)

0 commit comments

Comments
 (0)