Googolplexic commited on
Commit
bd9da91
·
1 Parent(s): eec7a66

Refactor API key retrieval to use environment variable directly and improve validation logic

Browse files
Files changed (1) hide show
  1. api_monitor.py +8 -6
api_monitor.py CHANGED
@@ -55,7 +55,7 @@ def verify_mcp_api_key(api_key):
55
  """
56
  try:
57
  # Get the key server URL from environment or use default
58
- key_server_url = os.getenv("KEY_SERVER_URL", "http://localhost:3001")
59
 
60
  response = requests.post(
61
  f"{key_server_url}/api/verifyKey",
@@ -183,11 +183,13 @@ def validate_api_configuration(
183
  try:
184
  # Validate input parameters
185
  if not mcp_api_key or not mcp_api_key.strip():
186
- return {
187
- "success": False,
188
- "message": "MCP API key is required",
189
- "config_id": None,
190
- } # Verify the MCP API key with the key generation server
 
 
191
  key_verification = verify_mcp_api_key(mcp_api_key)
192
  if not key_verification["success"]:
193
  return {
 
55
  """
56
  try:
57
  # Get the key server URL from environment or use default
58
+ key_server_url = os.getenv("KEY_SERVER_URL")
59
 
60
  response = requests.post(
61
  f"{key_server_url}/api/verifyKey",
 
183
  try:
184
  # Validate input parameters
185
  if not mcp_api_key or not mcp_api_key.strip():
186
+ mcp_api_key = os.environ["MCP_API_KEY"]
187
+ if not mcp_api_key or not mcp_api_key.strip():
188
+ return {
189
+ "success": False,
190
+ "message": "MCP API key is required",
191
+ "config_id": None,
192
+ } # Verify the MCP API key with the key generation server
193
  key_verification = verify_mcp_api_key(mcp_api_key)
194
  if not key_verification["success"]:
195
  return {