mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-07-17 05:07:23 +08:00
d03b09b6a3
## Summary - Align Go WenCai and SearXNG behavior, schemas, and node parameters with Python. - Add the `WenCai` and `SearXNG` Canvas components and register their tool factories. - Match Python's current WenCai behavior by returning an empty report while its upstream request is disabled. - Add SearXNG request validation, SSRF-safe DNS pinning, raw result preservation, and reference rendering. - Support context cancellation, error envelopes, and lock-safe retrieval references. ## Tests Passed: - `bash build.sh --test ./internal/agent/tool/...` - `bash build.sh --test ./internal/agent/component/...` - `bash build.sh --test ./internal/agent/runtime/...` - `bash build.sh --test ./internal/agent/...` - `cd web && npm run type-check` <img width="1900" height="1102" alt="image" src="https://github.com/user-attachments/assets/ec77d217-d9fd-455a-96ec-9aabf6841109" /> <img width="1900" height="1102" alt="image" src="https://github.com/user-attachments/assets/52ac129f-cb65-453d-ae48-cc518803ac23" />
110 lines
2.9 KiB
Go
110 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 (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
einotool "github.com/cloudwego/eino/components/tool"
|
|
|
|
agenttool "ragflow/internal/agent/tool"
|
|
)
|
|
|
|
type wencaiInvoker interface {
|
|
InvokableRun(ctx context.Context, argsJSON string, opts ...einotool.Option) (string, error)
|
|
}
|
|
|
|
type wencaiComponent struct {
|
|
inner wencaiInvoker
|
|
}
|
|
|
|
func newWencaiComponent(params map[string]any) (Component, error) {
|
|
toolParams := make(map[string]any, 2)
|
|
for _, key := range []string{"top_n", "query_type"} {
|
|
if value, ok := params[key]; ok {
|
|
toolParams[key] = value
|
|
}
|
|
}
|
|
inner, err := agenttool.BuildByName("wencai", toolParams)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
invoker, ok := inner.(wencaiInvoker)
|
|
if !ok {
|
|
return nil, fmt.Errorf("WenCai: tool does not implement InvokableRun")
|
|
}
|
|
return newWencaiComponentWithInvoker(invoker), nil
|
|
}
|
|
|
|
func newWencaiComponentWithInvoker(inner wencaiInvoker) Component {
|
|
return &wencaiComponent{inner: inner}
|
|
}
|
|
|
|
func (c *wencaiComponent) Name() string { return "WenCai" }
|
|
|
|
func (c *wencaiComponent) Inputs() map[string]string {
|
|
return map[string]string{
|
|
"query": "The question/conditions to select stocks.",
|
|
}
|
|
}
|
|
|
|
func (c *wencaiComponent) Outputs() map[string]string {
|
|
return map[string]string{
|
|
"report": "WenCai query report.",
|
|
}
|
|
}
|
|
|
|
func (c *wencaiComponent) GetInputForm() map[string]any {
|
|
return map[string]any{
|
|
"query": map[string]any{
|
|
"name": "Query",
|
|
"type": "line",
|
|
},
|
|
}
|
|
}
|
|
|
|
func (c *wencaiComponent) Invoke(ctx context.Context, inputs map[string]any) (map[string]any, error) {
|
|
query := stringParam(inputs["query"])
|
|
if query == "" {
|
|
return map[string]any{"report": ""}, nil
|
|
}
|
|
argsJSON, err := json.Marshal(map[string]any{"query": query})
|
|
if err != nil {
|
|
return nil, fmt.Errorf("canvas: WenCai: encode query: %w", err)
|
|
}
|
|
out, err := c.inner.InvokableRun(ctx, string(argsJSON))
|
|
decoded := parseToolEnvelope(out)
|
|
report, _ := decoded["report"].(string)
|
|
if message, _ := decoded["_ERROR"].(string); message != "" {
|
|
return map[string]any{"report": report, "_ERROR": message}, nil
|
|
}
|
|
if err != nil {
|
|
return nil, fmt.Errorf("canvas: WenCai: %w", err)
|
|
}
|
|
return map[string]any{"report": report}, nil
|
|
}
|
|
|
|
func (c *wencaiComponent) Stream(_ context.Context, _ map[string]any) (<-chan map[string]any, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
func init() {
|
|
Register("WenCai", newWencaiComponent)
|
|
}
|