Skip to content

chore(internal): codegen related update #146

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
stainless-app merged 1 commit into from
Mar 12, 2025
Merged

chore(internal): codegen related update #146

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
The table of contents is too big for display.
Diff view
Diff view
  •  
  •  
  •  
306 changes: 114 additions & 192 deletions ...ent-okhttp/src/main/kotlin/com/braintrustdata/api/client/okhttp/BraintrustOkHttpClient.kt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -19,15 +19,10 @@ class BraintrustOkHttpClient private constructor() {

companion object {

/**
* Returns a mutable builder for constructing an instance of
* [BraintrustOkHttpClient].
*/
@JvmStatic
fun builder() = Builder()
/** Returns a mutable builder for constructing an instance of [BraintrustOkHttpClient]. */
@JvmStatic fun builder() = Builder()

@JvmStatic
fun fromEnv(): BraintrustClient = builder().fromEnv().build()
@JvmStatic fun fromEnv(): BraintrustClient = builder().fromEnv().build()
}

/** A builder for [BraintrustOkHttpClient]. */
Expand All @@ -38,163 +33,99 @@ class BraintrustOkHttpClient private constructor() {
private var timeout: Timeout = Timeout.default()
private var proxy: Proxy? = null

fun baseUrl(baseUrl: String) =
apply {
clientOptions.baseUrl(baseUrl)
this.baseUrl = baseUrl
}

fun jsonMapper(jsonMapper: JsonMapper) =
apply {
clientOptions.jsonMapper(jsonMapper)
}

fun clock(clock: Clock) =
apply {
clientOptions.clock(clock)
}

fun headers(headers: Headers) =
apply {
clientOptions.headers(headers)
}

fun headers(headers: Map<String, Iterable<String>>) =
apply {
clientOptions.headers(headers)
}

fun putHeader(name: String, value: String) =
apply {
clientOptions.putHeader(
name, value
)
}

fun putHeaders(name: String, values: Iterable<String>) =
apply {
clientOptions.putHeaders(
name, values
)
}

fun putAllHeaders(headers: Headers) =
apply {
clientOptions.putAllHeaders(headers)
}

fun putAllHeaders(headers: Map<String, Iterable<String>>) =
apply {
clientOptions.putAllHeaders(headers)
}

fun replaceHeaders(name: String, value: String) =
apply {
clientOptions.replaceHeaders(
name, value
)
}

fun replaceHeaders(name: String, values: Iterable<String>) =
apply {
clientOptions.replaceHeaders(
name, values
)
}

fun replaceAllHeaders(headers: Headers) =
apply {
clientOptions.replaceAllHeaders(headers)
}

fun replaceAllHeaders(headers: Map<String, Iterable<String>>) =
apply {
clientOptions.replaceAllHeaders(headers)
}

fun removeHeaders(name: String) =
apply {
clientOptions.removeHeaders(name)
}

fun removeAllHeaders(names: Set<String>) =
apply {
clientOptions.removeAllHeaders(names)
}

fun queryParams(queryParams: QueryParams) =
apply {
clientOptions.queryParams(queryParams)
}

fun queryParams(queryParams: Map<String, Iterable<String>>) =
apply {
clientOptions.queryParams(queryParams)
}

fun putQueryParam(key: String, value: String) =
apply {
clientOptions.putQueryParam(
key, value
)
}

fun putQueryParams(key: String, values: Iterable<String>) =
apply {
clientOptions.putQueryParams(
key, values
)
}

fun putAllQueryParams(queryParams: QueryParams) =
apply {
clientOptions.putAllQueryParams(queryParams)
}

fun putAllQueryParams(queryParams: Map<String, Iterable<String>>) =
apply {
clientOptions.putAllQueryParams(queryParams)
}

fun replaceQueryParams(key: String, value: String) =
apply {
clientOptions.replaceQueryParams(
key, value
)
}

fun replaceQueryParams(key: String, values: Iterable<String>) =
apply {
clientOptions.replaceQueryParams(
key, values
)
}

fun replaceAllQueryParams(queryParams: QueryParams) =
apply {
clientOptions.replaceAllQueryParams(queryParams)
}

fun replaceAllQueryParams(queryParams: Map<String, Iterable<String>>) =
apply {
clientOptions.replaceAllQueryParams(queryParams)
}

fun removeQueryParams(key: String) =
apply {
clientOptions.removeQueryParams(key)
}

fun removeAllQueryParams(keys: Set<String>) =
apply {
clientOptions.removeAllQueryParams(keys)
}

fun timeout(timeout: Timeout) =
apply {
clientOptions.timeout(timeout)
this.timeout = timeout
}
fun baseUrl(baseUrl: String) = apply {
clientOptions.baseUrl(baseUrl)
this.baseUrl = baseUrl
}

fun jsonMapper(jsonMapper: JsonMapper) = apply { clientOptions.jsonMapper(jsonMapper) }

fun clock(clock: Clock) = apply { clientOptions.clock(clock) }

fun headers(headers: Headers) = apply { clientOptions.headers(headers) }

fun headers(headers: Map<String, Iterable<String>>) = apply {
clientOptions.headers(headers)
}

fun putHeader(name: String, value: String) = apply { clientOptions.putHeader(name, value) }

fun putHeaders(name: String, values: Iterable<String>) = apply {
clientOptions.putHeaders(name, values)
}

fun putAllHeaders(headers: Headers) = apply { clientOptions.putAllHeaders(headers) }

fun putAllHeaders(headers: Map<String, Iterable<String>>) = apply {
clientOptions.putAllHeaders(headers)
}

fun replaceHeaders(name: String, value: String) = apply {
clientOptions.replaceHeaders(name, value)
}

fun replaceHeaders(name: String, values: Iterable<String>) = apply {
clientOptions.replaceHeaders(name, values)
}

fun replaceAllHeaders(headers: Headers) = apply { clientOptions.replaceAllHeaders(headers) }

fun replaceAllHeaders(headers: Map<String, Iterable<String>>) = apply {
clientOptions.replaceAllHeaders(headers)
}

fun removeHeaders(name: String) = apply { clientOptions.removeHeaders(name) }

fun removeAllHeaders(names: Set<String>) = apply { clientOptions.removeAllHeaders(names) }

fun queryParams(queryParams: QueryParams) = apply { clientOptions.queryParams(queryParams) }

fun queryParams(queryParams: Map<String, Iterable<String>>) = apply {
clientOptions.queryParams(queryParams)
}

fun putQueryParam(key: String, value: String) = apply {
clientOptions.putQueryParam(key, value)
}

fun putQueryParams(key: String, values: Iterable<String>) = apply {
clientOptions.putQueryParams(key, values)
}

fun putAllQueryParams(queryParams: QueryParams) = apply {
clientOptions.putAllQueryParams(queryParams)
}

fun putAllQueryParams(queryParams: Map<String, Iterable<String>>) = apply {
clientOptions.putAllQueryParams(queryParams)
}

fun replaceQueryParams(key: String, value: String) = apply {
clientOptions.replaceQueryParams(key, value)
}

fun replaceQueryParams(key: String, values: Iterable<String>) = apply {
clientOptions.replaceQueryParams(key, values)
}

fun replaceAllQueryParams(queryParams: QueryParams) = apply {
clientOptions.replaceAllQueryParams(queryParams)
}

fun replaceAllQueryParams(queryParams: Map<String, Iterable<String>>) = apply {
clientOptions.replaceAllQueryParams(queryParams)
}

fun removeQueryParams(key: String) = apply { clientOptions.removeQueryParams(key) }

fun removeAllQueryParams(keys: Set<String>) = apply {
clientOptions.removeAllQueryParams(keys)
}

fun timeout(timeout: Timeout) = apply {
clientOptions.timeout(timeout)
this.timeout = timeout
}

/**
* Sets the maximum time allowed for a complete HTTP call, not including retries.
Expand All @@ -205,40 +136,31 @@ class BraintrustOkHttpClient private constructor() {
*/
fun timeout(timeout: Duration) = timeout(Timeout.builder().request(timeout).build())

fun maxRetries(maxRetries: Int) =
apply {
clientOptions.maxRetries(maxRetries)
}
fun maxRetries(maxRetries: Int) = apply { clientOptions.maxRetries(maxRetries) }

fun proxy(proxy: Proxy) =
apply {
this.proxy = proxy
}
fun proxy(proxy: Proxy) = apply { this.proxy = proxy }

fun responseValidation(responseValidation: Boolean) =
apply {
clientOptions.responseValidation(responseValidation)
}
fun responseValidation(responseValidation: Boolean) = apply {
clientOptions.responseValidation(responseValidation)
}

fun apiKey(apiKey: String?) =
apply {
clientOptions.apiKey(apiKey)
}
fun apiKey(apiKey: String?) = apply { clientOptions.apiKey(apiKey) }

fun apiKey(apiKey: Optional<String>) = apiKey(apiKey.getOrNull())

fun fromEnv() =
apply {
clientOptions.fromEnv()
}
fun fromEnv() = apply { clientOptions.fromEnv() }

fun build(): BraintrustClient =
BraintrustClientImpl(clientOptions
.httpClient(OkHttpClient.builder()
.baseUrl(baseUrl)
.timeout(timeout)
.proxy(proxy)
.build())
.build())
BraintrustClientImpl(
clientOptions
.httpClient(
OkHttpClient.builder()
.baseUrl(baseUrl)
.timeout(timeout)
.proxy(proxy)
.build()
)
.build()
)
}
}
Loading