pipeline

Documentation

Overview

Package pipeline provides a way to chain multiple llm executions.

Index

• Constants
• Variables
• func NewSplitter(llm Llm, splitterFn SplitterFn) *splitter
• func SetNextTube(output types.M, nextTube int) types.M
• func SetNextTubeExit(output types.M) types.M
• type Decoder
• type Llm
• type LlmEngine
• type LlmMode
• type Memory
• type Pipe
• type Pipeline
  • func New(pipes ...Pipe) *Pipeline
  • func (p Pipeline) Run(ctx context.Context, input types.M) (types.M, error)
  • func (p *Pipeline) WithPostCallbacks(callbacks ...PipelineCallback) *Pipeline
  • func (p *Pipeline) WithPreCallbacks(callbacks ...PipelineCallback) *Pipeline
• type PipelineCallback
• type Prompt
• type SplitterFn
• type Tube
  • func NewTube(llm Llm) *Tube
  • func (t *Tube) Namespace() string
  • func (s *Tube) Run(ctx context.Context, input types.M) (types.M, error)
  • func (t *Tube) WithDecoder(decoder Decoder) *Tube
  • func (t *Tube) WithMemory(namespace string, memory Memory) *Tube

Constants

const (
	NextTubeKey  = "next_tube"
	NextTubeExit = -1
)

Variables

var (
	ErrDecoding       = errors.New("decoding input error")
	ErrInvalidLmmMode = errors.New("invalid LLM mode")
	ErrLLMExecution   = errors.New("llm execution error")
)

var (
	ErrSplitFunction = fmt.Errorf("split function error")
)

Functions

func NewSplitter

func NewSplitter(
	llm Llm,
	splitterFn SplitterFn,
) *splitter

func SetNextTube

func SetNextTube(output types.M, nextTube int) types.M

func SetNextTubeExit

func SetNextTubeExit(output types.M) types.M

Types

type Decoder

type Decoder interface {
	Decode(input string) (types.M, error)
}

type Llm

type Llm struct {
	LlmEngine LlmEngine
	LlmMode   LlmMode
	Prompt    Prompt
	Chat      *chat.Chat
}

type LlmEngine

type LlmEngine interface {
	Completion(ctx context.Context, prompt string) (string, error)
	Chat(ctx context.Context, chat *chat.Chat) (string, error)
}

type LlmMode

type LlmMode int

const (
	LlmModeChat LlmMode = iota
	LlmModeCompletion
)

type Memory

type Memory interface {
	Get(key string) interface{}
	Set(key string, value interface{}) error
	All() types.M
	Delete(key string) error
	Clear() error
}

type Pipe

type Pipe interface {
	Run(ctx context.Context, input types.M) (types.M, error)
}

type Pipeline

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

func New

func New(pipes ...Pipe) *Pipeline

func (Pipeline) Run

func (p Pipeline) Run(ctx context.Context, input types.M) (types.M, error)

Run chains the steps of the pipeline and returns the output of the last step.

func (*Pipeline) WithPostCallbacks

func (p *Pipeline) WithPostCallbacks(callbacks ...PipelineCallback) *Pipeline

func (*Pipeline) WithPreCallbacks

func (p *Pipeline) WithPreCallbacks(callbacks ...PipelineCallback) *Pipeline

type PipelineCallback

type PipelineCallback func(values types.M) (types.M, error)

type Prompt

type Prompt interface {
	String() string
	Format(input types.M) error
}

type SplitterFn

type SplitterFn func(input types.M) ([]types.M, error)

type Tube

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

func NewTube

func NewTube(
	llm Llm,
) *Tube

func (*Tube) Namespace

func (t *Tube) Namespace() string

func (*Tube) Run

func (s *Tube) Run(ctx context.Context, input types.M) (types.M, error)

Run execute the step and return the output. The prompt is formatted with the input and the output of the prompt is used as input for the LLM. If the step has a memory, the output is stored in the memory.

func (*Tube) WithDecoder

func (t *Tube) WithDecoder(decoder Decoder) *Tube

func (*Tube) WithMemory

func (t *Tube) WithMemory(namespace string, memory Memory) *Tube