// Code generated by protoc-gen-go. DO NOT EDIT.
// versions:
// 	protoc-gen-go v1.36.11
// 	protoc        v3.21.12
// source: yandex/cloud/ai/foundation_models/v1/embedding/embedding_service.proto

package foundation_models

import (
	_ "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 obtain text embeddings.
type TextEmbeddingRequest struct {
	state protoimpl.MessageState `protogen:"open.v1"`
	// The [model URI](/docs/foundation-models/concepts/embeddings) to be used for obtaining text embeddings.
	ModelUri string `protobuf:"bytes,1,opt,name=model_uri,json=modelUri,proto3" json:"model_uri,omitempty"`
	// The input text for which the embedding is requested.
	Text string `protobuf:"bytes,2,opt,name=text,proto3" json:"text,omitempty"`
	// Optional parameter to specify embedding dimension for models that support multi-dimensional outputs
	Dim           *wrapperspb.Int64Value `protobuf:"bytes,3,opt,name=dim,proto3" json:"dim,omitempty"`
	unknownFields protoimpl.UnknownFields
	sizeCache     protoimpl.SizeCache
}

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

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

func (*TextEmbeddingRequest) ProtoMessage() {}

func (x *TextEmbeddingRequest) ProtoReflect() protoreflect.Message {
	mi := &file_yandex_cloud_ai_foundation_models_v1_embedding_embedding_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 TextEmbeddingRequest.ProtoReflect.Descriptor instead.
func (*TextEmbeddingRequest) Descriptor() ([]byte, []int) {
	return file_yandex_cloud_ai_foundation_models_v1_embedding_embedding_service_proto_rawDescGZIP(), []int{0}
}

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

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

func (x *TextEmbeddingRequest) GetDim() *wrapperspb.Int64Value {
	if x != nil {
		return x.Dim
	}
	return nil
}

// Response containing generated text embedding.
type TextEmbeddingResponse struct {
	state protoimpl.MessageState `protogen:"open.v1"`
	// A repeated list of double values representing the embedding.
	Embedding []float64 `protobuf:"fixed64,1,rep,packed,name=embedding,proto3" json:"embedding,omitempty"`
	// The number of tokens in the input text.
	NumTokens int64 `protobuf:"varint,2,opt,name=num_tokens,json=numTokens,proto3" json:"num_tokens,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 *TextEmbeddingResponse) Reset() {
	*x = TextEmbeddingResponse{}
	mi := &file_yandex_cloud_ai_foundation_models_v1_embedding_embedding_service_proto_msgTypes[1]
	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
	ms.StoreMessageInfo(mi)
}

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

func (*TextEmbeddingResponse) ProtoMessage() {}

func (x *TextEmbeddingResponse) ProtoReflect() protoreflect.Message {
	mi := &file_yandex_cloud_ai_foundation_models_v1_embedding_embedding_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 TextEmbeddingResponse.ProtoReflect.Descriptor instead.
func (*TextEmbeddingResponse) Descriptor() ([]byte, []int) {
	return file_yandex_cloud_ai_foundation_models_v1_embedding_embedding_service_proto_rawDescGZIP(), []int{1}
}

func (x *TextEmbeddingResponse) GetEmbedding() []float64 {
	if x != nil {
		return x.Embedding
	}
	return nil
}

func (x *TextEmbeddingResponse) GetNumTokens() int64 {
	if x != nil {
		return x.NumTokens
	}
	return 0
}

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

var File_yandex_cloud_ai_foundation_models_v1_embedding_embedding_service_proto protoreflect.FileDescriptor

const file_yandex_cloud_ai_foundation_models_v1_embedding_embedding_service_proto_rawDesc = "" +
	"\n" +
	"Fyandex/cloud/ai/foundation_models/v1/embedding/embedding_service.proto\x12$yandex.cloud.ai.foundation_models.v1\x1a\x1cgoogle/api/annotations.proto\x1a\x1egoogle/protobuf/wrappers.proto\"v\n" +
	"\x14TextEmbeddingRequest\x12\x1b\n" +
	"\tmodel_uri\x18\x01 \x01(\tR\bmodelUri\x12\x12\n" +
	"\x04text\x18\x02 \x01(\tR\x04text\x12-\n" +
	"\x03dim\x18\x03 \x01(\v2\x1b.google.protobuf.Int64ValueR\x03dim\"y\n" +
	"\x15TextEmbeddingResponse\x12\x1c\n" +
	"\tembedding\x18\x01 \x03(\x01R\tembedding\x12\x1d\n" +
	"\n" +
	"num_tokens\x18\x02 \x01(\x03R\tnumTokens\x12#\n" +
	"\rmodel_version\x18\x03 \x01(\tR\fmodelVersion2\xcd\x01\n" +
	"\x11EmbeddingsService\x12\xb7\x01\n" +
	"\rTextEmbedding\x12:.yandex.cloud.ai.foundation_models.v1.TextEmbeddingRequest\x1a;.yandex.cloud.ai.foundation_models.v1.TextEmbeddingResponse\"-\x82\xd3\xe4\x93\x02':\x01*\"\"/foundationModels/v1/textEmbeddingB\x90\x01\n" +
	"(yandex.cloud.api.ai.foundation_models.v1Zdgithub.com/yandex-cloud/go-genproto/yandex/cloud/ai/foundation_models/v1/embedding;foundation_modelsb\x06proto3"

var (
	file_yandex_cloud_ai_foundation_models_v1_embedding_embedding_service_proto_rawDescOnce sync.Once
	file_yandex_cloud_ai_foundation_models_v1_embedding_embedding_service_proto_rawDescData []byte
)

func file_yandex_cloud_ai_foundation_models_v1_embedding_embedding_service_proto_rawDescGZIP() []byte {
	file_yandex_cloud_ai_foundation_models_v1_embedding_embedding_service_proto_rawDescOnce.Do(func() {
		file_yandex_cloud_ai_foundation_models_v1_embedding_embedding_service_proto_rawDescData = protoimpl.X.CompressGZIP(unsafe.Slice(unsafe.StringData(file_yandex_cloud_ai_foundation_models_v1_embedding_embedding_service_proto_rawDesc), len(file_yandex_cloud_ai_foundation_models_v1_embedding_embedding_service_proto_rawDesc)))
	})
	return file_yandex_cloud_ai_foundation_models_v1_embedding_embedding_service_proto_rawDescData
}

var file_yandex_cloud_ai_foundation_models_v1_embedding_embedding_service_proto_msgTypes = make([]protoimpl.MessageInfo, 2)
var file_yandex_cloud_ai_foundation_models_v1_embedding_embedding_service_proto_goTypes = []any{
	(*TextEmbeddingRequest)(nil),  // 0: yandex.cloud.ai.foundation_models.v1.TextEmbeddingRequest
	(*TextEmbeddingResponse)(nil), // 1: yandex.cloud.ai.foundation_models.v1.TextEmbeddingResponse
	(*wrapperspb.Int64Value)(nil), // 2: google.protobuf.Int64Value
}
var file_yandex_cloud_ai_foundation_models_v1_embedding_embedding_service_proto_depIdxs = []int32{
	2, // 0: yandex.cloud.ai.foundation_models.v1.TextEmbeddingRequest.dim:type_name -> google.protobuf.Int64Value
	0, // 1: yandex.cloud.ai.foundation_models.v1.EmbeddingsService.TextEmbedding:input_type -> yandex.cloud.ai.foundation_models.v1.TextEmbeddingRequest
	1, // 2: yandex.cloud.ai.foundation_models.v1.EmbeddingsService.TextEmbedding:output_type -> yandex.cloud.ai.foundation_models.v1.TextEmbeddingResponse
	2, // [2:3] is the sub-list for method output_type
	1, // [1:2] is the sub-list for method input_type
	1, // [1:1] is the sub-list for extension type_name
	1, // [1:1] is the sub-list for extension extendee
	0, // [0:1] is the sub-list for field type_name
}

func init() { file_yandex_cloud_ai_foundation_models_v1_embedding_embedding_service_proto_init() }
func file_yandex_cloud_ai_foundation_models_v1_embedding_embedding_service_proto_init() {
	if File_yandex_cloud_ai_foundation_models_v1_embedding_embedding_service_proto != nil {
		return
	}
	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_embedding_embedding_service_proto_rawDesc), len(file_yandex_cloud_ai_foundation_models_v1_embedding_embedding_service_proto_rawDesc)),
			NumEnums:      0,
			NumMessages:   2,
			NumExtensions: 0,
			NumServices:   1,
		},
		GoTypes:           file_yandex_cloud_ai_foundation_models_v1_embedding_embedding_service_proto_goTypes,
		DependencyIndexes: file_yandex_cloud_ai_foundation_models_v1_embedding_embedding_service_proto_depIdxs,
		MessageInfos:      file_yandex_cloud_ai_foundation_models_v1_embedding_embedding_service_proto_msgTypes,
	}.Build()
	File_yandex_cloud_ai_foundation_models_v1_embedding_embedding_service_proto = out.File
	file_yandex_cloud_ai_foundation_models_v1_embedding_embedding_service_proto_goTypes = nil
	file_yandex_cloud_ai_foundation_models_v1_embedding_embedding_service_proto_depIdxs = nil
}
