waveai

Documentation

Index

• Constants
• func ConvertPrompt(prompt []wshrpc.OpenAIPromptMessageType) []openaiapi.ChatCompletionMessage
• func GetWSEndpoint() string
• func IsCloudAIRequest(opts *wshrpc.OpenAIOptsType) bool
• func MakeOpenAIPacket() *wshrpc.OpenAIPacketType
• func RunAICommand(ctx context.Context, request wshrpc.OpenAiStreamRequest) chan wshrpc.RespOrErrorUnion[wshrpc.OpenAIPacketType]
• func RunCloudCompletionStream(ctx context.Context, request wshrpc.OpenAiStreamRequest) chan wshrpc.RespOrErrorUnion[wshrpc.OpenAIPacketType]
• func RunLocalCompletionStream(ctx context.Context, request wshrpc.OpenAiStreamRequest) chan wshrpc.RespOrErrorUnion[wshrpc.OpenAIPacketType]
• type OpenAICloudReqPacketType
  • func MakeOpenAICloudReqPacket() *OpenAICloudReqPacketType
• type OpenAICmdInfoChatMessage
• type OpenAICmdInfoPacketOutputType

Constants

const CloudWebsocketConnectTimeout = 1 * time.Minute

const DefaultAzureAPIVersion = "2023-05-15"

const DefaultMaxTokens = 2048

const DefaultModel = "gpt-4o-mini"

const OpenAICloudReqStr = "openai-cloudreq"

const OpenAIPacketStr = "openai"

const PacketEOFStr = "EOF"

const WCloudWSEndpoint = "wss://wsapi.waveterm.dev/"

const WCloudWSEndpointVarName = "WCLOUD_WS_ENDPOINT"

Functions

func ConvertPrompt

func ConvertPrompt(prompt []wshrpc.OpenAIPromptMessageType) []openaiapi.ChatCompletionMessage

func GetWSEndpoint

func GetWSEndpoint() string

func IsCloudAIRequest

func IsCloudAIRequest(opts *wshrpc.OpenAIOptsType) bool

func MakeOpenAIPacket

func MakeOpenAIPacket() *wshrpc.OpenAIPacketType

func RunAICommand

func RunAICommand(ctx context.Context, request wshrpc.OpenAiStreamRequest) chan wshrpc.RespOrErrorUnion[wshrpc.OpenAIPacketType]

func RunCloudCompletionStream

func RunCloudCompletionStream(ctx context.Context, request wshrpc.OpenAiStreamRequest) chan wshrpc.RespOrErrorUnion[wshrpc.OpenAIPacketType]

func RunLocalCompletionStream

func RunLocalCompletionStream(ctx context.Context, request wshrpc.OpenAiStreamRequest) chan wshrpc.RespOrErrorUnion[wshrpc.OpenAIPacketType]

Types

type OpenAICloudReqPacketType

type OpenAICloudReqPacketType struct {
	Type       string                           `json:"type"`
	ClientId   string                           `json:"clientid"`
	Prompt     []wshrpc.OpenAIPromptMessageType `json:"prompt"`
	MaxTokens  int                              `json:"maxtokens,omitempty"`
	MaxChoices int                              `json:"maxchoices,omitempty"`
}

func MakeOpenAICloudReqPacket

func MakeOpenAICloudReqPacket() *OpenAICloudReqPacketType

type OpenAICmdInfoChatMessage

type OpenAICmdInfoChatMessage struct {
	MessageID           int                            `json:"messageid"`
	IsAssistantResponse bool                           `json:"isassistantresponse,omitempty"`
	AssistantResponse   *OpenAICmdInfoPacketOutputType `json:"assistantresponse,omitempty"`
	UserQuery           string                         `json:"userquery,omitempty"`
	UserEngineeredQuery string                         `json:"userengineeredquery,omitempty"`
}

type OpenAICmdInfoPacketOutputType

type OpenAICmdInfoPacketOutputType struct {
	Model        string `json:"model,omitempty"`
	Created      int64  `json:"created,omitempty"`
	FinishReason string `json:"finish_reason,omitempty"`
	Message      string `json:"message,omitempty"`
	Error        string `json:"error,omitempty"`
}