summaryrefslogtreecommitdiff
path: root/tools.go
diff options
context:
space:
mode:
Diffstat (limited to 'tools.go')
-rw-r--r--tools.go28
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",
+ },
+ },
+ },
+ },
+ },
+}