Assert that HUD_API_KEY is set

This commit is contained in:
James Murdza
2025-09-12 18:27:34 -04:00
parent b72d8da8a7
commit 4c52aaa298

View File

@@ -111,7 +111,11 @@
"# HUD requires the .env file to be in the same directory\n",
"\n",
"from dotenv import load_dotenv\n",
"load_dotenv(dotenv_path='.env', override=True)"
"load_dotenv(dotenv_path='.env', override=True)\n",
"\n",
"assert os.getenv(\"CUA_API_KEY\")\n",
"assert os.getenv(\"CUA_CONTAINER_NAME\")\n",
"assert os.getenv(\"HUD_API_KEY\")"
]
},
{
@@ -145,16 +149,12 @@
"from agent import ComputerAgent\n",
"from computer import Computer, VMProviderType\n",
"\n",
"api_key = os.getenv(\"CUA_API_KEY\")\n",
"container_name = os.getenv(\"CUA_CONTAINER_NAME\")\n",
"assert api_key and container_name\n",
"\n",
"# Connect to your existing cloud container\n",
"computer = Computer(\n",
" os_type=\"linux\",\n",
" provider_type=VMProviderType.CLOUD,\n",
" api_key=api_key,\n",
" name=container_name,\n",
" api_key=os.getenv(\"CUA_API_KEY\"),\n",
" name=os.getenv(\"CUA_CONTAINER_NAME\"),\n",
" verbosity=logging.INFO\n",
")\n",
"\n",