Function TestCodeModeTool_CallHello

pkg/tools/codemode/codemode_test.go:126–157

(t *testing.T)

}

func TestCodeModeTool_CallHello(t *testing.T) {
	t.Parallel()
	tool := Wrap(&testToolSet{
		tools: []tools.Tool{
			{
				Name: "hello_world",
				Handler: tools.NewHandler(func(ctx context.Context, args map[string]any) (*tools.ToolCallResult, error) {
					return tools.ResultSuccess("Hello, World!"), nil
				}),
			},
		},
	})

	allTools, err := tool.Tools(t.Context())
	require.NoError(t, err)
	require.Len(t, allTools, 1)

	result, err := allTools[0].Handler(t.Context(), tools.ToolCall{
		Function: tools.FunctionCall{
			Arguments: `{"script":"return hello_world();"}`,
		},
	})
	require.NoError(t, err)

	var scriptResult ScriptResult
	err = json.Unmarshal([]byte(result.Output), &scriptResult)
	require.NoError(t, err)

	require.Equal(t, "Hello, World!", scriptResult.Value)
	require.Empty(t, scriptResult.StdErr)
	require.Empty(t, scriptResult.StdOut)
}

func TestCodeModeTool_CallEcho(t *testing.T) {
	t.Parallel()