ai_content_moderation

Documentation

Index

• type Config
• type Plugin
  • func NewAIContentModerationPlugin(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"`
	StreamName  string `json:"stream_name" yaml:"stream_name"`
}

type Plugin

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

func NewAIContentModerationPlugin

func NewAIContentModerationPlugin(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)