refactor!: Migrate users + agent information into storage connectors (#785)
Co-authored-by: cpacker <packercharles@gmail.com>
This commit is contained in:
@@ -2,41 +2,45 @@ import memgpt.utils as utils
|
||||
|
||||
utils.DEBUG = True
|
||||
from memgpt.server.server import SyncServer
|
||||
from .utils import wipe_config, wipe_memgpt_home
|
||||
|
||||
|
||||
def test_server():
|
||||
wipe_memgpt_home()
|
||||
|
||||
user_id = "NULL"
|
||||
agent_id = "agent_26"
|
||||
|
||||
server = SyncServer()
|
||||
|
||||
try:
|
||||
server.user_message(user_id=user_id, agent_id=agent_id, message="Hello?")
|
||||
except ValueError as e:
|
||||
server.user_message(user_id=user_id, agent_id="agent no exist", message="Hello?")
|
||||
raise Exception("user_message call should have failed")
|
||||
except (KeyError, ValueError) as e:
|
||||
# Error is expected
|
||||
print(e)
|
||||
except:
|
||||
raise
|
||||
|
||||
agent_state = server.create_agent(
|
||||
user_id=user_id,
|
||||
agent_config=dict(
|
||||
preset="memgpt_chat",
|
||||
human="cs_phd",
|
||||
persona="sam_pov",
|
||||
),
|
||||
)
|
||||
print(f"Created agent\n{agent_state}")
|
||||
|
||||
try:
|
||||
server.user_message(user_id=user_id, agent_id=agent_id, message="/memory")
|
||||
server.user_message(user_id=user_id, agent_id=agent_state.id, message="/memory")
|
||||
raise Exception("user_message call should have failed")
|
||||
except ValueError as e:
|
||||
# Error is expected
|
||||
print(e)
|
||||
except:
|
||||
raise
|
||||
|
||||
try:
|
||||
print(server.run_command(user_id=user_id, agent_id=agent_id, command="/memory"))
|
||||
except ValueError as e:
|
||||
print(e)
|
||||
except:
|
||||
raise
|
||||
|
||||
try:
|
||||
server.user_message(user_id=user_id, agent_id="agent no-exist", message="Hello?")
|
||||
except ValueError as e:
|
||||
print(e)
|
||||
except:
|
||||
raise
|
||||
print(server.run_command(user_id=user_id, agent_id=agent_state.id, command="/memory"))
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
Reference in New Issue
Block a user