agents

Documentation

Index

• func ChatModel(ctx context.Context, modelName string) (model.ChatModel, error)
• func CountMessageTokens(messages []*schema.Message) (int, error)
• func CountToken(text string) int
• func DefaultMultiChatTemplate() prompt.ChatTemplate
• func DefaultTemplate() prompt.ChatTemplate
• func Generate(ctx context.Context, llm model.ChatModel, in []*schema.Message) (*schema.Message, error)
• func LLMGenerate(ctx context.Context, prompt string) (string, error)
• func Model() string
• func ReactAgent(ctx context.Context, tools []tool.BaseTool) (*react.Agent, error)
• func Stream(ctx context.Context, llm model.ChatModel, in []*schema.Message) (*schema.StreamReader[*schema.Message], error)
• func ToolcallModel() string

Functions

func ChatModel

func ChatModel(ctx context.Context, modelName string) (model.ChatModel, error)

func CountMessageTokens

func CountMessageTokens(messages []*schema.Message) (int, error)

func CountToken

func CountToken(text string) int

func DefaultMultiChatTemplate

func DefaultMultiChatTemplate() prompt.ChatTemplate

func DefaultTemplate

func DefaultTemplate() prompt.ChatTemplate

func Generate

func Generate(ctx context.Context, llm model.ChatModel, in []*schema.Message) (*schema.Message, error)

func LLMGenerate

func LLMGenerate(ctx context.Context, prompt string) (string, error)

func Model

func Model() string

func ReactAgent

func ReactAgent(ctx context.Context, tools []tool.BaseTool) (*react.Agent, error)

func Stream

func Stream(ctx context.Context, llm model.ChatModel, in []*schema.Message) (*schema.StreamReader[*schema.Message], error)

func ToolcallModel

func ToolcallModel() string