Update rxnim.py

rxnim.py

@@ -11,16 +11,12 @@ from getReaction import get_reaction
 
 
 class RXNIM:
-    def __init__(self, api_key_file='api_key.txt', api_version='2024-06-01', azure_endpoint='https://hkust.azure-api.net'):
-        # Read API key
-        with open(api_key_file, 'r') as api_key_file_handle:
-            self.API_KEY = api_key_file_handle.read().strip()
-
-    # def __init__(self, api_version='2024-06-01', azure_endpoint='https://hkust.azure-api.net'):
-    #     # 从环境变量读取 API Key
-    #     self.API_KEY = os.environ.get('key')
-    #     if not self.API_KEY:
-    #         raise ValueError("Environment variable 'KEY' not set.")
+
+    def __init__(self, api_version='2024-06-01', azure_endpoint='https://hkust.azure-api.net'):
+        # 从环境变量读取 API Key
+        self.API_KEY = os.environ.get('key')
+        if not self.API_KEY:
+            raise ValueError("Environment variable 'KEY' not set.")
 
         # Set up client
         self.client = AzureOpenAI(