更新配置

This commit is contained in:
Hua
2025-02-19 08:46:26 +08:00
parent e0b133ea1e
commit 7ec5abe67f
3 changed files with 21 additions and 5 deletions

View File

@ -6,8 +6,9 @@ import (
"code-review/services"
"code-review/services/ai"
"fmt"
"github.com/gin-gonic/gin"
"log"
"github.com/gin-gonic/gin"
)
func main() {
@ -22,10 +23,10 @@ func main() {
}
// 初始化 AI 负载均衡器
aiBalancer := config.NewAIBalancer(cfg.AIs)
config.ResetAIBalancer(cfg.AIs)
// 获取一个 AI 配置
aiConfig, err := aiBalancer.Next()
aiConfig, err := config.GetAIBalancer().Next()
if err != nil {
log.Fatalf("获取 AI 配置失败: %v", err)
}