aispec

type AIClient interface {
	Chatter
	FunctionCaller
	Configurable
	StructuredStreamer
	ModelListCaller
}

type AIConfig struct {
	// gateway network config
	BaseURL string
	Domain  string `json:"domain" app:"name:domain,verbose:第三方加速域名,id:4"`
	NoHttps bool   `json:"no_https" app:"name:no_https,verbose:NoHttps,desc:是否禁用使用https请求api,id:3"`

	// basic model
	Model    string  `json:"model" app:"name:model,verbose:模型名称,id:2,type:list"`
	Timeout  float64 // `app:"name:请求超时时长"`
	Deadline time.Time

	APIKey string `json:"api_key" app:"name:api_key,verbose:ApiKey,desc:APIKey / Token,required:true,id:1"`
	Proxy  string `json:"proxy" app:"name:proxy,verbose:代理地址,id:5"`
	Host   string
	Port   int

	StreamHandler       func(io.Reader)
	ReasonStreamHandler func(reader io.Reader)
	Type                string `json:"Type"`
	Context             context.Context

	FunctionCallRetryTimes int

	HTTPErrorHandler func(error)

	Images []*ImageDescription

	Headers             []*ypb.HTTPHeader
	EnableThinking      bool
	EnableThinkingField string
	EnableThinkingValue any
}

type AIConfigOption func(*AIConfig)

type ChatBaseContext struct {
	PoCOptionGenerator  func() ([]poc.PocConfigOption, error)
	EnableThinking      bool
	EnableThinkingField string
	EnableThinkingValue any
	ThinkingBudget      int64
	StreamHandler       func(io.Reader)
	ReasonStreamHandler func(reader io.Reader)
	ErrHandler          func(err error)
	ImageUrls           []*ImageDescription
	DisableStream       bool
}

type ChatBaseOption func(c *ChatBaseContext)

type ChatChoice struct {
	Index        int        `json:"index"`
	Message      ChatDetail `json:"message"`
	FinishReason string     `json:"finish_reason"`
}

type ChatCompletion struct {
	ID      string       `json:"id"`
	Object  string       `json:"object"`
	Created int64        `json:"created"`
	Choices []ChatChoice `json:"choices"`
	Usage   ChatUsage    `json:"usage"`
	Error   *ChatError   `json:"error,omitempty"`
}

type ChatContent struct {
	Type     string `json:"type"` // text / image_url
	Text     string `json:"text,omitempty"`
	ImageUrl any    `json:"image_url,omitempty"`
}

type ChatDetail struct {
	Role         string        `json:"role"`
	Name         string        `json:"name,omitempty"`
	Content      any           `json:"content"`
	ToolCalls    []*ToolCall   `json:"tool_calls,omitempty"`
	ToolCallID   string        `json:"tool_call_id,omitempty"`
	FunctionCall *FunctionCall `json:"function_call,omitempty"`
}

type ChatDetails []ChatDetail

type ChatError struct {
	Message string `json:"message"`
	Type    string `json:"type"`
	Param   string `json:"param"`
	Code    string `json:"code"`
}

type ChatMessage struct {
	Model          string       `json:"model"`
	Messages       []ChatDetail `json:"messages"`
	Stream         bool         `json:"stream"`
	EnableThinking bool         `json:"enable_thinking,omitempty"`
}

type ChatUsage struct {
	PromptTokens     int `json:"prompt_tokens"`
	CompletionTokens int `json:"completion_tokens"`
	TotalTokens      int `json:"total_tokens"`
}

type Chatter interface {
	Chat(string, ...any) (string, error)
	ChatStream(string) (io.Reader, error)
}

type Configurable interface {
	LoadOption(opt ...AIConfigOption)
	BuildHTTPOptions() ([]poc.PocConfigOption, error)
	CheckValid() error
}

type EmbeddingCaller interface {
	Embedding(string) ([]float32, error)
	// EmbeddingRaw 返回原始的 embedding 结果，可能包含多个向量
	// 如果服务器返回单个向量（[]float32），则返回 [][]float32{vector}
	// 如果服务器返回多个向量（[][]float32），则原样返回
	EmbeddingRaw(string) ([][]float32, error)
}

type FuncReturn struct {
	Name      string `json:"name"`
	Arguments string `json:"arguments"`
}

type FunctionCall struct {
	Name      string `json:"name"`
	Arguments string `json:"arguments"`
}

type FunctionCaller interface {
	ExtractData(data string, desc string, fields map[string]any) (map[string]any, error)
}

type GeneralChatter func(string, ...AIConfigOption) (string, error)

type ImageDescription struct {
	Url string `json:"url"`
}

type ModelListCaller interface {
	GetModelList() ([]*ModelMeta, error)
}

type ModelMeta struct {
	Id      string `json:"id"`
	Object  string `json:"object,omitempty"`
	Created int64  `json:"created,omitempty"`
	OwnedBy string `json:"owned_by,omitempty"`
}

type Parameters struct {
	Type       string              `json:"type"`
	Properties map[string]Property `json:"properties"`
	Required   []string            `json:"required"`
}

type Property struct {
	Type        string   `json:"type"`
	Description string   `json:"description"`
	Enum        []string `json:"enum,omitempty"`
}

type StructuredData struct {
	Id             string
	Event          string
	DataSourceType string
	DataRaw        []byte

	IsParsed   bool
	IsResponse bool
	// parsed from node
	OutputNodeName     string
	OutputNodeStatus   string
	OutputNodeId       string
	OutputNodeType     string
	OutputNodeExecTime string
	OutputText         string
	OutputReason       string

	HaveUsage  bool
	ModelUsage []UsageStatsInfo
}

type StructuredStreamer interface {
	SupportedStructuredStream() bool
	StructuredStream(string, ...any) (chan *StructuredData, error)
}

type ToolCall struct {
	ID       string     `json:"id"`
	Type     string     `json:"type"`
	Function FuncReturn `json:"function"`
}

type UsageStatsInfo struct {
	Model       string
	InputToken  int
	OutputToken int
}