Add long term memory

2024-12-17 19:53:07 +01:00
parent d053dc249c
commit 2561f2f175
6 changed files with 104 additions and 47 deletions
--- a/agent/options.go
+++ b/agent/options.go
@@ -14,12 +14,12 @@ type llmOptions struct {
 }

 type options struct {
-	LLMAPI                                            llmOptions
-	character                                         Character
-	randomIdentityGuidance                            string
-	randomIdentity                                    bool
-	userActions                                       Actions
-	enableHUD, standaloneJob, showCharacter, enableKB bool
+	LLMAPI                                                                                       llmOptions
+	character                                                                                    Character
+	randomIdentityGuidance                                                                       string
+	randomIdentity                                                                               bool
+	userActions                                                                                  Actions
+	enableHUD, standaloneJob, showCharacter, enableKB, enableSummaryMemory, enableLongTermMemory bool

 	canStopItself         bool
 	initiateConversations bool
@@ -127,6 +127,16 @@ var EnablePersonality = func(o *options) error {
 	return nil
 }

+var EnableSummaryMemory = func(o *options) error {
+	o.enableSummaryMemory = true
+	return nil
+}
+
+var EnableLongTermMemory = func(o *options) error {
+	o.enableLongTermMemory = true
+	return nil
+}
+
 func WithRAGDB(db RAGDB) Option {
 	return func(o *options) error {
 		o.ragdb = db