Enha: openai tools [WIP]

1 files changed, 27 insertions, 1 deletions

diff --git a/tools.go b/tools.go
index fe95ce5..01ab32b 100644
--- a/tools.go
+++ b/tools.go
@@ -1,8 +1,8 @@
 package main
 
 import (
-	"gf-lt/models"
 	"fmt"
+	"gf-lt/models"
 	"regexp"
 	"strings"
 	"time"
@@ -141,3 +141,29 @@ var fnMap = map[string]fnSig{
 	"recall_topics": recallTopics,
 	"memorise":      memorise,
 }
+
+// openai style def
+var baseTools = []models.Tool{
+	// memorise
+	models.Tool{
+		Type: "function",
+		Function: models.ToolFunc{
+			Name:        "memorise",
+			Description: "save topic-data in key-value cache",
+			Parameters: models.ToolFuncParams{
+				Type:     "object",
+				Required: []string{"topic", "data"},
+				Properties: map[string]models.ToolArgProps{
+					"topic": models.ToolArgProps{
+						Type:        "string",
+						Description: "topic is the key under which data is saved",
+					},
+					"data": models.ToolArgProps{
+						Type:        "string",
+						Description: "data is the value that is saved under the topic-key",
+					},
+				},
+			},
+		},
+	},
+}