62 lines
1.3 KiB
Go
62 lines
1.3 KiB
Go
package tools
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"math"
|
|
"strconv"
|
|
|
|
"github.com/expr-lang/expr"
|
|
)
|
|
|
|
type CalculatorTool struct{}
|
|
|
|
func (CalculatorTool) Name() string { return "calculate" }
|
|
func (CalculatorTool) Description() string {
|
|
return "计算数学表达式，如 \"(12 + 3) * 4\"、\"2^10\"、\"sqrt(9)\""
|
|
}
|
|
func (CalculatorTool) Parameters() map[string]any {
|
|
return map[string]any{
|
|
"type": "object",
|
|
"properties": map[string]any{
|
|
"expression": map[string]any{"type": "string", "description": "要计算的数学表达式"},
|
|
},
|
|
"required": []string{"expression"},
|
|
}
|
|
}
|
|
|
|
func (CalculatorTool) Execute(args json.RawMessage) (string, error) {
|
|
var p struct {
|
|
Expression string `json:"expression"`
|
|
}
|
|
if err := json.Unmarshal(args, &p); err != nil {
|
|
return "", err
|
|
}
|
|
if p.Expression == "" {
|
|
return "", fmt.Errorf("expression 不能为空")
|
|
}
|
|
out, err := expr.Eval(p.Expression, nil)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
switch v := out.(type) {
|
|
case float64:
|
|
return strconv.FormatFloat(round(v), 'f', -1, 64), nil
|
|
case int:
|
|
return strconv.Itoa(v), nil
|
|
case bool:
|
|
return strconv.FormatBool(v), nil
|
|
case string:
|
|
return v, nil
|
|
default:
|
|
return fmt.Sprintf("%v", v), nil
|
|
}
|
|
}
|
|
|
|
func round(v float64) float64 {
|
|
if math.IsInf(v, 0) || math.IsNaN(v) {
|
|
return v
|
|
}
|
|
return math.Round(v*1e8) / 1e8
|
|
}
|