mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-07-19 06:01:05 +08:00
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{}
|
||
|
|
}
|
||
|
|
}
|