ben
/
coze-studio
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: Fix bug in initializing database tool for agent (#569)

This commit is contained in:
liuyunchao-1998 2025-08-05 15:30:50 +08:00 committed by GitHub
parent bb6ff0026b
commit d4d0ba6b9a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
backend/domain/agent/singleagent/internal/agentflow/node_tool_database.go
View File

@ -112,22 +112,21 @@ func newDatabaseTools(ctx context.Context, conf *databaseConfig) ([]tool.Invokab
dbInfos := conf.databaseConf dbInfos := conf.databaseConf
d := &databaseTool{
spaceID: conf.spaceID,
connectorUID: conf.userID,
agentIdentity: conf.agentIdentity,
}
tools := make([]tool.InvokableTool, 0, len(dbInfos)) tools := make([]tool.InvokableTool, 0, len(dbInfos))
for _, dbInfo := range dbInfos { for _, dbInfo := range dbInfos {
tID, err := strconv.ParseInt(dbInfo.GetTableId(), 10, 64) tID, err := strconv.ParseInt(dbInfo.GetTableId(), 10, 64)
if err != nil { if err != nil {
return nil, err 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) dbTool, err := utils.InferTool(dbInfo.GetTableName(), buildDatabaseToolDescription(dbInfo), d.Invoke)
if err != nil { if err != nil {
return nil, err return nil, err