From 45e6d5793d46e188699c497e1d4cc73063fdcb72 Mon Sep 17 00:00:00 2001 From: Sarah Wooders Date: Wed, 3 Jan 2024 19:01:27 -0800 Subject: [PATCH] Modify tests to use openai quickstart --- tests/test_base_functions.py | 2 +- tests/test_client.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_base_functions.py b/tests/test_base_functions.py index d2a3f695..8e30761b 100644 --- a/tests/test_base_functions.py +++ b/tests/test_base_functions.py @@ -14,7 +14,7 @@ def create_test_agent(): """Create a test agent that we can call functions on""" wipe_config() global client - client = MemGPT(quickstart="memgpt_hosted") + client = MemGPT(quickstart="openai") agent_id = client.create_agent( agent_config={ diff --git a/tests/test_client.py b/tests/test_client.py index 1e969fdc..5ef097dc 100644 --- a/tests/test_client.py +++ b/tests/test_client.py @@ -11,7 +11,7 @@ client = None def test_create_agent(): wipe_config() global client - client = MemGPT(quickstart="memgpt_hosted") + client = MemGPT(quickstart="openai") agent_id = client.create_agent( agent_config={