// Code generated by protoc-gen-go. DO NOT EDIT.
// versions:
// 	protoc-gen-go v1.36.6
// 	protoc        v3.21.12
// source: yandex/cloud/ai/foundation_models/v1/text_generation/text_generation_service.proto

package foundation_models

import (
	v1 "github.com/yandex-cloud/go-genproto/yandex/cloud/ai/foundation_models/v1"
	_ "github.com/yandex-cloud/go-genproto/yandex/cloud/api"
	operation "github.com/yandex-cloud/go-genproto/yandex/cloud/operation"
	_ "google.golang.org/genproto/googleapis/api/annotations"
	protoreflect "google.golang.org/protobuf/reflect/protoreflect"
	protoimpl "google.golang.org/protobuf/runtime/protoimpl"
	wrapperspb "google.golang.org/protobuf/types/known/wrapperspb"
	reflect "reflect"
	sync "sync"
	unsafe "unsafe"
)

const (
	// Verify that this generated code is sufficiently up-to-date.
	_ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion)
	// Verify that runtime/protoimpl is sufficiently up-to-date.
	_ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20)
)

// Request for the service to generate text completion.
type CompletionRequest struct {
	state protoimpl.MessageState `protogen:"open.v1"`
	// The [ID of the model](/docs/foundation-models/concepts/yandexgpt/models) to be used for completion generation.
	ModelUri string `protobuf:"bytes,1,opt,name=model_uri,json=modelUri,proto3" json:"model_uri,omitempty"`
	// Configuration options for completion generation.
	CompletionOptions *v1.CompletionOptions `protobuf:"bytes,2,opt,name=completion_options,json=completionOptions,proto3" json:"completion_options,omitempty"`
	// A list of messages representing the context for the completion model.
	Messages []*v1.Message `protobuf:"bytes,3,rep,name=messages,proto3" json:"messages,omitempty"`
	// List of tools that are available for the model to invoke during the completion generation.
	// Note: This parameter is not yet supported and will be ignored if provided.
	Tools []*v1.Tool `protobuf:"bytes,4,rep,name=tools,proto3" json:"tools,omitempty"`
	// Specifies the format of the model's response.
	//
	// Types that are valid to be assigned to ResponseFormat:
	//
	//	*CompletionRequest_JsonObject
	//	*CompletionRequest_JsonSchema
	ResponseFormat isCompletionRequest_ResponseFormat `protobuf_oneof:"ResponseFormat"`
	// Controls whether the model can generate multiple tool calls in a single response. Defaults to true.
	ParallelToolCalls *wrapperspb.BoolValue `protobuf:"bytes,7,opt,name=parallel_tool_calls,json=parallelToolCalls,proto3" json:"parallel_tool_calls,omitempty"`
	// Specifies how the model should select which tool (or tools) to use when generating a response.
	ToolChoice    *v1.ToolChoice `protobuf:"bytes,8,opt,name=tool_choice,json=toolChoice,proto3" json:"tool_choice,omitempty"`
	unknownFields protoimpl.UnknownFields
	sizeCache     protoimpl.SizeCache
}

func (x *CompletionRequest) Reset() {
	*x = CompletionRequest{}
	mi := &file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_msgTypes[0]
	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
	ms.StoreMessageInfo(mi)
}

func (x *CompletionRequest) String() string {
	return protoimpl.X.MessageStringOf(x)
}

func (*CompletionRequest) ProtoMessage() {}

func (x *CompletionRequest) ProtoReflect() protoreflect.Message {
	mi := &file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_msgTypes[0]
	if x != nil {
		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
		if ms.LoadMessageInfo() == nil {
			ms.StoreMessageInfo(mi)
		}
		return ms
	}
	return mi.MessageOf(x)
}

// Deprecated: Use CompletionRequest.ProtoReflect.Descriptor instead.
func (*CompletionRequest) Descriptor() ([]byte, []int) {
	return file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_rawDescGZIP(), []int{0}
}

func (x *CompletionRequest) GetModelUri() string {
	if x != nil {
		return x.ModelUri
	}
	return ""
}

func (x *CompletionRequest) GetCompletionOptions() *v1.CompletionOptions {
	if x != nil {
		return x.CompletionOptions
	}
	return nil
}

func (x *CompletionRequest) GetMessages() []*v1.Message {
	if x != nil {
		return x.Messages
	}
	return nil
}

func (x *CompletionRequest) GetTools() []*v1.Tool {
	if x != nil {
		return x.Tools
	}
	return nil
}

func (x *CompletionRequest) GetResponseFormat() isCompletionRequest_ResponseFormat {
	if x != nil {
		return x.ResponseFormat
	}
	return nil
}

func (x *CompletionRequest) GetJsonObject() bool {
	if x != nil {
		if x, ok := x.ResponseFormat.(*CompletionRequest_JsonObject); ok {
			return x.JsonObject
		}
	}
	return false
}

func (x *CompletionRequest) GetJsonSchema() *v1.JsonSchema {
	if x != nil {
		if x, ok := x.ResponseFormat.(*CompletionRequest_JsonSchema); ok {
			return x.JsonSchema
		}
	}
	return nil
}

func (x *CompletionRequest) GetParallelToolCalls() *wrapperspb.BoolValue {
	if x != nil {
		return x.ParallelToolCalls
	}
	return nil
}

func (x *CompletionRequest) GetToolChoice() *v1.ToolChoice {
	if x != nil {
		return x.ToolChoice
	}
	return nil
}

type isCompletionRequest_ResponseFormat interface {
	isCompletionRequest_ResponseFormat()
}

type CompletionRequest_JsonObject struct {
	// When set to true, the model will respond with a valid JSON object.
	// Be sure to explicitly ask the model for JSON.
	// Otherwise, it may generate excessive whitespace and run indefinitely until it reaches the token limit.
	JsonObject bool `protobuf:"varint,5,opt,name=json_object,json=jsonObject,proto3,oneof"`
}

type CompletionRequest_JsonSchema struct {
	// Enforces a specific JSON structure for the model's response based on a provided schema.
	JsonSchema *v1.JsonSchema `protobuf:"bytes,6,opt,name=json_schema,json=jsonSchema,proto3,oneof"`
}

func (*CompletionRequest_JsonObject) isCompletionRequest_ResponseFormat() {}

func (*CompletionRequest_JsonSchema) isCompletionRequest_ResponseFormat() {}

// Response containing generated text completions.
type CompletionResponse struct {
	state protoimpl.MessageState `protogen:"open.v1"`
	// A list of generated completion alternatives.
	Alternatives []*v1.Alternative `protobuf:"bytes,1,rep,name=alternatives,proto3" json:"alternatives,omitempty"`
	// A set of statistics describing the number of content tokens used by the completion model.
	Usage *v1.ContentUsage `protobuf:"bytes,2,opt,name=usage,proto3" json:"usage,omitempty"`
	// The model version changes with each new releases.
	ModelVersion  string `protobuf:"bytes,3,opt,name=model_version,json=modelVersion,proto3" json:"model_version,omitempty"`
	unknownFields protoimpl.UnknownFields
	sizeCache     protoimpl.SizeCache
}

func (x *CompletionResponse) Reset() {
	*x = CompletionResponse{}
	mi := &file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_msgTypes[1]
	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
	ms.StoreMessageInfo(mi)
}

func (x *CompletionResponse) String() string {
	return protoimpl.X.MessageStringOf(x)
}

func (*CompletionResponse) ProtoMessage() {}

func (x *CompletionResponse) ProtoReflect() protoreflect.Message {
	mi := &file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_msgTypes[1]
	if x != nil {
		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
		if ms.LoadMessageInfo() == nil {
			ms.StoreMessageInfo(mi)
		}
		return ms
	}
	return mi.MessageOf(x)
}

// Deprecated: Use CompletionResponse.ProtoReflect.Descriptor instead.
func (*CompletionResponse) Descriptor() ([]byte, []int) {
	return file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_rawDescGZIP(), []int{1}
}

func (x *CompletionResponse) GetAlternatives() []*v1.Alternative {
	if x != nil {
		return x.Alternatives
	}
	return nil
}

func (x *CompletionResponse) GetUsage() *v1.ContentUsage {
	if x != nil {
		return x.Usage
	}
	return nil
}

func (x *CompletionResponse) GetModelVersion() string {
	if x != nil {
		return x.ModelVersion
	}
	return ""
}

// Request for the service to generate batch text completion.
type BatchCompletionRequest struct {
	state protoimpl.MessageState `protogen:"open.v1"`
	// The [ID of the model](/docs/foundation-models/concepts/yandexgpt/models) to be used for batch completion generation.
	ModelUri string `protobuf:"bytes,1,opt,name=model_uri,json=modelUri,proto3" json:"model_uri,omitempty"`
	// Configuration options for completion generation.
	CompletionOptions *v1.CompletionOptions `protobuf:"bytes,2,opt,name=completion_options,json=completionOptions,proto3" json:"completion_options,omitempty"`
	// Specifies the format of the request.
	//
	// Types that are valid to be assigned to RequestFormat:
	//
	//	*BatchCompletionRequest_SourceDatasetId
	RequestFormat isBatchCompletionRequest_RequestFormat `protobuf_oneof:"request_format"`
	// Specifies the format of the model's response.
	//
	// Types that are valid to be assigned to ResponseFormat:
	//
	//	*BatchCompletionRequest_JsonObject
	//	*BatchCompletionRequest_JsonSchema
	ResponseFormat isBatchCompletionRequest_ResponseFormat `protobuf_oneof:"ResponseFormat"`
	unknownFields  protoimpl.UnknownFields
	sizeCache      protoimpl.SizeCache
}

func (x *BatchCompletionRequest) Reset() {
	*x = BatchCompletionRequest{}
	mi := &file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_msgTypes[2]
	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
	ms.StoreMessageInfo(mi)
}

func (x *BatchCompletionRequest) String() string {
	return protoimpl.X.MessageStringOf(x)
}

func (*BatchCompletionRequest) ProtoMessage() {}

func (x *BatchCompletionRequest) ProtoReflect() protoreflect.Message {
	mi := &file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_msgTypes[2]
	if x != nil {
		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
		if ms.LoadMessageInfo() == nil {
			ms.StoreMessageInfo(mi)
		}
		return ms
	}
	return mi.MessageOf(x)
}

// Deprecated: Use BatchCompletionRequest.ProtoReflect.Descriptor instead.
func (*BatchCompletionRequest) Descriptor() ([]byte, []int) {
	return file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_rawDescGZIP(), []int{2}
}

func (x *BatchCompletionRequest) GetModelUri() string {
	if x != nil {
		return x.ModelUri
	}
	return ""
}

func (x *BatchCompletionRequest) GetCompletionOptions() *v1.CompletionOptions {
	if x != nil {
		return x.CompletionOptions
	}
	return nil
}

func (x *BatchCompletionRequest) GetRequestFormat() isBatchCompletionRequest_RequestFormat {
	if x != nil {
		return x.RequestFormat
	}
	return nil
}

func (x *BatchCompletionRequest) GetSourceDatasetId() string {
	if x != nil {
		if x, ok := x.RequestFormat.(*BatchCompletionRequest_SourceDatasetId); ok {
			return x.SourceDatasetId
		}
	}
	return ""
}

func (x *BatchCompletionRequest) GetResponseFormat() isBatchCompletionRequest_ResponseFormat {
	if x != nil {
		return x.ResponseFormat
	}
	return nil
}

func (x *BatchCompletionRequest) GetJsonObject() bool {
	if x != nil {
		if x, ok := x.ResponseFormat.(*BatchCompletionRequest_JsonObject); ok {
			return x.JsonObject
		}
	}
	return false
}

func (x *BatchCompletionRequest) GetJsonSchema() *v1.JsonSchema {
	if x != nil {
		if x, ok := x.ResponseFormat.(*BatchCompletionRequest_JsonSchema); ok {
			return x.JsonSchema
		}
	}
	return nil
}

type isBatchCompletionRequest_RequestFormat interface {
	isBatchCompletionRequest_RequestFormat()
}

type BatchCompletionRequest_SourceDatasetId struct {
	// ID of the dataset containing the context for the completion model.
	SourceDatasetId string `protobuf:"bytes,3,opt,name=source_dataset_id,json=sourceDatasetId,proto3,oneof"`
}

func (*BatchCompletionRequest_SourceDatasetId) isBatchCompletionRequest_RequestFormat() {}

type isBatchCompletionRequest_ResponseFormat interface {
	isBatchCompletionRequest_ResponseFormat()
}

type BatchCompletionRequest_JsonObject struct {
	// When set to true, the model will respond with a valid JSON object.
	// Be sure to explicitly ask the model for JSON.
	// Otherwise, it may generate excessive whitespace and run indefinitely until it reaches the token limit.
	JsonObject bool `protobuf:"varint,4,opt,name=json_object,json=jsonObject,proto3,oneof"`
}

type BatchCompletionRequest_JsonSchema struct {
	// Enforces a specific JSON structure for the model's response based on a provided schema.
	JsonSchema *v1.JsonSchema `protobuf:"bytes,5,opt,name=json_schema,json=jsonSchema,proto3,oneof"`
}

func (*BatchCompletionRequest_JsonObject) isBatchCompletionRequest_ResponseFormat() {}

func (*BatchCompletionRequest_JsonSchema) isBatchCompletionRequest_ResponseFormat() {}

// Metadata of the batch completion operation.
type BatchCompletionMetadata struct {
	state protoimpl.MessageState `protogen:"open.v1"`
	// The ID of the batch completion task.
	TaskId string `protobuf:"bytes,1,opt,name=task_id,json=taskId,proto3" json:"task_id,omitempty"`
	// The status of the batch completion task.
	TaskStatus v1.BatchInferenceTaskStatus `protobuf:"varint,2,opt,name=task_status,json=taskStatus,proto3,enum=yandex.cloud.ai.foundation_models.v1.BatchInferenceTaskStatus" json:"task_status,omitempty"`
	// A number of currently completed batches of the completion task.
	CompletedBatches int64 `protobuf:"varint,3,opt,name=completed_batches,json=completedBatches,proto3" json:"completed_batches,omitempty"`
	// A number of total batches of the completion task.
	TotalBatches  int64 `protobuf:"varint,4,opt,name=total_batches,json=totalBatches,proto3" json:"total_batches,omitempty"`
	unknownFields protoimpl.UnknownFields
	sizeCache     protoimpl.SizeCache
}

func (x *BatchCompletionMetadata) Reset() {
	*x = BatchCompletionMetadata{}
	mi := &file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_msgTypes[3]
	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
	ms.StoreMessageInfo(mi)
}

func (x *BatchCompletionMetadata) String() string {
	return protoimpl.X.MessageStringOf(x)
}

func (*BatchCompletionMetadata) ProtoMessage() {}

func (x *BatchCompletionMetadata) ProtoReflect() protoreflect.Message {
	mi := &file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_msgTypes[3]
	if x != nil {
		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
		if ms.LoadMessageInfo() == nil {
			ms.StoreMessageInfo(mi)
		}
		return ms
	}
	return mi.MessageOf(x)
}

// Deprecated: Use BatchCompletionMetadata.ProtoReflect.Descriptor instead.
func (*BatchCompletionMetadata) Descriptor() ([]byte, []int) {
	return file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_rawDescGZIP(), []int{3}
}

func (x *BatchCompletionMetadata) GetTaskId() string {
	if x != nil {
		return x.TaskId
	}
	return ""
}

func (x *BatchCompletionMetadata) GetTaskStatus() v1.BatchInferenceTaskStatus {
	if x != nil {
		return x.TaskStatus
	}
	return v1.BatchInferenceTaskStatus(0)
}

func (x *BatchCompletionMetadata) GetCompletedBatches() int64 {
	if x != nil {
		return x.CompletedBatches
	}
	return 0
}

func (x *BatchCompletionMetadata) GetTotalBatches() int64 {
	if x != nil {
		return x.TotalBatches
	}
	return 0
}

// Response containing information about completion task.
type BatchCompletionResponse struct {
	state protoimpl.MessageState `protogen:"open.v1"`
	// The ID of the batch completion task.
	TaskId string `protobuf:"bytes,1,opt,name=task_id,json=taskId,proto3" json:"task_id,omitempty"`
	// The status of the batch completion task.
	TaskStatus v1.BatchInferenceTaskStatus `protobuf:"varint,2,opt,name=task_status,json=taskStatus,proto3,enum=yandex.cloud.ai.foundation_models.v1.BatchInferenceTaskStatus" json:"task_status,omitempty"`
	// The ID of the dataset containing completion results.
	ResultDatasetId string `protobuf:"bytes,3,opt,name=result_dataset_id,json=resultDatasetId,proto3" json:"result_dataset_id,omitempty"`
	unknownFields   protoimpl.UnknownFields
	sizeCache       protoimpl.SizeCache
}

func (x *BatchCompletionResponse) Reset() {
	*x = BatchCompletionResponse{}
	mi := &file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_msgTypes[4]
	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
	ms.StoreMessageInfo(mi)
}

func (x *BatchCompletionResponse) String() string {
	return protoimpl.X.MessageStringOf(x)
}

func (*BatchCompletionResponse) ProtoMessage() {}

func (x *BatchCompletionResponse) ProtoReflect() protoreflect.Message {
	mi := &file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_msgTypes[4]
	if x != nil {
		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
		if ms.LoadMessageInfo() == nil {
			ms.StoreMessageInfo(mi)
		}
		return ms
	}
	return mi.MessageOf(x)
}

// Deprecated: Use BatchCompletionResponse.ProtoReflect.Descriptor instead.
func (*BatchCompletionResponse) Descriptor() ([]byte, []int) {
	return file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_rawDescGZIP(), []int{4}
}

func (x *BatchCompletionResponse) GetTaskId() string {
	if x != nil {
		return x.TaskId
	}
	return ""
}

func (x *BatchCompletionResponse) GetTaskStatus() v1.BatchInferenceTaskStatus {
	if x != nil {
		return x.TaskStatus
	}
	return v1.BatchInferenceTaskStatus(0)
}

func (x *BatchCompletionResponse) GetResultDatasetId() string {
	if x != nil {
		return x.ResultDatasetId
	}
	return ""
}

// Request for the service to tokenize input text.
type TokenizeRequest struct {
	state protoimpl.MessageState `protogen:"open.v1"`
	// The identifier of the model to be used for tokenization.
	ModelUri string `protobuf:"bytes,1,opt,name=model_uri,json=modelUri,proto3" json:"model_uri,omitempty"`
	// Text to be tokenized.
	Text          string `protobuf:"bytes,2,opt,name=text,proto3" json:"text,omitempty"`
	unknownFields protoimpl.UnknownFields
	sizeCache     protoimpl.SizeCache
}

func (x *TokenizeRequest) Reset() {
	*x = TokenizeRequest{}
	mi := &file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_msgTypes[5]
	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
	ms.StoreMessageInfo(mi)
}

func (x *TokenizeRequest) String() string {
	return protoimpl.X.MessageStringOf(x)
}

func (*TokenizeRequest) ProtoMessage() {}

func (x *TokenizeRequest) ProtoReflect() protoreflect.Message {
	mi := &file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_msgTypes[5]
	if x != nil {
		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
		if ms.LoadMessageInfo() == nil {
			ms.StoreMessageInfo(mi)
		}
		return ms
	}
	return mi.MessageOf(x)
}

// Deprecated: Use TokenizeRequest.ProtoReflect.Descriptor instead.
func (*TokenizeRequest) Descriptor() ([]byte, []int) {
	return file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_rawDescGZIP(), []int{5}
}

func (x *TokenizeRequest) GetModelUri() string {
	if x != nil {
		return x.ModelUri
	}
	return ""
}

func (x *TokenizeRequest) GetText() string {
	if x != nil {
		return x.Text
	}
	return ""
}

// Response containing tokenized content from request.
type TokenizeResponse struct {
	state protoimpl.MessageState `protogen:"open.v1"`
	// A list of tokens obtained from tokenization.
	Tokens []*v1.Token `protobuf:"bytes,1,rep,name=tokens,proto3" json:"tokens,omitempty"`
	// Model version (changes with model releases).
	ModelVersion  string `protobuf:"bytes,2,opt,name=model_version,json=modelVersion,proto3" json:"model_version,omitempty"`
	unknownFields protoimpl.UnknownFields
	sizeCache     protoimpl.SizeCache
}

func (x *TokenizeResponse) Reset() {
	*x = TokenizeResponse{}
	mi := &file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_msgTypes[6]
	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
	ms.StoreMessageInfo(mi)
}

func (x *TokenizeResponse) String() string {
	return protoimpl.X.MessageStringOf(x)
}

func (*TokenizeResponse) ProtoMessage() {}

func (x *TokenizeResponse) ProtoReflect() protoreflect.Message {
	mi := &file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_msgTypes[6]
	if x != nil {
		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
		if ms.LoadMessageInfo() == nil {
			ms.StoreMessageInfo(mi)
		}
		return ms
	}
	return mi.MessageOf(x)
}

// Deprecated: Use TokenizeResponse.ProtoReflect.Descriptor instead.
func (*TokenizeResponse) Descriptor() ([]byte, []int) {
	return file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_rawDescGZIP(), []int{6}
}

func (x *TokenizeResponse) GetTokens() []*v1.Token {
	if x != nil {
		return x.Tokens
	}
	return nil
}

func (x *TokenizeResponse) GetModelVersion() string {
	if x != nil {
		return x.ModelVersion
	}
	return ""
}

var File_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto protoreflect.FileDescriptor

const file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_rawDesc = "" +
	"\n" +
	"Ryandex/cloud/ai/foundation_models/v1/text_generation/text_generation_service.proto\x12$yandex.cloud.ai.foundation_models.v1\x1a\x1cgoogle/api/annotations.proto\x1a6yandex/cloud/ai/foundation_models/v1/text_common.proto\x1aFyandex/cloud/ai/foundation_models/v1/batch_inference_task_status.proto\x1a yandex/cloud/api/operation.proto\x1a&yandex/cloud/operation/operation.proto\x1a\x1egoogle/protobuf/wrappers.proto\"\xce\x04\n" +
	"\x11CompletionRequest\x12\x1b\n" +
	"\tmodel_uri\x18\x01 \x01(\tR\bmodelUri\x12f\n" +
	"\x12completion_options\x18\x02 \x01(\v27.yandex.cloud.ai.foundation_models.v1.CompletionOptionsR\x11completionOptions\x12I\n" +
	"\bmessages\x18\x03 \x03(\v2-.yandex.cloud.ai.foundation_models.v1.MessageR\bmessages\x12@\n" +
	"\x05tools\x18\x04 \x03(\v2*.yandex.cloud.ai.foundation_models.v1.ToolR\x05tools\x12!\n" +
	"\vjson_object\x18\x05 \x01(\bH\x00R\n" +
	"jsonObject\x12S\n" +
	"\vjson_schema\x18\x06 \x01(\v20.yandex.cloud.ai.foundation_models.v1.JsonSchemaH\x00R\n" +
	"jsonSchema\x12J\n" +
	"\x13parallel_tool_calls\x18\a \x01(\v2\x1a.google.protobuf.BoolValueR\x11parallelToolCalls\x12Q\n" +
	"\vtool_choice\x18\b \x01(\v20.yandex.cloud.ai.foundation_models.v1.ToolChoiceR\n" +
	"toolChoiceB\x10\n" +
	"\x0eResponseFormat\"\xda\x01\n" +
	"\x12CompletionResponse\x12U\n" +
	"\falternatives\x18\x01 \x03(\v21.yandex.cloud.ai.foundation_models.v1.AlternativeR\falternatives\x12H\n" +
	"\x05usage\x18\x02 \x01(\v22.yandex.cloud.ai.foundation_models.v1.ContentUsageR\x05usage\x12#\n" +
	"\rmodel_version\x18\x03 \x01(\tR\fmodelVersion\"\xe7\x02\n" +
	"\x16BatchCompletionRequest\x12\x1b\n" +
	"\tmodel_uri\x18\x01 \x01(\tR\bmodelUri\x12f\n" +
	"\x12completion_options\x18\x02 \x01(\v27.yandex.cloud.ai.foundation_models.v1.CompletionOptionsR\x11completionOptions\x12,\n" +
	"\x11source_dataset_id\x18\x03 \x01(\tH\x00R\x0fsourceDatasetId\x12!\n" +
	"\vjson_object\x18\x04 \x01(\bH\x01R\n" +
	"jsonObject\x12S\n" +
	"\vjson_schema\x18\x05 \x01(\v20.yandex.cloud.ai.foundation_models.v1.JsonSchemaH\x01R\n" +
	"jsonSchemaB\x10\n" +
	"\x0erequest_formatB\x10\n" +
	"\x0eResponseFormat\"\xe5\x01\n" +
	"\x17BatchCompletionMetadata\x12\x17\n" +
	"\atask_id\x18\x01 \x01(\tR\x06taskId\x12_\n" +
	"\vtask_status\x18\x02 \x01(\x0e2>.yandex.cloud.ai.foundation_models.v1.BatchInferenceTaskStatusR\n" +
	"taskStatus\x12+\n" +
	"\x11completed_batches\x18\x03 \x01(\x03R\x10completedBatches\x12#\n" +
	"\rtotal_batches\x18\x04 \x01(\x03R\ftotalBatches\"\xbf\x01\n" +
	"\x17BatchCompletionResponse\x12\x17\n" +
	"\atask_id\x18\x01 \x01(\tR\x06taskId\x12_\n" +
	"\vtask_status\x18\x02 \x01(\x0e2>.yandex.cloud.ai.foundation_models.v1.BatchInferenceTaskStatusR\n" +
	"taskStatus\x12*\n" +
	"\x11result_dataset_id\x18\x03 \x01(\tR\x0fresultDatasetId\"B\n" +
	"\x0fTokenizeRequest\x12\x1b\n" +
	"\tmodel_uri\x18\x01 \x01(\tR\bmodelUri\x12\x12\n" +
	"\x04text\x18\x02 \x01(\tR\x04text\"|\n" +
	"\x10TokenizeResponse\x12C\n" +
	"\x06tokens\x18\x01 \x03(\v2+.yandex.cloud.ai.foundation_models.v1.TokenR\x06tokens\x12#\n" +
	"\rmodel_version\x18\x02 \x01(\tR\fmodelVersion2\xc7\x01\n" +
	"\x15TextGenerationService\x12\xad\x01\n" +
	"\n" +
	"Completion\x127.yandex.cloud.ai.foundation_models.v1.CompletionRequest\x1a8.yandex.cloud.ai.foundation_models.v1.CompletionResponse\"*\x82\xd3\xe4\x93\x02$:\x01*\"\x1f/foundationModels/v1/completion0\x012\xd0\x01\n" +
	"\x1aTextGenerationAsyncService\x12\xb1\x01\n" +
	"\n" +
	"Completion\x127.yandex.cloud.ai.foundation_models.v1.CompletionRequest\x1a!.yandex.cloud.operation.Operation\"G\xb2\xd2*\x14\x12\x12CompletionResponse\x82\xd3\xe4\x93\x02):\x01*\"$/foundationModels/v1/completionAsync2\xf3\x01\n" +
	"\x1aTextGenerationBatchService\x12\xd4\x01\n" +
	"\n" +
	"Completion\x12<.yandex.cloud.ai.foundation_models.v1.BatchCompletionRequest\x1a!.yandex.cloud.operation.Operation\"e\xb2\xd2*2\n" +
	"\x17BatchCompletionMetadata\x12\x17BatchCompletionResponse\x82\xd3\xe4\x93\x02):\x01*\"$/foundationModels/v1/completionBatch2\xf4\x02\n" +
	"\x10TokenizerService\x12\xa3\x01\n" +
	"\bTokenize\x125.yandex.cloud.ai.foundation_models.v1.TokenizeRequest\x1a6.yandex.cloud.ai.foundation_models.v1.TokenizeResponse\"(\x82\xd3\xe4\x93\x02\":\x01*\"\x1d/foundationModels/v1/tokenize\x12\xb9\x01\n" +
	"\x12TokenizeCompletion\x127.yandex.cloud.ai.foundation_models.v1.CompletionRequest\x1a6.yandex.cloud.ai.foundation_models.v1.TokenizeResponse\"2\x82\xd3\xe4\x93\x02,:\x01*\"'/foundationModels/v1/tokenizeCompletionB\x96\x01\n" +
	"(yandex.cloud.api.ai.foundation_models.v1Zjgithub.com/yandex-cloud/go-genproto/yandex/cloud/ai/foundation_models/v1/text_generation;foundation_modelsb\x06proto3"

var (
	file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_rawDescOnce sync.Once
	file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_rawDescData []byte
)

func file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_rawDescGZIP() []byte {
	file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_rawDescOnce.Do(func() {
		file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_rawDescData = protoimpl.X.CompressGZIP(unsafe.Slice(unsafe.StringData(file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_rawDesc), len(file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_rawDesc)))
	})
	return file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_rawDescData
}

var file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_msgTypes = make([]protoimpl.MessageInfo, 7)
var file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_goTypes = []any{
	(*CompletionRequest)(nil),        // 0: yandex.cloud.ai.foundation_models.v1.CompletionRequest
	(*CompletionResponse)(nil),       // 1: yandex.cloud.ai.foundation_models.v1.CompletionResponse
	(*BatchCompletionRequest)(nil),   // 2: yandex.cloud.ai.foundation_models.v1.BatchCompletionRequest
	(*BatchCompletionMetadata)(nil),  // 3: yandex.cloud.ai.foundation_models.v1.BatchCompletionMetadata
	(*BatchCompletionResponse)(nil),  // 4: yandex.cloud.ai.foundation_models.v1.BatchCompletionResponse
	(*TokenizeRequest)(nil),          // 5: yandex.cloud.ai.foundation_models.v1.TokenizeRequest
	(*TokenizeResponse)(nil),         // 6: yandex.cloud.ai.foundation_models.v1.TokenizeResponse
	(*v1.CompletionOptions)(nil),     // 7: yandex.cloud.ai.foundation_models.v1.CompletionOptions
	(*v1.Message)(nil),               // 8: yandex.cloud.ai.foundation_models.v1.Message
	(*v1.Tool)(nil),                  // 9: yandex.cloud.ai.foundation_models.v1.Tool
	(*v1.JsonSchema)(nil),            // 10: yandex.cloud.ai.foundation_models.v1.JsonSchema
	(*wrapperspb.BoolValue)(nil),     // 11: google.protobuf.BoolValue
	(*v1.ToolChoice)(nil),            // 12: yandex.cloud.ai.foundation_models.v1.ToolChoice
	(*v1.Alternative)(nil),           // 13: yandex.cloud.ai.foundation_models.v1.Alternative
	(*v1.ContentUsage)(nil),          // 14: yandex.cloud.ai.foundation_models.v1.ContentUsage
	(v1.BatchInferenceTaskStatus)(0), // 15: yandex.cloud.ai.foundation_models.v1.BatchInferenceTaskStatus
	(*v1.Token)(nil),                 // 16: yandex.cloud.ai.foundation_models.v1.Token
	(*operation.Operation)(nil),      // 17: yandex.cloud.operation.Operation
}
var file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_depIdxs = []int32{
	7,  // 0: yandex.cloud.ai.foundation_models.v1.CompletionRequest.completion_options:type_name -> yandex.cloud.ai.foundation_models.v1.CompletionOptions
	8,  // 1: yandex.cloud.ai.foundation_models.v1.CompletionRequest.messages:type_name -> yandex.cloud.ai.foundation_models.v1.Message
	9,  // 2: yandex.cloud.ai.foundation_models.v1.CompletionRequest.tools:type_name -> yandex.cloud.ai.foundation_models.v1.Tool
	10, // 3: yandex.cloud.ai.foundation_models.v1.CompletionRequest.json_schema:type_name -> yandex.cloud.ai.foundation_models.v1.JsonSchema
	11, // 4: yandex.cloud.ai.foundation_models.v1.CompletionRequest.parallel_tool_calls:type_name -> google.protobuf.BoolValue
	12, // 5: yandex.cloud.ai.foundation_models.v1.CompletionRequest.tool_choice:type_name -> yandex.cloud.ai.foundation_models.v1.ToolChoice
	13, // 6: yandex.cloud.ai.foundation_models.v1.CompletionResponse.alternatives:type_name -> yandex.cloud.ai.foundation_models.v1.Alternative
	14, // 7: yandex.cloud.ai.foundation_models.v1.CompletionResponse.usage:type_name -> yandex.cloud.ai.foundation_models.v1.ContentUsage
	7,  // 8: yandex.cloud.ai.foundation_models.v1.BatchCompletionRequest.completion_options:type_name -> yandex.cloud.ai.foundation_models.v1.CompletionOptions
	10, // 9: yandex.cloud.ai.foundation_models.v1.BatchCompletionRequest.json_schema:type_name -> yandex.cloud.ai.foundation_models.v1.JsonSchema
	15, // 10: yandex.cloud.ai.foundation_models.v1.BatchCompletionMetadata.task_status:type_name -> yandex.cloud.ai.foundation_models.v1.BatchInferenceTaskStatus
	15, // 11: yandex.cloud.ai.foundation_models.v1.BatchCompletionResponse.task_status:type_name -> yandex.cloud.ai.foundation_models.v1.BatchInferenceTaskStatus
	16, // 12: yandex.cloud.ai.foundation_models.v1.TokenizeResponse.tokens:type_name -> yandex.cloud.ai.foundation_models.v1.Token
	0,  // 13: yandex.cloud.ai.foundation_models.v1.TextGenerationService.Completion:input_type -> yandex.cloud.ai.foundation_models.v1.CompletionRequest
	0,  // 14: yandex.cloud.ai.foundation_models.v1.TextGenerationAsyncService.Completion:input_type -> yandex.cloud.ai.foundation_models.v1.CompletionRequest
	2,  // 15: yandex.cloud.ai.foundation_models.v1.TextGenerationBatchService.Completion:input_type -> yandex.cloud.ai.foundation_models.v1.BatchCompletionRequest
	5,  // 16: yandex.cloud.ai.foundation_models.v1.TokenizerService.Tokenize:input_type -> yandex.cloud.ai.foundation_models.v1.TokenizeRequest
	0,  // 17: yandex.cloud.ai.foundation_models.v1.TokenizerService.TokenizeCompletion:input_type -> yandex.cloud.ai.foundation_models.v1.CompletionRequest
	1,  // 18: yandex.cloud.ai.foundation_models.v1.TextGenerationService.Completion:output_type -> yandex.cloud.ai.foundation_models.v1.CompletionResponse
	17, // 19: yandex.cloud.ai.foundation_models.v1.TextGenerationAsyncService.Completion:output_type -> yandex.cloud.operation.Operation
	17, // 20: yandex.cloud.ai.foundation_models.v1.TextGenerationBatchService.Completion:output_type -> yandex.cloud.operation.Operation
	6,  // 21: yandex.cloud.ai.foundation_models.v1.TokenizerService.Tokenize:output_type -> yandex.cloud.ai.foundation_models.v1.TokenizeResponse
	6,  // 22: yandex.cloud.ai.foundation_models.v1.TokenizerService.TokenizeCompletion:output_type -> yandex.cloud.ai.foundation_models.v1.TokenizeResponse
	18, // [18:23] is the sub-list for method output_type
	13, // [13:18] is the sub-list for method input_type
	13, // [13:13] is the sub-list for extension type_name
	13, // [13:13] is the sub-list for extension extendee
	0,  // [0:13] is the sub-list for field type_name
}

func init() {
	file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_init()
}
func file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_init() {
	if File_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto != nil {
		return
	}
	file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_msgTypes[0].OneofWrappers = []any{
		(*CompletionRequest_JsonObject)(nil),
		(*CompletionRequest_JsonSchema)(nil),
	}
	file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_msgTypes[2].OneofWrappers = []any{
		(*BatchCompletionRequest_SourceDatasetId)(nil),
		(*BatchCompletionRequest_JsonObject)(nil),
		(*BatchCompletionRequest_JsonSchema)(nil),
	}
	type x struct{}
	out := protoimpl.TypeBuilder{
		File: protoimpl.DescBuilder{
			GoPackagePath: reflect.TypeOf(x{}).PkgPath(),
			RawDescriptor: unsafe.Slice(unsafe.StringData(file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_rawDesc), len(file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_rawDesc)),
			NumEnums:      0,
			NumMessages:   7,
			NumExtensions: 0,
			NumServices:   4,
		},
		GoTypes:           file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_goTypes,
		DependencyIndexes: file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_depIdxs,
		MessageInfos:      file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_msgTypes,
	}.Build()
	File_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto = out.File
	file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_goTypes = nil
	file_yandex_cloud_ai_foundation_models_v1_text_generation_text_generation_service_proto_depIdxs = nil
}
