mirror of
https://github.com/Tencent/WeKnora.git
synced 2026-09-01 14:53:07 +08:00
ttommybot
1d68b4dcd7
* fix: filter menus by deployment capabilities * fix: address deployment capabilities review feedback Restore Lite organizations hiding when the capabilities probe fails, align embed capability detection with route registration, move the handler into SystemHandler for Swagger/docs/client sync, and add cross-layer capability key contract tests. * style: gofmt deployment capabilities files Fix App CI formatting check so the vet/test step can run. --------- Co-authored-by: wizardchen <wizardchen@tencent.com>
105 lines
3.5 KiB
Go
105 lines
3.5 KiB
Go
package handler
|
|
|
|
import (
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
// DeploymentCapabilityKeys is the canonical capability key list shared with
|
|
// frontend/src/config/deploymentCapabilities.ts — keep both in sync.
|
|
var DeploymentCapabilityKeys = []string{
|
|
"organizations",
|
|
"agents",
|
|
"integrations.im",
|
|
"integrations.embed",
|
|
"integrations.api",
|
|
"settings.mcp",
|
|
"settings.websearch",
|
|
"settings.vectorstore",
|
|
"settings.storage",
|
|
"settings.sandbox",
|
|
}
|
|
|
|
// DeploymentCapability describes whether a deployment exposes a feature route.
|
|
type DeploymentCapability struct {
|
|
Supported bool `json:"supported"`
|
|
Reason string `json:"reason,omitempty"`
|
|
}
|
|
|
|
// DeploymentCapabilitiesData is returned by GET /system/capabilities.
|
|
type DeploymentCapabilitiesData struct {
|
|
Edition string `json:"edition"`
|
|
Capabilities map[string]DeploymentCapability `json:"capabilities"`
|
|
}
|
|
|
|
// DeploymentFeatureAvailability mirrors injected backend handlers/services.
|
|
type DeploymentFeatureAvailability struct {
|
|
Organizations bool
|
|
Agents bool
|
|
IM bool
|
|
Embed bool
|
|
API bool
|
|
MCP bool
|
|
WebSearch bool
|
|
VectorStore bool
|
|
Storage bool
|
|
Sandbox bool
|
|
}
|
|
|
|
func supportedDeploymentCapability(supported bool) DeploymentCapability {
|
|
if supported {
|
|
return DeploymentCapability{Supported: true}
|
|
}
|
|
return DeploymentCapability{Supported: false, Reason: "route_not_registered"}
|
|
}
|
|
|
|
// BuildDeploymentCapabilities derives the deployment capability snapshot.
|
|
func BuildDeploymentCapabilities(
|
|
edition string,
|
|
available DeploymentFeatureAvailability,
|
|
) DeploymentCapabilitiesData {
|
|
isLite := strings.EqualFold(strings.TrimSpace(edition), "lite")
|
|
organizations := supportedDeploymentCapability(available.Organizations && !isLite)
|
|
if isLite {
|
|
organizations.Reason = "not_supported_in_lite"
|
|
}
|
|
|
|
return DeploymentCapabilitiesData{
|
|
Edition: edition,
|
|
Capabilities: map[string]DeploymentCapability{
|
|
"organizations": organizations,
|
|
"agents": supportedDeploymentCapability(available.Agents),
|
|
"integrations.im": supportedDeploymentCapability(available.IM),
|
|
"integrations.embed": supportedDeploymentCapability(available.Embed),
|
|
"integrations.api": supportedDeploymentCapability(available.API),
|
|
"settings.mcp": supportedDeploymentCapability(available.MCP),
|
|
"settings.websearch": supportedDeploymentCapability(available.WebSearch),
|
|
"settings.vectorstore": supportedDeploymentCapability(available.VectorStore),
|
|
"settings.storage": supportedDeploymentCapability(available.Storage),
|
|
"settings.sandbox": supportedDeploymentCapability(available.Sandbox),
|
|
},
|
|
}
|
|
}
|
|
|
|
// BindDeploymentCapabilities stores the startup snapshot used by GetDeploymentCapabilities.
|
|
func (h *SystemHandler) BindDeploymentCapabilities(data DeploymentCapabilitiesData) {
|
|
h.deploymentCapabilities = data
|
|
}
|
|
|
|
// GetDeploymentCapabilities godoc
|
|
// @Summary 获取部署能力清单
|
|
// @Description 返回当前部署版本及实际注册的后端路由所对应的功能能力；仅 supported=false 表示入口应隐藏
|
|
// @Tags 系统
|
|
// @Produce json
|
|
// @Success 200 {object} map[string]interface{} "标准 code/msg/data 包装，data 为 DeploymentCapabilitiesData"
|
|
// @Router /system/capabilities [get]
|
|
func (h *SystemHandler) GetDeploymentCapabilities(c *gin.Context) {
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"code": 0,
|
|
"msg": "success",
|
|
"data": h.deploymentCapabilities,
|
|
})
|
|
}
|