tokenizerpb

package
v0.9.0 Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: Jun 16, 2026 License: Apache-2.0 Imports: 9 Imported by: 0

Documentation

Index

Constants

View Source
const (
	TokenizationService_Tokenize_FullMethodName             = "/tokenization.TokenizationService/Tokenize"
	TokenizationService_RenderChatTemplate_FullMethodName   = "/tokenization.TokenizationService/RenderChatTemplate"
	TokenizationService_InitializeTokenizer_FullMethodName  = "/tokenization.TokenizationService/InitializeTokenizer"
	TokenizationService_RenderChatCompletion_FullMethodName = "/tokenization.TokenizationService/RenderChatCompletion"
	TokenizationService_RenderCompletion_FullMethodName     = "/tokenization.TokenizationService/RenderCompletion"
)

Variables

View Source
var File_api_tokenizerpb_tokenizer_proto protoreflect.FileDescriptor
View Source
var TokenizationService_ServiceDesc = grpc.ServiceDesc{
	ServiceName: "tokenization.TokenizationService",
	HandlerType: (*TokenizationServiceServer)(nil),
	Methods: []grpc.MethodDesc{
		{
			MethodName: "Tokenize",
			Handler:    _TokenizationService_Tokenize_Handler,
		},
		{
			MethodName: "RenderChatTemplate",
			Handler:    _TokenizationService_RenderChatTemplate_Handler,
		},
		{
			MethodName: "InitializeTokenizer",
			Handler:    _TokenizationService_InitializeTokenizer_Handler,
		},
		{
			MethodName: "RenderChatCompletion",
			Handler:    _TokenizationService_RenderChatCompletion_Handler,
		},
		{
			MethodName: "RenderCompletion",
			Handler:    _TokenizationService_RenderCompletion_Handler,
		},
	},
	Streams:  []grpc.StreamDesc{},
	Metadata: "api/tokenizerpb/tokenizer.proto",
}

TokenizationService_ServiceDesc is the grpc.ServiceDesc for TokenizationService service. It's only intended for direct use with grpc.RegisterService, and not to be introspected or modified (even as a copy)

Functions

func RegisterTokenizationServiceServer

func RegisterTokenizationServiceServer(s grpc.ServiceRegistrar, srv TokenizationServiceServer)

Types

type ChatMessage

type ChatMessage struct {
	Role          string         `protobuf:"bytes,1,opt,name=role,proto3" json:"role,omitempty"`                                                // Role of the message (e.g., "user", "assistant", "system")
	Content       *string        `protobuf:"bytes,2,opt,name=content,proto3,oneof" json:"content,omitempty"`                                    // Plain-text content (Content.Raw)
	ContentParts  []*ContentPart `protobuf:"bytes,3,rep,name=content_parts,json=contentParts,proto3" json:"content_parts,omitempty"`            // Multimodal content blocks (Content.Structured)
	ToolCallsJson *string        `protobuf:"bytes,4,opt,name=tool_calls_json,json=toolCallsJson,proto3,oneof" json:"tool_calls_json,omitempty"` // Assistant tool calls, encoded as a JSON array
	// contains filtered or unexported fields
}

ChatMessage represents a single message in a conversation. Mirrors Content in pkg/tokenization/types: set content for plain text, content_parts for multimodal (image_url blocks). Exactly one should be set.

func (*ChatMessage) Descriptor deprecated

func (*ChatMessage) Descriptor() ([]byte, []int)

Deprecated: Use ChatMessage.ProtoReflect.Descriptor instead.

func (*ChatMessage) GetContent

func (x *ChatMessage) GetContent() string

func (*ChatMessage) GetContentParts added in v0.7.0

func (x *ChatMessage) GetContentParts() []*ContentPart

func (*ChatMessage) GetRole

func (x *ChatMessage) GetRole() string

func (*ChatMessage) GetToolCallsJson added in v0.8.1

func (x *ChatMessage) GetToolCallsJson() string

func (*ChatMessage) ProtoMessage

func (*ChatMessage) ProtoMessage()

func (*ChatMessage) ProtoReflect

func (x *ChatMessage) ProtoReflect() protoreflect.Message

func (*ChatMessage) Reset

func (x *ChatMessage) Reset()

func (*ChatMessage) String

func (x *ChatMessage) String() string

type ChatTemplateRequest

type ChatTemplateRequest struct {
	ConversationTurns         []*ConversationTurn `protobuf:"bytes,1,rep,name=conversation_turns,json=conversationTurns,proto3" json:"conversation_turns,omitempty"` // The conversation turns (batches of messages)
	Tools                     []*ToolDescription  `protobuf:"bytes,2,rep,name=tools,proto3" json:"tools,omitempty"`                                                  // Tools available to the conversation
	Documents                 []*Document         `protobuf:"bytes,3,rep,name=documents,proto3" json:"documents,omitempty"`                                          // Documents related to the conversation
	ChatTemplate              string              `protobuf:"bytes,4,opt,name=chat_template,json=chatTemplate,proto3" json:"chat_template,omitempty"`                // The chat template to use
	ReturnAssistantTokensMask bool                ``                                                                                                                 // Whether to return assistant token mask
	/* 141-byte string literal not displayed */
	ContinueFinalMessage bool              `protobuf:"varint,6,opt,name=continue_final_message,json=continueFinalMessage,proto3" json:"continue_final_message,omitempty"` // Whether to continue the final message
	AddGenerationPrompt  bool              `protobuf:"varint,7,opt,name=add_generation_prompt,json=addGenerationPrompt,proto3" json:"add_generation_prompt,omitempty"`    // Whether to add generation prompt
	ChatTemplateKwargs   map[string]*Value ``                                                                                                                             // Additional chat template arguments
	/* 191-byte string literal not displayed */
	ModelName string `protobuf:"bytes,9,opt,name=model_name,json=modelName,proto3" json:"model_name,omitempty"` // The name of the model to use for tokenization
	// contains filtered or unexported fields
}

ChatTemplateRequest represents a request to render a chat template

func (*ChatTemplateRequest) Descriptor deprecated

func (*ChatTemplateRequest) Descriptor() ([]byte, []int)

Deprecated: Use ChatTemplateRequest.ProtoReflect.Descriptor instead.

func (*ChatTemplateRequest) GetAddGenerationPrompt

func (x *ChatTemplateRequest) GetAddGenerationPrompt() bool

func (*ChatTemplateRequest) GetChatTemplate

func (x *ChatTemplateRequest) GetChatTemplate() string

func (*ChatTemplateRequest) GetChatTemplateKwargs

func (x *ChatTemplateRequest) GetChatTemplateKwargs() map[string]*Value

func (*ChatTemplateRequest) GetContinueFinalMessage

func (x *ChatTemplateRequest) GetContinueFinalMessage() bool

func (*ChatTemplateRequest) GetConversationTurns

func (x *ChatTemplateRequest) GetConversationTurns() []*ConversationTurn

func (*ChatTemplateRequest) GetDocuments

func (x *ChatTemplateRequest) GetDocuments() []*Document

func (*ChatTemplateRequest) GetModelName

func (x *ChatTemplateRequest) GetModelName() string

func (*ChatTemplateRequest) GetReturnAssistantTokensMask

func (x *ChatTemplateRequest) GetReturnAssistantTokensMask() bool

func (*ChatTemplateRequest) GetTools

func (x *ChatTemplateRequest) GetTools() []*ToolDescription

func (*ChatTemplateRequest) ProtoMessage

func (*ChatTemplateRequest) ProtoMessage()

func (*ChatTemplateRequest) ProtoReflect

func (x *ChatTemplateRequest) ProtoReflect() protoreflect.Message

func (*ChatTemplateRequest) Reset

func (x *ChatTemplateRequest) Reset()

func (*ChatTemplateRequest) String

func (x *ChatTemplateRequest) String() string

type ChatTemplateResponse

type ChatTemplateResponse struct {
	RenderedPrompt string `protobuf:"bytes,1,opt,name=rendered_prompt,json=renderedPrompt,proto3" json:"rendered_prompt,omitempty"` // The rendered chat template prompt
	Success        bool   `protobuf:"varint,2,opt,name=success,proto3" json:"success,omitempty"`                                    // Whether the request was successful
	ErrorMessage   string `protobuf:"bytes,3,opt,name=error_message,json=errorMessage,proto3" json:"error_message,omitempty"`       // Error message if the request failed
	// contains filtered or unexported fields
}

ChatTemplateResponse represents the response from rendering a chat template

func (*ChatTemplateResponse) Descriptor deprecated

func (*ChatTemplateResponse) Descriptor() ([]byte, []int)

Deprecated: Use ChatTemplateResponse.ProtoReflect.Descriptor instead.

func (*ChatTemplateResponse) GetErrorMessage

func (x *ChatTemplateResponse) GetErrorMessage() string

func (*ChatTemplateResponse) GetRenderedPrompt

func (x *ChatTemplateResponse) GetRenderedPrompt() string

func (*ChatTemplateResponse) GetSuccess

func (x *ChatTemplateResponse) GetSuccess() bool

func (*ChatTemplateResponse) ProtoMessage

func (*ChatTemplateResponse) ProtoMessage()

func (*ChatTemplateResponse) ProtoReflect

func (x *ChatTemplateResponse) ProtoReflect() protoreflect.Message

func (*ChatTemplateResponse) Reset

func (x *ChatTemplateResponse) Reset()

func (*ChatTemplateResponse) String

func (x *ChatTemplateResponse) String() string

type ContentPart added in v0.7.0

type ContentPart struct {
	Type     string    `protobuf:"bytes,1,opt,name=type,proto3" json:"type,omitempty"`                               // "text" or "image_url"
	Text     *string   `protobuf:"bytes,2,opt,name=text,proto3,oneof" json:"text,omitempty"`                         // Text content (type == "text")
	ImageUrl *ImageUrl `protobuf:"bytes,3,opt,name=image_url,json=imageUrl,proto3,oneof" json:"image_url,omitempty"` // Image content (type == "image_url")
	// contains filtered or unexported fields
}

ContentPart represents a single part of a multimodal message content block

func (*ContentPart) Descriptor deprecated added in v0.7.0

func (*ContentPart) Descriptor() ([]byte, []int)

Deprecated: Use ContentPart.ProtoReflect.Descriptor instead.

func (*ContentPart) GetImageUrl added in v0.7.0

func (x *ContentPart) GetImageUrl() *ImageUrl

func (*ContentPart) GetText added in v0.7.0

func (x *ContentPart) GetText() string

func (*ContentPart) GetType added in v0.7.0

func (x *ContentPart) GetType() string

func (*ContentPart) ProtoMessage added in v0.7.0

func (*ContentPart) ProtoMessage()

func (*ContentPart) ProtoReflect added in v0.7.0

func (x *ContentPart) ProtoReflect() protoreflect.Message

func (*ContentPart) Reset added in v0.7.0

func (x *ContentPart) Reset()

func (*ContentPart) String added in v0.7.0

func (x *ContentPart) String() string

type ConversationTurn

type ConversationTurn struct {
	Messages []*ChatMessage `protobuf:"bytes,1,rep,name=messages,proto3" json:"messages,omitempty"` // The messages in this turn
	// contains filtered or unexported fields
}

ConversationTurn represents a single turn in a conversation (a single message or multiple messages per turn)

func (*ConversationTurn) Descriptor deprecated

func (*ConversationTurn) Descriptor() ([]byte, []int)

Deprecated: Use ConversationTurn.ProtoReflect.Descriptor instead.

func (*ConversationTurn) GetMessages

func (x *ConversationTurn) GetMessages() []*ChatMessage

func (*ConversationTurn) ProtoMessage

func (*ConversationTurn) ProtoMessage()

func (*ConversationTurn) ProtoReflect

func (x *ConversationTurn) ProtoReflect() protoreflect.Message

func (*ConversationTurn) Reset

func (x *ConversationTurn) Reset()

func (*ConversationTurn) String

func (x *ConversationTurn) String() string

type Document

type Document struct {
	Document map[string]*Value `` // Document definition
	/* 143-byte string literal not displayed */
	// contains filtered or unexported fields
}

Document represents a document

func (*Document) Descriptor deprecated

func (*Document) Descriptor() ([]byte, []int)

Deprecated: Use Document.ProtoReflect.Descriptor instead.

func (*Document) GetDocument

func (x *Document) GetDocument() map[string]*Value

func (*Document) ProtoMessage

func (*Document) ProtoMessage()

func (*Document) ProtoReflect

func (x *Document) ProtoReflect() protoreflect.Message

func (*Document) Reset

func (x *Document) Reset()

func (*Document) String

func (x *Document) String() string

type ImageUrl added in v0.7.0

type ImageUrl struct {
	Url string `protobuf:"bytes,1,opt,name=url,proto3" json:"url,omitempty"` // URL of the image (http/https or data URI)
	// contains filtered or unexported fields
}

ImageUrl represents the image_url content block field

func (*ImageUrl) Descriptor deprecated added in v0.7.0

func (*ImageUrl) Descriptor() ([]byte, []int)

Deprecated: Use ImageUrl.ProtoReflect.Descriptor instead.

func (*ImageUrl) GetUrl added in v0.7.0

func (x *ImageUrl) GetUrl() string

func (*ImageUrl) ProtoMessage added in v0.7.0

func (*ImageUrl) ProtoMessage()

func (*ImageUrl) ProtoReflect added in v0.7.0

func (x *ImageUrl) ProtoReflect() protoreflect.Message

func (*ImageUrl) Reset added in v0.7.0

func (x *ImageUrl) Reset()

func (*ImageUrl) String added in v0.7.0

func (x *ImageUrl) String() string

type InitializeTokenizerRequest

type InitializeTokenizerRequest struct {
	ModelName           string `protobuf:"bytes,1,opt,name=model_name,json=modelName,proto3" json:"model_name,omitempty"`                                  // The name of the model to initialize
	EnableThinking      bool   `protobuf:"varint,2,opt,name=enable_thinking,json=enableThinking,proto3" json:"enable_thinking,omitempty"`                  // Whether to enable thinking tokens
	AddGenerationPrompt bool   `protobuf:"varint,3,opt,name=add_generation_prompt,json=addGenerationPrompt,proto3" json:"add_generation_prompt,omitempty"` // Whether to add generation prompt
	// contains filtered or unexported fields
}

InitializeTokenizerRequest represents a request to initialize tokenizer for a model

func (*InitializeTokenizerRequest) Descriptor deprecated

func (*InitializeTokenizerRequest) Descriptor() ([]byte, []int)

Deprecated: Use InitializeTokenizerRequest.ProtoReflect.Descriptor instead.

func (*InitializeTokenizerRequest) GetAddGenerationPrompt

func (x *InitializeTokenizerRequest) GetAddGenerationPrompt() bool

func (*InitializeTokenizerRequest) GetEnableThinking

func (x *InitializeTokenizerRequest) GetEnableThinking() bool

func (*InitializeTokenizerRequest) GetModelName

func (x *InitializeTokenizerRequest) GetModelName() string

func (*InitializeTokenizerRequest) ProtoMessage

func (*InitializeTokenizerRequest) ProtoMessage()

func (*InitializeTokenizerRequest) ProtoReflect

func (*InitializeTokenizerRequest) Reset

func (x *InitializeTokenizerRequest) Reset()

func (*InitializeTokenizerRequest) String

func (x *InitializeTokenizerRequest) String() string

type InitializeTokenizerResponse

type InitializeTokenizerResponse struct {
	Success      bool   `protobuf:"varint,1,opt,name=success,proto3" json:"success,omitempty"`                              // Whether the initialization was successful
	ErrorMessage string `protobuf:"bytes,2,opt,name=error_message,json=errorMessage,proto3" json:"error_message,omitempty"` // Error message if initialization failed
	// contains filtered or unexported fields
}

InitializeTokenizerResponse represents the response from tokenizer initialization

func (*InitializeTokenizerResponse) Descriptor deprecated

func (*InitializeTokenizerResponse) Descriptor() ([]byte, []int)

Deprecated: Use InitializeTokenizerResponse.ProtoReflect.Descriptor instead.

func (*InitializeTokenizerResponse) GetErrorMessage

func (x *InitializeTokenizerResponse) GetErrorMessage() string

func (*InitializeTokenizerResponse) GetSuccess

func (x *InitializeTokenizerResponse) GetSuccess() bool

func (*InitializeTokenizerResponse) ProtoMessage

func (*InitializeTokenizerResponse) ProtoMessage()

func (*InitializeTokenizerResponse) ProtoReflect

func (*InitializeTokenizerResponse) Reset

func (x *InitializeTokenizerResponse) Reset()

func (*InitializeTokenizerResponse) String

func (x *InitializeTokenizerResponse) String() string

type ListValue

type ListValue struct {
	Values []*Value `protobuf:"bytes,1,rep,name=values,proto3" json:"values,omitempty"`
	// contains filtered or unexported fields
}

ListValue represents a list of values

func (*ListValue) Descriptor deprecated

func (*ListValue) Descriptor() ([]byte, []int)

Deprecated: Use ListValue.ProtoReflect.Descriptor instead.

func (*ListValue) GetValues

func (x *ListValue) GetValues() []*Value

func (*ListValue) ProtoMessage

func (*ListValue) ProtoMessage()

func (*ListValue) ProtoReflect

func (x *ListValue) ProtoReflect() protoreflect.Message

func (*ListValue) Reset

func (x *ListValue) Reset()

func (*ListValue) String

func (x *ListValue) String() string

type MultiModalFeatures added in v0.7.0

type MultiModalFeatures struct {
	MmHashes map[string]*StringList `` // Per-modality item hashes
	/* 159-byte string literal not displayed */
	MmPlaceholders map[string]*PlaceholderRangeList `` // Per-modality placeholder ranges
	/* 177-byte string literal not displayed */
	// contains filtered or unexported fields
}

MultiModalFeatures contains multimodal metadata produced by the render step

func (*MultiModalFeatures) Descriptor deprecated added in v0.7.0

func (*MultiModalFeatures) Descriptor() ([]byte, []int)

Deprecated: Use MultiModalFeatures.ProtoReflect.Descriptor instead.

func (*MultiModalFeatures) GetMmHashes added in v0.7.0

func (x *MultiModalFeatures) GetMmHashes() map[string]*StringList

func (*MultiModalFeatures) GetMmPlaceholders added in v0.7.0

func (x *MultiModalFeatures) GetMmPlaceholders() map[string]*PlaceholderRangeList

func (*MultiModalFeatures) ProtoMessage added in v0.7.0

func (*MultiModalFeatures) ProtoMessage()

func (*MultiModalFeatures) ProtoReflect added in v0.7.0

func (x *MultiModalFeatures) ProtoReflect() protoreflect.Message

func (*MultiModalFeatures) Reset added in v0.7.0

func (x *MultiModalFeatures) Reset()

func (*MultiModalFeatures) String added in v0.7.0

func (x *MultiModalFeatures) String() string

type PlaceholderRange added in v0.7.0

type PlaceholderRange struct {
	Offset int32 `protobuf:"varint,1,opt,name=offset,proto3" json:"offset,omitempty"` // Start index of the placeholder tokens in the prompt
	Length int32 `protobuf:"varint,2,opt,name=length,proto3" json:"length,omitempty"` // Number of placeholder tokens
	// contains filtered or unexported fields
}

PlaceholderRange represents a range of placeholder tokens for a multimodal item

func (*PlaceholderRange) Descriptor deprecated added in v0.7.0

func (*PlaceholderRange) Descriptor() ([]byte, []int)

Deprecated: Use PlaceholderRange.ProtoReflect.Descriptor instead.

func (*PlaceholderRange) GetLength added in v0.7.0

func (x *PlaceholderRange) GetLength() int32

func (*PlaceholderRange) GetOffset added in v0.7.0

func (x *PlaceholderRange) GetOffset() int32

func (*PlaceholderRange) ProtoMessage added in v0.7.0

func (*PlaceholderRange) ProtoMessage()

func (*PlaceholderRange) ProtoReflect added in v0.7.0

func (x *PlaceholderRange) ProtoReflect() protoreflect.Message

func (*PlaceholderRange) Reset added in v0.7.0

func (x *PlaceholderRange) Reset()

func (*PlaceholderRange) String added in v0.7.0

func (x *PlaceholderRange) String() string

type PlaceholderRangeList added in v0.7.0

type PlaceholderRangeList struct {
	Ranges []*PlaceholderRange `protobuf:"bytes,1,rep,name=ranges,proto3" json:"ranges,omitempty"`
	// contains filtered or unexported fields
}

PlaceholderRangeList holds a list of PlaceholderRange (used as map values)

func (*PlaceholderRangeList) Descriptor deprecated added in v0.7.0

func (*PlaceholderRangeList) Descriptor() ([]byte, []int)

Deprecated: Use PlaceholderRangeList.ProtoReflect.Descriptor instead.

func (*PlaceholderRangeList) GetRanges added in v0.7.0

func (x *PlaceholderRangeList) GetRanges() []*PlaceholderRange

func (*PlaceholderRangeList) ProtoMessage added in v0.7.0

func (*PlaceholderRangeList) ProtoMessage()

func (*PlaceholderRangeList) ProtoReflect added in v0.7.0

func (x *PlaceholderRangeList) ProtoReflect() protoreflect.Message

func (*PlaceholderRangeList) Reset added in v0.7.0

func (x *PlaceholderRangeList) Reset()

func (*PlaceholderRangeList) String added in v0.7.0

func (x *PlaceholderRangeList) String() string

type RenderChatCompletionRequest added in v0.7.0

type RenderChatCompletionRequest struct {
	ModelName           string         `protobuf:"bytes,1,opt,name=model_name,json=modelName,proto3" json:"model_name,omitempty"`          // Model name to use for renderer selection
	Messages            []*ChatMessage `protobuf:"bytes,2,rep,name=messages,proto3" json:"messages,omitempty"`                             // Conversation messages
	ToolsJson           *string        `protobuf:"bytes,3,opt,name=tools_json,json=toolsJson,proto3,oneof" json:"tools_json,omitempty"`    // Tools available to the model, as a JSON array
	ChatTemplate        string         `protobuf:"bytes,4,opt,name=chat_template,json=chatTemplate,proto3" json:"chat_template,omitempty"` // Override chat template (optional)
	AddGenerationPrompt *bool          ``                                                                                                  // Whether to append the generation prompt (default: true)
	/* 127-byte string literal not displayed */
	ContinueFinalMessage bool    `protobuf:"varint,6,opt,name=continue_final_message,json=continueFinalMessage,proto3" json:"continue_final_message,omitempty"` // Whether to continue the final message
	ChatTemplateKwargs   *string `protobuf:"bytes,7,opt,name=chat_template_kwargs,json=chatTemplateKwargs,proto3,oneof" json:"chat_template_kwargs,omitempty"`  // Additional chat template arguments as JSON
	// contains filtered or unexported fields
}

RenderChatCompletionRequest contains the fields from an OpenAI chat completion request that are relevant to rendering (chat template application + tokenization).

func (*RenderChatCompletionRequest) Descriptor deprecated added in v0.7.0

func (*RenderChatCompletionRequest) Descriptor() ([]byte, []int)

Deprecated: Use RenderChatCompletionRequest.ProtoReflect.Descriptor instead.

func (*RenderChatCompletionRequest) GetAddGenerationPrompt added in v0.7.0

func (x *RenderChatCompletionRequest) GetAddGenerationPrompt() bool

func (*RenderChatCompletionRequest) GetChatTemplate added in v0.7.0

func (x *RenderChatCompletionRequest) GetChatTemplate() string

func (*RenderChatCompletionRequest) GetChatTemplateKwargs added in v0.7.0

func (x *RenderChatCompletionRequest) GetChatTemplateKwargs() string

func (*RenderChatCompletionRequest) GetContinueFinalMessage added in v0.7.0

func (x *RenderChatCompletionRequest) GetContinueFinalMessage() bool

func (*RenderChatCompletionRequest) GetMessages added in v0.7.0

func (x *RenderChatCompletionRequest) GetMessages() []*ChatMessage

func (*RenderChatCompletionRequest) GetModelName added in v0.7.0

func (x *RenderChatCompletionRequest) GetModelName() string

func (*RenderChatCompletionRequest) GetToolsJson added in v0.7.0

func (x *RenderChatCompletionRequest) GetToolsJson() string

func (*RenderChatCompletionRequest) ProtoMessage added in v0.7.0

func (*RenderChatCompletionRequest) ProtoMessage()

func (*RenderChatCompletionRequest) ProtoReflect added in v0.7.0

func (*RenderChatCompletionRequest) Reset added in v0.7.0

func (x *RenderChatCompletionRequest) Reset()

func (*RenderChatCompletionRequest) String added in v0.7.0

func (x *RenderChatCompletionRequest) String() string

type RenderChatCompletionResponse added in v0.7.0

type RenderChatCompletionResponse struct {
	RequestId    string              `protobuf:"bytes,1,opt,name=request_id,json=requestId,proto3" json:"request_id,omitempty"`          // Request ID from the render response
	TokenIds     []uint32            `protobuf:"varint,2,rep,packed,name=token_ids,json=tokenIds,proto3" json:"token_ids,omitempty"`     // Token IDs for the rendered input
	Features     *MultiModalFeatures `protobuf:"bytes,3,opt,name=features,proto3" json:"features,omitempty"`                             // Multimodal features (populated for MM inputs)
	Success      bool                `protobuf:"varint,4,opt,name=success,proto3" json:"success,omitempty"`                              // Whether the request was successful
	ErrorMessage string              `protobuf:"bytes,5,opt,name=error_message,json=errorMessage,proto3" json:"error_message,omitempty"` // Error message if the request failed
	// contains filtered or unexported fields
}

RenderChatCompletionResponse contains the rendered output from OpenAIServingRender

func (*RenderChatCompletionResponse) Descriptor deprecated added in v0.7.0

func (*RenderChatCompletionResponse) Descriptor() ([]byte, []int)

Deprecated: Use RenderChatCompletionResponse.ProtoReflect.Descriptor instead.

func (*RenderChatCompletionResponse) GetErrorMessage added in v0.7.0

func (x *RenderChatCompletionResponse) GetErrorMessage() string

func (*RenderChatCompletionResponse) GetFeatures added in v0.7.0

func (*RenderChatCompletionResponse) GetRequestId added in v0.7.0

func (x *RenderChatCompletionResponse) GetRequestId() string

func (*RenderChatCompletionResponse) GetSuccess added in v0.7.0

func (x *RenderChatCompletionResponse) GetSuccess() bool

func (*RenderChatCompletionResponse) GetTokenIds added in v0.7.0

func (x *RenderChatCompletionResponse) GetTokenIds() []uint32

func (*RenderChatCompletionResponse) ProtoMessage added in v0.7.0

func (*RenderChatCompletionResponse) ProtoMessage()

func (*RenderChatCompletionResponse) ProtoReflect added in v0.7.0

func (*RenderChatCompletionResponse) Reset added in v0.7.0

func (x *RenderChatCompletionResponse) Reset()

func (*RenderChatCompletionResponse) String added in v0.7.0

type RenderCompletionRequest added in v0.7.0

type RenderCompletionRequest struct {
	ModelName string `protobuf:"bytes,1,opt,name=model_name,json=modelName,proto3" json:"model_name,omitempty"` // Model name to use for renderer selection
	Prompt    string `protobuf:"bytes,2,opt,name=prompt,proto3" json:"prompt,omitempty"`                        // Text prompt to render
	// contains filtered or unexported fields
}

RenderCompletionRequest contains the fields from an OpenAI completion request that are relevant to rendering (validation + tokenization).

func (*RenderCompletionRequest) Descriptor deprecated added in v0.7.0

func (*RenderCompletionRequest) Descriptor() ([]byte, []int)

Deprecated: Use RenderCompletionRequest.ProtoReflect.Descriptor instead.

func (*RenderCompletionRequest) GetModelName added in v0.7.0

func (x *RenderCompletionRequest) GetModelName() string

func (*RenderCompletionRequest) GetPrompt added in v0.7.0

func (x *RenderCompletionRequest) GetPrompt() string

func (*RenderCompletionRequest) ProtoMessage added in v0.7.0

func (*RenderCompletionRequest) ProtoMessage()

func (*RenderCompletionRequest) ProtoReflect added in v0.7.0

func (x *RenderCompletionRequest) ProtoReflect() protoreflect.Message

func (*RenderCompletionRequest) Reset added in v0.7.0

func (x *RenderCompletionRequest) Reset()

func (*RenderCompletionRequest) String added in v0.7.0

func (x *RenderCompletionRequest) String() string

type RenderCompletionResponse added in v0.7.0

type RenderCompletionResponse struct {
	RequestId    string   `protobuf:"bytes,1,opt,name=request_id,json=requestId,proto3" json:"request_id,omitempty"`          // Request ID from the render response
	TokenIds     []uint32 `protobuf:"varint,2,rep,packed,name=token_ids,json=tokenIds,proto3" json:"token_ids,omitempty"`     // Token IDs for the rendered prompt
	Success      bool     `protobuf:"varint,3,opt,name=success,proto3" json:"success,omitempty"`                              // Whether the request was successful
	ErrorMessage string   `protobuf:"bytes,4,opt,name=error_message,json=errorMessage,proto3" json:"error_message,omitempty"` // Error message if the request failed
	// contains filtered or unexported fields
}

RenderCompletionResponse contains the rendered output for a completion request.

func (*RenderCompletionResponse) Descriptor deprecated added in v0.7.0

func (*RenderCompletionResponse) Descriptor() ([]byte, []int)

Deprecated: Use RenderCompletionResponse.ProtoReflect.Descriptor instead.

func (*RenderCompletionResponse) GetErrorMessage added in v0.7.0

func (x *RenderCompletionResponse) GetErrorMessage() string

func (*RenderCompletionResponse) GetRequestId added in v0.7.0

func (x *RenderCompletionResponse) GetRequestId() string

func (*RenderCompletionResponse) GetSuccess added in v0.7.0

func (x *RenderCompletionResponse) GetSuccess() bool

func (*RenderCompletionResponse) GetTokenIds added in v0.7.0

func (x *RenderCompletionResponse) GetTokenIds() []uint32

func (*RenderCompletionResponse) ProtoMessage added in v0.7.0

func (*RenderCompletionResponse) ProtoMessage()

func (*RenderCompletionResponse) ProtoReflect added in v0.7.0

func (x *RenderCompletionResponse) ProtoReflect() protoreflect.Message

func (*RenderCompletionResponse) Reset added in v0.7.0

func (x *RenderCompletionResponse) Reset()

func (*RenderCompletionResponse) String added in v0.7.0

func (x *RenderCompletionResponse) String() string

type StringList added in v0.7.0

type StringList struct {
	Values []string `protobuf:"bytes,1,rep,name=values,proto3" json:"values,omitempty"`
	// contains filtered or unexported fields
}

StringList holds a list of strings (used as map values)

func (*StringList) Descriptor deprecated added in v0.7.0

func (*StringList) Descriptor() ([]byte, []int)

Deprecated: Use StringList.ProtoReflect.Descriptor instead.

func (*StringList) GetValues added in v0.7.0

func (x *StringList) GetValues() []string

func (*StringList) ProtoMessage added in v0.7.0

func (*StringList) ProtoMessage()

func (*StringList) ProtoReflect added in v0.7.0

func (x *StringList) ProtoReflect() protoreflect.Message

func (*StringList) Reset added in v0.7.0

func (x *StringList) Reset()

func (*StringList) String added in v0.7.0

func (x *StringList) String() string

type StructValue

type StructValue struct {
	Fields map[string]*Value `` /* 139-byte string literal not displayed */
	// contains filtered or unexported fields
}

StructValue represents a structured value (key-value pairs)

func (*StructValue) Descriptor deprecated

func (*StructValue) Descriptor() ([]byte, []int)

Deprecated: Use StructValue.ProtoReflect.Descriptor instead.

func (*StructValue) GetFields

func (x *StructValue) GetFields() map[string]*Value

func (*StructValue) ProtoMessage

func (*StructValue) ProtoMessage()

func (*StructValue) ProtoReflect

func (x *StructValue) ProtoReflect() protoreflect.Message

func (*StructValue) Reset

func (x *StructValue) Reset()

func (*StructValue) String

func (x *StructValue) String() string

type TokenizationServiceClient

type TokenizationServiceClient interface {
	// Tokenize converts a text input to token IDs
	Tokenize(ctx context.Context, in *TokenizeRequest, opts ...grpc.CallOption) (*TokenizeResponse, error)
	// Deprecated: use RenderChatCompletion instead.
	RenderChatTemplate(ctx context.Context, in *ChatTemplateRequest, opts ...grpc.CallOption) (*ChatTemplateResponse, error)
	// InitializeTokenizer initializes the tokenizer for a specific model
	InitializeTokenizer(ctx context.Context, in *InitializeTokenizerRequest, opts ...grpc.CallOption) (*InitializeTokenizerResponse, error)
	// RenderChatCompletion renders an OpenAI chat completion request via OpenAIServingRender,
	// returning token IDs and multimodal features as a GenerateRequest-compatible response
	RenderChatCompletion(ctx context.Context, in *RenderChatCompletionRequest, opts ...grpc.CallOption) (*RenderChatCompletionResponse, error)
	// RenderCompletion renders an OpenAI completion request via OpenAIServingRender,
	// returning the rendered output for the given prompt
	RenderCompletion(ctx context.Context, in *RenderCompletionRequest, opts ...grpc.CallOption) (*RenderCompletionResponse, error)
}

TokenizationServiceClient is the client API for TokenizationService service.

For semantics around ctx use and closing/ending streaming RPCs, please refer to https://pkg.go.dev/google.golang.org/grpc/?tab=doc#ClientConn.NewStream.

TokenizationService defines the gRPC service for tokenization

type TokenizationServiceServer

type TokenizationServiceServer interface {
	// Tokenize converts a text input to token IDs
	Tokenize(context.Context, *TokenizeRequest) (*TokenizeResponse, error)
	// Deprecated: use RenderChatCompletion instead.
	RenderChatTemplate(context.Context, *ChatTemplateRequest) (*ChatTemplateResponse, error)
	// InitializeTokenizer initializes the tokenizer for a specific model
	InitializeTokenizer(context.Context, *InitializeTokenizerRequest) (*InitializeTokenizerResponse, error)
	// RenderChatCompletion renders an OpenAI chat completion request via OpenAIServingRender,
	// returning token IDs and multimodal features as a GenerateRequest-compatible response
	RenderChatCompletion(context.Context, *RenderChatCompletionRequest) (*RenderChatCompletionResponse, error)
	// RenderCompletion renders an OpenAI completion request via OpenAIServingRender,
	// returning the rendered output for the given prompt
	RenderCompletion(context.Context, *RenderCompletionRequest) (*RenderCompletionResponse, error)
	// contains filtered or unexported methods
}

TokenizationServiceServer is the server API for TokenizationService service. All implementations must embed UnimplementedTokenizationServiceServer for forward compatibility.

TokenizationService defines the gRPC service for tokenization

type TokenizeRequest

type TokenizeRequest struct {
	Input            string `protobuf:"bytes,1,opt,name=input,proto3" json:"input,omitempty"`                                                  // The text input to tokenize
	ModelName        string `protobuf:"bytes,2,opt,name=model_name,json=modelName,proto3" json:"model_name,omitempty"`                         // The name of the model to use for tokenization
	AddSpecialTokens bool   `protobuf:"varint,3,opt,name=add_special_tokens,json=addSpecialTokens,proto3" json:"add_special_tokens,omitempty"` // Whether to add special tokens during tokenization
	// contains filtered or unexported fields
}

TokenizeRequest represents a request to tokenize a text input

func (*TokenizeRequest) Descriptor deprecated

func (*TokenizeRequest) Descriptor() ([]byte, []int)

Deprecated: Use TokenizeRequest.ProtoReflect.Descriptor instead.

func (*TokenizeRequest) GetAddSpecialTokens

func (x *TokenizeRequest) GetAddSpecialTokens() bool

func (*TokenizeRequest) GetInput

func (x *TokenizeRequest) GetInput() string

func (*TokenizeRequest) GetModelName

func (x *TokenizeRequest) GetModelName() string

func (*TokenizeRequest) ProtoMessage

func (*TokenizeRequest) ProtoMessage()

func (*TokenizeRequest) ProtoReflect

func (x *TokenizeRequest) ProtoReflect() protoreflect.Message

func (*TokenizeRequest) Reset

func (x *TokenizeRequest) Reset()

func (*TokenizeRequest) String

func (x *TokenizeRequest) String() string

type TokenizeResponse

type TokenizeResponse struct {
	InputIds     []uint32 `protobuf:"varint,1,rep,packed,name=input_ids,json=inputIds,proto3" json:"input_ids,omitempty"`     // Token IDs for the input
	Success      bool     `protobuf:"varint,2,opt,name=success,proto3" json:"success,omitempty"`                              // Whether the request was successful
	ErrorMessage string   `protobuf:"bytes,3,opt,name=error_message,json=errorMessage,proto3" json:"error_message,omitempty"` // Error message if the request failed
	// Direct array of [start, end] pairs
	OffsetPairs []uint32 `protobuf:"varint,4,rep,packed,name=offset_pairs,json=offsetPairs,proto3" json:"offset_pairs,omitempty"` // Flattened array of [start, end, start, end, ...]
	// contains filtered or unexported fields
}

TokenizeResponse represents the response from tokenization

func (*TokenizeResponse) Descriptor deprecated

func (*TokenizeResponse) Descriptor() ([]byte, []int)

Deprecated: Use TokenizeResponse.ProtoReflect.Descriptor instead.

func (*TokenizeResponse) GetErrorMessage

func (x *TokenizeResponse) GetErrorMessage() string

func (*TokenizeResponse) GetInputIds

func (x *TokenizeResponse) GetInputIds() []uint32

func (*TokenizeResponse) GetOffsetPairs

func (x *TokenizeResponse) GetOffsetPairs() []uint32

func (*TokenizeResponse) GetSuccess

func (x *TokenizeResponse) GetSuccess() bool

func (*TokenizeResponse) ProtoMessage

func (*TokenizeResponse) ProtoMessage()

func (*TokenizeResponse) ProtoReflect

func (x *TokenizeResponse) ProtoReflect() protoreflect.Message

func (*TokenizeResponse) Reset

func (x *TokenizeResponse) Reset()

func (*TokenizeResponse) String

func (x *TokenizeResponse) String() string

type ToolDescription

type ToolDescription struct {
	Tool map[string]*Value `` // Tool definition
	/* 135-byte string literal not displayed */
	// contains filtered or unexported fields
}

ToolDescription represents a description of a tool

func (*ToolDescription) Descriptor deprecated

func (*ToolDescription) Descriptor() ([]byte, []int)

Deprecated: Use ToolDescription.ProtoReflect.Descriptor instead.

func (*ToolDescription) GetTool

func (x *ToolDescription) GetTool() map[string]*Value

func (*ToolDescription) ProtoMessage

func (*ToolDescription) ProtoMessage()

func (*ToolDescription) ProtoReflect

func (x *ToolDescription) ProtoReflect() protoreflect.Message

func (*ToolDescription) Reset

func (x *ToolDescription) Reset()

func (*ToolDescription) String

func (x *ToolDescription) String() string

type UnimplementedTokenizationServiceServer

type UnimplementedTokenizationServiceServer struct{}

UnimplementedTokenizationServiceServer must be embedded to have forward compatible implementations.

NOTE: this should be embedded by value instead of pointer to avoid a nil pointer dereference when methods are called.

func (UnimplementedTokenizationServiceServer) InitializeTokenizer

func (UnimplementedTokenizationServiceServer) RenderChatCompletion added in v0.7.0

func (UnimplementedTokenizationServiceServer) RenderChatTemplate

func (UnimplementedTokenizationServiceServer) RenderCompletion added in v0.7.0

func (UnimplementedTokenizationServiceServer) Tokenize

type UnsafeTokenizationServiceServer

type UnsafeTokenizationServiceServer interface {
	// contains filtered or unexported methods
}

UnsafeTokenizationServiceServer may be embedded to opt out of forward compatibility for this service. Use of this interface is not recommended, as added methods to TokenizationServiceServer will result in compilation errors.

type Value

type Value struct {

	// Types that are valid to be assigned to Value:
	//
	//	*Value_StringValue
	//	*Value_NumberValue
	//	*Value_BoolValue
	//	*Value_ListValue
	//	*Value_StructValue
	Value isValue_Value `protobuf_oneof:"value"`
	// contains filtered or unexported fields
}

Value represents a generic value that can be string, number, bool, or list

func (*Value) Descriptor deprecated

func (*Value) Descriptor() ([]byte, []int)

Deprecated: Use Value.ProtoReflect.Descriptor instead.

func (*Value) GetBoolValue

func (x *Value) GetBoolValue() bool

func (*Value) GetListValue

func (x *Value) GetListValue() *ListValue

func (*Value) GetNumberValue

func (x *Value) GetNumberValue() float64

func (*Value) GetStringValue

func (x *Value) GetStringValue() string

func (*Value) GetStructValue

func (x *Value) GetStructValue() *StructValue

func (*Value) GetValue

func (x *Value) GetValue() isValue_Value

func (*Value) ProtoMessage

func (*Value) ProtoMessage()

func (*Value) ProtoReflect

func (x *Value) ProtoReflect() protoreflect.Message

func (*Value) Reset

func (x *Value) Reset()

func (*Value) String

func (x *Value) String() string

type Value_BoolValue

type Value_BoolValue struct {
	BoolValue bool `protobuf:"varint,3,opt,name=bool_value,json=boolValue,proto3,oneof"`
}

type Value_ListValue

type Value_ListValue struct {
	ListValue *ListValue `protobuf:"bytes,4,opt,name=list_value,json=listValue,proto3,oneof"`
}

type Value_NumberValue

type Value_NumberValue struct {
	NumberValue float64 `protobuf:"fixed64,2,opt,name=number_value,json=numberValue,proto3,oneof"`
}

type Value_StringValue

type Value_StringValue struct {
	StringValue string `protobuf:"bytes,1,opt,name=string_value,json=stringValue,proto3,oneof"`
}

type Value_StructValue

type Value_StructValue struct {
	StructValue *StructValue `protobuf:"bytes,5,opt,name=struct_value,json=structValue,proto3,oneof"`
}

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL