mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-06-29 23:41:12 +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>
86 lines
2.9 KiB
Go
86 lines
2.9 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"
|
|
)
|
|
|
|
// TestSplitSentences_Empty: empty input returns a single empty
|
|
// element (never nil, callers can always range).
|
|
func TestSplitSentences_Empty(t *testing.T) {
|
|
out := splitSentences("")
|
|
if len(out) != 1 {
|
|
t.Fatalf("expected 1 element, got %d", len(out))
|
|
}
|
|
if out[0] != "" {
|
|
t.Errorf("expected empty string, got %q", out[0])
|
|
}
|
|
}
|
|
|
|
// TestSplitSentences_NoBoundary: text without sentence boundaries
|
|
// returns the whole text as a single element.
|
|
func TestSplitSentences_NoBoundary(t *testing.T) {
|
|
out := splitSentences("just a fragment")
|
|
if len(out) != 1 {
|
|
t.Fatalf("expected 1 element, got %d", len(out))
|
|
}
|
|
if out[0] != "just a fragment" {
|
|
t.Errorf("got %q, want full text", out[0])
|
|
}
|
|
}
|
|
|
|
// TestSplitSentences_DotBoundary: a period+space split.
|
|
func TestSplitSentences_DotBoundary(t *testing.T) {
|
|
out := splitSentences("first. second. third.")
|
|
if len(out) != 3 {
|
|
t.Fatalf("expected 3 sentences, got %d: %+v", len(out), out)
|
|
}
|
|
if out[0] != "first." || out[1] != "second." || out[2] != "third." {
|
|
t.Errorf("unexpected split: %+v", out)
|
|
}
|
|
}
|
|
|
|
// TestSplitSentences_MixedPunctuation: "!" and "?" are also
|
|
// sentence boundaries.
|
|
func TestSplitSentences_MixedPunctuation(t *testing.T) {
|
|
out := splitSentences("wow! really? ok.")
|
|
if len(out) != 3 {
|
|
t.Fatalf("expected 3, got %d: %+v", len(out), out)
|
|
}
|
|
if !strings.HasSuffix(out[0], "!") || !strings.HasSuffix(out[1], "?") || !strings.HasSuffix(out[2], ".") {
|
|
t.Errorf("wrong trailing punctuation: %+v", out)
|
|
}
|
|
}
|
|
|
|
// TestMessage_Stream_MultiSentence: a resolved content with
|
|
// multiple sentences emits N content chunks + 1 done chunk.
|
|
// The end-to-end message_test.go already exercises the single-
|
|
// sentence happy path; this test pins the multi-sentence path
|
|
// through splitSentences directly (Stream integration is via the
|
|
// existing test).
|
|
func TestMessage_Stream_MultiSentence(t *testing.T) {
|
|
// Direct unit test of splitSentences; the Stream() method uses
|
|
// this internally. The full Stream() integration is covered by
|
|
// TestMessage_Stream in message_test.go.
|
|
sentences := splitSentences("First sentence. Second sentence. Third.")
|
|
if len(sentences) != 3 {
|
|
t.Fatalf("expected 3 sentences, got %d: %+v", len(sentences), sentences)
|
|
}
|
|
}
|