api

type Accumulator struct {
	openai.ChatCompletionAccumulator
	Content   string
	Reasoning string
	// contains filtered or unexported fields
}

type CallbackFunc func(channel, content string, index int, end bool)

type Client struct {
	openai.Client
	Server         Server
	BaseURL        string
	ModelName      string
	ReasoningField string
	ContextLength  int
}

type Config struct {
	SystemPrompt      string       `json:"system_prompt"`
	ReasoningEffort   string       `json:"reasoning_effort"` // low | medium | high | none
	Tools             []ToolConfig `json:"tools,omitzero"`
	Temperature       float64      `json:"temperature,omitzero"`
	TopP              float64      `json:"top_p,omitzero"`
	TopK              int          `json:"top_k,omitzero"`
	PresencePenalty   float64      `json:"presence_penalty,omitzero"`
	RepetitionPenalty float64      `json:"repetition_penalty,omitzero"`
	CompactThreshold  float64      `json:"compact_threshold,omitzero"` // if set then apply message compaction if hit this fraction of model context length
}

type Conversation struct {
	ID        string                     `json:"id"`
	Config    Config                     `json:"config"`
	Messages  []Message                  `json:"messages"`
	NumTokens int                        `json:"num_tokens"`
	ToolData  map[string]json.RawMessage `json:"tool_data,omitzero"`
}

type Item struct {
	ID      string `json:"id"`
	Summary string `json:"summary"`
}

type Message struct {
	Role            string          `json:"role"`            // user | assistant | tool
	Update          bool            `json:"update,omitzero"` // true if update to existing message
	End             bool            `json:"end,omitzero"`    // true if update and message is now complete
	Content         string          `json:"content"`
	Reasoning       string          `json:"reasoning,omitzero"`
	ToolCall        json.RawMessage `json:"tool_call,omitzero"`
	ToolCallID      string          `json:"tool_call_id,omitzero"`
	ContentTokens   int             `json:"content_tokens,omitzero"`
	ReasoningTokens int             `json:"reasoning_tokens,omitzero"`
	Excluded        bool            `json:"excluded,omitzero"` // message is ignored by NewRequest if this is set
}

type Request struct {
	Action  string  `json:"action"`           // add | list | load | delete | config
	ID      string  `json:"id,omitzero"`      // if action=load,delete uuid format
	Message Message `json:"message,omitzero"` // if action=add
	Config  *Config `json:"config,omitzero"`  // if action=config
}

type Response struct {
	Action       string       `json:"action"`                // add | list | load | config | stats
	Message      Message      `json:"message,omitzero"`      // if action=add
	Conversation Conversation `json:"conversation,omitzero"` // if action=load
	List         []Item       `json:"list,omitzero"`         // if action=list
	Config       Config       `json:"config,omitzero"`       // if action=config
	Stats        Stats        `json:"stats,omitzero"`        // if action=stats
}

type Server int

type Stats struct {
	Model            string         `json:"model"`             // model name
	ApiCalls         int            `json:"api_calls"`         // total number of API calls
	ApiTime          int            `json:"api_time"`          // total elapsed time in API calls in msec
	CompletionTokens int            `json:"completion_tokens"` // no. of completion tokens generated
	PromptTokens     int            `json:"prompt_tokens"`     // max prompt length
	ToolCalls        int            `json:"tool_calls"`        // total number of tool calls
	Functions        map[string]int `json:"functions"`         // numer of tool calls by function name
	ToolTime         int            `json:"tool_time"`         // total elapsed time in tool calls in msec
}

type ToolConfig struct {
	Name    string `json:"name"`
	Enabled bool   `json:"enabled"`
}

type ToolFunction interface {
	// function name and argument schema
	Definition() shared.FunctionDefinitionParam
	// call with args in JSON format - if err is set it is treated as fatal, else the model can retry with new args
	Call(args string) (req, resp string, err error)
}