mirror of
https://github.com/cpacker/MemGPT.git
synced 2025-06-03 04:30:22 +00:00
b83d1c599f
* init server refactor * refactored websockets server/client code to use internal server API * added intentional fail on test * update workflow to try and get test to pass remotely * refactor to put websocket code in a separate subdirectory * added fastapi rest server * add error handling * modified interface return style * disabled certain tests on remote * added SSE response option for user_message * fix ws interface test * fallback for oai key * add soft fail for test when localhost is borked * add step_yield for all server related interfaces * extra catch * update toml + lock with server add-ons (add uvicorn+fastapi, move websockets to server extra) * regen lock file * added pytest-asyncio as an extra in dev * add pydantic to deps * renamed CreateConfig to CreateAgentConfig * fixed POST request for creating agent + tested it
44 lines
903 B
Python
44 lines
903 B
Python
import memgpt.utils as utils
|
|
|
|
utils.DEBUG = True
|
|
from memgpt.server.server import SyncServer
|
|
|
|
|
|
def test_server():
|
|
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:
|
|
print(e)
|
|
except:
|
|
raise
|
|
|
|
try:
|
|
server.user_message(user_id=user_id, agent_id=agent_id, message="/memory")
|
|
except ValueError as e:
|
|
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
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_server()
|