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 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 ClaudeModelProvider struct {
	// contains filtered or unexported fields
}

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
}

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 StepFunModelProvider struct {
	// contains filtered or unexported fields
}

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

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

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

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
}