llm

type BaseTool interface {
	Info(ctx context.Context) (*ToolInfo, error)
	InvokableRun(ctx context.Context, input string) (string, error)
}

type ChatTemplate interface {
	Format(ctx context.Context, data map[string]any) ([]*Message, error)
}

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

type FunctionTool struct {
	Name        string
	Description string
	Parameters  *ParamsOneOf
	Execute     func(ctx context.Context, input string) (string, error)
}

type GenerateRequest struct {
	Model       string
	Messages    []*Message
	Tools       []Tool
	MaxTokens   int
	Temperature float64
}

type InvokableTool interface {
	BaseTool
}

type Message struct {
	Role    string
	Content string
	Name    string
}

type MessageFragment struct {
	Content string
	Done    bool
	Err     error
}

type ParamsOneOf struct {
	OneOf []Schema
}

type Provider interface {
	Name() string
	Generate(ctx context.Context, req *GenerateRequest) (*Message, error)
	GenerateStream(ctx context.Context, req *GenerateRequest) (<-chan MessageFragment, error)
}

type Role = string

type Schema struct {
	Type        string
	Description string
	Properties  map[string]Schema
	Required    []string
}

type Tool interface {
	Name() string
	Description() string
	Parameters() *ParamsOneOf
	Execute(ctx context.Context, input string) (string, error)
}

type ToolCall struct {
	ID       string
	Type     string
	Function ToolCallFunction
}

type ToolCallFunction struct {
	Name      string
	Arguments string
}

type ToolInfo struct {
	Name        string
	Desc        string
	ParamsOneOf *ParamsOneOf
}