fix(internal): fix imports and tests

David Meadows · stainless-app[bot] · commit f5e58d1c7129 · 2024-10-01T16:46:03.000Z
diff --git a/braintrust-kotlin-core/src/main/kotlin/com/braintrustdata/api/models/FunctionCreateParams.kt b/braintrust-kotlin-core/src/main/kotlin/com/braintrustdata/api/models/FunctionCreateParams.kt
@@ -13,7 +13,7 @@ import com.braintrustdata.api.core.NoAutoDetect
 import com.braintrustdata.api.core.getOrThrow
 import com.braintrustdata.api.core.toUnmodifiable
 import com.braintrustdata.api.errors.BraintrustInvalidDataException
-import com.braintrustdata.api.models.*
+import com.braintrustdata.api.models.CodeBundle.*
 import com.fasterxml.jackson.annotation.JsonAnyGetter
 import com.fasterxml.jackson.annotation.JsonAnySetter
 import com.fasterxml.jackson.annotation.JsonCreator
diff --git a/braintrust-kotlin-core/src/main/kotlin/com/braintrustdata/api/models/FunctionReplaceParams.kt b/braintrust-kotlin-core/src/main/kotlin/com/braintrustdata/api/models/FunctionReplaceParams.kt
@@ -13,7 +13,7 @@ import com.braintrustdata.api.core.NoAutoDetect
 import com.braintrustdata.api.core.getOrThrow
 import com.braintrustdata.api.core.toUnmodifiable
 import com.braintrustdata.api.errors.BraintrustInvalidDataException
-import com.braintrustdata.api.models.*
+import com.braintrustdata.api.models.CodeBundle.*
 import com.fasterxml.jackson.annotation.JsonAnyGetter
 import com.fasterxml.jackson.annotation.JsonAnySetter
 import com.fasterxml.jackson.annotation.JsonCreator
diff --git a/braintrust-kotlin-core/src/main/kotlin/com/braintrustdata/api/models/FunctionUpdateParams.kt b/braintrust-kotlin-core/src/main/kotlin/com/braintrustdata/api/models/FunctionUpdateParams.kt
@@ -13,7 +13,7 @@ import com.braintrustdata.api.core.NoAutoDetect
 import com.braintrustdata.api.core.getOrThrow
 import com.braintrustdata.api.core.toUnmodifiable
 import com.braintrustdata.api.errors.BraintrustInvalidDataException
-import com.braintrustdata.api.models.*
+import com.braintrustdata.api.models.CodeBundle.*
 import com.fasterxml.jackson.annotation.JsonAnyGetter
 import com.fasterxml.jackson.annotation.JsonAnySetter
 import com.fasterxml.jackson.annotation.JsonCreator
diff --git a/braintrust-kotlin-core/src/main/kotlin/com/braintrustdata/api/models/ProjectScoreCreateParams.kt b/braintrust-kotlin-core/src/main/kotlin/com/braintrustdata/api/models/ProjectScoreCreateParams.kt
@@ -10,7 +10,6 @@ import com.braintrustdata.api.core.NoAutoDetect
 import com.braintrustdata.api.core.getOrThrow
 import com.braintrustdata.api.core.toUnmodifiable
 import com.braintrustdata.api.errors.BraintrustInvalidDataException
-import com.braintrustdata.api.models.*
 import com.fasterxml.jackson.annotation.JsonAnyGetter
 import com.fasterxml.jackson.annotation.JsonAnySetter
 import com.fasterxml.jackson.annotation.JsonProperty
@@ -294,7 +293,7 @@ constructor(
         fun categories(categories: Categories) = apply { this.categories = categories }
 
         /** For categorical-type project scores, the list of all categories */
-        fun categories(projectScoreCategories: List<ProjectScoreCategory>) = apply {
+        fun projectScoreCategories(projectScoreCategories: List<ProjectScoreCategory>) = apply {
             this.categories = Categories.ofProjectScoreCategories(projectScoreCategories)
         }
 
diff --git a/braintrust-kotlin-core/src/main/kotlin/com/braintrustdata/api/models/ProjectScoreReplaceParams.kt b/braintrust-kotlin-core/src/main/kotlin/com/braintrustdata/api/models/ProjectScoreReplaceParams.kt
@@ -10,7 +10,6 @@ import com.braintrustdata.api.core.NoAutoDetect
 import com.braintrustdata.api.core.getOrThrow
 import com.braintrustdata.api.core.toUnmodifiable
 import com.braintrustdata.api.errors.BraintrustInvalidDataException
-import com.braintrustdata.api.models.*
 import com.fasterxml.jackson.annotation.JsonAnyGetter
 import com.fasterxml.jackson.annotation.JsonAnySetter
 import com.fasterxml.jackson.annotation.JsonProperty
@@ -294,7 +293,7 @@ constructor(
         fun categories(categories: Categories) = apply { this.categories = categories }
 
         /** For categorical-type project scores, the list of all categories */
-        fun categories(projectScoreCategories: List<ProjectScoreCategory>) = apply {
+        fun projectScoreCategories(projectScoreCategories: List<ProjectScoreCategory>) = apply {
             this.categories = Categories.ofProjectScoreCategories(projectScoreCategories)
         }
 
diff --git a/braintrust-kotlin-core/src/main/kotlin/com/braintrustdata/api/models/ProjectScoreUpdateParams.kt b/braintrust-kotlin-core/src/main/kotlin/com/braintrustdata/api/models/ProjectScoreUpdateParams.kt
@@ -10,7 +10,6 @@ import com.braintrustdata.api.core.NoAutoDetect
 import com.braintrustdata.api.core.getOrThrow
 import com.braintrustdata.api.core.toUnmodifiable
 import com.braintrustdata.api.errors.BraintrustInvalidDataException
-import com.braintrustdata.api.models.*
 import com.fasterxml.jackson.annotation.JsonAnyGetter
 import com.fasterxml.jackson.annotation.JsonAnySetter
 import com.fasterxml.jackson.annotation.JsonProperty
@@ -281,7 +280,7 @@ constructor(
         fun categories(categories: Categories) = apply { this.categories = categories }
 
         /** For categorical-type project scores, the list of all categories */
-        fun categories(projectScoreCategories: List<ProjectScoreCategory>) = apply {
+        fun projectScoreCategories(projectScoreCategories: List<ProjectScoreCategory>) = apply {
             this.categories = Categories.ofProjectScoreCategories(projectScoreCategories)
         }
 
diff --git a/braintrust-kotlin-core/src/main/kotlin/com/braintrustdata/api/models/PromptData.kt b/braintrust-kotlin-core/src/main/kotlin/com/braintrustdata/api/models/PromptData.kt
@@ -932,7 +932,7 @@ private constructor(
                             tryDeserialize(node, jacksonTypeRef<Auto>())?.let {
                                 return FunctionCall(auto = it, _json = json)
                             }
-                            tryDeserialize(node, jacksonTypeRef<None>())?.let {
+                            tryDeserialize(node, jacksonTypeRef<FunctionCall.None>())?.let {
                                 return FunctionCall(none = it, _json = json)
                             }
                             tryDeserialize(node, jacksonTypeRef<Function>()) { it.validate() }
diff --git a/braintrust-kotlin-core/src/main/kotlin/com/braintrustdata/api/models/ToolChoice.kt b/braintrust-kotlin-core/src/main/kotlin/com/braintrustdata/api/models/ToolChoice.kt
@@ -135,7 +135,7 @@ private constructor(
             tryDeserialize(node, jacksonTypeRef<Auto>())?.let {
                 return ToolChoice(auto = it, _json = json)
             }
-            tryDeserialize(node, jacksonTypeRef<None>())?.let {
+            tryDeserialize(node, jacksonTypeRef<ToolChoice.None>())?.let {
                 return ToolChoice(none = it, _json = json)
             }
             tryDeserialize(node, jacksonTypeRef<Function>()) { it.validate() }
diff --git a/braintrust-kotlin-core/src/test/kotlin/com/braintrustdata/api/models/ProjectScoreCreateParamsTest.kt b/braintrust-kotlin-core/src/test/kotlin/com/braintrustdata/api/models/ProjectScoreCreateParamsTest.kt
@@ -14,10 +14,8 @@ class ProjectScoreCreateParamsTest {
             .name("name")
             .projectId("182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e")
             .scoreType(ProjectScoreType.SLIDER)
-            .categories(
-                ProjectScoreCreateParams.Categories.ofList<ProjectScoreCategory>(
-                    listOf(ProjectScoreCategory.builder().name("name").value(42.23).build())
-                )
+            .projectScoreCategories(
+                listOf(ProjectScoreCategory.builder().name("name").value(42.23).build())
             )
             .config(
                 ProjectScoreConfig.builder()
@@ -53,10 +51,8 @@ class ProjectScoreCreateParamsTest {
                 .name("name")
                 .projectId("182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e")
                 .scoreType(ProjectScoreType.SLIDER)
-                .categories(
-                    ProjectScoreCreateParams.Categories.ofList<ProjectScoreCategory>(
-                        listOf(ProjectScoreCategory.builder().name("name").value(42.23).build())
-                    )
+                .projectScoreCategories(
+                    listOf(ProjectScoreCategory.builder().name("name").value(42.23).build())
                 )
                 .config(
                     ProjectScoreConfig.builder()
@@ -92,7 +88,7 @@ class ProjectScoreCreateParamsTest {
         assertThat(body.scoreType()).isEqualTo(ProjectScoreType.SLIDER)
         assertThat(body.categories())
             .isEqualTo(
-                ProjectScoreCreateParams.Categories.ofList<ProjectScoreCategory>(
+                ProjectScoreCreateParams.Categories.ofProjectScoreCategories(
                     listOf(ProjectScoreCategory.builder().name("name").value(42.23).build())
                 )
             )
diff --git a/braintrust-kotlin-core/src/test/kotlin/com/braintrustdata/api/models/ProjectScoreListParamsTest.kt b/braintrust-kotlin-core/src/test/kotlin/com/braintrustdata/api/models/ProjectScoreListParamsTest.kt
@@ -56,9 +56,7 @@ class ProjectScoreListParamsTest {
         expected.put(
             "score_type",
             listOf(
-                ProjectScoreListParams.ScoreType.ofProjectScoreType(
-                        ProjectScoreType.SLIDER.toString()
-                    )
+                ProjectScoreListParams.ScoreType.ofProjectScoreType(ProjectScoreType.SLIDER)
                     .toString()
             )
         )
diff --git a/braintrust-kotlin-core/src/test/kotlin/com/braintrustdata/api/models/ProjectScoreReplaceParamsTest.kt b/braintrust-kotlin-core/src/test/kotlin/com/braintrustdata/api/models/ProjectScoreReplaceParamsTest.kt
@@ -14,10 +14,8 @@ class ProjectScoreReplaceParamsTest {
             .name("name")
             .projectId("182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e")
             .scoreType(ProjectScoreType.SLIDER)
-            .categories(
-                ProjectScoreReplaceParams.Categories.ofList<ProjectScoreCategory>(
-                    listOf(ProjectScoreCategory.builder().name("name").value(42.23).build())
-                )
+            .projectScoreCategories(
+                listOf(ProjectScoreCategory.builder().name("name").value(42.23).build())
             )
             .config(
                 ProjectScoreConfig.builder()
@@ -53,10 +51,8 @@ class ProjectScoreReplaceParamsTest {
                 .name("name")
                 .projectId("182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e")
                 .scoreType(ProjectScoreType.SLIDER)
-                .categories(
-                    ProjectScoreReplaceParams.Categories.ofList<ProjectScoreCategory>(
-                        listOf(ProjectScoreCategory.builder().name("name").value(42.23).build())
-                    )
+                .projectScoreCategories(
+                    listOf(ProjectScoreCategory.builder().name("name").value(42.23).build())
                 )
                 .config(
                     ProjectScoreConfig.builder()
@@ -92,7 +88,7 @@ class ProjectScoreReplaceParamsTest {
         assertThat(body.scoreType()).isEqualTo(ProjectScoreType.SLIDER)
         assertThat(body.categories())
             .isEqualTo(
-                ProjectScoreReplaceParams.Categories.ofList<ProjectScoreCategory>(
+                ProjectScoreReplaceParams.Categories.ofProjectScoreCategories(
                     listOf(ProjectScoreCategory.builder().name("name").value(42.23).build())
                 )
             )
diff --git a/braintrust-kotlin-core/src/test/kotlin/com/braintrustdata/api/models/ProjectScoreTest.kt b/braintrust-kotlin-core/src/test/kotlin/com/braintrustdata/api/models/ProjectScoreTest.kt
@@ -18,7 +18,7 @@ class ProjectScoreTest {
                 .scoreType(ProjectScoreType.SLIDER)
                 .userId("182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e")
                 .categories(
-                    ProjectScore.Categories.ofList<ProjectScoreCategory>(
+                    ProjectScore.Categories.ofProjectScoreCategories(
                         listOf(ProjectScoreCategory.builder().name("name").value(42.23).build())
                     )
                 )
@@ -59,7 +59,7 @@ class ProjectScoreTest {
         assertThat(projectScore.userId()).isEqualTo("182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e")
         assertThat(projectScore.categories())
             .isEqualTo(
-                ProjectScore.Categories.ofList<ProjectScoreCategory>(
+                ProjectScore.Categories.ofProjectScoreCategories(
                     listOf(ProjectScoreCategory.builder().name("name").value(42.23).build())
                 )
             )
diff --git a/braintrust-kotlin-core/src/test/kotlin/com/braintrustdata/api/models/ProjectScoreUpdateParamsTest.kt b/braintrust-kotlin-core/src/test/kotlin/com/braintrustdata/api/models/ProjectScoreUpdateParamsTest.kt
@@ -13,7 +13,7 @@ class ProjectScoreUpdateParamsTest {
         ProjectScoreUpdateParams.builder()
             .projectScoreId("182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e")
             .categories(
-                ProjectScoreUpdateParams.Categories.ofList<ProjectScoreCategory>(
+                ProjectScoreUpdateParams.Categories.ofProjectScoreCategories(
                     listOf(ProjectScoreCategory.builder().name("name").value(42.23).build())
                 )
             )
@@ -52,7 +52,7 @@ class ProjectScoreUpdateParamsTest {
             ProjectScoreUpdateParams.builder()
                 .projectScoreId("182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e")
                 .categories(
-                    ProjectScoreUpdateParams.Categories.ofList<ProjectScoreCategory>(
+                    ProjectScoreUpdateParams.Categories.ofProjectScoreCategories(
                         listOf(ProjectScoreCategory.builder().name("name").value(42.23).build())
                     )
                 )
@@ -89,7 +89,7 @@ class ProjectScoreUpdateParamsTest {
         assertThat(body).isNotNull
         assertThat(body.categories())
             .isEqualTo(
-                ProjectScoreUpdateParams.Categories.ofList<ProjectScoreCategory>(
+                ProjectScoreUpdateParams.Categories.ofProjectScoreCategories(
                     listOf(ProjectScoreCategory.builder().name("name").value(42.23).build())
                 )
             )
diff --git a/braintrust-kotlin-core/src/test/kotlin/com/braintrustdata/api/services/blocking/EnvVarServiceTest.kt b/braintrust-kotlin-core/src/test/kotlin/com/braintrustdata/api/services/blocking/EnvVarServiceTest.kt
@@ -82,7 +82,6 @@ class EnvVarServiceTest {
             envVarService.list(
                 EnvVarListParams.builder()
                     .envVarName("env_var_name")
-                    .ids(EnvVarListParams.Ids.ofString("182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e"))
                     .limit(123L)
                     .objectId("182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e")
                     .objectType(EnvVarListParams.ObjectType.ORGANIZATION)
diff --git a/braintrust-kotlin-core/src/test/kotlin/com/braintrustdata/api/services/blocking/ProjectScoreServiceTest.kt b/braintrust-kotlin-core/src/test/kotlin/com/braintrustdata/api/services/blocking/ProjectScoreServiceTest.kt
@@ -27,7 +27,7 @@ class ProjectScoreServiceTest {
                     .projectId("182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e")
                     .scoreType(ProjectScoreType.SLIDER)
                     .categories(
-                        ProjectScoreCreateParams.Categories.ofList<ProjectScoreCategory>(
+                        ProjectScoreCreateParams.Categories.ofProjectScoreCategories(
                             listOf(ProjectScoreCategory.builder().name("name").value(42.23).build())
                         )
                     )
@@ -95,7 +95,7 @@ class ProjectScoreServiceTest {
                 ProjectScoreUpdateParams.builder()
                     .projectScoreId("182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e")
                     .categories(
-                        ProjectScoreUpdateParams.Categories.ofList<ProjectScoreCategory>(
+                        ProjectScoreUpdateParams.Categories.ofProjectScoreCategories(
                             listOf(ProjectScoreCategory.builder().name("name").value(42.23).build())
                         )
                     )
@@ -180,7 +180,7 @@ class ProjectScoreServiceTest {
                     .projectId("182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e")
                     .scoreType(ProjectScoreType.SLIDER)
                     .categories(
-                        ProjectScoreReplaceParams.Categories.ofList<ProjectScoreCategory>(
+                        ProjectScoreReplaceParams.Categories.ofProjectScoreCategories(
                             listOf(ProjectScoreCategory.builder().name("name").value(42.23).build())
                         )
                     )

Original file line number	Diff line number	Diff line change
`@@ -932,7 +932,7 @@ private constructor(`
`932`	`932`	`tryDeserialize(node, jacksonTypeRef<Auto>())?.let {`
`933`	`933`	`return FunctionCall(auto = it, _json = json)`
`934`	`934`	`}`
`935`		`- tryDeserialize(node, jacksonTypeRef<None>())?.let {`
	`935`	`+ tryDeserialize(node, jacksonTypeRef<FunctionCall.None>())?.let {`
`936`	`936`	`return FunctionCall(none = it, _json = json)`
`937`	`937`	`}`
`938`	`938`	`tryDeserialize(node, jacksonTypeRef<Function>()) { it.validate() }`
Original file line number	Diff line number	Diff line change
`@@ -135,7 +135,7 @@ private constructor(`
`135`	`135`	`tryDeserialize(node, jacksonTypeRef<Auto>())?.let {`
`136`	`136`	`return ToolChoice(auto = it, _json = json)`
`137`	`137`	`}`
`138`		`- tryDeserialize(node, jacksonTypeRef<None>())?.let {`
	`138`	`+ tryDeserialize(node, jacksonTypeRef<ToolChoice.None>())?.let {`
`139`	`139`	`return ToolChoice(none = it, _json = json)`
`140`	`140`	`}`
`141`	`141`	`tryDeserialize(node, jacksonTypeRef<Function>()) { it.validate() }`
Original file line number	Diff line number	Diff line change
`@@ -56,9 +56,7 @@ class ProjectScoreListParamsTest {`
`56`	`56`	`expected.put(`
`57`	`57`	`"score_type",`
`58`	`58`	`listOf(`
`59`		`- ProjectScoreListParams.ScoreType.ofProjectScoreType(`
`60`		`- ProjectScoreType.SLIDER.toString()`
`61`		`- )`
	`59`	`+ ProjectScoreListParams.ScoreType.ofProjectScoreType(ProjectScoreType.SLIDER)`
`62`	`60`	`.toString()`
`63`	`61`	`)`
`64`	`62`	`)`