Fix config tests (#343)

Co-authored-by: Vivian Fang <hi@vivi.sh>

.github/workflows/tests.yml

@@ -24,14 +24,13 @@ jobs:
     steps:
     - uses: actions/checkout@v1
 
-    - name: Install poetry
-      run: pipx install poetry
-
     - name: Set up Python
       uses: actions/setup-python@v4
       with:
         python-version: "3.10"
-        cache: "poetry"
+
+    - name: Install poetry
+      run: pip install poetry
 
     - name: Set Poetry config
       run: |

memgpt/cli/cli_config.py

@@ -76,18 +76,11 @@ def configure():
         endpoint_options += ["openai"]
 
     assert len(endpoint_options) > 0, "No endpoints found. Please enable OpenAI, Azure, or set OPENAI_API_BASE."
-    if len(endpoint_options) == 1:
-        default_endpoint = endpoint_options[0]
-    else:
-        default_endpoint = questionary.select("Select default inference endpoint:", endpoint_options).ask()
+    default_endpoint = questionary.select("Select default inference endpoint:", endpoint_options).ask()
 
     # configure embedding provider
     endpoint_options.append("local")  # can compute embeddings locally
-    if len(endpoint_options) == 1:
-        default_embedding_endpoint = endpoint_options[0]
-        print(f"Using embedding endpoint {default_embedding_endpoint}")
-    else:
-        default_embedding_endpoint = questionary.select("Select default embedding endpoint:", endpoint_options).ask()
+    default_embedding_endpoint = questionary.select("Select default embedding endpoint:", endpoint_options).ask()
 
     # configure embedding dimentions
     default_embedding_dim = 1536

tests/test_cli.py

@@ -22,7 +22,6 @@ def test_save_load():
     child.expect("Empty input received. Try again!", timeout=TIMEOUT)
     child.sendline("/save")
 
-    child.expect("Saved local", timeout=TIMEOUT)
     child.expect("Enter your message:", timeout=TIMEOUT)
     child.sendline("/exit")
 

tests/utils.py

@@ -18,6 +18,12 @@ def configure_memgpt(enable_openai=True, enable_azure=False):
     else:
         child.sendline("n")
 
+    child.expect("Select default inference endpoint:", timeout=TIMEOUT)
+    child.sendline()
+
+    child.expect("Select default embedding endpoint:", timeout=TIMEOUT)
+    child.sendline()
+
     child.expect("Select default preset:", timeout=TIMEOUT)
     child.sendline()