Refresh API Key Store on API Key Creation

.gitignore

@@ -181,6 +181,5 @@ core
 app.config.js
 wandb
 .vscode
-api_keys.json
-prompting/api/api_keys.json
+**/api_keys.json
 weights.csv

validator_api/api_management.py

@@ -45,9 +45,11 @@ def validate_api_key(api_key: str = Header(...)):
 @router.post("/create-api-key/")
 def create_api_key(rate_limit: int, admin_key: str = Depends(validate_admin_key)):
     """Creates a new API key with a specified rate limit."""
+    global _keys
     new_api_key = secrets.token_hex(16)
     _keys[new_api_key] = {"rate_limit": rate_limit, "usage": 0}
     save_api_keys(_keys)
+    _keys = load_api_keys()
     return {"message": "API key created", "api_key": new_api_key}
 
 

validator_api/gpt_endpoints.py

@@ -10,16 +10,13 @@
 from shared.epistula import SynapseStreamResult, query_miners
 from shared.settings import shared_settings
 from shared.uids import get_uids
+from validator_api.api_management import _keys
 from validator_api.chat_completion import chat_completion
 from validator_api.mixture_of_miners import mixture_of_miners
 from validator_api.utils import forward_response
 
 router = APIRouter()
 
-# load api keys from api_keys.json
-with open("api_keys.json", "r") as f:
-    _keys = json.load(f)
-
 
 def validate_api_key(api_key: str = Header(...)):
     if api_key not in _keys: