mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-07-16 20:57:21 +08:00
a7da78d0d7
## Summary
This PR consolidates Eino-backed Agent tools behind the shared
ToolBackedComponent implementation and aligns their Canvas
configuration,
runtime inputs, output conversion, validation, and registration.
## What changed
- Migrated these Canvas components to the unified tool-backed path:
- Tavily Search and Extract
- Execute SQL
- Google
- Yahoo Finance
- Email
- DuckDuckGo
- Wikipedia
- Google Scholar
- ArXiv
- PubMed
- BGPT
- GitHub
- WenCai
- SearXNG
- Keenable Search
- Removed superseded component wrappers and their duplicate tests.
- Added dedicated registry builders for node-level configuration and
validation.
- Kept model-emitted runtime inputs separate from Canvas node
configuration.
- Moved Email defaults, template resolution, recipient parsing, SMTP
execution, and output conversion into the owning tool.
- Added complete ToolComponent specifications for Canvas inputs,
outputs, and input forms.
- Preserved raw upstream fields where downstream workflows may depend on
them.
- Added workflow registration coverage for all migrated component names.
- Kept HTTP Request, Docs Generator, and Browser as standalone
components because they are not Eino-backed tools.
## Testing
Passed:
bash build.sh --test ./internal/agent/tool/...
bash build.sh --test ./internal/agent/component/...
bash build.sh --test ./internal/agent/runtime/...
<img width="2057" height="1111" alt="image"
src="https://github.com/user-attachments/assets/c728d7a3-9d15-4c5c-b0eb-6b77ad0e41ac"
/>
<img width="2057" height="1111" alt="image"
src="https://github.com/user-attachments/assets/ef13119a-ef92-4b43-9061-eee9511bf492"
/>
<img width="2057" height="1111" alt="image"
src="https://github.com/user-attachments/assets/f50f2c30-eab0-463c-b0b3-0a02523219f1"
/>
69 lines
2.1 KiB
Go
69 lines
2.1 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"
|
|
|
|
einotool "github.com/cloudwego/eino/components/tool"
|
|
)
|
|
|
|
// ToolInvoker is the invocation seam shared by Eino tools and Canvas
|
|
// components backed by those tools.
|
|
type ToolInvoker interface {
|
|
InvokableRun(ctx context.Context, argsJSON string, opts ...einotool.Option) (string, error)
|
|
}
|
|
|
|
// ComponentSpec describes the Canvas-facing surface of a tool. It is kept
|
|
// separate from Info(), whose schema contains only model-emitted arguments.
|
|
type ComponentSpec struct {
|
|
Inputs map[string]string
|
|
Outputs map[string]string
|
|
InputForm map[string]any
|
|
}
|
|
|
|
// ToolComponent is the required Canvas adaptation contract implemented by a
|
|
// tool that can back a Canvas component.
|
|
type ToolComponent interface {
|
|
ToolInvoker
|
|
ComponentSpec() ComponentSpec
|
|
// BuildComponentOutputs converts the complete decoded tool envelope into
|
|
// the component's public Canvas outputs.
|
|
BuildComponentOutputs(envelope map[string]any) map[string]any
|
|
}
|
|
|
|
// ReferenceBuilder is an optional capability for tools that add retrieval
|
|
// references to Canvas state.
|
|
type ReferenceBuilder interface {
|
|
BuildReferences(ctx context.Context, envelope map[string]any) (chunks []map[string]any, docAggs []map[string]any)
|
|
}
|
|
|
|
func envelopeSlice(envelope map[string]any, key string) []any {
|
|
switch values := envelope[key].(type) {
|
|
case []any:
|
|
return values
|
|
case []map[string]any:
|
|
result := make([]any, 0, len(values))
|
|
for _, value := range values {
|
|
result = append(result, value)
|
|
}
|
|
return result
|
|
default:
|
|
return []any{}
|
|
}
|
|
}
|