linting

environments/community/protein_design/utils/api_utils.py

@@ -0,0 +1,26 @@
+import logging
+import os
+from typing import Optional
+
+from dotenv import load_dotenv
+
+load_dotenv()
+logger = logging.getLogger(__name__)
+
+
+def load_api_key() -> Optional[str]:
+    """
+    Load the NVIDIA NIM API key from environment variables.
+
+    Returns:
+        The API key from environment variables, or None if not found
+    """
+    api_key = os.environ.get("NVIDIA_NIM_API_KEY")
+    if not api_key:
+        logger.error(
+            "NVIDIA_NIM_API_KEY not found in environment variables. "
+            "Please set it in your .env file."
+        )
+        return None
+
+    return api_key