feat(tools): return error to agent if any happened
This commit is contained in:
@ -39,16 +39,20 @@ func (handler ToolsHandlers) Handle(info ToolHandlerInfo, toolCallMessage AgentA
|
|||||||
}
|
}
|
||||||
|
|
||||||
res, err := fnHandler.Fn(info, arguments, toolCallMessage.ToolCalls[0])
|
res, err := fnHandler.Fn(info, arguments, toolCallMessage.ToolCalls[0])
|
||||||
if err != nil {
|
|
||||||
return []AgentTextMessage{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
responses[i] = AgentTextMessage{
|
responseMessage := AgentTextMessage{
|
||||||
Role: "tool",
|
Role: "tool",
|
||||||
Name: fnName,
|
Name: fnName,
|
||||||
Content: res,
|
|
||||||
ToolCallId: toolCall.Id,
|
ToolCallId: toolCall.Id,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
responseMessage.Content = err.Error()
|
||||||
|
} else {
|
||||||
|
responseMessage.Content = res
|
||||||
|
}
|
||||||
|
|
||||||
|
responses[i] = responseMessage
|
||||||
}
|
}
|
||||||
|
|
||||||
return responses, nil
|
return responses, nil
|
||||||
|
@ -1,6 +1,7 @@
|
|||||||
package client
|
package client
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"errors"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"github.com/google/uuid"
|
"github.com/google/uuid"
|
||||||
@ -19,7 +20,11 @@ func (suite *ToolTestSuite) SetupTest() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
suite.handler.AddTool("a", func(info ToolHandlerInfo, args string, call ToolCall) (any, error) {
|
suite.handler.AddTool("a", func(info ToolHandlerInfo, args string, call ToolCall) (any, error) {
|
||||||
return true, nil
|
return args, nil
|
||||||
|
})
|
||||||
|
|
||||||
|
suite.handler.AddTool("error", func(info ToolHandlerInfo, args string, call ToolCall) (any, error) {
|
||||||
|
return false, errors.New("I will always error")
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -40,7 +45,7 @@ func (suite *ToolTestSuite) TestSingleToolCall() {
|
|||||||
Id: "1",
|
Id: "1",
|
||||||
Function: FunctionCall{
|
Function: FunctionCall{
|
||||||
Name: "a",
|
Name: "a",
|
||||||
Arguments: "",
|
Arguments: "return",
|
||||||
},
|
},
|
||||||
}},
|
}},
|
||||||
})
|
})
|
||||||
@ -49,7 +54,7 @@ func (suite *ToolTestSuite) TestSingleToolCall() {
|
|||||||
|
|
||||||
assert.EqualValues(response, []AgentTextMessage{{
|
assert.EqualValues(response, []AgentTextMessage{{
|
||||||
Role: "tool",
|
Role: "tool",
|
||||||
Content: "true",
|
Content: "\"return\"",
|
||||||
ToolCallId: "1",
|
ToolCallId: "1",
|
||||||
Name: "a",
|
Name: "a",
|
||||||
}})
|
}})
|
||||||
@ -90,7 +95,7 @@ func (suite *ToolTestSuite) TestMultipleToolCalls() {
|
|||||||
Id: "1",
|
Id: "1",
|
||||||
Function: FunctionCall{
|
Function: FunctionCall{
|
||||||
Name: "a",
|
Name: "a",
|
||||||
Arguments: "",
|
Arguments: "first-call",
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
@ -98,7 +103,7 @@ func (suite *ToolTestSuite) TestMultipleToolCalls() {
|
|||||||
Id: "2",
|
Id: "2",
|
||||||
Function: FunctionCall{
|
Function: FunctionCall{
|
||||||
Name: "a",
|
Name: "a",
|
||||||
Arguments: "",
|
Arguments: "second-call",
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
@ -109,13 +114,63 @@ func (suite *ToolTestSuite) TestMultipleToolCalls() {
|
|||||||
assert.EqualValues(response, []AgentTextMessage{
|
assert.EqualValues(response, []AgentTextMessage{
|
||||||
{
|
{
|
||||||
Role: "tool",
|
Role: "tool",
|
||||||
Content: "true",
|
Content: "\"first-call\"",
|
||||||
ToolCallId: "1",
|
ToolCallId: "1",
|
||||||
Name: "a",
|
Name: "a",
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
Role: "tool",
|
Role: "tool",
|
||||||
Content: "true",
|
Content: "\"second-call\"",
|
||||||
|
ToolCallId: "2",
|
||||||
|
Name: "a",
|
||||||
|
},
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (suite *ToolTestSuite) TestMultipleToolCallsWithErrors() {
|
||||||
|
assert := suite.Assert()
|
||||||
|
require := suite.Require()
|
||||||
|
|
||||||
|
response, err := suite.handler.Handle(
|
||||||
|
ToolHandlerInfo{
|
||||||
|
UserId: uuid.Nil,
|
||||||
|
ImageId: uuid.Nil,
|
||||||
|
},
|
||||||
|
AgentAssistantToolCall{
|
||||||
|
Role: "assistant",
|
||||||
|
Content: "",
|
||||||
|
ToolCalls: []ToolCall{
|
||||||
|
{
|
||||||
|
Index: 0,
|
||||||
|
Id: "1",
|
||||||
|
Function: FunctionCall{
|
||||||
|
Name: "error",
|
||||||
|
Arguments: "",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Index: 1,
|
||||||
|
Id: "2",
|
||||||
|
Function: FunctionCall{
|
||||||
|
Name: "a",
|
||||||
|
Arguments: "no-error",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
require.NoError(err, "Tool call shouldnt return an error")
|
||||||
|
|
||||||
|
assert.EqualValues(response, []AgentTextMessage{
|
||||||
|
{
|
||||||
|
Role: "tool",
|
||||||
|
Content: "I will always error",
|
||||||
|
ToolCallId: "1",
|
||||||
|
Name: "error",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Role: "tool",
|
||||||
|
Content: "\"no-error\"",
|
||||||
ToolCallId: "2",
|
ToolCallId: "2",
|
||||||
Name: "a",
|
Name: "a",
|
||||||
},
|
},
|
||||||
|
Reference in New Issue
Block a user