diff options
Diffstat (limited to 'tools.go')
| -rw-r--r-- | tools.go | 114 |
1 files changed, 62 insertions, 52 deletions
@@ -1,25 +1,32 @@ package main import ( - "elefant/models" - "encoding/json" + "gf-lt/models" + "fmt" "regexp" + "strings" "time" ) var ( - // TODO: form that message based on existing funcs + toolCallRE = regexp.MustCompile(`__tool_call__\s*([\s\S]*?)__tool_call__`) + quotesRE = regexp.MustCompile(`(".*?")`) + starRE = regexp.MustCompile(`(\*.*?\*)`) + thinkRE = regexp.MustCompile(`(<think>\s*([\s\S]*?)</think>)`) + codeBlockRE = regexp.MustCompile(`(?s)\x60{3}(?:.*?)\n(.*?)\n\s*\x60{3}\s*`) + rpDefenitionSysMsg = ` +For this roleplay immersion is at most importance. +Every character thinks and acts based on their personality and setting of the roleplay. +Meta discussions outside of roleplay is allowed if clearly labeled as out of character, for example: (ooc: {msg}) or <ooc>{msg}</ooc>. +` basicSysMsg = `Large Language Model that helps user with any of his requests.` - toolCallRE = regexp.MustCompile(`__tool_call__\s*([\s\S]*?)__tool_call__`) - toolSysMsg = `You're a helpful assistant. -# Tools -You can do functions call if needed. + toolSysMsg = `You can do functions call if needed. Your current tools: <tools> [ { "name":"recall", -"args": "topic", +"args": ["topic"], "when_to_use": "when asked about topic that user previously asked to memorise" }, { @@ -29,25 +36,44 @@ Your current tools: }, { "name":"recall_topics", -"args": null, -"when_to_use": "once in a while" +"args": [], +"when_to_use": "to see what topics are saved in memory" } ] </tools> To make a function call return a json object within __tool_call__ tags; -Example: +<example_request> __tool_call__ { "name":"recall", -"args": "Adam" +"args": ["Adam's number"] } __tool_call__ -When done right, tool call will be delivered to the 'tool' agent. 'tool' agent will respond with the results of the call. +</example_request> +Tool call is addressed to the tool agent, avoid sending more info than tool call itself, while making a call. +When done right, tool call will be delivered to the tool agent. tool agent will respond with the results of the call. +<example_response> +tool: +under the topic: Adam's number is stored: +559-996 +</example_response> After that you are free to respond to the user. ` - systemMsg = toolSysMsg - sysMap = map[string]string{"basic_sys": basicSysMsg, "tool_sys": toolSysMsg} - sysLabels = []string{"cancel", "basic_sys", "tool_sys"} + basicCard = &models.CharCard{ + SysPrompt: basicSysMsg, + FirstMsg: defaultFirstMsg, + Role: "", + FilePath: "", + } + toolCard = &models.CharCard{ + SysPrompt: toolSysMsg, + FirstMsg: defaultFirstMsg, + Role: "", + FilePath: "", + } + // sysMap = map[string]string{"basic_sys": basicSysMsg, "tool_sys": toolSysMsg} + sysMap = map[string]*models.CharCard{"basic_sys": basicCard, "tool_sys": toolCard} + sysLabels = []string{"basic_sys", "tool_sys"} ) /* @@ -59,10 +85,11 @@ also: - others do; */ func memorise(args ...string) []byte { - agent := assistantRole + agent := cfg.AssistantRole if len(args) < 2 { - logger.Warn("not enough args to call memorise tool") - return nil + msg := "not enough args to call memorise tool; need topic and data to remember" + logger.Error(msg) + return []byte(msg) } memory := &models.Memory{ Agent: agent, @@ -70,63 +97,46 @@ func memorise(args ...string) []byte { Mind: args[1], UpdatedAt: time.Now(), } - store.Memorise(memory) - return nil + if _, err := store.Memorise(memory); err != nil { + logger.Error("failed to save memory", "err", err, "memoory", memory) + return []byte("failed to save info") + } + msg := "info saved under the topic:" + args[0] + return []byte(msg) } func recall(args ...string) []byte { - agent := assistantRole + agent := cfg.AssistantRole if len(args) < 1 { logger.Warn("not enough args to call recall tool") return nil } mind, err := store.Recall(agent, args[0]) if err != nil { - logger.Error("failed to use tool", "error", err, "args", args) - return nil + msg := fmt.Sprintf("failed to recall; error: %v; args: %v", err, args) + logger.Error(msg) + return []byte(msg) } - return []byte(mind) + answer := fmt.Sprintf("under the topic: %s is stored:\n%s", args[0], mind) + return []byte(answer) } func recallTopics(args ...string) []byte { - agent := assistantRole + agent := cfg.AssistantRole topics, err := store.RecallTopics(agent) if err != nil { logger.Error("failed to use tool", "error", err, "args", args) return nil } - data, err := json.Marshal(topics) - if err != nil { - logger.Error("failed to use tool", "error", err, "args", args) - return nil - } - return data + joinedS := strings.Join(topics, ";") + return []byte(joinedS) } -func fullMemoryLoad() {} - -// predifine funcs -func getUserDetails(args ...string) []byte { - // db query - // return DB[id[0]] - m := map[string]any{ - "username": "fm11", - "id": 24983, - "reputation": 911, - "balance": 214.73, - } - data, err := json.Marshal(m) - if err != nil { - logger.Error("failed to use tool", "error", err, "args", args) - return nil - } - return data -} +// func fullMemoryLoad() {} type fnSig func(...string) []byte var fnMap = map[string]fnSig{ - "get_id": getUserDetails, "recall": recall, "recall_topics": recallTopics, "memorise": memorise, |