Assert Cua API keys exist in notebook

This commit is contained in:
James Murdza
2025-09-12 12:55:22 -04:00
parent 1aca043006
commit 68ecdcc99a

View File

@@ -106,18 +106,21 @@
"outputs": [],
"source": [
"import logging\n",
"from pathlib import Path\n",
"import os\n",
"\n",
"from pathlib import Path\n",
"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=os.getenv(\"CUA_API_KEY\"),\n",
" name=os.getenv(\"CUA_CONTAINER_NAME\"),\n",
" api_key=api_key,\n",
" name=container_name,\n",
" verbosity=logging.INFO\n",
")\n",
"\n",