40 lines
1.3 KiB
Go
40 lines
1.3 KiB
Go
package agent
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
"github.com/gogf/gf/v2/os/gcache"
|
|
)
|
|
|
|
var modelCfgCache = gcache.New()
|
|
|
|
// GetModelConfig 从 config.yml 读取 LLM 配置（缓存 60s），未配置返回明确错误
|
|
func GetModelConfig(ctx context.Context) (*ModelConfig, error) {
|
|
cacheKey := "llm:model_config"
|
|
v, err := modelCfgCache.Get(ctx, cacheKey)
|
|
if err == nil && !v.IsNil() {
|
|
if cfg, ok := v.Val().(*ModelConfig); ok {
|
|
return cfg, nil
|
|
}
|
|
}
|
|
cfg := &ModelConfig{
|
|
BaseURL: g.Cfg().MustGet(ctx, "llm.base_url", "").String(),
|
|
APIKey: g.Cfg().MustGet(ctx, "llm.api_key", "").String(),
|
|
ModelName: g.Cfg().MustGet(ctx, "llm.model_name", "").String(),
|
|
MaxTokens: g.Cfg().MustGet(ctx, "llm.max_tokens", 4096).Int(),
|
|
Temperature: g.Cfg().MustGet(ctx, "llm.temperature", 0.8).Float32(),
|
|
Timeout: time.Duration(g.Cfg().MustGet(ctx, "chat.timeout", 300).Int()) * time.Second,
|
|
MaxRetries: g.Cfg().MustGet(ctx, "chat.max_retries", 3).Int(),
|
|
}
|
|
if cfg.APIKey == "" || cfg.ModelName == "" || cfg.BaseURL == "" {
|
|
return nil, fmt.Errorf("LLM 未配置：请在 config.yml 设置 llm.base_url / llm.api_key / llm.model_name")
|
|
}
|
|
if err := modelCfgCache.Set(ctx, cacheKey, cfg, 60*time.Second); err != nil {
|
|
g.Log().Warningf(ctx, "写入 LLM 配置缓存失败: %v", err)
|
|
}
|
|
return cfg, nil
|
|
}
|