// Code generated by protoc-gen-goext. DO NOT EDIT.

package foundation_models

import (
	v1 "github.com/yandex-cloud/go-genproto/yandex/cloud/ai/foundation_models/v1"
	wrapperspb "google.golang.org/protobuf/types/known/wrapperspb"
)

type CompletionRequest_ResponseFormat = isCompletionRequest_ResponseFormat

func (m *CompletionRequest) SetResponseFormat(v CompletionRequest_ResponseFormat) {
	m.ResponseFormat = v
}

func (m *CompletionRequest) SetModelUri(v string) {
	m.ModelUri = v
}

func (m *CompletionRequest) SetCompletionOptions(v *v1.CompletionOptions) {
	m.CompletionOptions = v
}

func (m *CompletionRequest) SetMessages(v []*v1.Message) {
	m.Messages = v
}

func (m *CompletionRequest) SetTools(v []*v1.Tool) {
	m.Tools = v
}

func (m *CompletionRequest) SetJsonObject(v bool) {
	m.ResponseFormat = &CompletionRequest_JsonObject{
		JsonObject: v,
	}
}

func (m *CompletionRequest) SetJsonSchema(v *v1.JsonSchema) {
	m.ResponseFormat = &CompletionRequest_JsonSchema{
		JsonSchema: v,
	}
}

func (m *CompletionRequest) SetParallelToolCalls(v *wrapperspb.BoolValue) {
	m.ParallelToolCalls = v
}

func (m *CompletionRequest) SetToolChoice(v *v1.ToolChoice) {
	m.ToolChoice = v
}

func (m *CompletionResponse) SetAlternatives(v []*v1.Alternative) {
	m.Alternatives = v
}

func (m *CompletionResponse) SetUsage(v *v1.ContentUsage) {
	m.Usage = v
}

func (m *CompletionResponse) SetModelVersion(v string) {
	m.ModelVersion = v
}

type BatchCompletionRequest_RequestFormat = isBatchCompletionRequest_RequestFormat

func (m *BatchCompletionRequest) SetRequestFormat(v BatchCompletionRequest_RequestFormat) {
	m.RequestFormat = v
}

type BatchCompletionRequest_ResponseFormat = isBatchCompletionRequest_ResponseFormat

func (m *BatchCompletionRequest) SetResponseFormat(v BatchCompletionRequest_ResponseFormat) {
	m.ResponseFormat = v
}

func (m *BatchCompletionRequest) SetModelUri(v string) {
	m.ModelUri = v
}

func (m *BatchCompletionRequest) SetCompletionOptions(v *v1.CompletionOptions) {
	m.CompletionOptions = v
}

func (m *BatchCompletionRequest) SetSourceDatasetId(v string) {
	m.RequestFormat = &BatchCompletionRequest_SourceDatasetId{
		SourceDatasetId: v,
	}
}

func (m *BatchCompletionRequest) SetJsonObject(v bool) {
	m.ResponseFormat = &BatchCompletionRequest_JsonObject{
		JsonObject: v,
	}
}

func (m *BatchCompletionRequest) SetJsonSchema(v *v1.JsonSchema) {
	m.ResponseFormat = &BatchCompletionRequest_JsonSchema{
		JsonSchema: v,
	}
}

func (m *BatchCompletionMetadata) SetTaskId(v string) {
	m.TaskId = v
}

func (m *BatchCompletionMetadata) SetTaskStatus(v v1.BatchInferenceTaskStatus) {
	m.TaskStatus = v
}

func (m *BatchCompletionMetadata) SetCompletedBatches(v int64) {
	m.CompletedBatches = v
}

func (m *BatchCompletionMetadata) SetTotalBatches(v int64) {
	m.TotalBatches = v
}

func (m *BatchCompletionResponse) SetTaskId(v string) {
	m.TaskId = v
}

func (m *BatchCompletionResponse) SetTaskStatus(v v1.BatchInferenceTaskStatus) {
	m.TaskStatus = v
}

func (m *BatchCompletionResponse) SetResultDatasetId(v string) {
	m.ResultDatasetId = v
}

func (m *TokenizeRequest) SetModelUri(v string) {
	m.ModelUri = v
}

func (m *TokenizeRequest) SetText(v string) {
	m.Text = v
}

func (m *TokenizeResponse) SetTokens(v []*v1.Token) {
	m.Tokens = v
}

func (m *TokenizeResponse) SetModelVersion(v string) {
	m.ModelVersion = v
}
