From d0ef339a4a59e20719ef26288aba624871236ac5 Mon Sep 17 00:00:00 2001
From: cpacker <packercharles@gmail.com>
Date: Tue, 2 Jan 2024 22:18:17 -0800
Subject: [PATCH] use --all-extras in poetry install for tests

---
 .github/workflows/tests.yml | 2 +-
 tests/utils.py              | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index c98b1507..d64e3b0b 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -47,7 +47,7 @@ jobs:
 
         OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
       run: |
-        poetry install -E dev -E postgres -E server
+        poetry install --all-extras
 
     - name: Set Poetry config
       env:
diff --git a/tests/utils.py b/tests/utils.py
index b80faa38..c08c0088 100644
--- a/tests/utils.py
+++ b/tests/utils.py
@@ -1,5 +1,4 @@
 import os
-import pexpect
 
 from memgpt.config import MemGPTConfig
 
@@ -18,6 +17,8 @@ def wipe_config():
 
 
 def configure_memgpt_localllm():
+    import pexpect
+
     wipe_config()
     child = pexpect.spawn("memgpt configure")