diff --git a/backend/domain/agent/singleagent/internal/agentflow/node_tool_database.go b/backend/domain/agent/singleagent/internal/agentflow/node_tool_database.go index 75aefb1b..3af89bfa 100644 --- a/backend/domain/agent/singleagent/internal/agentflow/node_tool_database.go +++ b/backend/domain/agent/singleagent/internal/agentflow/node_tool_database.go @@ -112,22 +112,21 @@ func newDatabaseTools(ctx context.Context, conf *databaseConfig) ([]tool.Invokab dbInfos := conf.databaseConf - d := &databaseTool{ - spaceID: conf.spaceID, - connectorUID: conf.userID, - agentIdentity: conf.agentIdentity, - } - tools := make([]tool.InvokableTool, 0, len(dbInfos)) for _, dbInfo := range dbInfos { tID, err := strconv.ParseInt(dbInfo.GetTableId(), 10, 64) if err != nil { return nil, err } + d := &databaseTool{ + spaceID: conf.spaceID, + connectorUID: conf.userID, + agentIdentity: conf.agentIdentity, + promptDisabled: dbInfo.GetPromptDisabled(), + name: dbInfo.GetTableName(), + databaseID: tID, + } - d.databaseID = tID - d.promptDisabled = dbInfo.GetPromptDisabled() - d.name = dbInfo.GetTableName() dbTool, err := utils.InferTool(dbInfo.GetTableName(), buildDatabaseToolDescription(dbInfo), d.Invoke) if err != nil { return nil, err