openai

Documentation

Index

• type Config
• type Plugin
  • func NewOpenAIPlugin(appctx *stream_context.Context, config Config) (*Plugin, error)
  • func (p *Plugin) EvolveSchema(streamSchema *schema.StreamSchemaObj) error
  • func (p *Plugin) Process(context context.Context, msg *message.Message) (*message.Message, error)

Types

type Config

type Config struct {
	ApiKey         string `json:"api_key" yaml:"api_key"`
	SourceField    string `json:"source_field" yaml:"source_field"`
	TargetField    string `json:"target_field" yaml:"target_field"`
	Model          string `json:"model" yaml:"model"`
	Prompt         string `json:"prompt" yaml:"prompt"`
	StreamName     string `json:"stream_name" yaml:"stream_name"`
	LimitPerMinute int64  `json:"limit_per_minute" yaml:"limit_per_minute"`
}

type Plugin

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

func NewOpenAIPlugin

func NewOpenAIPlugin(appctx *stream_context.Context, config Config) (*Plugin, error)

func (*Plugin) EvolveSchema

func (p *Plugin) EvolveSchema(streamSchema *schema.StreamSchemaObj) error

EvolveSchema will add a string column to the schema in order to store OpenAI response

func (*Plugin) Process

func (p *Plugin) Process(context context.Context, msg *message.Message) (*message.Message, error)