diff --git a/src/libs/Baseten/Generated/Baseten.BasetenClient.CreateTrainersSessions.g.cs b/src/libs/Baseten/Generated/Baseten.BasetenClient.CreateTrainerSessions.g.cs
similarity index 89%
rename from src/libs/Baseten/Generated/Baseten.BasetenClient.CreateTrainersSessions.g.cs
rename to src/libs/Baseten/Generated/Baseten.BasetenClient.CreateTrainerSessions.g.cs
index 34b43d3..39c7e46 100644
--- a/src/libs/Baseten/Generated/Baseten.BasetenClient.CreateTrainersSessions.g.cs
+++ b/src/libs/Baseten/Generated/Baseten.BasetenClient.CreateTrainerSessions.g.cs
@@ -7,7 +7,7 @@ public partial class BasetenClient
{
- private static readonly global::Baseten.EndPointSecurityRequirement s_CreateTrainersSessionsSecurityRequirement0 =
+ private static readonly global::Baseten.EndPointSecurityRequirement s_CreateTrainerSessionsSecurityRequirement0 =
new global::Baseten.EndPointSecurityRequirement
{
Authorizations = new global::Baseten.EndPointAuthorizationRequirement[]
@@ -21,22 +21,22 @@ public partial class BasetenClient
},
},
};
- private static readonly global::Baseten.EndPointSecurityRequirement[] s_CreateTrainersSessionsSecurityRequirements =
+ private static readonly global::Baseten.EndPointSecurityRequirement[] s_CreateTrainerSessionsSecurityRequirements =
new global::Baseten.EndPointSecurityRequirement[]
- { s_CreateTrainersSessionsSecurityRequirement0,
+ { s_CreateTrainerSessionsSecurityRequirement0,
};
- partial void PrepareCreateTrainersSessionsArguments(
+ partial void PrepareCreateTrainerSessionsArguments(
global::System.Net.Http.HttpClient httpClient,
global::Baseten.CreateTrainerSessionRequestV1 request);
- partial void PrepareCreateTrainersSessionsRequest(
+ partial void PrepareCreateTrainerSessionsRequest(
global::System.Net.Http.HttpClient httpClient,
global::System.Net.Http.HttpRequestMessage httpRequestMessage,
global::Baseten.CreateTrainerSessionRequestV1 request);
- partial void ProcessCreateTrainersSessionsResponse(
+ partial void ProcessCreateTrainerSessionsResponse(
global::System.Net.Http.HttpClient httpClient,
global::System.Net.Http.HttpResponseMessage httpResponseMessage);
- partial void ProcessCreateTrainersSessionsResponseContent(
+ partial void ProcessCreateTrainerSessionsResponseContent(
global::System.Net.Http.HttpClient httpClient,
global::System.Net.Http.HttpResponseMessage httpResponseMessage,
ref string content);
@@ -51,19 +51,19 @@ partial void ProcessCreateTrainersSessionsResponseContent(
///
///
/// curl --request POST \
- /// --url https://api.baseten.co/v1/trainers/sessions \
+ /// --url https://api.baseten.co/v1/trainer_sessions \
/// --header "Authorization: Api-Key $BASETEN_API_KEY" \
/// --data '{
/// "training_project_id": null
/// }'
///
- public async global::System.Threading.Tasks.Task CreateTrainersSessionsAsync(
+ public async global::System.Threading.Tasks.Task CreateTrainerSessionsAsync(
global::Baseten.CreateTrainerSessionRequestV1 request,
global::Baseten.AutoSDKRequestOptions? requestOptions = default,
global::System.Threading.CancellationToken cancellationToken = default)
{
- var __response = await CreateTrainersSessionsAsResponseAsync(
+ var __response = await CreateTrainerSessionsAsResponseAsync(
request: request,
requestOptions: requestOptions,
@@ -82,13 +82,13 @@ partial void ProcessCreateTrainersSessionsResponseContent(
///
///
/// curl --request POST \
- /// --url https://api.baseten.co/v1/trainers/sessions \
+ /// --url https://api.baseten.co/v1/trainer_sessions \
/// --header "Authorization: Api-Key $BASETEN_API_KEY" \
/// --data '{
/// "training_project_id": null
/// }'
///
- public async global::System.Threading.Tasks.Task> CreateTrainersSessionsAsResponseAsync(
+ public async global::System.Threading.Tasks.Task> CreateTrainerSessionsAsResponseAsync(
global::Baseten.CreateTrainerSessionRequestV1 request,
global::Baseten.AutoSDKRequestOptions? requestOptions = default,
@@ -98,15 +98,15 @@ partial void ProcessCreateTrainersSessionsResponseContent(
PrepareArguments(
client: HttpClient);
- PrepareCreateTrainersSessionsArguments(
+ PrepareCreateTrainerSessionsArguments(
httpClient: HttpClient,
request: request);
var __authorizations = global::Baseten.EndPointSecurityResolver.ResolveAuthorizations(
availableAuthorizations: Authorizations,
- securityRequirements: s_CreateTrainersSessionsSecurityRequirements,
- operationName: "CreateTrainersSessionsAsync");
+ securityRequirements: s_CreateTrainerSessionsSecurityRequirements,
+ operationName: "CreateTrainerSessionsAsync");
using var __timeoutCancellationTokenSource = global::Baseten.AutoSDKRequestOptionsSupport.CreateTimeoutCancellationTokenSource(
clientOptions: Options,
@@ -126,7 +126,7 @@ partial void ProcessCreateTrainersSessionsResponseContent(
{
var __pathBuilder = new global::Baseten.PathBuilder(
- path: "/v1/trainers/sessions",
+ path: "/v1/trainer_sessions",
baseUri: HttpClient.BaseAddress);
var __path = __pathBuilder.ToString();
__path = global::Baseten.AutoSDKRequestOptionsSupport.AppendQueryParameters(
@@ -171,7 +171,7 @@ partial void ProcessCreateTrainersSessionsResponseContent(
PrepareRequest(
client: HttpClient,
request: __httpRequest);
- PrepareCreateTrainersSessionsRequest(
+ PrepareCreateTrainerSessionsRequest(
httpClient: HttpClient,
httpRequestMessage: __httpRequest,
request: request);
@@ -191,9 +191,9 @@ partial void ProcessCreateTrainersSessionsResponseContent(
await global::Baseten.AutoSDKRequestOptionsSupport.OnBeforeRequestAsync(
clientOptions: Options,
context: global::Baseten.AutoSDKRequestOptionsSupport.CreateHookContext(
- operationId: "createTrainersSessions",
- methodName: "CreateTrainersSessionsAsync",
- pathTemplate: "\"/v1/trainers/sessions\"",
+ operationId: "createTrainerSessions",
+ methodName: "CreateTrainerSessionsAsync",
+ pathTemplate: "\"/v1/trainer_sessions\"",
httpMethod: "POST",
baseUri: BaseUri,
request: __httpRequest!,
@@ -225,9 +225,9 @@ partial void ProcessCreateTrainersSessionsResponseContent(
await global::Baseten.AutoSDKRequestOptionsSupport.OnAfterErrorAsync(
clientOptions: Options,
context: global::Baseten.AutoSDKRequestOptionsSupport.CreateHookContext(
- operationId: "createTrainersSessions",
- methodName: "CreateTrainersSessionsAsync",
- pathTemplate: "\"/v1/trainers/sessions\"",
+ operationId: "createTrainerSessions",
+ methodName: "CreateTrainerSessionsAsync",
+ pathTemplate: "\"/v1/trainer_sessions\"",
httpMethod: "POST",
baseUri: BaseUri,
request: __httpRequest!,
@@ -266,9 +266,9 @@ partial void ProcessCreateTrainersSessionsResponseContent(
await global::Baseten.AutoSDKRequestOptionsSupport.OnAfterErrorAsync(
clientOptions: Options,
context: global::Baseten.AutoSDKRequestOptionsSupport.CreateHookContext(
- operationId: "createTrainersSessions",
- methodName: "CreateTrainersSessionsAsync",
- pathTemplate: "\"/v1/trainers/sessions\"",
+ operationId: "createTrainerSessions",
+ methodName: "CreateTrainerSessionsAsync",
+ pathTemplate: "\"/v1/trainer_sessions\"",
httpMethod: "POST",
baseUri: BaseUri,
request: __httpRequest!,
@@ -306,7 +306,7 @@ partial void ProcessCreateTrainersSessionsResponseContent(
ProcessResponse(
client: HttpClient,
response: __response);
- ProcessCreateTrainersSessionsResponse(
+ ProcessCreateTrainerSessionsResponse(
httpClient: HttpClient,
httpResponseMessage: __response);
if (__response.IsSuccessStatusCode)
@@ -314,9 +314,9 @@ partial void ProcessCreateTrainersSessionsResponseContent(
await global::Baseten.AutoSDKRequestOptionsSupport.OnAfterSuccessAsync(
clientOptions: Options,
context: global::Baseten.AutoSDKRequestOptionsSupport.CreateHookContext(
- operationId: "createTrainersSessions",
- methodName: "CreateTrainersSessionsAsync",
- pathTemplate: "\"/v1/trainers/sessions\"",
+ operationId: "createTrainerSessions",
+ methodName: "CreateTrainerSessionsAsync",
+ pathTemplate: "\"/v1/trainer_sessions\"",
httpMethod: "POST",
baseUri: BaseUri,
request: __httpRequest!,
@@ -336,9 +336,9 @@ partial void ProcessCreateTrainersSessionsResponseContent(
await global::Baseten.AutoSDKRequestOptionsSupport.OnAfterErrorAsync(
clientOptions: Options,
context: global::Baseten.AutoSDKRequestOptionsSupport.CreateHookContext(
- operationId: "createTrainersSessions",
- methodName: "CreateTrainersSessionsAsync",
- pathTemplate: "\"/v1/trainers/sessions\"",
+ operationId: "createTrainerSessions",
+ methodName: "CreateTrainerSessionsAsync",
+ pathTemplate: "\"/v1/trainer_sessions\"",
httpMethod: "POST",
baseUri: BaseUri,
request: __httpRequest!,
@@ -366,7 +366,7 @@ partial void ProcessCreateTrainersSessionsResponseContent(
client: HttpClient,
response: __response,
content: ref __content);
- ProcessCreateTrainersSessionsResponseContent(
+ ProcessCreateTrainerSessionsResponseContent(
httpClient: HttpClient,
httpResponseMessage: __response,
content: ref __content);
@@ -458,13 +458,14 @@ partial void ProcessCreateTrainersSessionsResponseContent(
/// Creates a TrainerSession for the given training project.
///
///
- /// ID of the training project to associate with.
+ /// ID of the training project to associate with. If omitted, a default project is created for the org.
+ /// Default Value: openapi-json-null-sentinel-value-2BF93600-0FE4-4250-987A-E5DDB203E464
///
/// Per-request overrides such as headers, query parameters, timeout, retries, and response buffering.
/// The token to cancel the operation with
///
- public async global::System.Threading.Tasks.Task CreateTrainersSessionsAsync(
- string trainingProjectId,
+ public async global::System.Threading.Tasks.Task CreateTrainerSessionsAsync(
+ string? trainingProjectId = default,
global::Baseten.AutoSDKRequestOptions? requestOptions = default,
global::System.Threading.CancellationToken cancellationToken = default)
{
@@ -473,7 +474,7 @@ partial void ProcessCreateTrainersSessionsResponseContent(
TrainingProjectId = trainingProjectId,
};
- return await CreateTrainersSessionsAsync(
+ return await CreateTrainerSessionsAsync(
request: __request,
requestOptions: requestOptions,
cancellationToken: cancellationToken).ConfigureAwait(false);
diff --git a/src/libs/Baseten/Generated/Baseten.BasetenClient.CreateTrainersSessionsBySessionIdSamplers.g.cs b/src/libs/Baseten/Generated/Baseten.BasetenClient.CreateTrainerSessionsBySessionIdSamplers.g.cs
similarity index 86%
rename from src/libs/Baseten/Generated/Baseten.BasetenClient.CreateTrainersSessionsBySessionIdSamplers.g.cs
rename to src/libs/Baseten/Generated/Baseten.BasetenClient.CreateTrainerSessionsBySessionIdSamplers.g.cs
index 2a24145..db51624 100644
--- a/src/libs/Baseten/Generated/Baseten.BasetenClient.CreateTrainersSessionsBySessionIdSamplers.g.cs
+++ b/src/libs/Baseten/Generated/Baseten.BasetenClient.CreateTrainerSessionsBySessionIdSamplers.g.cs
@@ -7,7 +7,7 @@ public partial class BasetenClient
{
- private static readonly global::Baseten.EndPointSecurityRequirement s_CreateTrainersSessionsBySessionIdSamplersSecurityRequirement0 =
+ private static readonly global::Baseten.EndPointSecurityRequirement s_CreateTrainerSessionsBySessionIdSamplersSecurityRequirement0 =
new global::Baseten.EndPointSecurityRequirement
{
Authorizations = new global::Baseten.EndPointAuthorizationRequirement[]
@@ -21,24 +21,24 @@ public partial class BasetenClient
},
},
};
- private static readonly global::Baseten.EndPointSecurityRequirement[] s_CreateTrainersSessionsBySessionIdSamplersSecurityRequirements =
+ private static readonly global::Baseten.EndPointSecurityRequirement[] s_CreateTrainerSessionsBySessionIdSamplersSecurityRequirements =
new global::Baseten.EndPointSecurityRequirement[]
- { s_CreateTrainersSessionsBySessionIdSamplersSecurityRequirement0,
+ { s_CreateTrainerSessionsBySessionIdSamplersSecurityRequirement0,
};
- partial void PrepareCreateTrainersSessionsBySessionIdSamplersArguments(
+ partial void PrepareCreateTrainerSessionsBySessionIdSamplersArguments(
global::System.Net.Http.HttpClient httpClient,
ref string sessionId,
global::Baseten.CreateSamplingServerRequestV1 request);
- partial void PrepareCreateTrainersSessionsBySessionIdSamplersRequest(
+ partial void PrepareCreateTrainerSessionsBySessionIdSamplersRequest(
global::System.Net.Http.HttpClient httpClient,
global::System.Net.Http.HttpRequestMessage httpRequestMessage,
string sessionId,
global::Baseten.CreateSamplingServerRequestV1 request);
- partial void ProcessCreateTrainersSessionsBySessionIdSamplersResponse(
+ partial void ProcessCreateTrainerSessionsBySessionIdSamplersResponse(
global::System.Net.Http.HttpClient httpClient,
global::System.Net.Http.HttpResponseMessage httpResponseMessage);
- partial void ProcessCreateTrainersSessionsBySessionIdSamplersResponseContent(
+ partial void ProcessCreateTrainerSessionsBySessionIdSamplersResponseContent(
global::System.Net.Http.HttpClient httpClient,
global::System.Net.Http.HttpResponseMessage httpResponseMessage,
ref string content);
@@ -54,21 +54,22 @@ partial void ProcessCreateTrainersSessionsBySessionIdSamplersResponseContent(
///
///
/// curl --request POST \
- /// --url https://api.baseten.co/v1/trainers/sessions/{session_id}/samplers \
+ /// --url https://api.baseten.co/v1/trainer_sessions/{session_id}/samplers \
/// --header "Authorization: Api-Key $BASETEN_API_KEY" \
/// --data '{
/// "model": null,
- /// "max_seq_length": null
+ /// "max_seq_length": null,
+ /// "checkpoint_path": "bt://trainers:k4q95w5/sampler_weights/step-100"
/// }'
///
- public async global::System.Threading.Tasks.Task CreateTrainersSessionsBySessionIdSamplersAsync(
+ public async global::System.Threading.Tasks.Task CreateTrainerSessionsBySessionIdSamplersAsync(
string sessionId,
global::Baseten.CreateSamplingServerRequestV1 request,
global::Baseten.AutoSDKRequestOptions? requestOptions = default,
global::System.Threading.CancellationToken cancellationToken = default)
{
- var __response = await CreateTrainersSessionsBySessionIdSamplersAsResponseAsync(
+ var __response = await CreateTrainerSessionsBySessionIdSamplersAsResponseAsync(
sessionId: sessionId,
request: request,
@@ -89,14 +90,15 @@ partial void ProcessCreateTrainersSessionsBySessionIdSamplersResponseContent(
///
///
/// curl --request POST \
- /// --url https://api.baseten.co/v1/trainers/sessions/{session_id}/samplers \
+ /// --url https://api.baseten.co/v1/trainer_sessions/{session_id}/samplers \
/// --header "Authorization: Api-Key $BASETEN_API_KEY" \
/// --data '{
/// "model": null,
- /// "max_seq_length": null
+ /// "max_seq_length": null,
+ /// "checkpoint_path": "bt://trainers:k4q95w5/sampler_weights/step-100"
/// }'
///
- public async global::System.Threading.Tasks.Task> CreateTrainersSessionsBySessionIdSamplersAsResponseAsync(
+ public async global::System.Threading.Tasks.Task> CreateTrainerSessionsBySessionIdSamplersAsResponseAsync(
string sessionId,
global::Baseten.CreateSamplingServerRequestV1 request,
@@ -107,7 +109,7 @@ partial void ProcessCreateTrainersSessionsBySessionIdSamplersResponseContent(
PrepareArguments(
client: HttpClient);
- PrepareCreateTrainersSessionsBySessionIdSamplersArguments(
+ PrepareCreateTrainerSessionsBySessionIdSamplersArguments(
httpClient: HttpClient,
sessionId: ref sessionId,
request: request);
@@ -115,8 +117,8 @@ partial void ProcessCreateTrainersSessionsBySessionIdSamplersResponseContent(
var __authorizations = global::Baseten.EndPointSecurityResolver.ResolveAuthorizations(
availableAuthorizations: Authorizations,
- securityRequirements: s_CreateTrainersSessionsBySessionIdSamplersSecurityRequirements,
- operationName: "CreateTrainersSessionsBySessionIdSamplersAsync");
+ securityRequirements: s_CreateTrainerSessionsBySessionIdSamplersSecurityRequirements,
+ operationName: "CreateTrainerSessionsBySessionIdSamplersAsync");
using var __timeoutCancellationTokenSource = global::Baseten.AutoSDKRequestOptionsSupport.CreateTimeoutCancellationTokenSource(
clientOptions: Options,
@@ -136,7 +138,7 @@ partial void ProcessCreateTrainersSessionsBySessionIdSamplersResponseContent(
{
var __pathBuilder = new global::Baseten.PathBuilder(
- path: $"/v1/trainers/sessions/{sessionId}/samplers",
+ path: $"/v1/trainer_sessions/{sessionId}/samplers",
baseUri: HttpClient.BaseAddress);
var __path = __pathBuilder.ToString();
__path = global::Baseten.AutoSDKRequestOptionsSupport.AppendQueryParameters(
@@ -181,7 +183,7 @@ partial void ProcessCreateTrainersSessionsBySessionIdSamplersResponseContent(
PrepareRequest(
client: HttpClient,
request: __httpRequest);
- PrepareCreateTrainersSessionsBySessionIdSamplersRequest(
+ PrepareCreateTrainerSessionsBySessionIdSamplersRequest(
httpClient: HttpClient,
httpRequestMessage: __httpRequest,
sessionId: sessionId!,
@@ -202,9 +204,9 @@ partial void ProcessCreateTrainersSessionsBySessionIdSamplersResponseContent(
await global::Baseten.AutoSDKRequestOptionsSupport.OnBeforeRequestAsync(
clientOptions: Options,
context: global::Baseten.AutoSDKRequestOptionsSupport.CreateHookContext(
- operationId: "createTrainersSessionsBySessionIdSamplers",
- methodName: "CreateTrainersSessionsBySessionIdSamplersAsync",
- pathTemplate: "$\"/v1/trainers/sessions/{sessionId}/samplers\"",
+ operationId: "createTrainerSessionsBySessionIdSamplers",
+ methodName: "CreateTrainerSessionsBySessionIdSamplersAsync",
+ pathTemplate: "$\"/v1/trainer_sessions/{sessionId}/samplers\"",
httpMethod: "POST",
baseUri: BaseUri,
request: __httpRequest!,
@@ -236,9 +238,9 @@ partial void ProcessCreateTrainersSessionsBySessionIdSamplersResponseContent(
await global::Baseten.AutoSDKRequestOptionsSupport.OnAfterErrorAsync(
clientOptions: Options,
context: global::Baseten.AutoSDKRequestOptionsSupport.CreateHookContext(
- operationId: "createTrainersSessionsBySessionIdSamplers",
- methodName: "CreateTrainersSessionsBySessionIdSamplersAsync",
- pathTemplate: "$\"/v1/trainers/sessions/{sessionId}/samplers\"",
+ operationId: "createTrainerSessionsBySessionIdSamplers",
+ methodName: "CreateTrainerSessionsBySessionIdSamplersAsync",
+ pathTemplate: "$\"/v1/trainer_sessions/{sessionId}/samplers\"",
httpMethod: "POST",
baseUri: BaseUri,
request: __httpRequest!,
@@ -277,9 +279,9 @@ partial void ProcessCreateTrainersSessionsBySessionIdSamplersResponseContent(
await global::Baseten.AutoSDKRequestOptionsSupport.OnAfterErrorAsync(
clientOptions: Options,
context: global::Baseten.AutoSDKRequestOptionsSupport.CreateHookContext(
- operationId: "createTrainersSessionsBySessionIdSamplers",
- methodName: "CreateTrainersSessionsBySessionIdSamplersAsync",
- pathTemplate: "$\"/v1/trainers/sessions/{sessionId}/samplers\"",
+ operationId: "createTrainerSessionsBySessionIdSamplers",
+ methodName: "CreateTrainerSessionsBySessionIdSamplersAsync",
+ pathTemplate: "$\"/v1/trainer_sessions/{sessionId}/samplers\"",
httpMethod: "POST",
baseUri: BaseUri,
request: __httpRequest!,
@@ -317,7 +319,7 @@ partial void ProcessCreateTrainersSessionsBySessionIdSamplersResponseContent(
ProcessResponse(
client: HttpClient,
response: __response);
- ProcessCreateTrainersSessionsBySessionIdSamplersResponse(
+ ProcessCreateTrainerSessionsBySessionIdSamplersResponse(
httpClient: HttpClient,
httpResponseMessage: __response);
if (__response.IsSuccessStatusCode)
@@ -325,9 +327,9 @@ partial void ProcessCreateTrainersSessionsBySessionIdSamplersResponseContent(
await global::Baseten.AutoSDKRequestOptionsSupport.OnAfterSuccessAsync(
clientOptions: Options,
context: global::Baseten.AutoSDKRequestOptionsSupport.CreateHookContext(
- operationId: "createTrainersSessionsBySessionIdSamplers",
- methodName: "CreateTrainersSessionsBySessionIdSamplersAsync",
- pathTemplate: "$\"/v1/trainers/sessions/{sessionId}/samplers\"",
+ operationId: "createTrainerSessionsBySessionIdSamplers",
+ methodName: "CreateTrainerSessionsBySessionIdSamplersAsync",
+ pathTemplate: "$\"/v1/trainer_sessions/{sessionId}/samplers\"",
httpMethod: "POST",
baseUri: BaseUri,
request: __httpRequest!,
@@ -347,9 +349,9 @@ partial void ProcessCreateTrainersSessionsBySessionIdSamplersResponseContent(
await global::Baseten.AutoSDKRequestOptionsSupport.OnAfterErrorAsync(
clientOptions: Options,
context: global::Baseten.AutoSDKRequestOptionsSupport.CreateHookContext(
- operationId: "createTrainersSessionsBySessionIdSamplers",
- methodName: "CreateTrainersSessionsBySessionIdSamplersAsync",
- pathTemplate: "$\"/v1/trainers/sessions/{sessionId}/samplers\"",
+ operationId: "createTrainerSessionsBySessionIdSamplers",
+ methodName: "CreateTrainerSessionsBySessionIdSamplersAsync",
+ pathTemplate: "$\"/v1/trainer_sessions/{sessionId}/samplers\"",
httpMethod: "POST",
baseUri: BaseUri,
request: __httpRequest!,
@@ -377,7 +379,7 @@ partial void ProcessCreateTrainersSessionsBySessionIdSamplersResponseContent(
client: HttpClient,
response: __response,
content: ref __content);
- ProcessCreateTrainersSessionsBySessionIdSamplersResponseContent(
+ ProcessCreateTrainerSessionsBySessionIdSamplersResponseContent(
httpClient: HttpClient,
httpResponseMessage: __response,
content: ref __content);
@@ -476,13 +478,18 @@ partial void ProcessCreateTrainersSessionsBySessionIdSamplersResponseContent(
/// Minimum max-sequence-length the sampler must support. If set, only instance-type entries whose configured max_seq_length is >= this value are considered. Omit to accept any entry.
/// Default Value: openapi-json-null-sentinel-value-2BF93600-0FE4-4250-987A-E5DDB203E464
///
+ ///
+ /// Optional bt:// URI of an existing sampler-target checkpoint to load weights from on startup. Form: bt://trainers:<trainer_id>/sampler_weights/<checkpoint_name>.
+ /// Default Value: openapi-json-null-sentinel-value-2BF93600-0FE4-4250-987A-E5DDB203E464
+ ///
/// Per-request overrides such as headers, query parameters, timeout, retries, and response buffering.
/// The token to cancel the operation with
///
- public async global::System.Threading.Tasks.Task CreateTrainersSessionsBySessionIdSamplersAsync(
+ public async global::System.Threading.Tasks.Task CreateTrainerSessionsBySessionIdSamplersAsync(
string sessionId,
string model,
int? maxSeqLength = default,
+ string? checkpointPath = default,
global::Baseten.AutoSDKRequestOptions? requestOptions = default,
global::System.Threading.CancellationToken cancellationToken = default)
{
@@ -490,9 +497,10 @@ partial void ProcessCreateTrainersSessionsBySessionIdSamplersResponseContent(
{
Model = model,
MaxSeqLength = maxSeqLength,
+ CheckpointPath = checkpointPath,
};
- return await CreateTrainersSessionsBySessionIdSamplersAsync(
+ return await CreateTrainerSessionsBySessionIdSamplersAsync(
sessionId: sessionId,
request: __request,
requestOptions: requestOptions,
diff --git a/src/libs/Baseten/Generated/Baseten.BasetenClient.CreateTrainersSessionsBySessionIdTrainers.g.cs b/src/libs/Baseten/Generated/Baseten.BasetenClient.CreateTrainerSessionsBySessionIdTrainers.g.cs
similarity index 86%
rename from src/libs/Baseten/Generated/Baseten.BasetenClient.CreateTrainersSessionsBySessionIdTrainers.g.cs
rename to src/libs/Baseten/Generated/Baseten.BasetenClient.CreateTrainerSessionsBySessionIdTrainers.g.cs
index 7123e56..3b942fb 100644
--- a/src/libs/Baseten/Generated/Baseten.BasetenClient.CreateTrainersSessionsBySessionIdTrainers.g.cs
+++ b/src/libs/Baseten/Generated/Baseten.BasetenClient.CreateTrainerSessionsBySessionIdTrainers.g.cs
@@ -7,7 +7,7 @@ public partial class BasetenClient
{
- private static readonly global::Baseten.EndPointSecurityRequirement s_CreateTrainersSessionsBySessionIdTrainersSecurityRequirement0 =
+ private static readonly global::Baseten.EndPointSecurityRequirement s_CreateTrainerSessionsBySessionIdTrainersSecurityRequirement0 =
new global::Baseten.EndPointSecurityRequirement
{
Authorizations = new global::Baseten.EndPointAuthorizationRequirement[]
@@ -21,24 +21,24 @@ public partial class BasetenClient
},
},
};
- private static readonly global::Baseten.EndPointSecurityRequirement[] s_CreateTrainersSessionsBySessionIdTrainersSecurityRequirements =
+ private static readonly global::Baseten.EndPointSecurityRequirement[] s_CreateTrainerSessionsBySessionIdTrainersSecurityRequirements =
new global::Baseten.EndPointSecurityRequirement[]
- { s_CreateTrainersSessionsBySessionIdTrainersSecurityRequirement0,
+ { s_CreateTrainerSessionsBySessionIdTrainersSecurityRequirement0,
};
- partial void PrepareCreateTrainersSessionsBySessionIdTrainersArguments(
+ partial void PrepareCreateTrainerSessionsBySessionIdTrainersArguments(
global::System.Net.Http.HttpClient httpClient,
ref string sessionId,
global::Baseten.CreateTrainerServerRequestV1 request);
- partial void PrepareCreateTrainersSessionsBySessionIdTrainersRequest(
+ partial void PrepareCreateTrainerSessionsBySessionIdTrainersRequest(
global::System.Net.Http.HttpClient httpClient,
global::System.Net.Http.HttpRequestMessage httpRequestMessage,
string sessionId,
global::Baseten.CreateTrainerServerRequestV1 request);
- partial void ProcessCreateTrainersSessionsBySessionIdTrainersResponse(
+ partial void ProcessCreateTrainerSessionsBySessionIdTrainersResponse(
global::System.Net.Http.HttpClient httpClient,
global::System.Net.Http.HttpResponseMessage httpResponseMessage);
- partial void ProcessCreateTrainersSessionsBySessionIdTrainersResponseContent(
+ partial void ProcessCreateTrainerSessionsBySessionIdTrainersResponseContent(
global::System.Net.Http.HttpClient httpClient,
global::System.Net.Http.HttpResponseMessage httpResponseMessage,
ref string content);
@@ -54,24 +54,25 @@ partial void ProcessCreateTrainersSessionsBySessionIdTrainersResponseContent(
///
///
/// curl --request POST \
- /// --url https://api.baseten.co/v1/trainers/sessions/{session_id}/trainers \
+ /// --url https://api.baseten.co/v1/trainer_sessions/{session_id}/trainers \
/// --header "Authorization: Api-Key $BASETEN_API_KEY" \
/// --data '{
/// "model": null,
/// "max_seq_len": null,
/// "lora_rank": null,
/// "seed": null,
- /// "scale_down_delay_seconds": null
+ /// "scale_down_delay_seconds": null,
+ /// "checkpoint_path": "bt://trainers:k4q95w5/weights/step-100"
/// }'
///
- public async global::System.Threading.Tasks.Task CreateTrainersSessionsBySessionIdTrainersAsync(
+ public async global::System.Threading.Tasks.Task CreateTrainerSessionsBySessionIdTrainersAsync(
string sessionId,
global::Baseten.CreateTrainerServerRequestV1 request,
global::Baseten.AutoSDKRequestOptions? requestOptions = default,
global::System.Threading.CancellationToken cancellationToken = default)
{
- var __response = await CreateTrainersSessionsBySessionIdTrainersAsResponseAsync(
+ var __response = await CreateTrainerSessionsBySessionIdTrainersAsResponseAsync(
sessionId: sessionId,
request: request,
@@ -92,17 +93,18 @@ partial void ProcessCreateTrainersSessionsBySessionIdTrainersResponseContent(
///
///
/// curl --request POST \
- /// --url https://api.baseten.co/v1/trainers/sessions/{session_id}/trainers \
+ /// --url https://api.baseten.co/v1/trainer_sessions/{session_id}/trainers \
/// --header "Authorization: Api-Key $BASETEN_API_KEY" \
/// --data '{
/// "model": null,
/// "max_seq_len": null,
/// "lora_rank": null,
/// "seed": null,
- /// "scale_down_delay_seconds": null
+ /// "scale_down_delay_seconds": null,
+ /// "checkpoint_path": "bt://trainers:k4q95w5/weights/step-100"
/// }'
///
- public async global::System.Threading.Tasks.Task> CreateTrainersSessionsBySessionIdTrainersAsResponseAsync(
+ public async global::System.Threading.Tasks.Task> CreateTrainerSessionsBySessionIdTrainersAsResponseAsync(
string sessionId,
global::Baseten.CreateTrainerServerRequestV1 request,
@@ -113,7 +115,7 @@ partial void ProcessCreateTrainersSessionsBySessionIdTrainersResponseContent(
PrepareArguments(
client: HttpClient);
- PrepareCreateTrainersSessionsBySessionIdTrainersArguments(
+ PrepareCreateTrainerSessionsBySessionIdTrainersArguments(
httpClient: HttpClient,
sessionId: ref sessionId,
request: request);
@@ -121,8 +123,8 @@ partial void ProcessCreateTrainersSessionsBySessionIdTrainersResponseContent(
var __authorizations = global::Baseten.EndPointSecurityResolver.ResolveAuthorizations(
availableAuthorizations: Authorizations,
- securityRequirements: s_CreateTrainersSessionsBySessionIdTrainersSecurityRequirements,
- operationName: "CreateTrainersSessionsBySessionIdTrainersAsync");
+ securityRequirements: s_CreateTrainerSessionsBySessionIdTrainersSecurityRequirements,
+ operationName: "CreateTrainerSessionsBySessionIdTrainersAsync");
using var __timeoutCancellationTokenSource = global::Baseten.AutoSDKRequestOptionsSupport.CreateTimeoutCancellationTokenSource(
clientOptions: Options,
@@ -142,7 +144,7 @@ partial void ProcessCreateTrainersSessionsBySessionIdTrainersResponseContent(
{
var __pathBuilder = new global::Baseten.PathBuilder(
- path: $"/v1/trainers/sessions/{sessionId}/trainers",
+ path: $"/v1/trainer_sessions/{sessionId}/trainers",
baseUri: HttpClient.BaseAddress);
var __path = __pathBuilder.ToString();
__path = global::Baseten.AutoSDKRequestOptionsSupport.AppendQueryParameters(
@@ -187,7 +189,7 @@ partial void ProcessCreateTrainersSessionsBySessionIdTrainersResponseContent(
PrepareRequest(
client: HttpClient,
request: __httpRequest);
- PrepareCreateTrainersSessionsBySessionIdTrainersRequest(
+ PrepareCreateTrainerSessionsBySessionIdTrainersRequest(
httpClient: HttpClient,
httpRequestMessage: __httpRequest,
sessionId: sessionId!,
@@ -208,9 +210,9 @@ partial void ProcessCreateTrainersSessionsBySessionIdTrainersResponseContent(
await global::Baseten.AutoSDKRequestOptionsSupport.OnBeforeRequestAsync(
clientOptions: Options,
context: global::Baseten.AutoSDKRequestOptionsSupport.CreateHookContext(
- operationId: "createTrainersSessionsBySessionIdTrainers",
- methodName: "CreateTrainersSessionsBySessionIdTrainersAsync",
- pathTemplate: "$\"/v1/trainers/sessions/{sessionId}/trainers\"",
+ operationId: "createTrainerSessionsBySessionIdTrainers",
+ methodName: "CreateTrainerSessionsBySessionIdTrainersAsync",
+ pathTemplate: "$\"/v1/trainer_sessions/{sessionId}/trainers\"",
httpMethod: "POST",
baseUri: BaseUri,
request: __httpRequest!,
@@ -242,9 +244,9 @@ partial void ProcessCreateTrainersSessionsBySessionIdTrainersResponseContent(
await global::Baseten.AutoSDKRequestOptionsSupport.OnAfterErrorAsync(
clientOptions: Options,
context: global::Baseten.AutoSDKRequestOptionsSupport.CreateHookContext(
- operationId: "createTrainersSessionsBySessionIdTrainers",
- methodName: "CreateTrainersSessionsBySessionIdTrainersAsync",
- pathTemplate: "$\"/v1/trainers/sessions/{sessionId}/trainers\"",
+ operationId: "createTrainerSessionsBySessionIdTrainers",
+ methodName: "CreateTrainerSessionsBySessionIdTrainersAsync",
+ pathTemplate: "$\"/v1/trainer_sessions/{sessionId}/trainers\"",
httpMethod: "POST",
baseUri: BaseUri,
request: __httpRequest!,
@@ -283,9 +285,9 @@ partial void ProcessCreateTrainersSessionsBySessionIdTrainersResponseContent(
await global::Baseten.AutoSDKRequestOptionsSupport.OnAfterErrorAsync(
clientOptions: Options,
context: global::Baseten.AutoSDKRequestOptionsSupport.CreateHookContext(
- operationId: "createTrainersSessionsBySessionIdTrainers",
- methodName: "CreateTrainersSessionsBySessionIdTrainersAsync",
- pathTemplate: "$\"/v1/trainers/sessions/{sessionId}/trainers\"",
+ operationId: "createTrainerSessionsBySessionIdTrainers",
+ methodName: "CreateTrainerSessionsBySessionIdTrainersAsync",
+ pathTemplate: "$\"/v1/trainer_sessions/{sessionId}/trainers\"",
httpMethod: "POST",
baseUri: BaseUri,
request: __httpRequest!,
@@ -323,7 +325,7 @@ partial void ProcessCreateTrainersSessionsBySessionIdTrainersResponseContent(
ProcessResponse(
client: HttpClient,
response: __response);
- ProcessCreateTrainersSessionsBySessionIdTrainersResponse(
+ ProcessCreateTrainerSessionsBySessionIdTrainersResponse(
httpClient: HttpClient,
httpResponseMessage: __response);
if (__response.IsSuccessStatusCode)
@@ -331,9 +333,9 @@ partial void ProcessCreateTrainersSessionsBySessionIdTrainersResponseContent(
await global::Baseten.AutoSDKRequestOptionsSupport.OnAfterSuccessAsync(
clientOptions: Options,
context: global::Baseten.AutoSDKRequestOptionsSupport.CreateHookContext(
- operationId: "createTrainersSessionsBySessionIdTrainers",
- methodName: "CreateTrainersSessionsBySessionIdTrainersAsync",
- pathTemplate: "$\"/v1/trainers/sessions/{sessionId}/trainers\"",
+ operationId: "createTrainerSessionsBySessionIdTrainers",
+ methodName: "CreateTrainerSessionsBySessionIdTrainersAsync",
+ pathTemplate: "$\"/v1/trainer_sessions/{sessionId}/trainers\"",
httpMethod: "POST",
baseUri: BaseUri,
request: __httpRequest!,
@@ -353,9 +355,9 @@ partial void ProcessCreateTrainersSessionsBySessionIdTrainersResponseContent(
await global::Baseten.AutoSDKRequestOptionsSupport.OnAfterErrorAsync(
clientOptions: Options,
context: global::Baseten.AutoSDKRequestOptionsSupport.CreateHookContext(
- operationId: "createTrainersSessionsBySessionIdTrainers",
- methodName: "CreateTrainersSessionsBySessionIdTrainersAsync",
- pathTemplate: "$\"/v1/trainers/sessions/{sessionId}/trainers\"",
+ operationId: "createTrainerSessionsBySessionIdTrainers",
+ methodName: "CreateTrainerSessionsBySessionIdTrainersAsync",
+ pathTemplate: "$\"/v1/trainer_sessions/{sessionId}/trainers\"",
httpMethod: "POST",
baseUri: BaseUri,
request: __httpRequest!,
@@ -383,7 +385,7 @@ partial void ProcessCreateTrainersSessionsBySessionIdTrainersResponseContent(
client: HttpClient,
response: __response,
content: ref __content);
- ProcessCreateTrainersSessionsBySessionIdTrainersResponseContent(
+ ProcessCreateTrainerSessionsBySessionIdTrainersResponseContent(
httpClient: HttpClient,
httpResponseMessage: __response,
content: ref __content);
@@ -494,16 +496,21 @@ partial void ProcessCreateTrainersSessionsBySessionIdTrainersResponseContent(
/// Seconds of inactivity before the trainer scales to zero. Must be positive. Defaults to 3600 (1 hour).
/// Default Value: 3600
///
+ ///
+ /// Optional bt:// URI of an existing trainer-target checkpoint to resume training from. Form: bt://trainers:<trainer_id>/weights/<checkpoint_name>.
+ /// Default Value: openapi-json-null-sentinel-value-2BF93600-0FE4-4250-987A-E5DDB203E464
+ ///
/// Per-request overrides such as headers, query parameters, timeout, retries, and response buffering.
/// The token to cancel the operation with
///
- public async global::System.Threading.Tasks.Task CreateTrainersSessionsBySessionIdTrainersAsync(
+ public async global::System.Threading.Tasks.Task CreateTrainerSessionsBySessionIdTrainersAsync(
string sessionId,
string model,
int? maxSeqLen = default,
int? loraRank = default,
int? seed = default,
int? scaleDownDelaySeconds = default,
+ string? checkpointPath = default,
global::Baseten.AutoSDKRequestOptions? requestOptions = default,
global::System.Threading.CancellationToken cancellationToken = default)
{
@@ -514,9 +521,10 @@ partial void ProcessCreateTrainersSessionsBySessionIdTrainersResponseContent(
LoraRank = loraRank,
Seed = seed,
ScaleDownDelaySeconds = scaleDownDelaySeconds,
+ CheckpointPath = checkpointPath,
};
- return await CreateTrainersSessionsBySessionIdTrainersAsync(
+ return await CreateTrainerSessionsBySessionIdTrainersAsync(
sessionId: sessionId,
request: __request,
requestOptions: requestOptions,
diff --git a/src/libs/Baseten/Generated/Baseten.BasetenClient.CreateTrainersCheckpointsSearch.g.cs b/src/libs/Baseten/Generated/Baseten.BasetenClient.CreateTrainersCheckpointsSearch.g.cs
new file mode 100644
index 0000000..e00f446
--- /dev/null
+++ b/src/libs/Baseten/Generated/Baseten.BasetenClient.CreateTrainersCheckpointsSearch.g.cs
@@ -0,0 +1,482 @@
+
+#nullable enable
+
+namespace Baseten
+{
+ public partial class BasetenClient
+ {
+
+
+ private static readonly global::Baseten.EndPointSecurityRequirement s_CreateTrainersCheckpointsSearchSecurityRequirement0 =
+ new global::Baseten.EndPointSecurityRequirement
+ {
+ Authorizations = new global::Baseten.EndPointAuthorizationRequirement[]
+ { new global::Baseten.EndPointAuthorizationRequirement
+ {
+ Type = "Http",
+ SchemeId = "HttpBearer",
+ Location = "Header",
+ Name = "Bearer",
+ FriendlyName = "Bearer",
+ },
+ },
+ };
+ private static readonly global::Baseten.EndPointSecurityRequirement[] s_CreateTrainersCheckpointsSearchSecurityRequirements =
+ new global::Baseten.EndPointSecurityRequirement[]
+ { s_CreateTrainersCheckpointsSearchSecurityRequirement0,
+ };
+ partial void PrepareCreateTrainersCheckpointsSearchArguments(
+ global::System.Net.Http.HttpClient httpClient,
+ global::Baseten.CheckpointSearchRequestV1 request);
+ partial void PrepareCreateTrainersCheckpointsSearchRequest(
+ global::System.Net.Http.HttpClient httpClient,
+ global::System.Net.Http.HttpRequestMessage httpRequestMessage,
+ global::Baseten.CheckpointSearchRequestV1 request);
+ partial void ProcessCreateTrainersCheckpointsSearchResponse(
+ global::System.Net.Http.HttpClient httpClient,
+ global::System.Net.Http.HttpResponseMessage httpResponseMessage);
+
+ partial void ProcessCreateTrainersCheckpointsSearchResponseContent(
+ global::System.Net.Http.HttpClient httpClient,
+ global::System.Net.Http.HttpResponseMessage httpResponseMessage,
+ ref string content);
+
+ ///
+ /// Look up trainer checkpoint info by bt:// URI.
+ /// Look up the metadata for a trainer checkpoint by its bt://trainers:<trainer_id>/(weights|sampler_weights)/<name> URI.
+ ///
+ ///
+ /// Per-request overrides such as headers, query parameters, timeout, retries, and response buffering.
+ /// The token to cancel the operation with
+ ///
+ ///
+ /// curl --request POST \
+ /// --url https://api.baseten.co/v1/trainers/checkpoints/search \
+ /// --header "Authorization: Api-Key $BASETEN_API_KEY" \
+ /// --data '{
+ /// "checkpoint_path": "bt://trainers:k4q95w5/sampler_weights/step-100"
+ /// }'
+ ///
+ public async global::System.Threading.Tasks.Task CreateTrainersCheckpointsSearchAsync(
+
+ global::Baseten.CheckpointSearchRequestV1 request,
+ global::Baseten.AutoSDKRequestOptions? requestOptions = default,
+ global::System.Threading.CancellationToken cancellationToken = default)
+ {
+ var __response = await CreateTrainersCheckpointsSearchAsResponseAsync(
+
+ request: request,
+ requestOptions: requestOptions,
+ cancellationToken: cancellationToken
+ ).ConfigureAwait(false);
+
+ return __response.Body;
+ }
+ ///
+ /// Look up trainer checkpoint info by bt:// URI.
+ /// Look up the metadata for a trainer checkpoint by its bt://trainers:<trainer_id>/(weights|sampler_weights)/<name> URI.
+ ///
+ ///
+ /// Per-request overrides such as headers, query parameters, timeout, retries, and response buffering.
+ /// The token to cancel the operation with
+ ///
+ ///
+ /// curl --request POST \
+ /// --url https://api.baseten.co/v1/trainers/checkpoints/search \
+ /// --header "Authorization: Api-Key $BASETEN_API_KEY" \
+ /// --data '{
+ /// "checkpoint_path": "bt://trainers:k4q95w5/sampler_weights/step-100"
+ /// }'
+ ///
+ public async global::System.Threading.Tasks.Task> CreateTrainersCheckpointsSearchAsResponseAsync(
+
+ global::Baseten.CheckpointSearchRequestV1 request,
+ global::Baseten.AutoSDKRequestOptions? requestOptions = default,
+ global::System.Threading.CancellationToken cancellationToken = default)
+ {
+ request = request ?? throw new global::System.ArgumentNullException(nameof(request));
+
+ PrepareArguments(
+ client: HttpClient);
+ PrepareCreateTrainersCheckpointsSearchArguments(
+ httpClient: HttpClient,
+ request: request);
+
+
+ var __authorizations = global::Baseten.EndPointSecurityResolver.ResolveAuthorizations(
+ availableAuthorizations: Authorizations,
+ securityRequirements: s_CreateTrainersCheckpointsSearchSecurityRequirements,
+ operationName: "CreateTrainersCheckpointsSearchAsync");
+
+ using var __timeoutCancellationTokenSource = global::Baseten.AutoSDKRequestOptionsSupport.CreateTimeoutCancellationTokenSource(
+ clientOptions: Options,
+ requestOptions: requestOptions,
+ cancellationToken: cancellationToken);
+ var __effectiveCancellationToken = __timeoutCancellationTokenSource?.Token ?? cancellationToken;
+ var __effectiveReadResponseAsString = global::Baseten.AutoSDKRequestOptionsSupport.GetReadResponseAsString(
+ clientOptions: Options,
+ requestOptions: requestOptions,
+ fallbackValue: ReadResponseAsString);
+ var __maxAttempts = global::Baseten.AutoSDKRequestOptionsSupport.GetMaxAttempts(
+ clientOptions: Options,
+ requestOptions: requestOptions,
+ supportsRetry: true);
+
+ global::System.Net.Http.HttpRequestMessage __CreateHttpRequest()
+ {
+
+ var __pathBuilder = new global::Baseten.PathBuilder(
+ path: "/v1/trainers/checkpoints/search",
+ baseUri: HttpClient.BaseAddress);
+ var __path = __pathBuilder.ToString();
+ __path = global::Baseten.AutoSDKRequestOptionsSupport.AppendQueryParameters(
+ path: __path,
+ clientParameters: Options.QueryParameters,
+ requestParameters: requestOptions?.QueryParameters);
+ var __httpRequest = new global::System.Net.Http.HttpRequestMessage(
+ method: global::System.Net.Http.HttpMethod.Post,
+ requestUri: new global::System.Uri(__path, global::System.UriKind.RelativeOrAbsolute));
+#if NET6_0_OR_GREATER
+ __httpRequest.Version = global::System.Net.HttpVersion.Version11;
+ __httpRequest.VersionPolicy = global::System.Net.Http.HttpVersionPolicy.RequestVersionOrHigher;
+#endif
+
+ foreach (var __authorization in __authorizations)
+ {
+ if (__authorization.Type == "Http" ||
+ __authorization.Type == "OAuth2" ||
+ __authorization.Type == "OpenIdConnect")
+ {
+ __httpRequest.Headers.Authorization = new global::System.Net.Http.Headers.AuthenticationHeaderValue(
+ scheme: __authorization.Name,
+ parameter: __authorization.Value);
+ }
+ else if (__authorization.Type == "ApiKey" &&
+ __authorization.Location == "Header")
+ {
+ __httpRequest.Headers.Add(__authorization.Name, __authorization.Value);
+ }
+ }
+ var __httpRequestContentBody = request.ToJson(JsonSerializerContext);
+ var __httpRequestContent = new global::System.Net.Http.StringContent(
+ content: __httpRequestContentBody,
+ encoding: global::System.Text.Encoding.UTF8,
+ mediaType: "application/json");
+ __httpRequest.Content = __httpRequestContent;
+ global::Baseten.AutoSDKRequestOptionsSupport.ApplyHeaders(
+ request: __httpRequest,
+ clientHeaders: Options.Headers,
+ requestHeaders: requestOptions?.Headers);
+
+ PrepareRequest(
+ client: HttpClient,
+ request: __httpRequest);
+ PrepareCreateTrainersCheckpointsSearchRequest(
+ httpClient: HttpClient,
+ httpRequestMessage: __httpRequest,
+ request: request);
+
+ return __httpRequest;
+ }
+
+ global::System.Net.Http.HttpRequestMessage? __httpRequest = null;
+ global::System.Net.Http.HttpResponseMessage? __response = null;
+ var __attemptNumber = 0;
+ try
+ {
+ for (var __attempt = 1; __attempt <= __maxAttempts; __attempt++)
+ {
+ __attemptNumber = __attempt;
+ __httpRequest = __CreateHttpRequest();
+ await global::Baseten.AutoSDKRequestOptionsSupport.OnBeforeRequestAsync(
+ clientOptions: Options,
+ context: global::Baseten.AutoSDKRequestOptionsSupport.CreateHookContext(
+ operationId: "createTrainersCheckpointsSearch",
+ methodName: "CreateTrainersCheckpointsSearchAsync",
+ pathTemplate: "\"/v1/trainers/checkpoints/search\"",
+ httpMethod: "POST",
+ baseUri: BaseUri,
+ request: __httpRequest!,
+ response: null,
+ exception: null,
+ clientOptions: Options,
+ requestOptions: requestOptions,
+ attempt: __attempt,
+ maxAttempts: __maxAttempts,
+ willRetry: false,
+ retryDelay: null,
+ retryReason: global::System.String.Empty,
+ cancellationToken: __effectiveCancellationToken)).ConfigureAwait(false);
+ try
+ {
+ __response = await HttpClient.SendAsync(
+ request: __httpRequest,
+ completionOption: global::System.Net.Http.HttpCompletionOption.ResponseContentRead,
+ cancellationToken: __effectiveCancellationToken).ConfigureAwait(false);
+ }
+ catch (global::System.Net.Http.HttpRequestException __exception)
+ {
+ var __retryDelay = global::Baseten.AutoSDKRequestOptionsSupport.GetRetryDelay(
+ clientOptions: Options,
+ requestOptions: requestOptions,
+ response: null,
+ attempt: __attempt);
+ var __willRetry = __attempt < __maxAttempts && !__effectiveCancellationToken.IsCancellationRequested;
+ await global::Baseten.AutoSDKRequestOptionsSupport.OnAfterErrorAsync(
+ clientOptions: Options,
+ context: global::Baseten.AutoSDKRequestOptionsSupport.CreateHookContext(
+ operationId: "createTrainersCheckpointsSearch",
+ methodName: "CreateTrainersCheckpointsSearchAsync",
+ pathTemplate: "\"/v1/trainers/checkpoints/search\"",
+ httpMethod: "POST",
+ baseUri: BaseUri,
+ request: __httpRequest!,
+ response: null,
+ exception: __exception,
+ clientOptions: Options,
+ requestOptions: requestOptions,
+ attempt: __attempt,
+ maxAttempts: __maxAttempts,
+ willRetry: __willRetry,
+ retryDelay: __willRetry ? __retryDelay : (global::System.TimeSpan?)null,
+ retryReason: "exception",
+ cancellationToken: __effectiveCancellationToken)).ConfigureAwait(false);
+ if (!__willRetry)
+ {
+ throw;
+ }
+
+ __httpRequest.Dispose();
+ __httpRequest = null;
+ await global::Baseten.AutoSDKRequestOptionsSupport.DelayBeforeRetryAsync(
+ retryDelay: __retryDelay,
+ cancellationToken: __effectiveCancellationToken).ConfigureAwait(false);
+ continue;
+ }
+
+ if (__response != null &&
+ __attempt < __maxAttempts &&
+ global::Baseten.AutoSDKRequestOptionsSupport.ShouldRetryStatusCode(__response.StatusCode))
+ {
+ var __retryDelay = global::Baseten.AutoSDKRequestOptionsSupport.GetRetryDelay(
+ clientOptions: Options,
+ requestOptions: requestOptions,
+ response: __response,
+ attempt: __attempt);
+ await global::Baseten.AutoSDKRequestOptionsSupport.OnAfterErrorAsync(
+ clientOptions: Options,
+ context: global::Baseten.AutoSDKRequestOptionsSupport.CreateHookContext(
+ operationId: "createTrainersCheckpointsSearch",
+ methodName: "CreateTrainersCheckpointsSearchAsync",
+ pathTemplate: "\"/v1/trainers/checkpoints/search\"",
+ httpMethod: "POST",
+ baseUri: BaseUri,
+ request: __httpRequest!,
+ response: __response,
+ exception: null,
+ clientOptions: Options,
+ requestOptions: requestOptions,
+ attempt: __attempt,
+ maxAttempts: __maxAttempts,
+ willRetry: true,
+ retryDelay: __retryDelay,
+ retryReason: "status:" + ((int)__response.StatusCode).ToString(global::System.Globalization.CultureInfo.InvariantCulture),
+ cancellationToken: __effectiveCancellationToken)).ConfigureAwait(false);
+ __response.Dispose();
+ __response = null;
+ __httpRequest.Dispose();
+ __httpRequest = null;
+ await global::Baseten.AutoSDKRequestOptionsSupport.DelayBeforeRetryAsync(
+ retryDelay: __retryDelay,
+ cancellationToken: __effectiveCancellationToken).ConfigureAwait(false);
+ continue;
+ }
+
+ break;
+ }
+
+ if (__response == null)
+ {
+ throw new global::System.InvalidOperationException("No response received.");
+ }
+
+ using (__response)
+ {
+
+ ProcessResponse(
+ client: HttpClient,
+ response: __response);
+ ProcessCreateTrainersCheckpointsSearchResponse(
+ httpClient: HttpClient,
+ httpResponseMessage: __response);
+ if (__response.IsSuccessStatusCode)
+ {
+ await global::Baseten.AutoSDKRequestOptionsSupport.OnAfterSuccessAsync(
+ clientOptions: Options,
+ context: global::Baseten.AutoSDKRequestOptionsSupport.CreateHookContext(
+ operationId: "createTrainersCheckpointsSearch",
+ methodName: "CreateTrainersCheckpointsSearchAsync",
+ pathTemplate: "\"/v1/trainers/checkpoints/search\"",
+ httpMethod: "POST",
+ baseUri: BaseUri,
+ request: __httpRequest!,
+ response: __response,
+ exception: null,
+ clientOptions: Options,
+ requestOptions: requestOptions,
+ attempt: __attemptNumber,
+ maxAttempts: __maxAttempts,
+ willRetry: false,
+ retryDelay: null,
+ retryReason: global::System.String.Empty,
+ cancellationToken: __effectiveCancellationToken)).ConfigureAwait(false);
+ }
+ else
+ {
+ await global::Baseten.AutoSDKRequestOptionsSupport.OnAfterErrorAsync(
+ clientOptions: Options,
+ context: global::Baseten.AutoSDKRequestOptionsSupport.CreateHookContext(
+ operationId: "createTrainersCheckpointsSearch",
+ methodName: "CreateTrainersCheckpointsSearchAsync",
+ pathTemplate: "\"/v1/trainers/checkpoints/search\"",
+ httpMethod: "POST",
+ baseUri: BaseUri,
+ request: __httpRequest!,
+ response: __response,
+ exception: null,
+ clientOptions: Options,
+ requestOptions: requestOptions,
+ attempt: __attemptNumber,
+ maxAttempts: __maxAttempts,
+ willRetry: false,
+ retryDelay: null,
+ retryReason: global::System.String.Empty,
+ cancellationToken: __effectiveCancellationToken)).ConfigureAwait(false);
+ }
+
+ if (__effectiveReadResponseAsString)
+ {
+ var __content = await __response.Content.ReadAsStringAsync(
+ #if NET5_0_OR_GREATER
+ __effectiveCancellationToken
+ #endif
+ ).ConfigureAwait(false);
+
+ ProcessResponseContent(
+ client: HttpClient,
+ response: __response,
+ content: ref __content);
+ ProcessCreateTrainersCheckpointsSearchResponseContent(
+ httpClient: HttpClient,
+ httpResponseMessage: __response,
+ content: ref __content);
+
+ try
+ {
+ __response.EnsureSuccessStatusCode();
+
+ var __value = global::Baseten.CheckpointSearchResponseV1.FromJson(__content, JsonSerializerContext) ??
+ throw new global::System.InvalidOperationException($"Response deserialization failed for \"{__content}\" ");
+ return new global::Baseten.AutoSDKHttpResponse(
+ statusCode: __response.StatusCode,
+ headers: global::Baseten.AutoSDKHttpResponse.CreateHeaders(__response),
+ requestUri: __response.RequestMessage?.RequestUri,
+ body: __value);
+ }
+ catch (global::System.Exception __ex)
+ {
+ throw new global::Baseten.ApiException(
+ message: __content ?? __response.ReasonPhrase ?? string.Empty,
+ innerException: __ex,
+ statusCode: __response.StatusCode)
+ {
+ ResponseBody = __content,
+ ResponseHeaders = global::System.Linq.Enumerable.ToDictionary(
+ __response.Headers,
+ h => h.Key,
+ h => h.Value),
+ };
+ }
+ }
+ else
+ {
+ try
+ {
+ __response.EnsureSuccessStatusCode();
+ using var __content = await __response.Content.ReadAsStreamAsync(
+ #if NET5_0_OR_GREATER
+ __effectiveCancellationToken
+ #endif
+ ).ConfigureAwait(false);
+
+ var __value = await global::Baseten.CheckpointSearchResponseV1.FromJsonStreamAsync(__content, JsonSerializerContext).ConfigureAwait(false) ??
+ throw new global::System.InvalidOperationException("Response deserialization failed.");
+ return new global::Baseten.AutoSDKHttpResponse(
+ statusCode: __response.StatusCode,
+ headers: global::Baseten.AutoSDKHttpResponse.CreateHeaders(__response),
+ requestUri: __response.RequestMessage?.RequestUri,
+ body: __value);
+ }
+ catch (global::System.Exception __ex)
+ {
+ string? __content = null;
+ try
+ {
+ __content = await __response.Content.ReadAsStringAsync(
+ #if NET5_0_OR_GREATER
+ __effectiveCancellationToken
+ #endif
+ ).ConfigureAwait(false);
+ }
+ catch (global::System.Exception)
+ {
+ }
+
+ throw new global::Baseten.ApiException(
+ message: __content ?? __response.ReasonPhrase ?? string.Empty,
+ innerException: __ex,
+ statusCode: __response.StatusCode)
+ {
+ ResponseBody = __content,
+ ResponseHeaders = global::System.Linq.Enumerable.ToDictionary(
+ __response.Headers,
+ h => h.Key,
+ h => h.Value),
+ };
+ }
+ }
+
+ }
+ }
+ finally
+ {
+ __httpRequest?.Dispose();
+ }
+ }
+ ///
+ /// Look up trainer checkpoint info by bt:// URI.
+ /// Look up the metadata for a trainer checkpoint by its bt://trainers:<trainer_id>/(weights|sampler_weights)/<name> URI.
+ ///
+ ///
+ /// bt:// URI of a trainer checkpoint. Form: bt://trainers:<trainer_id>/(weights|sampler_weights)/<checkpoint_name>.
+ ///
+ /// Per-request overrides such as headers, query parameters, timeout, retries, and response buffering.
+ /// The token to cancel the operation with
+ ///
+ public async global::System.Threading.Tasks.Task CreateTrainersCheckpointsSearchAsync(
+ string checkpointPath,
+ global::Baseten.AutoSDKRequestOptions? requestOptions = default,
+ global::System.Threading.CancellationToken cancellationToken = default)
+ {
+ var __request = new global::Baseten.CheckpointSearchRequestV1
+ {
+ CheckpointPath = checkpointPath,
+ };
+
+ return await CreateTrainersCheckpointsSearchAsync(
+ request: __request,
+ requestOptions: requestOptions,
+ cancellationToken: cancellationToken).ConfigureAwait(false);
+ }
+ }
+}
\ No newline at end of file
diff --git a/src/libs/Baseten/Generated/Baseten.IBasetenClient.CreateTrainersSessions.g.cs b/src/libs/Baseten/Generated/Baseten.IBasetenClient.CreateTrainerSessions.g.cs
similarity index 84%
rename from src/libs/Baseten/Generated/Baseten.IBasetenClient.CreateTrainersSessions.g.cs
rename to src/libs/Baseten/Generated/Baseten.IBasetenClient.CreateTrainerSessions.g.cs
index 2a2ae30..fbc4cd3 100644
--- a/src/libs/Baseten/Generated/Baseten.IBasetenClient.CreateTrainersSessions.g.cs
+++ b/src/libs/Baseten/Generated/Baseten.IBasetenClient.CreateTrainerSessions.g.cs
@@ -14,13 +14,13 @@ public partial interface IBasetenClient
///
///
/// curl --request POST \
- /// --url https://api.baseten.co/v1/trainers/sessions \
+ /// --url https://api.baseten.co/v1/trainer_sessions \
/// --header "Authorization: Api-Key $BASETEN_API_KEY" \
/// --data '{
/// "training_project_id": null
/// }'
///
- global::System.Threading.Tasks.Task CreateTrainersSessionsAsync(
+ global::System.Threading.Tasks.Task CreateTrainerSessionsAsync(
global::Baseten.CreateTrainerSessionRequestV1 request,
global::Baseten.AutoSDKRequestOptions? requestOptions = default,
@@ -35,13 +35,13 @@ public partial interface IBasetenClient
///
///
/// curl --request POST \
- /// --url https://api.baseten.co/v1/trainers/sessions \
+ /// --url https://api.baseten.co/v1/trainer_sessions \
/// --header "Authorization: Api-Key $BASETEN_API_KEY" \
/// --data '{
/// "training_project_id": null
/// }'
///
- global::System.Threading.Tasks.Task> CreateTrainersSessionsAsResponseAsync(
+ global::System.Threading.Tasks.Task> CreateTrainerSessionsAsResponseAsync(
global::Baseten.CreateTrainerSessionRequestV1 request,
global::Baseten.AutoSDKRequestOptions? requestOptions = default,
@@ -51,13 +51,14 @@ public partial interface IBasetenClient
/// Creates a TrainerSession for the given training project.
///
///
- /// ID of the training project to associate with.
+ /// ID of the training project to associate with. If omitted, a default project is created for the org.
+ /// Default Value: openapi-json-null-sentinel-value-2BF93600-0FE4-4250-987A-E5DDB203E464
///
/// Per-request overrides such as headers, query parameters, timeout, retries, and response buffering.
/// The token to cancel the operation with
///
- global::System.Threading.Tasks.Task CreateTrainersSessionsAsync(
- string trainingProjectId,
+ global::System.Threading.Tasks.Task CreateTrainerSessionsAsync(
+ string? trainingProjectId = default,
global::Baseten.AutoSDKRequestOptions? requestOptions = default,
global::System.Threading.CancellationToken cancellationToken = default);
}
diff --git a/src/libs/Baseten/Generated/Baseten.IBasetenClient.CreateTrainersSessionsBySessionIdSamplers.g.cs b/src/libs/Baseten/Generated/Baseten.IBasetenClient.CreateTrainerSessionsBySessionIdSamplers.g.cs
similarity index 78%
rename from src/libs/Baseten/Generated/Baseten.IBasetenClient.CreateTrainersSessionsBySessionIdSamplers.g.cs
rename to src/libs/Baseten/Generated/Baseten.IBasetenClient.CreateTrainerSessionsBySessionIdSamplers.g.cs
index bcd1680..be79595 100644
--- a/src/libs/Baseten/Generated/Baseten.IBasetenClient.CreateTrainersSessionsBySessionIdSamplers.g.cs
+++ b/src/libs/Baseten/Generated/Baseten.IBasetenClient.CreateTrainerSessionsBySessionIdSamplers.g.cs
@@ -15,14 +15,15 @@ public partial interface IBasetenClient
///
///
/// curl --request POST \
- /// --url https://api.baseten.co/v1/trainers/sessions/{session_id}/samplers \
+ /// --url https://api.baseten.co/v1/trainer_sessions/{session_id}/samplers \
/// --header "Authorization: Api-Key $BASETEN_API_KEY" \
/// --data '{
/// "model": null,
- /// "max_seq_length": null
+ /// "max_seq_length": null,
+ /// "checkpoint_path": "bt://trainers:k4q95w5/sampler_weights/step-100"
/// }'
///
- global::System.Threading.Tasks.Task CreateTrainersSessionsBySessionIdSamplersAsync(
+ global::System.Threading.Tasks.Task CreateTrainerSessionsBySessionIdSamplersAsync(
string sessionId,
global::Baseten.CreateSamplingServerRequestV1 request,
@@ -39,14 +40,15 @@ public partial interface IBasetenClient
///
///
/// curl --request POST \
- /// --url https://api.baseten.co/v1/trainers/sessions/{session_id}/samplers \
+ /// --url https://api.baseten.co/v1/trainer_sessions/{session_id}/samplers \
/// --header "Authorization: Api-Key $BASETEN_API_KEY" \
/// --data '{
/// "model": null,
- /// "max_seq_length": null
+ /// "max_seq_length": null,
+ /// "checkpoint_path": "bt://trainers:k4q95w5/sampler_weights/step-100"
/// }'
///
- global::System.Threading.Tasks.Task> CreateTrainersSessionsBySessionIdSamplersAsResponseAsync(
+ global::System.Threading.Tasks.Task> CreateTrainerSessionsBySessionIdSamplersAsResponseAsync(
string sessionId,
global::Baseten.CreateSamplingServerRequestV1 request,
@@ -64,13 +66,18 @@ public partial interface IBasetenClient
/// Minimum max-sequence-length the sampler must support. If set, only instance-type entries whose configured max_seq_length is >= this value are considered. Omit to accept any entry.
/// Default Value: openapi-json-null-sentinel-value-2BF93600-0FE4-4250-987A-E5DDB203E464
///
+ ///
+ /// Optional bt:// URI of an existing sampler-target checkpoint to load weights from on startup. Form: bt://trainers:<trainer_id>/sampler_weights/<checkpoint_name>.
+ /// Default Value: openapi-json-null-sentinel-value-2BF93600-0FE4-4250-987A-E5DDB203E464
+ ///
/// Per-request overrides such as headers, query parameters, timeout, retries, and response buffering.
/// The token to cancel the operation with
///
- global::System.Threading.Tasks.Task CreateTrainersSessionsBySessionIdSamplersAsync(
+ global::System.Threading.Tasks.Task CreateTrainerSessionsBySessionIdSamplersAsync(
string sessionId,
string model,
int? maxSeqLength = default,
+ string? checkpointPath = default,
global::Baseten.AutoSDKRequestOptions? requestOptions = default,
global::System.Threading.CancellationToken cancellationToken = default);
}
diff --git a/src/libs/Baseten/Generated/Baseten.IBasetenClient.CreateTrainersSessionsBySessionIdTrainers.g.cs b/src/libs/Baseten/Generated/Baseten.IBasetenClient.CreateTrainerSessionsBySessionIdTrainers.g.cs
similarity index 81%
rename from src/libs/Baseten/Generated/Baseten.IBasetenClient.CreateTrainersSessionsBySessionIdTrainers.g.cs
rename to src/libs/Baseten/Generated/Baseten.IBasetenClient.CreateTrainerSessionsBySessionIdTrainers.g.cs
index f6e4b19..b6f436e 100644
--- a/src/libs/Baseten/Generated/Baseten.IBasetenClient.CreateTrainersSessionsBySessionIdTrainers.g.cs
+++ b/src/libs/Baseten/Generated/Baseten.IBasetenClient.CreateTrainerSessionsBySessionIdTrainers.g.cs
@@ -15,17 +15,18 @@ public partial interface IBasetenClient
///
///
/// curl --request POST \
- /// --url https://api.baseten.co/v1/trainers/sessions/{session_id}/trainers \
+ /// --url https://api.baseten.co/v1/trainer_sessions/{session_id}/trainers \
/// --header "Authorization: Api-Key $BASETEN_API_KEY" \
/// --data '{
/// "model": null,
/// "max_seq_len": null,
/// "lora_rank": null,
/// "seed": null,
- /// "scale_down_delay_seconds": null
+ /// "scale_down_delay_seconds": null,
+ /// "checkpoint_path": "bt://trainers:k4q95w5/weights/step-100"
/// }'
///
- global::System.Threading.Tasks.Task CreateTrainersSessionsBySessionIdTrainersAsync(
+ global::System.Threading.Tasks.Task CreateTrainerSessionsBySessionIdTrainersAsync(
string sessionId,
global::Baseten.CreateTrainerServerRequestV1 request,
@@ -42,17 +43,18 @@ public partial interface IBasetenClient
///
///
/// curl --request POST \
- /// --url https://api.baseten.co/v1/trainers/sessions/{session_id}/trainers \
+ /// --url https://api.baseten.co/v1/trainer_sessions/{session_id}/trainers \
/// --header "Authorization: Api-Key $BASETEN_API_KEY" \
/// --data '{
/// "model": null,
/// "max_seq_len": null,
/// "lora_rank": null,
/// "seed": null,
- /// "scale_down_delay_seconds": null
+ /// "scale_down_delay_seconds": null,
+ /// "checkpoint_path": "bt://trainers:k4q95w5/weights/step-100"
/// }'
///
- global::System.Threading.Tasks.Task> CreateTrainersSessionsBySessionIdTrainersAsResponseAsync(
+ global::System.Threading.Tasks.Task> CreateTrainerSessionsBySessionIdTrainersAsResponseAsync(
string sessionId,
global::Baseten.CreateTrainerServerRequestV1 request,
@@ -82,16 +84,21 @@ public partial interface IBasetenClient
/// Seconds of inactivity before the trainer scales to zero. Must be positive. Defaults to 3600 (1 hour).
/// Default Value: 3600
///
+ ///
+ /// Optional bt:// URI of an existing trainer-target checkpoint to resume training from. Form: bt://trainers:<trainer_id>/weights/<checkpoint_name>.
+ /// Default Value: openapi-json-null-sentinel-value-2BF93600-0FE4-4250-987A-E5DDB203E464
+ ///
/// Per-request overrides such as headers, query parameters, timeout, retries, and response buffering.
/// The token to cancel the operation with
///
- global::System.Threading.Tasks.Task CreateTrainersSessionsBySessionIdTrainersAsync(
+ global::System.Threading.Tasks.Task CreateTrainerSessionsBySessionIdTrainersAsync(
string sessionId,
string model,
int? maxSeqLen = default,
int? loraRank = default,
int? seed = default,
int? scaleDownDelaySeconds = default,
+ string? checkpointPath = default,
global::Baseten.AutoSDKRequestOptions? requestOptions = default,
global::System.Threading.CancellationToken cancellationToken = default);
}
diff --git a/src/libs/Baseten/Generated/Baseten.IBasetenClient.CreateTrainersCheckpointsSearch.g.cs b/src/libs/Baseten/Generated/Baseten.IBasetenClient.CreateTrainersCheckpointsSearch.g.cs
new file mode 100644
index 0000000..468f31b
--- /dev/null
+++ b/src/libs/Baseten/Generated/Baseten.IBasetenClient.CreateTrainersCheckpointsSearch.g.cs
@@ -0,0 +1,64 @@
+#nullable enable
+
+namespace Baseten
+{
+ public partial interface IBasetenClient
+ {
+ ///
+ /// Look up trainer checkpoint info by bt:// URI.
+ /// Look up the metadata for a trainer checkpoint by its bt://trainers:<trainer_id>/(weights|sampler_weights)/<name> URI.
+ ///
+ ///
+ /// Per-request overrides such as headers, query parameters, timeout, retries, and response buffering.
+ /// The token to cancel the operation with
+ ///
+ ///
+ /// curl --request POST \
+ /// --url https://api.baseten.co/v1/trainers/checkpoints/search \
+ /// --header "Authorization: Api-Key $BASETEN_API_KEY" \
+ /// --data '{
+ /// "checkpoint_path": "bt://trainers:k4q95w5/sampler_weights/step-100"
+ /// }'
+ ///
+ global::System.Threading.Tasks.Task CreateTrainersCheckpointsSearchAsync(
+
+ global::Baseten.CheckpointSearchRequestV1 request,
+ global::Baseten.AutoSDKRequestOptions? requestOptions = default,
+ global::System.Threading.CancellationToken cancellationToken = default);
+ ///
+ /// Look up trainer checkpoint info by bt:// URI.
+ /// Look up the metadata for a trainer checkpoint by its bt://trainers:<trainer_id>/(weights|sampler_weights)/<name> URI.
+ ///
+ ///
+ /// Per-request overrides such as headers, query parameters, timeout, retries, and response buffering.
+ /// The token to cancel the operation with
+ ///
+ ///
+ /// curl --request POST \
+ /// --url https://api.baseten.co/v1/trainers/checkpoints/search \
+ /// --header "Authorization: Api-Key $BASETEN_API_KEY" \
+ /// --data '{
+ /// "checkpoint_path": "bt://trainers:k4q95w5/sampler_weights/step-100"
+ /// }'
+ ///
+ global::System.Threading.Tasks.Task> CreateTrainersCheckpointsSearchAsResponseAsync(
+
+ global::Baseten.CheckpointSearchRequestV1 request,
+ global::Baseten.AutoSDKRequestOptions? requestOptions = default,
+ global::System.Threading.CancellationToken cancellationToken = default);
+ ///
+ /// Look up trainer checkpoint info by bt:// URI.
+ /// Look up the metadata for a trainer checkpoint by its bt://trainers:<trainer_id>/(weights|sampler_weights)/<name> URI.
+ ///
+ ///
+ /// bt:// URI of a trainer checkpoint. Form: bt://trainers:<trainer_id>/(weights|sampler_weights)/<checkpoint_name>.
+ ///
+ /// Per-request overrides such as headers, query parameters, timeout, retries, and response buffering.
+ /// The token to cancel the operation with
+ ///
+ global::System.Threading.Tasks.Task CreateTrainersCheckpointsSearchAsync(
+ string checkpointPath,
+ global::Baseten.AutoSDKRequestOptions? requestOptions = default,
+ global::System.Threading.CancellationToken cancellationToken = default);
+ }
+}
\ No newline at end of file
diff --git a/src/libs/Baseten/Generated/Baseten.JsonConverters.TrainerCheckpointTarget.g.cs b/src/libs/Baseten/Generated/Baseten.JsonConverters.TrainerCheckpointTarget.g.cs
new file mode 100644
index 0000000..ab448a5
--- /dev/null
+++ b/src/libs/Baseten/Generated/Baseten.JsonConverters.TrainerCheckpointTarget.g.cs
@@ -0,0 +1,53 @@
+#nullable enable
+
+namespace Baseten.JsonConverters
+{
+ ///
+ public sealed class TrainerCheckpointTargetJsonConverter : global::System.Text.Json.Serialization.JsonConverter
+ {
+ ///
+ public override global::Baseten.TrainerCheckpointTarget Read(
+ ref global::System.Text.Json.Utf8JsonReader reader,
+ global::System.Type typeToConvert,
+ global::System.Text.Json.JsonSerializerOptions options)
+ {
+ switch (reader.TokenType)
+ {
+ case global::System.Text.Json.JsonTokenType.String:
+ {
+ var stringValue = reader.GetString();
+ if (stringValue != null)
+ {
+ return global::Baseten.TrainerCheckpointTargetExtensions.ToEnum(stringValue) ?? default;
+ }
+
+ break;
+ }
+ case global::System.Text.Json.JsonTokenType.Number:
+ {
+ var numValue = reader.GetInt32();
+ return (global::Baseten.TrainerCheckpointTarget)numValue;
+ }
+ case global::System.Text.Json.JsonTokenType.Null:
+ {
+ return default(global::Baseten.TrainerCheckpointTarget);
+ }
+ default:
+ throw new global::System.ArgumentOutOfRangeException(nameof(reader));
+ }
+
+ return default;
+ }
+
+ ///
+ public override void Write(
+ global::System.Text.Json.Utf8JsonWriter writer,
+ global::Baseten.TrainerCheckpointTarget value,
+ global::System.Text.Json.JsonSerializerOptions options)
+ {
+ writer = writer ?? throw new global::System.ArgumentNullException(nameof(writer));
+
+ writer.WriteStringValue(global::Baseten.TrainerCheckpointTargetExtensions.ToValueString(value));
+ }
+ }
+}
diff --git a/src/libs/Baseten/Generated/Baseten.JsonConverters.TrainerCheckpointTargetNullable.g.cs b/src/libs/Baseten/Generated/Baseten.JsonConverters.TrainerCheckpointTargetNullable.g.cs
new file mode 100644
index 0000000..c95287f
--- /dev/null
+++ b/src/libs/Baseten/Generated/Baseten.JsonConverters.TrainerCheckpointTargetNullable.g.cs
@@ -0,0 +1,60 @@
+#nullable enable
+
+namespace Baseten.JsonConverters
+{
+ ///
+ public sealed class TrainerCheckpointTargetNullableJsonConverter : global::System.Text.Json.Serialization.JsonConverter
+ {
+ ///
+ public override global::Baseten.TrainerCheckpointTarget? Read(
+ ref global::System.Text.Json.Utf8JsonReader reader,
+ global::System.Type typeToConvert,
+ global::System.Text.Json.JsonSerializerOptions options)
+ {
+ switch (reader.TokenType)
+ {
+ case global::System.Text.Json.JsonTokenType.String:
+ {
+ var stringValue = reader.GetString();
+ if (stringValue != null)
+ {
+ return global::Baseten.TrainerCheckpointTargetExtensions.ToEnum(stringValue);
+ }
+
+ break;
+ }
+ case global::System.Text.Json.JsonTokenType.Number:
+ {
+ var numValue = reader.GetInt32();
+ return (global::Baseten.TrainerCheckpointTarget)numValue;
+ }
+ case global::System.Text.Json.JsonTokenType.Null:
+ {
+ return default(global::Baseten.TrainerCheckpointTarget?);
+ }
+ default:
+ throw new global::System.ArgumentOutOfRangeException(nameof(reader));
+ }
+
+ return default;
+ }
+
+ ///
+ public override void Write(
+ global::System.Text.Json.Utf8JsonWriter writer,
+ global::Baseten.TrainerCheckpointTarget? value,
+ global::System.Text.Json.JsonSerializerOptions options)
+ {
+ writer = writer ?? throw new global::System.ArgumentNullException(nameof(writer));
+
+ if (value == null)
+ {
+ writer.WriteNullValue();
+ }
+ else
+ {
+ writer.WriteStringValue(global::Baseten.TrainerCheckpointTargetExtensions.ToValueString(value.Value));
+ }
+ }
+ }
+}
diff --git a/src/libs/Baseten/Generated/Baseten.JsonSerializerContext.g.cs b/src/libs/Baseten/Generated/Baseten.JsonSerializerContext.g.cs
index 13575d2..277036a 100644
--- a/src/libs/Baseten/Generated/Baseten.JsonSerializerContext.g.cs
+++ b/src/libs/Baseten/Generated/Baseten.JsonSerializerContext.g.cs
@@ -65,6 +65,10 @@ namespace Baseten
typeof(global::Baseten.JsonConverters.V1InteractiveSessionTriggerNullableJsonConverter),
+ typeof(global::Baseten.JsonConverters.TrainerCheckpointTargetJsonConverter),
+
+ typeof(global::Baseten.JsonConverters.TrainerCheckpointTargetNullableJsonConverter),
+
typeof(global::Baseten.JsonConverters.APIKeyCategoryJsonConverter),
typeof(global::Baseten.JsonConverters.APIKeyCategoryNullableJsonConverter),
@@ -289,6 +293,10 @@ namespace Baseten
[global::System.Text.Json.Serialization.JsonSerializable(typeof(global::Baseten.SearchedTrainerV1))]
[global::System.Text.Json.Serialization.JsonSerializable(typeof(global::Baseten.SearchTrainersResponseV1))]
[global::System.Text.Json.Serialization.JsonSerializable(typeof(global::System.Collections.Generic.IList))]
+ [global::System.Text.Json.Serialization.JsonSerializable(typeof(global::Baseten.CheckpointSearchRequestV1))]
+ [global::System.Text.Json.Serialization.JsonSerializable(typeof(global::Baseten.TrainerCheckpointTarget), TypeInfoPropertyName = "TrainerCheckpointTarget2")]
+ [global::System.Text.Json.Serialization.JsonSerializable(typeof(global::Baseten.TrainerServerCheckpointV1))]
+ [global::System.Text.Json.Serialization.JsonSerializable(typeof(global::Baseten.CheckpointSearchResponseV1))]
[global::System.Text.Json.Serialization.JsonSerializable(typeof(global::Baseten.CreateTrainerSessionRequestV1))]
[global::System.Text.Json.Serialization.JsonSerializable(typeof(global::Baseten.TrainerSessionV1))]
[global::System.Text.Json.Serialization.JsonSerializable(typeof(global::Baseten.CreateTrainerSessionResponseV1))]
@@ -298,7 +306,6 @@ namespace Baseten
[global::System.Text.Json.Serialization.JsonSerializable(typeof(global::Baseten.CreateTrainerServerResponseV1))]
[global::System.Text.Json.Serialization.JsonSerializable(typeof(global::Baseten.CreateSamplingServerRequestV1))]
[global::System.Text.Json.Serialization.JsonSerializable(typeof(global::Baseten.CreateSamplingServerResponseV1))]
- [global::System.Text.Json.Serialization.JsonSerializable(typeof(global::Baseten.TrainerServerCheckpointV1))]
[global::System.Text.Json.Serialization.JsonSerializable(typeof(global::Baseten.GetTrainerServerCheckpointsResponseV1))]
[global::System.Text.Json.Serialization.JsonSerializable(typeof(global::System.Collections.Generic.IList))]
[global::System.Text.Json.Serialization.JsonSerializable(typeof(global::Baseten.GetTrainerServerCheckpointFilesResponseV1))]
diff --git a/src/libs/Baseten/Generated/Baseten.JsonSerializerContextTypes.g.cs b/src/libs/Baseten/Generated/Baseten.JsonSerializerContextTypes.g.cs
index 0fe9be3..426de02 100644
--- a/src/libs/Baseten/Generated/Baseten.JsonSerializerContextTypes.g.cs
+++ b/src/libs/Baseten/Generated/Baseten.JsonSerializerContextTypes.g.cs
@@ -744,259 +744,271 @@ public sealed partial class JsonSerializerContextTypes
///
///
///
- public global::Baseten.CreateTrainerSessionRequestV1? Type179 { get; set; }
+ public global::Baseten.CheckpointSearchRequestV1? Type179 { get; set; }
///
///
///
- public global::Baseten.TrainerSessionV1? Type180 { get; set; }
+ public global::Baseten.TrainerCheckpointTarget? Type180 { get; set; }
///
///
///
- public global::Baseten.CreateTrainerSessionResponseV1? Type181 { get; set; }
+ public global::Baseten.TrainerServerCheckpointV1? Type181 { get; set; }
///
///
///
- public global::Baseten.CreateTrainerServerRequestV1? Type182 { get; set; }
+ public global::Baseten.CheckpointSearchResponseV1? Type182 { get; set; }
///
///
///
- public global::Baseten.SamplingServerV1? Type183 { get; set; }
+ public global::Baseten.CreateTrainerSessionRequestV1? Type183 { get; set; }
///
///
///
- public global::Baseten.TrainerServerV1? Type184 { get; set; }
+ public global::Baseten.TrainerSessionV1? Type184 { get; set; }
///
///
///
- public global::Baseten.CreateTrainerServerResponseV1? Type185 { get; set; }
+ public global::Baseten.CreateTrainerSessionResponseV1? Type185 { get; set; }
///
///
///
- public global::Baseten.CreateSamplingServerRequestV1? Type186 { get; set; }
+ public global::Baseten.CreateTrainerServerRequestV1? Type186 { get; set; }
///
///
///
- public global::Baseten.CreateSamplingServerResponseV1? Type187 { get; set; }
+ public global::Baseten.SamplingServerV1? Type187 { get; set; }
///
///
///
- public global::Baseten.TrainerServerCheckpointV1? Type188 { get; set; }
+ public global::Baseten.TrainerServerV1? Type188 { get; set; }
///
///
///
- public global::Baseten.GetTrainerServerCheckpointsResponseV1? Type189 { get; set; }
+ public global::Baseten.CreateTrainerServerResponseV1? Type189 { get; set; }
///
///
///
- public global::System.Collections.Generic.IList? Type190 { get; set; }
+ public global::Baseten.CreateSamplingServerRequestV1? Type190 { get; set; }
///
///
///
- public global::Baseten.GetTrainerServerCheckpointFilesResponseV1? Type191 { get; set; }
+ public global::Baseten.CreateSamplingServerResponseV1? Type191 { get; set; }
///
///
///
- public global::Baseten.TrainingGpuCapacityItemV1? Type192 { get; set; }
+ public global::Baseten.GetTrainerServerCheckpointsResponseV1? Type192 { get; set; }
///
///
///
- public global::Baseten.GetTrainingGpuCapacityResponseV1? Type193 { get; set; }
+ public global::System.Collections.Generic.IList? Type193 { get; set; }
///
///
///
- public global::System.Collections.Generic.IList? Type194 { get; set; }
+ public global::Baseten.GetTrainerServerCheckpointFilesResponseV1? Type194 { get; set; }
///
///
///
- public global::Baseten.AWSCredentialsV1? Type195 { get; set; }
+ public global::Baseten.TrainingGpuCapacityItemV1? Type195 { get; set; }
///
///
///
- public global::Baseten.GetBlobCredentialsResponseV1? Type196 { get; set; }
+ public global::Baseten.GetTrainingGpuCapacityResponseV1? Type196 { get; set; }
///
///
///
- public global::Baseten.APIKeyCategory? Type197 { get; set; }
+ public global::System.Collections.Generic.IList? Type197 { get; set; }
///
///
///
- public global::Baseten.CreateAPIKeyRequestV1? Type198 { get; set; }
+ public global::Baseten.AWSCredentialsV1? Type198 { get; set; }
///
///
///
- public global::Baseten.APIKeyV1? Type199 { get; set; }
+ public global::Baseten.GetBlobCredentialsResponseV1? Type199 { get; set; }
///
///
///
- public global::Baseten.APIKeyInfoV1? Type200 { get; set; }
+ public global::Baseten.APIKeyCategory? Type200 { get; set; }
///
///
///
- public global::Baseten.APIKeysV1? Type201 { get; set; }
+ public global::Baseten.CreateAPIKeyRequestV1? Type201 { get; set; }
///
///
///
- public global::System.Collections.Generic.IList? Type202 { get; set; }
+ public global::Baseten.APIKeyV1? Type202 { get; set; }
///
///
///
- public global::Baseten.APIKeyTombstoneV1? Type203 { get; set; }
+ public global::Baseten.APIKeyInfoV1? Type203 { get; set; }
///
///
///
- public global::Baseten.ModelWeightSnapshotV1? Type204 { get; set; }
+ public global::Baseten.APIKeysV1? Type204 { get; set; }
///
///
///
- public global::Baseten.CreateModelWeightSnapshotRequestV1? Type205 { get; set; }
+ public global::System.Collections.Generic.IList? Type205 { get; set; }
///
///
///
- public global::Baseten.CreateLLMModelRequestV1? Type206 { get; set; }
+ public global::Baseten.APIKeyTombstoneV1? Type206 { get; set; }
///
///
///
- public global::System.Collections.Generic.IList