llm

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

type Client interface {
	Generate(ctx context.Context, msgs []Message, opts ...GenerateOption) (Response, error)
	Stream(ctx context.Context, msgs []Message, opts ...GenerateOption) (<-chan StreamChunk, error)
}

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

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

type GenerateOption func(*GenerateOptions)

type GenerateOptions struct {
	Temperature float32
	MaxTokens   int
	TopP        float32
	Stop        []string
	Schema      any
	Tools       []ToolDef
}

type Message struct {
	Role    Role   `json:"role"`
	Content string `json:"content"`
	Name    string `json:"name,omitempty"`
	ToolID  string `json:"tool_call_id,omitempty"`
}

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

type Response struct {
	Content   string
	ToolCalls []ToolCall
	Usage     Usage
	Raw       string
}

type Role string

type StreamChunk struct {
	Content  string
	ToolCall *ToolCall
	Done     bool
	Error    error
}

type ToolCall struct {
	ID       string `json:"id"`
	Type     string `json:"type"`
	FuncName string `json:"function_name"`
	FuncArgs string `json:"function_args"`
}

type ToolDef struct {
	Name        string `json:"name"`
	Description string `json:"description"`
	Parameters  any    `json:"parameters"`
}

type Usage struct {
	PromptTokens     int `json:"prompt_tokens"`
	CompletionTokens int `json:"completion_tokens"`
	TotalTokens      int `json:"total_tokens"`
}