From 14d706f94a45ab43c51f9c03abea7b3588c571c7 Mon Sep 17 00:00:00 2001
From: Grail Finder <wohilas@gmail.com>
Date: Wed, 27 Nov 2024 20:16:58 +0300
Subject: Feat: add toml config

---
 tools.go | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'tools.go')

diff --git a/tools.go b/tools.go
index b32d8cb..af4356e 100644
--- a/tools.go
+++ b/tools.go
@@ -61,7 +61,7 @@ also:
 - others do;
 */
 func memorise(args ...string) []byte {
-	agent := assistantRole
+	agent := cfg.AssistantRole
 	if len(args) < 2 {
 		msg := "not enough args to call memorise tool; need topic and data to remember"
 		logger.Error(msg)
@@ -79,7 +79,7 @@ func memorise(args ...string) []byte {
 }
 
 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
@@ -94,7 +94,7 @@ func recall(args ...string) []byte {
 }
 
 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)
-- 
cgit v1.2.3