Blizzard
609f7d06cf
- pets-fe: 微信原生小程序（首页/计划/记录/报告/社区/引导）， 服务端驱动、无假数据；弹层改用 scroll-view，打开时隐藏自定义 tabBar - pets-be: Gin + GORM(MySQL, sundynix_ 前缀) + MinIO，统一响应/分页， 微信 code2session 登录，provider-neutral AI(DeepSeek)，go:embed React 后台 - 修复：分段选择类型不匹配（字符串 vs 数字）导致选不中 Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
74 lines
2.0 KiB
Go
74 lines
2.0 KiB
Go
// 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)
|
||
}
|