chain

package

v0.2.0 Latest Latest Go to latest Published: Feb 28, 2024 License: Apache-2.0 Imports: 19 Imported by: 0

Details

Valid go.mod file
Redistributable license
Tagged version
Stable version
Learn more about best practices

Repository

github.com/kubeagi/arcadia

Links

Open Source Insights

Documentation ¶

Index ¶

Constants
func GetChainOptions(config v1alpha1.CommonChainConfig) []chains.ChainCallOption
type APIChain
- func NewAPIChain(baseNode base.BaseNode) *APIChain
type LLMChain
- func NewLLMChain(baseNode base.BaseNode) *LLMChain
type MapReduceChain
- func NewMapReduceChain(baseNode base.BaseNode) *MapReduceChain
type RetrievalQAChain
- func NewRetrievalQAChain(baseNode base.BaseNode) *RetrievalQAChain

Constants ¶

View Source

const (
	// For map-reduce
	DefaultPromptTemplateForMap = `
		{{.context}}

		With above content, please summarize it with only half content size of it.
		`
	DefaultPromptTemplatForReduce = `"{{.context}}"`

	// For post process the map-reduced summary
	DefaultPromptTemplateForPostMapReduce = `` /* 258-byte string literal not displayed */

	DefaultSummaryMaxNumberOfConcurrent = 2
	DefaultDocumentChunkSize            = 1024
	DefaultDocumentChunkOverlap         = 100
)

Variables ¶

This section is empty.

Functions ¶

func GetChainOptions ¶

func GetChainOptions(config v1alpha1.CommonChainConfig) []chains.ChainCallOption

Types ¶

type APIChain ¶

type APIChain struct {
	chains.APIChain
	base.BaseNode
	Instance *v1alpha1.APIChain
}

func NewAPIChain ¶

func NewAPIChain(baseNode base.BaseNode) *APIChain

func (*APIChain) Init ¶

func (l *APIChain) Init(ctx context.Context, cli client.Client, _ map[string]any) error

func (*APIChain) Ready ¶

func (l *APIChain) Ready() (isReady bool, msg string)

func (*APIChain) Run ¶

func (l *APIChain) Run(ctx context.Context, _ client.Client, args map[string]any) (map[string]any, error)

type LLMChain ¶

type LLMChain struct {
	chains.LLMChain
	base.BaseNode
	Instance *v1alpha1.LLMChain
}

func NewLLMChain ¶

func NewLLMChain(baseNode base.BaseNode) *LLMChain

func (*LLMChain) Init ¶

func (l *LLMChain) Init(ctx context.Context, cli client.Client, _ map[string]any) error

func (*LLMChain) Ready ¶

func (l *LLMChain) Ready() (isReady bool, msg string)

func (*LLMChain) Run ¶

func (l *LLMChain) Run(ctx context.Context, _ client.Client, args map[string]any) (outArgs map[string]any, err error)

type MapReduceChain ¶

type MapReduceChain struct {
	// BaseNode for this MapReduceChain
	// Only chain is allowed
	base.BaseNode

	// MapReduceDocuments used to generate summary
	chains.MapReduceDocuments
	// LLMChain used to
	chains.LLMChain
	// contains filtered or unexported fields
}

func NewMapReduceChain ¶

func NewMapReduceChain(baseNode base.BaseNode) *MapReduceChain

func (*MapReduceChain) Init ¶

func (l *MapReduceChain) Init(ctx context.Context, cli client.Client, args map[string]any) error

func (*MapReduceChain) Ready ¶

func (l *MapReduceChain) Ready() (bool, string)

func (*MapReduceChain) Run ¶

func (l *MapReduceChain) Run(ctx context.Context, cli client.Client, args map[string]any) (outArgs map[string]any, err error)

type RetrievalQAChain ¶

type RetrievalQAChain struct {
	chains.ConversationalRetrievalQA
	base.BaseNode
	Instance *v1alpha1.RetrievalQAChain
}

func NewRetrievalQAChain ¶

func NewRetrievalQAChain(baseNode base.BaseNode) *RetrievalQAChain

func (*RetrievalQAChain) Init ¶

func (l *RetrievalQAChain) Init(ctx context.Context, cli client.Client, _ map[string]any) error

func (*RetrievalQAChain) Ready ¶

func (l *RetrievalQAChain) Ready() (isReady bool, msg string)

func (*RetrievalQAChain) Run ¶

func (l *RetrievalQAChain) Run(ctx context.Context, _ client.Client, args map[string]any) (outArgs map[string]any, err error)

Source Files ¶

View all Source files

?	: This menu
/	: Search site
f or F	: Jump to
y or Y	: Canonical URL