Function resolveKnowledgeBaseIDs

api/controller/relay/relay_helper.go:496–530

resolveKnowledgeBaseIDs 解析知识库ID，支持 "all" 参数和加密的hashID

(eid int64, libraryHashIDs []string)

// resolveKnowledgeBaseIDs 解析知识库ID，支持 "all" 参数和加密的hashID
func resolveKnowledgeBaseIDs(eid int64, libraryHashIDs []string) ([]int64, error) {
	// 检查是否包含 "all" 标识
	for _, hashID := range libraryHashIDs {
		if hashID == "all" || hashID == "-1" {
			// 获取所有活跃状态的知识库
			libraries, err := model.GetLibrariesByEid(eid, nil)
			if err != nil {
				return nil, fmt.Errorf("获取知识库列表失败: %v", err)
			}
			var allIDs []int64
			for _, lib := range libraries {
				allIDs = append(allIDs, lib.ID)
			}
			return allIDs, nil
		}
	}

	// 解析加密的hashID
	var resolvedIDs []int64
	for _, hashID := range libraryHashIDs {
		// 尝试从hashID解码获取原始ID
		if originalID, err := hashids.Decode(hashID); err == nil {
			resolvedIDs = append(resolvedIDs, originalID)
		} else {
			// 如果解码失败，尝试直接解析为数字（向后兼容）
			if id, err := strconv.ParseInt(hashID, 10, 64); err == nil {
				resolvedIDs = append(resolvedIDs, id)
			} else {
				return nil, fmt.Errorf("无法解析知识库ID: %s", hashID)
			}
		}
	}

	return resolvedIDs, nil
}

// resolveFileIDs 解析文件ID，支持加密的hashID
func resolveFileIDs(eid int64, fileHashIDs []string) ([]int64, error) {