Feat: tools

tools.go

@@ -0,0 +1,65 @@
+package main
+
+import (
+	"grailbench/models"
+	"time"
+)
+
+func sendEmail(args map[string]string) []byte {
+	logger.Info("send-email is used", "args", args)
+	return nil
+}
+
+func getCurrentTimestamp(args map[string]string) []byte {
+	ts := time.Now()
+	return []byte(ts.Format(time.RFC3339))
+}
+
+type fnSig func(map[string]string) []byte
+
+var fnMap = map[string]fnSig{
+	"get_current_timestamp": getCurrentTimestamp,
+	"send_email":            sendEmail,
+}
+
+// openai style def
+var baseTools = []models.Tool{
+	// get_current_timestamp
+	models.Tool{
+		Type: "function",
+		Function: models.ToolFunc{
+			Name:        "get_current_timestamp",
+			Description: "Returns current timestamp in RFC3999 format",
+			Parameters: models.ToolFuncParams{
+				Type:     "object",
+				Required: []string{},
+			},
+		},
+	},
+	// send_email
+	models.Tool{
+		Type: "function",
+		Function: models.ToolFunc{
+			Name:        "send_email",
+			Description: "Sends email to a provided address with given message.",
+			Parameters: models.ToolFuncParams{
+				Type:     "object",
+				Required: []string{"address", "title", "body"},
+				Properties: map[string]models.ToolArgProps{
+					"address": models.ToolArgProps{
+						Type:        "string",
+						Description: "email address of the recipient",
+					},
+					"title": models.ToolArgProps{
+						Type:        "string",
+						Description: "",
+					},
+					"body": models.ToolArgProps{
+						Type:        "string",
+						Description: "email body, can be in form of html page, markdown or pure text.",
+					},
+				},
+			},
+		},
+	},
+}