mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-08-01 21:37:33 +08:00
29287ef74b
## Summary Refactor the Go `KnowledgeCompilerComponent` so its parameter is a **single string template id** instead of a DSL-level `variant` (or plural group id list). The `variant` is no longer in the DSL — it is now **derived at runtime from the resolved compilation template's `kind` field**. This aligns the Go ingestion port with the frontend Compiler operator, which emits a singular `compilation_template_group_id` and does not write `variant` into the generated `compiler.json`.
111 lines
4.2 KiB
Go
111 lines
4.2 KiB
Go
//
|
|
// Copyright 2026 The InfiniFlow Authors. All Rights Reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
//
|
|
|
|
package dao
|
|
|
|
import (
|
|
"context"
|
|
"ragflow/internal/common"
|
|
"ragflow/internal/entity"
|
|
|
|
"go.uber.org/zap"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/clause"
|
|
)
|
|
|
|
// BuiltinCompilationTemplateGroupID is the id of the system-provided
|
|
// compilation template group that compiler.json references as its default
|
|
// "create from template" group. It is a global (tenant-agnostic) catalogue so
|
|
// any tenant can resolve it without depending on Python-side initialization.
|
|
const BuiltinCompilationTemplateGroupID = "c3aa748c8b2111f191f3047c16ec874f"
|
|
|
|
// builtinCompilationTemplateKinds is the set of variants shipped in the
|
|
// built-in group. The Go KnowledgeCompiler derives the variant from each
|
|
// template's kind via common.KindToVariant.
|
|
var builtinCompilationTemplateKinds = []struct {
|
|
Kind string
|
|
Name string
|
|
}{
|
|
{Kind: "tree", Name: "Tree — RAPTOR-based document tree"},
|
|
{Kind: "mind_map", Name: "Mind map - Radial concept hierarchy"},
|
|
{Kind: "wiki", Name: "Wiki — Graph-based wiki"},
|
|
{Kind: "knowledge_graph", Name: "Knowledge graph"},
|
|
{Kind: "datasetnav", Name: "Dataset nav — top-down drill-down"},
|
|
{Kind: "page_index", Name: "Page index"},
|
|
{Kind: "session_essence", Name: "Session essence"},
|
|
{Kind: "session_graph", Name: "Session graph"},
|
|
{Kind: "timeline", Name: "Timeline"},
|
|
}
|
|
|
|
func strptr(s string) *string { return &s }
|
|
|
|
// SeedBuiltinCompilationTemplates provisions the built-in compilation template
|
|
// group (and its child templates) as a global, tenant-agnostic catalogue, so
|
|
// the Go compiler (and the frontend "create from template" catalogue) can
|
|
// resolve compiler.json's default group id out of the box. The group row uses
|
|
// an empty tenant_id and ResolveGroupTemplateIDs treats this built-in id as
|
|
// resolvable for every tenant.
|
|
func SeedBuiltinCompilationTemplates(ctx context.Context, db *gorm.DB) error {
|
|
if err := SeedBuiltinCompilationTemplatesForTenant(ctx, db, ""); err != nil {
|
|
common.Warn("failed to seed built-in compilation templates", zap.Error(err))
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// SeedBuiltinCompilationTemplatesForTenant upserts the built-in compilation
|
|
// template group and its child templates as a global, tenant-agnostic
|
|
// catalogue. The tenantID argument is accepted for API symmetry but the
|
|
// built-in entries always use an empty tenant_id, because ResolveGroupTemplateIDs
|
|
// resolves the built-in group id for every tenant. It is idempotent
|
|
// (OnConflict DO NOTHING on id) so it is safe to call repeatedly.
|
|
func SeedBuiltinCompilationTemplatesForTenant(ctx context.Context, db *gorm.DB, tenantID string) error {
|
|
valid := string(entity.StatusValid)
|
|
return db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
|
|
group := &entity.CompilationTemplateGroup{
|
|
ID: BuiltinCompilationTemplateGroupID,
|
|
TenantID: "", // global built-in catalogue
|
|
Name: "Built-in templates",
|
|
Status: strptr(valid),
|
|
}
|
|
if err := tx.WithContext(ctx).Clauses(clause.OnConflict{
|
|
Columns: []clause.Column{{Name: "id"}},
|
|
DoNothing: true,
|
|
}).Create(group).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, t := range builtinCompilationTemplateKinds {
|
|
tmpl := &entity.CompilationTemplate{
|
|
ID: BuiltinCompilationTemplateGroupID + "-" + t.Kind,
|
|
TenantID: nil, // global built-in catalogue
|
|
GroupID: strptr(BuiltinCompilationTemplateGroupID),
|
|
Name: t.Name,
|
|
Kind: t.Kind,
|
|
Config: entity.JSONMap{"kind": t.Kind},
|
|
Status: strptr(valid),
|
|
}
|
|
if err := tx.WithContext(ctx).Clauses(clause.OnConflict{
|
|
Columns: []clause.Column{{Name: "id"}},
|
|
DoNothing: true,
|
|
}).Create(tmpl).Error; err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
})
|
|
}
|