ai

type AI interface {
	LLMs() LLMs
	Model() string

	Limiter

	SetModel(string)
	Model

	Chatbot
	ChatSession() ChatSession

	Close() error
}

type Blob struct {
	MIMEType string
	Data     []byte
}

type ChatResponse interface {
	Raw() any
	Results() []string
	FunctionCalls() []FunctionCall
	TokenCount() TokenCount
}

type ChatSession interface {
	Chatbot
	History() []Content
}

type ChatStream interface {
	Next() (ChatResponse, error)
	Close() error
}

type Chatbot interface {
	Chat(context.Context, ...Part) (ChatResponse, error)
	ChatStream(context.Context, ...Part) (ChatStream, error)
}

type ClientConfig struct {
	LLMs LLMs

	APIKey   string
	Endpoint string
	Proxy    string

	Limit *int64

	Model       string
	ModelConfig ModelConfig
}

type ClientOption interface {
	Apply(*ClientConfig)
}

type Content struct {
	Parts []Part
	Role  string
}

type Function struct {
	Name        string
	Description string
	Parameters  Schema
}

type FunctionCall struct {
	ID        string
	Name      string
	Arguments string
}

type FunctionCallingMode int

type FunctionResponse struct {
	ID       string
	Response string
}

type Image string

type JSONSchema struct {
	Name        string
	Description string
	Schema      Schema
}

type LLMs string

type Limiter interface {
	SetLimit(rpm int64)
	Limit() (rpm int64)
}

type Model interface {
	SetFunctionCall([]Function, FunctionCallingMode)
	SetCount(x int64)
	SetMaxTokens(x int64)
	SetTemperature(x float64)
	SetTopP(x float64)
	SetJSONResponse(set bool, schema *JSONSchema)
}

type ModelConfig struct {
	Count        *int64
	MaxTokens    *int64
	Temperature  *float64
	TopP         *float64
	JSONResponse *bool
	JSONSchema   *JSONSchema
	Tools        []Function
	ToolConfig   FunctionCallingMode
}

type Part interface {
	// contains filtered or unexported methods
}

type Schema struct {
	Type       string         `json:"type"`
	Properties map[string]any `json:"properties,omitempty"`
	Enum       []string       `json:"enum,omitempty"`
	Items      *Schema        `json:"items,omitempty"`
	Required   []string       `json:"required,omitempty"`
}

type Text string

type TokenCount struct {
	Prompt int64
	Result int64
	Total  int64
}