mirror of
https://github.com/cpacker/MemGPT.git
synced 2025-06-03 04:30:22 +00:00
8736d46526
* First commit of memgpt client and some messy test code * rolled back unnecessary changes to abstract interface; switched client to always use Queueing Interface * Added missing interface clear() in run_command; added convenience method for checking if an agent exists, used that in create_agent * Formatting fixes * Fixed incorrect naming of get_agent_memory in rest server * Removed erroneous clear from client save method; Replaced print statements with appropriate logger calls in server * Updated readme with client usage instructions * added tests for Client * make printing to terminal togglable on queininginterface (should probably refactor this to a logger) * turn off printing to stdout via interface by default * allow importing the python client in a similar fashion to openai-python (see https://github.com/openai/openai-python) * Allowed quickstart on init of client; updated readme and test_client accordingly * oops, fixed name of openai_api_key config key * Fixed small typo * Fixed broken test by adding memgpt hosted model details to agent config * silence llamaindex 'LLM is explicitly disabled. Using MockLLM.' on server * default to openai if user's memgpt directory is empty (first time) * correct type hint * updated section on client in readme * added comment about how MemGPT config != Agent config * patch unrelated test * update wording on readme * patch another unrelated test * added python client to readme docs * Changed 'user' to 'human' in example; Defaulted AgentConfig.model to 'None'; Fixed issue in create_agent (accounting for dict config); matched test code to example * Fixed advanced example * patch test * patch --------- Co-authored-by: cpacker <packercharles@gmail.com>
36 lines
841 B
Python
36 lines
841 B
Python
from memgpt import MemGPT
|
|
from memgpt import constants
|
|
|
|
from .utils import wipe_config
|
|
|
|
|
|
test_agent_id = "test_client_agent"
|
|
client = None
|
|
|
|
|
|
def test_create_agent():
|
|
wipe_config()
|
|
global client
|
|
client = MemGPT(quickstart="memgpt_hosted")
|
|
|
|
agent_id = client.create_agent(
|
|
agent_config={
|
|
"name": test_agent_id,
|
|
"persona": constants.DEFAULT_PERSONA,
|
|
"human": constants.DEFAULT_HUMAN,
|
|
}
|
|
)
|
|
assert agent_id is not None
|
|
return client, agent_id
|
|
|
|
|
|
def test_user_message():
|
|
assert client is not None, "Run create_agent test first"
|
|
response = client.user_message(agent_id=test_agent_id, message="Hello my name is Test, Client Test")
|
|
assert response is not None and len(response) > 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_create_agent()
|
|
test_user_message()
|