mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-07-09 21:04:49 +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>
78 lines
2.6 KiB
Go
78 lines
2.6 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 tool
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
)
|
|
|
|
// TestRetrievalService_DefaultStub: with no SetRetrievalService
|
|
// call, the default stub returns ErrRetrievalServiceMissing.
|
|
func TestRetrievalService_DefaultStub(t *testing.T) {
|
|
svc := GetRetrievalService()
|
|
if svc == nil {
|
|
t.Fatal("GetRetrievalService returned nil")
|
|
}
|
|
_, err := svc.Search(context.Background(), RetrievalRequest{Query: "hi"})
|
|
if !errors.Is(err, ErrRetrievalServiceMissing) {
|
|
t.Errorf("err=%v, want ErrRetrievalServiceMissing", err)
|
|
}
|
|
}
|
|
|
|
// TestRetrievalService_RegisterAndRestore: a custom impl can be
|
|
// installed and later restored to the stub via SetRetrievalService(nil).
|
|
func TestRetrievalService_RegisterAndRestore(t *testing.T) {
|
|
prev := GetRetrievalService()
|
|
t.Cleanup(func() { SetRetrievalService(prev) })
|
|
|
|
// Install a fake.
|
|
SetRetrievalService(fakeRetrievalService{chunks: []RetrievalChunk{{ID: "1", Content: "fake"}}})
|
|
got, err := GetRetrievalService().Search(context.Background(), RetrievalRequest{Query: "x"})
|
|
if err != nil {
|
|
t.Fatalf("Search: %v", err)
|
|
}
|
|
if len(got) != 1 || got[0].ID != "1" {
|
|
t.Errorf("got %+v, want 1 chunk with id=1", got)
|
|
}
|
|
|
|
// Restore via nil.
|
|
SetRetrievalService(nil)
|
|
if _, err := GetRetrievalService().Search(context.Background(), RetrievalRequest{}); !errors.Is(err, ErrRetrievalServiceMissing) {
|
|
t.Errorf("after nil-reset: err=%v, want ErrRetrievalServiceMissing", err)
|
|
}
|
|
}
|
|
|
|
// fakeRetrievalService is a programmable stub for testing.
|
|
type fakeRetrievalService struct {
|
|
chunks []RetrievalChunk
|
|
err error
|
|
}
|
|
|
|
func (f fakeRetrievalService) Search(_ context.Context, _ RetrievalRequest) ([]RetrievalChunk, error) {
|
|
return f.chunks, f.err
|
|
}
|
|
|
|
// TestRetrievalChunk_Fields: round-trip the struct.
|
|
func TestRetrievalChunk_Fields(t *testing.T) {
|
|
c := RetrievalChunk{ID: "i", Content: "c", DocumentID: "d", Score: 0.5}
|
|
if c.ID != "i" || c.Content != "c" || c.DocumentID != "d" || c.Score != 0.5 {
|
|
t.Errorf("round-trip lost fields: %+v", c)
|
|
}
|
|
}
|