Add check for missing environment variable AICI_API_BASE

microsoft · Feb 2, 2024 · 09da6d2 · 09da6d2
1 parent f09a86b
commit 09da6d2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pyaici/cli.py b/pyaici/cli.py
@@ -340,6 +340,9 @@ def main_inner():
         runner_from_cli(args).bench()
         return
 
+    if os.environ.get(rest.BASE_URL_ENV, None) is None:
+        cli_error(f"environment variable {rest.BASE_URL_ENV} not set")
+
     if args.all_prefixes:
         prefixes = rest.detect_prefixes()
         if len(prefixes) <= 1: