diff --git a/backend/access_token.py b/backend/access_token.py index e61ef12..35fa657 100644 --- a/backend/access_token.py +++ b/backend/access_token.py @@ -18,10 +18,4 @@ async def validate_access_token(access_code: str): response = await client.post(url, content=data, headers=headers) response_data = response.json() - - if response_data["success"]: - print("Access token is valid.") - return True - else: - print(f"Access token validation failed: {response_data['failure_reason']}") - return False + return response_data diff --git a/backend/main.py b/backend/main.py index bd38807..cb8242a 100644 --- a/backend/main.py +++ b/backend/main.py @@ -97,13 +97,14 @@ async def stream_code(websocket: WebSocket): openai_api_key = None if "accessCode" in params and params["accessCode"]: print("Access code - using platform API key") - if await validate_access_token(params["accessCode"]): + res = await validate_access_token(params["accessCode"]) + if res["success"]: openai_api_key = os.environ.get("PLATFORM_OPENAI_API_KEY") else: await websocket.send_json( { "type": "error", - "value": "Invalid access code or you're out of credits. Please try again.", + "value": res["failure_reason"], } ) return