generate

Documentation

Index

• type Generate
  • func NewGenerate(ctx context.Context, promptName string, promptVars any, outputType string, ...) (*Generate, error)
  • func (g *Generate) BuildPrompt() ([]llms.MessageContent, error)
  • func (g *Generate) Generate() error
  • func (g *Generate) GenerateFileName() string
  • func (g *Generate) GetFileSuffix() string
  • func (g *Generate) GetOutputFile() string
  • func (g *Generate) GetOutputResults() string
  • func (g *Generate) SaveAndPrintResults() error
  • func (g *Generate) SaveFile(filename string, data []byte) error

Types

type Generate

type Generate struct {
	Context     context.Context
	AIClient    ai.IAI
	AIProvider  string
	PromptName  string
	PromptVars  any
	OutputType  string
	Output      any
	Results     string
	RunningTime string
}

func NewGenerate

func NewGenerate(ctx context.Context, promptName string, promptVars any, outputType string, output any) (*Generate, error)

func (*Generate) BuildPrompt

func (g *Generate) BuildPrompt() ([]llms.MessageContent, error)

BuildPrompt builds the prompt based on the prompt name and variables.

func (*Generate) Generate

func (g *Generate) Generate() error

Generate generates the text based on the prompt.

func (*Generate) GenerateFileName

func (g *Generate) GenerateFileName() string

GenerateFileName generates the file name based on the prompt variables.

func (*Generate) GetFileSuffix

func (g *Generate) GetFileSuffix() string

GetFileSuffix returns the file suffix based on the output format.

func (*Generate) GetOutputFile

func (g *Generate) GetOutputFile() string

GetOutputFile returns the output file path.

func (*Generate) GetOutputResults

func (g *Generate) GetOutputResults() string

GetOutputResults returns the results.

func (*Generate) SaveAndPrintResults

func (g *Generate) SaveAndPrintResults() error

SaveAndPrintResults saves the results and errors to the files and prints the errors.

func (*Generate) SaveFile

func (g *Generate) SaveFile(filename string, data []byte) error

SaveFile saves the data to the specified file.