mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-07-01 00:05:43 +08:00
e45659868a
Replaces the Python agent canvas runtime with a Go implementation that runs inside `cmd/server_main`. The canvas compiles into an eino Workflow that pauses on wait-for-user via native Interrupt/Resume (no sentinel flag) and resumes from a Redis-backed CheckPointStore. All 21 Python agent components and ~35 tools are ported with functional parity. Sandbox providers now read their JSON config from the admin-panel system_settings table with env fallback. 234 files / +35,413 / -6,111. All Go files are gofmt-clean (CI gate added); drops the v2 DSL E2E step and the gap-analysis plan (both redundant after the port ships). ## Type of change - [x] Refactoring - [x] New feature - [x] Bug fix 🤖 Generated with [Claude Code](https://claude.com/claude-code) --------- Co-authored-by: Claude <noreply@anthropic.com>
76 lines
2.4 KiB
Go
76 lines
2.4 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 component
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"ragflow/internal/agent/component/prompts"
|
|
)
|
|
|
|
// TestCitationPlusPrompt_EmptySources: no sources → render with empty
|
|
// sources block; IDs slice is empty.
|
|
func TestCitationPlusPrompt_EmptySources(t *testing.T) {
|
|
rendered, ids := prompts.CitationPlusPrompt(nil)
|
|
if len(ids) != 0 {
|
|
t.Errorf("expected empty ids, got %v", ids)
|
|
}
|
|
if !strings.Contains(rendered, "<context>") {
|
|
t.Error("expected <context> block in rendered prompt")
|
|
}
|
|
if !strings.Contains(rendered, "ID:") {
|
|
t.Error("expected 'ID:' prefix in sources block (empty section)")
|
|
}
|
|
}
|
|
|
|
// TestCitationPlusPrompt_WithSources: sources are rendered as
|
|
// `ID: <id>\n└── Content: <content>` blocks, ids are returned.
|
|
func TestCitationPlusPrompt_WithSources(t *testing.T) {
|
|
sources := []prompts.CitationSource{
|
|
{ID: "45", Content: "Smartphone market grew 7.8% in Q3 2024."},
|
|
{ID: "46", Content: "5G adoption reached 1.5B users."},
|
|
}
|
|
rendered, ids := prompts.CitationPlusPrompt(sources)
|
|
if len(ids) != 2 {
|
|
t.Fatalf("expected 2 ids, got %d", len(ids))
|
|
}
|
|
if ids[0] != "45" || ids[1] != "46" {
|
|
t.Errorf("ids=%v, want [45 46]", ids)
|
|
}
|
|
if !strings.Contains(rendered, "ID: 45") {
|
|
t.Error("expected ID: 45 in rendered prompt")
|
|
}
|
|
if !strings.Contains(rendered, "Smartphone market grew 7.8%") {
|
|
t.Error("expected source content in rendered prompt")
|
|
}
|
|
}
|
|
|
|
// TestCitationPlusPrompt_SkipsEmptyFields: sources with empty ID or
|
|
// content are filtered out.
|
|
func TestCitationPlusPrompt_SkipsEmptyFields(t *testing.T) {
|
|
sources := []prompts.CitationSource{
|
|
{ID: "", Content: "no id"},
|
|
{ID: "1", Content: ""},
|
|
{ID: "2", Content: "valid"},
|
|
}
|
|
_, ids := prompts.CitationPlusPrompt(sources)
|
|
if len(ids) != 1 || ids[0] != "2" {
|
|
t.Errorf("expected only valid source; got ids=%v", ids)
|
|
}
|
|
}
|