lingo

type ApiResponse struct {
	Code int         `json:"code"`
	Msg  string      `json:"msg"`
	Data interface{} `json:"data,omitempty"`
}

type Choice struct {
	Content      string
	FinishReason string
	ToolCalls    []chatToolCall // 当 finish_reason=tool_calls 时有值
}

type LLMOptions struct {
	Provider     string
	ApiKey       string
	BaseURL      string
	SystemPrompt string
	Logger       *zap.Logger
}

type LogConfig struct {
	Level      string
	Filename   string
	MaxSize    int
	MaxAge     int
	MaxBackups int
	Daily      bool
}

type ParseOptions struct {
	MaxTextLength      int
	PreserveLineBreaks bool
	IncludeTables      bool
}

type ParseRequest struct {
	FileName string
	Content  []byte
	FileType string
}

type ParseResult struct {
	Text     string
	FileType string
	FileName string
	ParsedAt time.Time
}

type ProviderHandler interface {
	ResetMemory()
	QueryWithOptions(userMessage string, opts *QueryOptions) (*QueryResult, error)
	// RecordToolResult adds a tool execution result to the history（用于多轮工具调用循环）。
	RecordToolResult(toolCallID string, content string)
}

type QueryOptions struct {
	Model       string
	Temperature float32
	MaxTokens   int
}

type QueryResult struct {
	Choices   []Choice
	Usage     *UsageInfo
	ToolCalls []chatToolCall // 当 finish_reason=tool_calls 时有值
}

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