增加ai办公，考勤管理
This commit is contained in:
hero920103 committed
1 parent
eb9b7beff5
commit
31ae07175d
46 files changed
+9250
-296
No files matched your search
@@ -13,6 +13,7 @@ import (
|
||||
"server/pkg/identity"
|
||||
"server/pkg/jwtutil"
|
||||
|
||||
"github.com/beego/beego/v2/client/orm"
|
||||
beego "github.com/beego/beego/v2/server/web"
|
||||
)
|
||||
|
||||
@@ -46,13 +47,22 @@ type aiProviderPayload struct {
|
||||
IsDefault int8 `json:"is_default"`
|
||||
Status int8 `json:"status"`
|
||||
Remark string `json:"remark"`
|
||||
// Scope "tenant" 表示租户级配置（user_id=0，全员共用，由「基础设置 → AI 设置」维护）；
|
||||
// 其它值/空表示个人配置，仅创建者可用。
|
||||
Scope string `json:"scope"`
|
||||
}
|
||||
|
||||
// clearTenantDefault 取消租户内所有默认配置
|
||||
func clearTenantDefault(tenantID string) {
|
||||
// aiTenantScope 本次请求是否在维护租户级配置
|
||||
func aiTenantScope(scope string) bool {
|
||||
return strings.EqualFold(strings.TrimSpace(scope), "tenant")
|
||||
}
|
||||
|
||||
// clearTenantDefault 取消同一归属范围内（租户级 / 同一用户）的其他默认配置
|
||||
func clearTenantDefault(tenantID string, userID uint64) {
|
||||
var defaults []models.BackendAiProvider
|
||||
_, _ = models.Orm.QueryTable(new(models.BackendAiProvider)).
|
||||
Filter("tenant_id", tenantID).
|
||||
Filter("user_id", userID).
|
||||
Filter("is_default", 1).
|
||||
All(&defaults)
|
||||
for _, d := range defaults {
|
||||
@@ -75,7 +85,9 @@ func parseProviderModels(m string) []string {
|
||||
return list
|
||||
}
|
||||
|
||||
// List GET /backend/ai/provider/list
|
||||
// List GET /backend/ai/provider/list?scope=tenant
|
||||
// scope=tenant 只返回租户级配置（user_id=0）；不传时返回「租户级 + 当前用户」，
|
||||
// 供 AI 广场选择可用接入。
|
||||
func (c *BackendAiProviderController) List() {
|
||||
claims, err := c.aiClaims()
|
||||
if err != nil {
|
||||
@@ -84,21 +96,34 @@ func (c *BackendAiProviderController) List() {
|
||||
return
|
||||
}
|
||||
|
||||
cond := orm.NewCondition().
|
||||
And("tenant_id", fmt.Sprintf("%d", claims.TenantId)).
|
||||
And("delete_time__isnull", true)
|
||||
if aiTenantScope(c.GetString("scope")) {
|
||||
cond = cond.And("user_id", 0)
|
||||
} else {
|
||||
cond = cond.AndCond(orm.NewCondition().
|
||||
Or("user_id", 0).
|
||||
Or("user_id", uint64(claims.UserID)))
|
||||
}
|
||||
|
||||
var list []models.BackendAiProvider
|
||||
_, err = models.Orm.QueryTable(new(models.BackendAiProvider)).
|
||||
Filter("tenant_id", fmt.Sprintf("%d", claims.TenantId)).
|
||||
Filter("user_id", uint64(claims.UserID)).
|
||||
Filter("delete_time__isnull", true).
|
||||
OrderBy("-id").
|
||||
SetCond(cond).
|
||||
OrderBy("user_id", "-id").
|
||||
All(&list)
|
||||
if err != nil {
|
||||
c.aiJsonErr(500, 500, "查询失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
// 解析模型列表（不脱敏api_key，前端需要完整api_key进行测试）
|
||||
// 解析模型列表。个人配置不回传脱敏 Key（本人可见）；
|
||||
// 租户级配置（user_id=0）全租户共用，Key 必须脱敏，测试走 test 接口的 id 模式。
|
||||
for i := range list {
|
||||
list[i].ModelsList = parseProviderModels(list[i].Models)
|
||||
if list[i].UserID == 0 {
|
||||
list[i].ApiKey = maskTenantAiKey(list[i].ApiKey)
|
||||
}
|
||||
}
|
||||
|
||||
c.aiOk(map[string]interface{}{"list": list})
|
||||
@@ -144,14 +169,20 @@ func (c *BackendAiProviderController) Create() {
|
||||
// 模型数组序列化为JSON
|
||||
modelsJSON, _ := json.Marshal(p.Models)
|
||||
|
||||
// 如果设为默认，先取消租户内其他默认
|
||||
// 租户级配置（user_id=0）由 OA 设置统一维护，全员可用
|
||||
ownerID := uint64(claims.UserID)
|
||||
if aiTenantScope(p.Scope) {
|
||||
ownerID = 0
|
||||
}
|
||||
|
||||
// 如果设为默认，先取消同一归属范围内的其他默认
|
||||
if p.IsDefault == 1 {
|
||||
clearTenantDefault(fmt.Sprintf("%d", claims.TenantId))
|
||||
clearTenantDefault(fmt.Sprintf("%d", claims.TenantId), ownerID)
|
||||
}
|
||||
|
||||
provider := models.BackendAiProvider{
|
||||
TenantID: fmt.Sprintf("%d", claims.TenantId),
|
||||
UserID: uint64(claims.UserID),
|
||||
UserID: ownerID,
|
||||
ProviderType: p.ProviderType,
|
||||
Name: strings.TrimSpace(p.Name),
|
||||
ApiBase: strings.TrimSpace(p.ApiBase),
|
||||
@@ -199,6 +230,11 @@ func (c *BackendAiProviderController) Update() {
|
||||
c.aiJsonErr(403, 403, "无权操作")
|
||||
return
|
||||
}
|
||||
// 租户级配置（user_id=0）允许本租户管理员在 AI 设置中维护，个人配置仅本人可改
|
||||
if provider.UserID != 0 && provider.UserID != uint64(claims.UserID) {
|
||||
c.aiJsonErr(403, 403, "无权操作")
|
||||
return
|
||||
}
|
||||
|
||||
body, err := io.ReadAll(c.Ctx.Request.Body)
|
||||
if err != nil {
|
||||
@@ -211,9 +247,9 @@ func (c *BackendAiProviderController) Update() {
|
||||
return
|
||||
}
|
||||
|
||||
// 如果设为默认，先取消租户内其他默认
|
||||
// 如果设为默认，先取消同一归属范围内的其他默认
|
||||
if p.IsDefault == 1 && provider.IsDefault != 1 {
|
||||
clearTenantDefault(fmt.Sprintf("%d", claims.TenantId))
|
||||
clearTenantDefault(fmt.Sprintf("%d", claims.TenantId), provider.UserID)
|
||||
}
|
||||
|
||||
provider.Name = strings.TrimSpace(p.Name)
|
||||
@@ -268,6 +304,10 @@ func (c *BackendAiProviderController) Delete() {
|
||||
c.aiJsonErr(403, 403, "无权操作")
|
||||
return
|
||||
}
|
||||
if provider.UserID != 0 && provider.UserID != uint64(claims.UserID) {
|
||||
c.aiJsonErr(403, 403, "无权操作")
|
||||
return
|
||||
}
|
||||
|
||||
now := time.Now()
|
||||
provider.DeleteTime = &now
|
||||
@@ -281,6 +321,8 @@ func (c *BackendAiProviderController) Delete() {
|
||||
}
|
||||
|
||||
type aiTestPayload struct {
|
||||
// ID 已保存的配置ID：不传 api_key（或传脱敏值）时用库里的密钥测试
|
||||
ID uint64 `json:"id"`
|
||||
ProviderType string `json:"provider_type"`
|
||||
ApiBase string `json:"api_base"`
|
||||
ApiKey string `json:"api_key"`
|
||||
@@ -293,10 +335,26 @@ type modelTestResult struct {
|
||||
Error string `json:"error"`
|
||||
}
|
||||
|
||||
// maskTenantAiKey 脱敏租户级配置的密钥（保留前后各 4 位）。
|
||||
// 注意：结果包含 ****，Update 遇到含 **** 的 key 会跳过更新，不会把脱敏值写回库。
|
||||
func maskTenantAiKey(key string) string {
|
||||
key = strings.TrimSpace(key)
|
||||
if key == "" {
|
||||
return ""
|
||||
}
|
||||
if len(key) <= 8 {
|
||||
return "****"
|
||||
}
|
||||
return key[:4] + "****" + key[len(key)-4:]
|
||||
}
|
||||
|
||||
// Test POST /backend/ai/provider/test
|
||||
// 批量测试模型连通性
|
||||
// 批量测试模型连通性。支持两种模式：
|
||||
// 1. 传 api_base + api_key：直接测试（新增/修改未保存的表单）；
|
||||
// 2. 传 id（api_key 留空或为脱敏值）：用已保存配置的密钥测试。
|
||||
func (c *BackendAiProviderController) Test() {
|
||||
if _, err := c.aiClaims(); err != nil {
|
||||
claims, err := c.aiClaims()
|
||||
if err != nil {
|
||||
code := identity.Status(err)
|
||||
c.aiJsonErr(code, code, err.Error())
|
||||
return
|
||||
@@ -313,7 +371,31 @@ func (c *BackendAiProviderController) Test() {
|
||||
return
|
||||
}
|
||||
|
||||
if strings.TrimSpace(p.ApiBase) == "" || strings.TrimSpace(p.ApiKey) == "" {
|
||||
apiKey := strings.TrimSpace(p.ApiKey)
|
||||
if p.ID > 0 && (apiKey == "" || strings.Contains(apiKey, "****")) {
|
||||
stored := models.BackendAiProvider{ID: p.ID}
|
||||
if err := models.Orm.Read(&stored); err != nil {
|
||||
c.aiJsonErr(404, 404, "配置不存在")
|
||||
return
|
||||
}
|
||||
if stored.TenantID != fmt.Sprintf("%d", claims.TenantId) {
|
||||
c.aiJsonErr(403, 403, "无权操作")
|
||||
return
|
||||
}
|
||||
if stored.UserID != 0 && stored.UserID != uint64(claims.UserID) {
|
||||
c.aiJsonErr(403, 403, "无权操作")
|
||||
return
|
||||
}
|
||||
apiKey = stored.ApiKey
|
||||
if strings.TrimSpace(p.ApiBase) == "" {
|
||||
p.ApiBase = stored.ApiBase
|
||||
}
|
||||
if strings.TrimSpace(p.ProviderType) == "" {
|
||||
p.ProviderType = stored.ProviderType
|
||||
}
|
||||
}
|
||||
|
||||
if strings.TrimSpace(p.ApiBase) == "" || apiKey == "" {
|
||||
c.aiJsonErr(400, 400, "接口地址和API Key不能为空")
|
||||
return
|
||||
}
|
||||
@@ -326,7 +408,7 @@ func (c *BackendAiProviderController) Test() {
|
||||
provider := models.BackendAiProvider{
|
||||
ProviderType: p.ProviderType,
|
||||
ApiBase: strings.TrimSpace(p.ApiBase),
|
||||
ApiKey: strings.TrimSpace(p.ApiKey),
|
||||
ApiKey: apiKey,
|
||||
}
|
||||
testMessages := []openaiMessage{{Role: "user", Content: "hi"}}
|
||||
|
||||
|
||||
Reference in new issue
Block a user