token

type Budget struct {
	ContextLimit    int64 `json:"context_limit,omitempty"`
	MaxOutputTokens int64 `json:"max_output_tokens,omitempty"`
	PromptLimit     int64 `json:"prompt_limit,omitempty"`
}

type Estimate struct {
	Provider         string `json:"provider,omitempty"`
	Model            string `json:"model"`
	InputTokens      int64  `json:"input_tokens"`
	SystemTokens     int64  `json:"system_tokens,omitempty"`
	MessageTokens    int64  `json:"message_tokens,omitempty"`
	ToolSchemaTokens int64  `json:"tool_schema_tokens,omitempty"`
	OverheadTokens   int64  `json:"overhead_tokens,omitempty"`
	ContextLimit     int64  `json:"context_limit,omitempty"`
	MaxOutputTokens  int64  `json:"max_output_tokens,omitempty"`
	RemainingTokens  int64  `json:"remaining_tokens,omitempty"`
	Source           Source `json:"source"`
	Estimated        bool   `json:"estimated"`
}

type EstimateRequest struct {
	Provider        string            `json:"provider,omitempty"`
	Model           string            `json:"model"`
	Messages        []Message         `json:"messages"`
	Tools           []ToolSpec        `json:"tools,omitempty"`
	MaxOutputTokens int64             `json:"max_output_tokens,omitempty"`
	ContextLimit    int64             `json:"context_limit,omitempty"`
	Metadata        map[string]string `json:"metadata,omitempty"`
}

type Estimator interface {
	CountPromptTokens(context.Context, EstimateRequest) (Estimate, error)
}

type InputImage struct {
	URL      string `json:"url,omitempty"`
	Data     []byte `json:"data,omitempty"`
	MIMEType string `json:"mime_type,omitempty"`
}

type InputPart struct {
	Type  InputPartType `json:"type"`
	Text  string        `json:"text,omitempty"`
	Image *InputImage   `json:"image,omitempty"`
}

type InputPartType string

type Message struct {
	Role       string      `json:"role"`
	Content    string      `json:"content,omitempty"`
	Input      []InputPart `json:"input,omitempty"`
	Name       string      `json:"name,omitempty"`
	ToolCallID string      `json:"tool_call_id,omitempty"`
	ToolCalls  []ToolCall  `json:"tool_calls,omitempty"`
}

type Source string

type TiktokenEstimator struct {
	Encoding string
}

type ToolCall struct {
	ID        string          `json:"id"`
	Name      string          `json:"name"`
	Arguments json.RawMessage `json:"arguments,omitempty"`
}

type ToolSpec struct {
	Name        string          `json:"name"`
	Description string          `json:"description,omitempty"`
	InputSchema json.RawMessage `json:"input_schema,omitempty"`
}