modules

package

v0.15.0 Latest Latest Go to latest Published: Jan 10, 2025 License: MIT Imports: 8 Imported by: 1

Details

This section is empty.

This section is empty.

This section is empty.

type ChainOfThought struct {
	Predict *Predict
}

func NewChainOfThought(signature core.Signature) *ChainOfThought

func (c *ChainOfThought) Clone() core.Module

func (c *ChainOfThought) Compose(next core.Module) core.Module

func (c *ChainOfThought) GetSignature() core.Signature

func (c *ChainOfThought) GetSubModules() []core.Module

func (c *ChainOfThought) Process(ctx context.Context, inputs map[string]any) (map[string]any, error)

func (c *ChainOfThought) SetLLM(llm core.LLM)

func (c *ChainOfThought) SetSubModules(modules []core.Module)

type Module interface {
	Forward(inputs map[string]interface{}) (Predict, error)
}

type Predict struct {
	core.BaseModule
	Demos []core.Example
	LLM   core.LLM
}

func NewPredict(signature core.Signature) *Predict

func (p *Predict) Clone() core.Module

func (p *Predict) FormatOutputs(outputs map[string]interface{}) map[string]interface{}

func (p *Predict) GetDemos() []core.Example

func (p *Predict) GetSignature() core.Signature

func (p *Predict) Process(ctx context.Context, inputs map[string]interface{}) (map[string]interface{}, error)

func (p *Predict) SetDemos(demos []core.Example)

func (p *Predict) SetLLM(llm core.LLM)

func (p *Predict) ValidateInputs(inputs map[string]interface{}) error

type ReAct struct {
	core.BaseModule
	Predict  *Predict
	Tools    []Tool
	MaxIters int
}

func NewReAct(signature core.Signature, tools []Tool, maxIters int) *ReAct

func (r *ReAct) Clone() core.Module

func (r *ReAct) Process(ctx context.Context, inputs map[string]any) (map[string]any, error)

func (r *ReAct) SetLLM(llm core.LLM)

type Tool interface {
	CanHandle(action string) bool
	Execute(ctx context.Context, action string) (string, error)
}