alihan/MemGPT-llm
1
0
Fork 0
mirror of https://github.com/cpacker/MemGPT.git synced 2023-10-17 01:28:22 +03:00
Code Issues Packages Projects Releases Wiki Activity

support .txt extension

Browse Source
This commit is contained in:
Charles Packer
2023-10-13 01:34:33 -07:00
parent 2fb1a1edbc
commit fd8deaba84
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
memgpt/humans/humans.py
View File

@@ -3,7 +3,7 @@ import os
DEFAULT = 'cs_phd'
def get_human_text(key=DEFAULT):
filename = f'{key}.txt'
filename = key if key.endswith('.txt') else f'{key}.txt'
file_path = os.path.join(os.path.dirname(__file__), 'examples', filename)
if os.path.exists(file_path):

4
memgpt/personas/personas.py
View File

@@ -4,9 +4,9 @@ DEFAULT = 'sam'
def get_persona_text(key=DEFAULT):
filename = f'{key}.txt'
filename = key if key.endswith('.txt') else f'{key}.txt'
file_path = os.path.join(os.path.dirname(__file__), 'examples', filename)
if os.path.exists(file_path):
with open(file_path, 'r') as file:
return file.read().strip()