model

type AgentInfo struct {
	AgentClients  *agent.AgentClients
	AgentMessages *AgentMessages
}

type AgentMessages struct {
	Messages  []*RawMessage
	ToolCalls any
}

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

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

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

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

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

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

type ChatMessage struct {
	Role    string  `json:"role"`
	Message string  `json:"message"`
	User    *string `json:"user,omitempty"`
}

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

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

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

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

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

type GeneratedImage struct {
	URL     string
	B64JSON string
}

type GitHubModelProvider struct {
	*LocalModelProvider
}

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

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

type ImageGenRequest struct {
	// UpstreamModel is the do-ai/fal model_id (e.g. "fal-ai/flux/schnell").
	UpstreamModel string
	Prompt        string
	// N is the number of images (fal "num_images"); clamped to [1,10].
	N int
	// Size is an OpenAI-style "WxH" (e.g. "1024x1024"); passed through to fal
	// as image_size when set. Empty means the model default.
	Size string
	// OutputFormat is fal's format hint (e.g. "jpeg", "png"); optional.
	OutputFormat string
}

type ImageGenResult struct {
	Images []GeneratedImage
}

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

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

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

type ModelProvider interface {
	GetPricing() string
	QueryText(question string, writer io.Writer, history []*RawMessage, prompt string, knowledgeMessages []*RawMessage, agentInfo *AgentInfo, lang string) (*ModelResult, error)
}

type ModelResult struct {
	PromptTokenCount   int
	ResponseTokenCount int
	TotalTokenCount    int
	ImageCount         int
	TotalPrice         float64
	Currency           string

	// Unpriced says NO price was stated for this call, so TotalPrice is an absence
	// rather than an amount.
	//
	// A client prices from a table of models it knows, and it cannot know a model
	// served from our own hardware or from a vendor's free tier. Reporting nothing
	// for those used to mean returning an error, which failed the whole request —
	// the call succeeded upstream and the caller got a pricing message instead of an
	// answer. Reporting a bare zero would be worse: an empty Currency reads as USD
	// downstream, so an unknown model would bill as costing nothing rather than as
	// unknown.
	//
	// Stated as the NEGATIVE deliberately, unlike a route's Priced. The safe zero
	// value differs because the risk does: a route that defaulted to priced-at-zero
	// would give inference away, while a result that defaulted to unpriced would
	// stop billing every client that never sets it. Zero here means priced, which is
	// what every existing client already means. The usage warehouse has carried this
	// same word all along and can now be told the truth.
	Unpriced bool

	// CacheReadTokenCount and CacheWriteTokenCount are the prompt tokens a
	// provider served from, and wrote to, its prompt cache.
	//
	// They are part of PromptTokenCount, not additional to it — a provider
	// reports the cached portion of the prompt it already charged for, at a
	// different rate (a read is typically a tenth of the input price, a write
	// a quarter more). So they price the SAME tokens differently; adding them
	// to a total would count those tokens twice.
	//
	// Everything downstream of here was already built to carry them — the
	// usage record, the ClickHouse columns, the per-rate pricing, the gen_ai
	// span attributes. This struct was the one place they had nowhere to sit,
	// so every request reported zero reads and zero writes however well the
	// provider cached, and the saving was invisible to billing and to o11y
	// alike. Reading them is the whole fix.
	CacheReadTokenCount  int
	CacheWriteTokenCount int
}

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

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

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

type RawMessage struct {
	Text           string
	Author         string
	TextTokenCount int
	ToolCall       openai.ToolCall
	ToolCallID     string
}

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

type SearchResult struct {
	Icon     string `json:"icon,omitempty"`
	SiteName string `json:"site_name,omitempty"`
	Index    int    `json:"index"`
	Title    string `json:"title"`
	URL      string `json:"url"`
}

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

type Spec struct {
	Upstream Upstream
	Model    string // the vendor's own name for it, e.g. claude-opus-4-5

	ClientID string
	Secret   string
	UserKey  string

	URL        string
	APIVersion string
	Compatible string // the API this upstream speaks when it is not its own

	Temperature float32
	TopP        float32
	TopK        int
	Frequency   float32
	Presence    float32
	Thinking    bool

	InputPrice  float64 // per thousand tokens
	OutputPrice float64 // per thousand tokens
	Currency    string
}

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

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

type ThreeDGenRequest struct {
	UpstreamModel string `json:"model"`
	Prompt        string `json:"prompt"`
	ImageUrl      string `json:"image_url,omitempty"`
	Format        string `json:"format,omitempty"` // "splat" | "ply" | "glb" | "obj"
	Quality       string `json:"quality,omitempty"`
}

type ThreeDStatus struct {
	Id         string `json:"id"`
	Model      string `json:"model"`
	Status     string `json:"status"` // "queued" | "in_progress" | "completed" | "failed"
	Url        string `json:"url,omitempty"`
	Format     string `json:"format,omitempty"`
	SplatCount int    `json:"splat_count,omitempty"`
	Vertices   int    `json:"vertices,omitempty"`
	Error      string `json:"error,omitempty"`
}

type ToolCall struct {
	Name      string `json:"name"`
	Arguments string `json:"arguments"`
	Content   string `json:"content"`
}

type ToolCallResponse struct {
	Success  bool   `json:"success"`
	Data     any    `json:"data"`
	Error    string `json:"error,omitempty"`
	ToolName string `json:"toolName"`
}

type Upstream string

type VideoGenRequest struct {
	// UpstreamModel is the upstream model id (e.g. "wan2-2-t2v-a14b").
	UpstreamModel string
	Prompt        string
	// Size is an optional OpenAI-style "WxH" hint passed through when set.
	Size string
	// Seconds is an optional clip-length hint (upstream default when 0).
	Seconds int
}

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

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

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