aljaz/Auto-GPT
1
0
Fork 0
mirror of https://github.com/aljazceru/Auto-GPT.git synced 2025-12-18 06:24:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

Save redis memory state, with the default being to wipe on start still.

Browse Source
This commit is contained in:
BillSchumacher
2023-04-07 00:48:27 -05:00
parent 5a1d9e6d0a
commit cce79695fa
2 changed files with 11 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
scripts/memory/redismem.py
View File

@@ -44,7 +44,8 @@ class RedisMemory(MemoryProviderSingleton):
password=redis_password,
db=0 # Cannot be changed
)
self.redis.flushall()
if cfg.wipe_redis_on_start:
self.redis.flushall()
try:
self.redis.ft("gpt").create_index(
fields=SCHEMA,
@@ -55,7 +56,9 @@ class RedisMemory(MemoryProviderSingleton):
)
except Exception as e:
print("Error creating Redis search index: ", e)
self.vec_num = 0
existing_vec_num = self.redis.get('vec_num')
self.vec_num = int(existing_vec_num.decode('utf-8')) if\
existing_vec_num else 0
def add(self, data: str) -> str:
"""
@@ -72,10 +75,13 @@ class RedisMemory(MemoryProviderSingleton):
b"data": data,
"embedding": vector
}
self.redis.hset(f"gpt:{self.vec_num}", mapping=data_dict)
pipe = self.redis.pipeline()
pipe.hset(f"gpt:{self.vec_num}", mapping=data_dict)
_text = f"Inserting data into memory at index: {self.vec_num}:\n"\
f"data: {data}"
self.vec_num += 1
pipe.set('vec_num', self.vec_num)
pipe.execute()
return _text
def get(self, data: str) -> Optional[List[Any]]: