Update helper.py

helper.py

@@ -1,13 +1,14 @@
 import os
 from dotenv import load_dotenv, find_dotenv
-import json
 
-# Load environment variables
 def load_env():
+    """Load environment variables from the .env file."""
     _ = load_dotenv(find_dotenv())
 
-# Load the Together API key
-def get_together_api_key():
-     load_env()
-     together_api_key = os.getenv("TOGETHER_API_KEY")
-     return together_api_key
+def get_together_api_key() -> str:
+    """Retrieve the Together API key from the environment variables."""
+    load_env()
+    api_key = os.getenv("TOGETHER_API_KEY")
+    if not api_key:
+        raise ValueError("TOGETHER_API_KEY is not set in the environment variables.")
+    return api_key