add errors to make sure envs set correctly (#418)

letta-ai · Nov 10, 2023 · 3b3e8b4 · 3b3e8b4
1 parent e0455b7
commit 3b3e8b4
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/memgpt/cli/cli.py b/memgpt/cli/cli.py
@@ -170,6 +170,16 @@ def run(
     if config.model_endpoint == "azure":
         configure_azure_support()
 
+    # TODO: remove once model calling logic is cleaner
+    if memgpt_agent.model != "local":
+        assert (
+            os.getenv("OPENAI_API_BASE") is None and os.getenv("BACKEND_TYPE") is None
+        ), f"Please make sure your enviornment variables OPENAI_API_BASE and BACKEND_TYPE are unset"
+    else:
+        assert os.getenv("OPENAI_API_BASE") and os.getenv(
+            "BACKEND_TYPE"
+        ), f"Please make sure your enviornment variables OPENAI_API_BASE and BACKEND_TYPE are set to run a model with a local endpoint"
+
     run_agent_loop(memgpt_agent, first, no_verify, config)  # TODO: add back no_verify