Skip to content

Commit 4fd6c23

Browse files
stainless-botStainless Bot
authored andcommitted
refactor(client)!: rename header and query params builder methods
chore: unknown commit message
1 parent ed455e3 commit 4fd6c23

24 files changed

+1028
-860
lines changed

openai-java-core/src/main/kotlin/com/openai/models/BatchCancelParams.kt

Lines changed: 43 additions & 36 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,8 @@
22

33
package com.openai.models
44

5+
import com.google.common.collect.ArrayListMultimap
6+
import com.google.common.collect.ListMultimap
57
import com.openai.core.JsonValue
68
import com.openai.core.NoAutoDetect
79
import com.openai.core.toImmutable
@@ -12,8 +14,8 @@ import java.util.Optional
1214
class BatchCancelParams
1315
constructor(
1416
private val batchId: String,
15-
private val additionalQueryParams: Map<String, List<String>>,
1617
private val additionalHeaders: Map<String, List<String>>,
18+
private val additionalQueryParams: Map<String, List<String>>,
1719
private val additionalBodyProperties: Map<String, JsonValue>,
1820
) {
1921

@@ -24,37 +26,37 @@ constructor(
2426
return Optional.ofNullable(additionalBodyProperties.ifEmpty { null })
2527
}
2628

27-
@JvmSynthetic internal fun getQueryParams(): Map<String, List<String>> = additionalQueryParams
28-
2929
@JvmSynthetic internal fun getHeaders(): Map<String, List<String>> = additionalHeaders
3030

31+
@JvmSynthetic internal fun getQueryParams(): Map<String, List<String>> = additionalQueryParams
32+
3133
fun getPathParam(index: Int): String {
3234
return when (index) {
3335
0 -> batchId
3436
else -> ""
3537
}
3638
}
3739

38-
fun _additionalQueryParams(): Map<String, List<String>> = additionalQueryParams
39-
4040
fun _additionalHeaders(): Map<String, List<String>> = additionalHeaders
4141

42+
fun _additionalQueryParams(): Map<String, List<String>> = additionalQueryParams
43+
4244
fun _additionalBodyProperties(): Map<String, JsonValue> = additionalBodyProperties
4345

4446
override fun equals(other: Any?): Boolean {
4547
if (this === other) {
4648
return true
4749
}
4850

49-
return /* spotless:off */ other is BatchCancelParams && this.batchId == other.batchId && this.additionalQueryParams == other.additionalQueryParams && this.additionalHeaders == other.additionalHeaders && this.additionalBodyProperties == other.additionalBodyProperties /* spotless:on */
51+
return /* spotless:off */ other is BatchCancelParams && this.batchId == other.batchId && this.additionalHeaders == other.additionalHeaders && this.additionalQueryParams == other.additionalQueryParams && this.additionalBodyProperties == other.additionalBodyProperties /* spotless:on */
5052
}
5153

5254
override fun hashCode(): Int {
53-
return /* spotless:off */ Objects.hash(batchId, additionalQueryParams, additionalHeaders, additionalBodyProperties) /* spotless:on */
55+
return /* spotless:off */ Objects.hash(batchId, additionalHeaders, additionalQueryParams, additionalBodyProperties) /* spotless:on */
5456
}
5557

5658
override fun toString() =
57-
"BatchCancelParams{batchId=$batchId, additionalQueryParams=$additionalQueryParams, additionalHeaders=$additionalHeaders, additionalBodyProperties=$additionalBodyProperties}"
59+
"BatchCancelParams{batchId=$batchId, additionalHeaders=$additionalHeaders, additionalQueryParams=$additionalQueryParams, additionalBodyProperties=$additionalBodyProperties}"
5860

5961
fun toBuilder() = Builder().from(this)
6062

@@ -67,59 +69,58 @@ constructor(
6769
class Builder {
6870

6971
private var batchId: String? = null
70-
private var additionalQueryParams: MutableMap<String, MutableList<String>> = mutableMapOf()
71-
private var additionalHeaders: MutableMap<String, MutableList<String>> = mutableMapOf()
72+
private var additionalHeaders: ListMultimap<String, String> = ArrayListMultimap.create()
73+
private var additionalQueryParams: ListMultimap<String, String> = ArrayListMultimap.create()
7274
private var additionalBodyProperties: MutableMap<String, JsonValue> = mutableMapOf()
7375

7476
@JvmSynthetic
7577
internal fun from(batchCancelParams: BatchCancelParams) = apply {
7678
this.batchId = batchCancelParams.batchId
77-
additionalQueryParams(batchCancelParams.additionalQueryParams)
7879
additionalHeaders(batchCancelParams.additionalHeaders)
80+
additionalQueryParams(batchCancelParams.additionalQueryParams)
7981
additionalBodyProperties(batchCancelParams.additionalBodyProperties)
8082
}
8183

8284
fun batchId(batchId: String) = apply { this.batchId = batchId }
8385

84-
fun additionalQueryParams(additionalQueryParams: Map<String, List<String>>) = apply {
85-
this.additionalQueryParams.clear()
86-
putAllQueryParams(additionalQueryParams)
86+
fun additionalHeaders(additionalHeaders: Map<String, Iterable<String>>) = apply {
87+
this.additionalHeaders.clear()
88+
putAllAdditionalHeaders(additionalHeaders)
8789
}
8890

89-
fun putQueryParam(name: String, value: String) = apply {
90-
this.additionalQueryParams.getOrPut(name) { mutableListOf() }.add(value)
91+
fun putAdditionalHeader(name: String, value: String) = apply {
92+
additionalHeaders.put(name, value)
9193
}
9294

93-
fun putQueryParams(name: String, values: Iterable<String>) = apply {
94-
this.additionalQueryParams.getOrPut(name) { mutableListOf() }.addAll(values)
95+
fun putAdditionalHeaders(name: String, values: Iterable<String>) = apply {
96+
additionalHeaders.putAll(name, values)
9597
}
9698

97-
fun putAllQueryParams(additionalQueryParams: Map<String, Iterable<String>>) = apply {
98-
additionalQueryParams.forEach(this::putQueryParams)
99+
fun putAllAdditionalHeaders(additionalHeaders: Map<String, Iterable<String>>) = apply {
100+
additionalHeaders.forEach(::putAdditionalHeaders)
99101
}
100102

101-
fun removeQueryParam(name: String) = apply {
102-
this.additionalQueryParams.put(name, mutableListOf())
103-
}
103+
fun removeAdditionalHeader(name: String) = apply { additionalHeaders.removeAll(name) }
104104

105-
fun additionalHeaders(additionalHeaders: Map<String, Iterable<String>>) = apply {
106-
this.additionalHeaders.clear()
107-
putAllHeaders(additionalHeaders)
105+
fun additionalQueryParams(additionalQueryParams: Map<String, Iterable<String>>) = apply {
106+
this.additionalQueryParams.clear()
107+
putAllAdditionalQueryParams(additionalQueryParams)
108108
}
109109

110-
fun putHeader(name: String, value: String) = apply {
111-
this.additionalHeaders.getOrPut(name) { mutableListOf() }.add(value)
110+
fun putAdditionalQueryParam(key: String, value: String) = apply {
111+
additionalQueryParams.put(key, value)
112112
}
113113

114-
fun putHeaders(name: String, values: Iterable<String>) = apply {
115-
this.additionalHeaders.getOrPut(name) { mutableListOf() }.addAll(values)
114+
fun putAdditionalQueryParams(key: String, values: Iterable<String>) = apply {
115+
additionalQueryParams.putAll(key, values)
116116
}
117117

118-
fun putAllHeaders(additionalHeaders: Map<String, Iterable<String>>) = apply {
119-
additionalHeaders.forEach(this::putHeaders)
120-
}
118+
fun putAllAdditionalQueryParams(additionalQueryParams: Map<String, Iterable<String>>) =
119+
apply {
120+
additionalQueryParams.forEach(::putAdditionalQueryParams)
121+
}
121122

122-
fun removeHeader(name: String) = apply { this.additionalHeaders.put(name, mutableListOf()) }
123+
fun removeAdditionalQueryParam(key: String) = apply { additionalQueryParams.removeAll(key) }
123124

124125
fun additionalBodyProperties(additionalBodyProperties: Map<String, JsonValue>) = apply {
125126
this.additionalBodyProperties.clear()
@@ -138,8 +139,14 @@ constructor(
138139
fun build(): BatchCancelParams =
139140
BatchCancelParams(
140141
checkNotNull(batchId) { "`batchId` is required but was not set" },
141-
additionalQueryParams.mapValues { it.value.toImmutable() }.toImmutable(),
142-
additionalHeaders.mapValues { it.value.toImmutable() }.toImmutable(),
142+
additionalHeaders
143+
.asMap()
144+
.mapValues { it.value.toList().toImmutable() }
145+
.toImmutable(),
146+
additionalQueryParams
147+
.asMap()
148+
.mapValues { it.value.toList().toImmutable() }
149+
.toImmutable(),
143150
additionalBodyProperties.toImmutable(),
144151
)
145152
}

openai-java-core/src/main/kotlin/com/openai/models/BatchCreateParams.kt

Lines changed: 43 additions & 36 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,8 @@ import com.fasterxml.jackson.annotation.JsonAnySetter
77
import com.fasterxml.jackson.annotation.JsonCreator
88
import com.fasterxml.jackson.annotation.JsonProperty
99
import com.fasterxml.jackson.databind.annotation.JsonDeserialize
10+
import com.google.common.collect.ArrayListMultimap
11+
import com.google.common.collect.ListMultimap
1012
import com.openai.core.Enum
1113
import com.openai.core.ExcludeMissing
1214
import com.openai.core.JsonField
@@ -24,8 +26,8 @@ constructor(
2426
private val endpoint: Endpoint,
2527
private val inputFileId: String,
2628
private val metadata: Metadata?,
27-
private val additionalQueryParams: Map<String, List<String>>,
2829
private val additionalHeaders: Map<String, List<String>>,
30+
private val additionalQueryParams: Map<String, List<String>>,
2931
private val additionalBodyProperties: Map<String, JsonValue>,
3032
) {
3133

@@ -48,10 +50,10 @@ constructor(
4850
)
4951
}
5052

51-
@JvmSynthetic internal fun getQueryParams(): Map<String, List<String>> = additionalQueryParams
52-
5353
@JvmSynthetic internal fun getHeaders(): Map<String, List<String>> = additionalHeaders
5454

55+
@JvmSynthetic internal fun getQueryParams(): Map<String, List<String>> = additionalQueryParams
56+
5557
@JsonDeserialize(builder = BatchCreateBody.Builder::class)
5658
@NoAutoDetect
5759
class BatchCreateBody
@@ -205,26 +207,26 @@ constructor(
205207
"BatchCreateBody{completionWindow=$completionWindow, endpoint=$endpoint, inputFileId=$inputFileId, metadata=$metadata, additionalProperties=$additionalProperties}"
206208
}
207209

208-
fun _additionalQueryParams(): Map<String, List<String>> = additionalQueryParams
209-
210210
fun _additionalHeaders(): Map<String, List<String>> = additionalHeaders
211211

212+
fun _additionalQueryParams(): Map<String, List<String>> = additionalQueryParams
213+
212214
fun _additionalBodyProperties(): Map<String, JsonValue> = additionalBodyProperties
213215

214216
override fun equals(other: Any?): Boolean {
215217
if (this === other) {
216218
return true
217219
}
218220

219-
return /* spotless:off */ other is BatchCreateParams && this.completionWindow == other.completionWindow && this.endpoint == other.endpoint && this.inputFileId == other.inputFileId && this.metadata == other.metadata && this.additionalQueryParams == other.additionalQueryParams && this.additionalHeaders == other.additionalHeaders && this.additionalBodyProperties == other.additionalBodyProperties /* spotless:on */
221+
return /* spotless:off */ other is BatchCreateParams && this.completionWindow == other.completionWindow && this.endpoint == other.endpoint && this.inputFileId == other.inputFileId && this.metadata == other.metadata && this.additionalHeaders == other.additionalHeaders && this.additionalQueryParams == other.additionalQueryParams && this.additionalBodyProperties == other.additionalBodyProperties /* spotless:on */
220222
}
221223

222224
override fun hashCode(): Int {
223-
return /* spotless:off */ Objects.hash(completionWindow, endpoint, inputFileId, metadata, additionalQueryParams, additionalHeaders, additionalBodyProperties) /* spotless:on */
225+
return /* spotless:off */ Objects.hash(completionWindow, endpoint, inputFileId, metadata, additionalHeaders, additionalQueryParams, additionalBodyProperties) /* spotless:on */
224226
}
225227

226228
override fun toString() =
227-
"BatchCreateParams{completionWindow=$completionWindow, endpoint=$endpoint, inputFileId=$inputFileId, metadata=$metadata, additionalQueryParams=$additionalQueryParams, additionalHeaders=$additionalHeaders, additionalBodyProperties=$additionalBodyProperties}"
229+
"BatchCreateParams{completionWindow=$completionWindow, endpoint=$endpoint, inputFileId=$inputFileId, metadata=$metadata, additionalHeaders=$additionalHeaders, additionalQueryParams=$additionalQueryParams, additionalBodyProperties=$additionalBodyProperties}"
228230

229231
fun toBuilder() = Builder().from(this)
230232

@@ -240,8 +242,8 @@ constructor(
240242
private var endpoint: Endpoint? = null
241243
private var inputFileId: String? = null
242244
private var metadata: Metadata? = null
243-
private var additionalQueryParams: MutableMap<String, MutableList<String>> = mutableMapOf()
244-
private var additionalHeaders: MutableMap<String, MutableList<String>> = mutableMapOf()
245+
private var additionalHeaders: ListMultimap<String, String> = ArrayListMultimap.create()
246+
private var additionalQueryParams: ListMultimap<String, String> = ArrayListMultimap.create()
245247
private var additionalBodyProperties: MutableMap<String, JsonValue> = mutableMapOf()
246248

247249
@JvmSynthetic
@@ -250,8 +252,8 @@ constructor(
250252
this.endpoint = batchCreateParams.endpoint
251253
this.inputFileId = batchCreateParams.inputFileId
252254
this.metadata = batchCreateParams.metadata
253-
additionalQueryParams(batchCreateParams.additionalQueryParams)
254255
additionalHeaders(batchCreateParams.additionalHeaders)
256+
additionalQueryParams(batchCreateParams.additionalQueryParams)
255257
additionalBodyProperties(batchCreateParams.additionalBodyProperties)
256258
}
257259

@@ -287,45 +289,44 @@ constructor(
287289
/** Optional custom metadata for the batch. */
288290
fun metadata(metadata: Metadata) = apply { this.metadata = metadata }
289291

290-
fun additionalQueryParams(additionalQueryParams: Map<String, List<String>>) = apply {
291-
this.additionalQueryParams.clear()
292-
putAllQueryParams(additionalQueryParams)
292+
fun additionalHeaders(additionalHeaders: Map<String, Iterable<String>>) = apply {
293+
this.additionalHeaders.clear()
294+
putAllAdditionalHeaders(additionalHeaders)
293295
}
294296

295-
fun putQueryParam(name: String, value: String) = apply {
296-
this.additionalQueryParams.getOrPut(name) { mutableListOf() }.add(value)
297+
fun putAdditionalHeader(name: String, value: String) = apply {
298+
additionalHeaders.put(name, value)
297299
}
298300

299-
fun putQueryParams(name: String, values: Iterable<String>) = apply {
300-
this.additionalQueryParams.getOrPut(name) { mutableListOf() }.addAll(values)
301+
fun putAdditionalHeaders(name: String, values: Iterable<String>) = apply {
302+
additionalHeaders.putAll(name, values)
301303
}
302304

303-
fun putAllQueryParams(additionalQueryParams: Map<String, Iterable<String>>) = apply {
304-
additionalQueryParams.forEach(this::putQueryParams)
305+
fun putAllAdditionalHeaders(additionalHeaders: Map<String, Iterable<String>>) = apply {
306+
additionalHeaders.forEach(::putAdditionalHeaders)
305307
}
306308

307-
fun removeQueryParam(name: String) = apply {
308-
this.additionalQueryParams.put(name, mutableListOf())
309-
}
309+
fun removeAdditionalHeader(name: String) = apply { additionalHeaders.removeAll(name) }
310310

311-
fun additionalHeaders(additionalHeaders: Map<String, Iterable<String>>) = apply {
312-
this.additionalHeaders.clear()
313-
putAllHeaders(additionalHeaders)
311+
fun additionalQueryParams(additionalQueryParams: Map<String, Iterable<String>>) = apply {
312+
this.additionalQueryParams.clear()
313+
putAllAdditionalQueryParams(additionalQueryParams)
314314
}
315315

316-
fun putHeader(name: String, value: String) = apply {
317-
this.additionalHeaders.getOrPut(name) { mutableListOf() }.add(value)
316+
fun putAdditionalQueryParam(key: String, value: String) = apply {
317+
additionalQueryParams.put(key, value)
318318
}
319319

320-
fun putHeaders(name: String, values: Iterable<String>) = apply {
321-
this.additionalHeaders.getOrPut(name) { mutableListOf() }.addAll(values)
320+
fun putAdditionalQueryParams(key: String, values: Iterable<String>) = apply {
321+
additionalQueryParams.putAll(key, values)
322322
}
323323

324-
fun putAllHeaders(additionalHeaders: Map<String, Iterable<String>>) = apply {
325-
additionalHeaders.forEach(this::putHeaders)
326-
}
324+
fun putAllAdditionalQueryParams(additionalQueryParams: Map<String, Iterable<String>>) =
325+
apply {
326+
additionalQueryParams.forEach(::putAdditionalQueryParams)
327+
}
327328

328-
fun removeHeader(name: String) = apply { this.additionalHeaders.put(name, mutableListOf()) }
329+
fun removeAdditionalQueryParam(key: String) = apply { additionalQueryParams.removeAll(key) }
329330

330331
fun additionalBodyProperties(additionalBodyProperties: Map<String, JsonValue>) = apply {
331332
this.additionalBodyProperties.clear()
@@ -347,8 +348,14 @@ constructor(
347348
checkNotNull(endpoint) { "`endpoint` is required but was not set" },
348349
checkNotNull(inputFileId) { "`inputFileId` is required but was not set" },
349350
metadata,
350-
additionalQueryParams.mapValues { it.value.toImmutable() }.toImmutable(),
351-
additionalHeaders.mapValues { it.value.toImmutable() }.toImmutable(),
351+
additionalHeaders
352+
.asMap()
353+
.mapValues { it.value.toList().toImmutable() }
354+
.toImmutable(),
355+
additionalQueryParams
356+
.asMap()
357+
.mapValues { it.value.toList().toImmutable() }
358+
.toImmutable(),
352359
additionalBodyProperties.toImmutable(),
353360
)
354361
}

0 commit comments

Comments
 (0)