feat: manually mirror opencoze's code from bytedance

Change-Id: I09a73aadda978ad9511264a756b2ce51f5761adf

backend/domain/agent/singleagent/internal/agentflow/node_tool_knowledge.go

@@ -0,0 +1,114 @@
+/*
+ * Copyright 2025 coze-dev Authors
+ *
+ * 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 agentflow
+
+import (
+	"context"
+	"fmt"
+	"reflect"
+	"strconv"
+
+	"github.com/cloudwego/eino/components/tool"
+	"github.com/cloudwego/eino/components/tool/utils"
+	"github.com/cloudwego/eino/schema"
+	"github.com/getkin/kin-openapi/openapi3"
+
+	"github.com/coze-dev/coze-studio/backend/api/model/ocean/cloud/bot_common"
+	"github.com/coze-dev/coze-studio/backend/crossdomain/contract/crossknowledge"
+	knowledgeEntity "github.com/coze-dev/coze-studio/backend/domain/knowledge/entity"
+)
+
+const (
+	knowledgeToolName = "recallKnowledge"
+	knowledgeDesc     = `Provides multiple retrieval methods to search for content fragments stored in the knowledge base, helping the large model obtain more accurate and reliable context information`
+)
+
+type knowledgeConfig struct {
+	knowledgeInfos  []*knowledgeEntity.Knowledge
+	knowledgeConfig *bot_common.Knowledge
+	Input           *schema.Message
+	GetHistory      func() []*schema.Message
+}
+
+func newKnowledgeTool(ctx context.Context, conf *knowledgeConfig) (tool.InvokableTool, error) {
+	kl := &knowledgeTool{
+		knowledgeConfig: conf.knowledgeConfig,
+		Input:           conf.Input,
+		GetHistory:      conf.GetHistory,
+	}
+
+	customTagsFn := func(name string, t reflect.Type, tag reflect.StructTag,
+		schema *openapi3.Schema,
+	) error {
+		// Process KnowledgeIDs field only
+		if name != "KnowledgeIDs" {
+			return nil
+		}
+
+		// Build knowledge base description
+		desc := "Available Knowledge Base List as format knowledge_id: knowledge_name - knowledge_description: \n"
+		for _, k := range conf.knowledgeInfos {
+			desc += fmt.Sprintf("- %d: %s - %s\n", k.ID, k.Name, k.Description)
+		}
+
+		schema.Type = openapi3.TypeArray
+		schema.Items = &openapi3.SchemaRef{
+			Value: &openapi3.Schema{
+				Type: openapi3.TypeInteger,
+			},
+		}
+		// 设置字段描述和枚举值
+		schema.Description = desc
+		schema.Enum = make([]interface{}, 0, len(conf.knowledgeInfos))
+		for _, k := range conf.knowledgeInfos {
+			schema.Enum = append(schema.Enum, strconv.FormatInt(k.ID, 10))
+		}
+
+		return nil
+	}
+
+	return utils.InferTool(knowledgeToolName, knowledgeDesc, kl.Retrieve, utils.WithSchemaCustomizer(customTagsFn))
+}
+
+type RetrieveRequest struct {
+	KnowledgeIDs []int64 `json:"knowledge_ids" jsonschema:"description="`
+}
+
+type knowledgeTool struct {
+	knowledgeConfig *bot_common.Knowledge
+	Input           *schema.Message
+	GetHistory      func() []*schema.Message
+}
+
+func (k *knowledgeTool) Retrieve(ctx context.Context, req *RetrieveRequest) ([]*schema.Document, error) {
+	rr, err := genKnowledgeRequest(ctx, req.KnowledgeIDs, k.knowledgeConfig, k.Input.Content, k.GetHistory())
+	if err != nil {
+		return nil, err
+	}
+
+	resp, err := crossknowledge.DefaultSVC().Retrieve(ctx, rr)
+	if err != nil {
+		return nil, err
+	}
+
+	docs, err := convertDocument(ctx, resp.RetrieveSlices)
+	if err != nil {
+		return nil, err
+	}
+
+	return docs, nil
+}