custom

package
v0.0.0-...-2ec7614 Latest Latest
Warning

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

Go to latest
Published: Apr 23, 2024 License: MIT Imports: 14 Imported by: 0

Documentation

Index

Constants

View Source
const (
	Completer_Complete_FullMethodName = "/completer.completer/Complete"
)

Variables

View Source
var (
	Role_name = map[int32]string{
		0: "ROLE_UNSPECIFIED",
		1: "ROLE_SYSTEM",
		2: "ROLE_USER",
		3: "ROLE_ASSISTANT",
	}
	Role_value = map[string]int32{
		"ROLE_UNSPECIFIED": 0,
		"ROLE_SYSTEM":      1,
		"ROLE_USER":        2,
		"ROLE_ASSISTANT":   3,
	}
)

Enum value maps for Role.

View Source
var Completer_ServiceDesc = grpc.ServiceDesc{
	ServiceName: "completer.completer",
	HandlerType: (*CompleterServer)(nil),
	Methods:     []grpc.MethodDesc{},
	Streams: []grpc.StreamDesc{
		{
			StreamName:    "Complete",
			Handler:       _Completer_Complete_Handler,
			ServerStreams: true,
		},
	},
	Metadata: "completer.proto",
}

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

View Source
var File_completer_proto protoreflect.FileDescriptor

Functions

func RegisterCompleterServer

func RegisterCompleterServer(s grpc.ServiceRegistrar, srv CompleterServer)

Types

type Client

type Client struct {
	// contains filtered or unexported fields
}

func New

func New(url string, options ...Option) (*Client, error)

func (*Client) Complete

func (c *Client) Complete(ctx context.Context, messages []provider.Message, options *provider.CompleteOptions) (*provider.Completion, error)

type CompleterClient

type CompleterClient interface {
	Complete(ctx context.Context, in *CompletionRequest, opts ...grpc.CallOption) (Completer_CompleteClient, error)
}

CompleterClient is the client API for Completer 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.

func NewCompleterClient

func NewCompleterClient(cc grpc.ClientConnInterface) CompleterClient

type CompleterServer

type CompleterServer interface {
	Complete(*CompletionRequest, Completer_CompleteServer) error
	// contains filtered or unexported methods
}

CompleterServer is the server API for Completer service. All implementations must embed UnimplementedCompleterServer for forward compatibility

type Completer_CompleteClient

type Completer_CompleteClient interface {
	Recv() (*Completion, error)
	grpc.ClientStream
}

type Completer_CompleteServer

type Completer_CompleteServer interface {
	Send(*Completion) error
	grpc.ServerStream
}

type Completion

type Completion struct {
	Id      string   `protobuf:"bytes,1,opt,name=id,proto3" json:"id,omitempty"`
	Message *Message `protobuf:"bytes,2,opt,name=message,proto3" json:"message,omitempty"`
	// contains filtered or unexported fields
}

func (*Completion) Descriptor deprecated

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

Deprecated: Use Completion.ProtoReflect.Descriptor instead.

func (*Completion) GetId

func (x *Completion) GetId() string

func (*Completion) GetMessage

func (x *Completion) GetMessage() *Message

func (*Completion) ProtoMessage

func (*Completion) ProtoMessage()

func (*Completion) ProtoReflect

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

func (*Completion) Reset

func (x *Completion) Reset()

func (*Completion) String

func (x *Completion) String() string

type CompletionRequest

type CompletionRequest struct {
	Model       string     `protobuf:"bytes,1,opt,name=model,proto3" json:"model,omitempty"`
	Messages    []*Message `protobuf:"bytes,2,rep,name=messages,proto3" json:"messages,omitempty"`
	Stop        []string   `protobuf:"bytes,5,rep,name=stop,proto3" json:"stop,omitempty"`
	MaxTokens   *int32     `protobuf:"varint,4,opt,name=max_tokens,json=maxTokens,proto3,oneof" json:"max_tokens,omitempty"`
	Temperature *float32   `protobuf:"fixed32,3,opt,name=temperature,proto3,oneof" json:"temperature,omitempty"`
	// contains filtered or unexported fields
}

func (*CompletionRequest) Descriptor deprecated

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

Deprecated: Use CompletionRequest.ProtoReflect.Descriptor instead.

func (*CompletionRequest) GetMaxTokens

func (x *CompletionRequest) GetMaxTokens() int32

func (*CompletionRequest) GetMessages

func (x *CompletionRequest) GetMessages() []*Message

func (*CompletionRequest) GetModel

func (x *CompletionRequest) GetModel() string

func (*CompletionRequest) GetStop

func (x *CompletionRequest) GetStop() []string

func (*CompletionRequest) GetTemperature

func (x *CompletionRequest) GetTemperature() float32

func (*CompletionRequest) ProtoMessage

func (*CompletionRequest) ProtoMessage()

func (*CompletionRequest) ProtoReflect

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

func (*CompletionRequest) Reset

func (x *CompletionRequest) Reset()

func (*CompletionRequest) String

func (x *CompletionRequest) String() string

type Message

type Message struct {
	Role    Role   `protobuf:"varint,1,opt,name=role,proto3,enum=completer.Role" json:"role,omitempty"`
	Content string `protobuf:"bytes,2,opt,name=content,proto3" json:"content,omitempty"`
	// contains filtered or unexported fields
}

func (*Message) Descriptor deprecated

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

Deprecated: Use Message.ProtoReflect.Descriptor instead.

func (*Message) GetContent

func (x *Message) GetContent() string

func (*Message) GetRole

func (x *Message) GetRole() Role

func (*Message) ProtoMessage

func (*Message) ProtoMessage()

func (*Message) ProtoReflect

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

func (*Message) Reset

func (x *Message) Reset()

func (*Message) String

func (x *Message) String() string

type Option

type Option func(*Client)

func WithModel

func WithModel(model string) Option

type Role

type Role int32
const (
	Role_ROLE_UNSPECIFIED Role = 0
	Role_ROLE_SYSTEM      Role = 1
	Role_ROLE_USER        Role = 2
	Role_ROLE_ASSISTANT   Role = 3
)

func (Role) Descriptor

func (Role) Descriptor() protoreflect.EnumDescriptor

func (Role) Enum

func (x Role) Enum() *Role

func (Role) EnumDescriptor deprecated

func (Role) EnumDescriptor() ([]byte, []int)

Deprecated: Use Role.Descriptor instead.

func (Role) Number

func (x Role) Number() protoreflect.EnumNumber

func (Role) String

func (x Role) String() string

func (Role) Type

func (Role) Type() protoreflect.EnumType

type UnimplementedCompleterServer

type UnimplementedCompleterServer struct {
}

UnimplementedCompleterServer must be embedded to have forward compatible implementations.

func (UnimplementedCompleterServer) Complete

type UnsafeCompleterServer

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

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

Jump to

Keyboard shortcuts

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