// Package ai 提供 provider 中立的大模型调用能力。
// 目前内置 openai 兼容 provider（适配 DeepSeek / Qwen 兼容模式 / Gemini 兼容端点等），
// 以及规则化 mock。新增其它厂商只需实现 Provider 接口。
package ai

import (
	"github.com/sundynix/pets-be/internal/config"
)

// Message 一条对话消息
type Message struct {
	Role    string // system / user / assistant
	Content string
}

// Options 单次调用参数
type Options struct {
	JSON        bool    // 要求返回 JSON（结构化输出）
	MaxTokens   int     // 0 用配置默认
	Temperature float64 // <0 用配置默认
}

// Provider 大模型供应商接口
type Provider interface {
	Name() string
	// Complete 传入 system 提示与多轮消息，返回助手回复文本
	Complete(system string, messages []Message, opts Options) (string, error)
}

// Engine 对外统一入口，持有当前 provider 与开关
type Engine struct {
	provider Provider
	cfg      config.AIConfig
}

// New 根据配置构建引擎。未启用或缺 key 时 Enabled()=false，业务侧回退规则化文案。
func New(cfg config.AIConfig) *Engine {
	e := &Engine{cfg: cfg}
	if !cfg.Enabled || cfg.APIKey == "" {
		return e
	}
	switch cfg.Provider {
	case "", "openai":
		e.provider = newOpenAICompatProvider(cfg)
	default:
		// 未知 provider 视为未启用，回退规则化
	}
	return e
}

// Enabled 是否可用真实模型
func (e *Engine) Enabled() bool {
	return e.provider != nil
}

// ProviderName 当前供应商名（诊断用）
func (e *Engine) ProviderName() string {
	if e.provider == nil {
		return "disabled"
	}
	return e.provider.Name()
}

// Complete 代理到当前 provider（调用前请先判断 Enabled）
func (e *Engine) Complete(system string, messages []Message, opts Options) (string, error) {
	if opts.MaxTokens == 0 {
		opts.MaxTokens = e.cfg.MaxTokens
	}
	if opts.Temperature < 0 {
		opts.Temperature = e.cfg.Temperature
	}
	return e.provider.Complete(system, messages, opts)
}