From 1d8f6641a2ea7e43b913b7068330cff94a065a3f Mon Sep 17 00:00:00 2001 From: Abi Raja Date: Tue, 19 Mar 2024 13:38:09 -0400 Subject: [PATCH] do not allow non-subscribers to use Claude --- backend/routes/generate_code.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/backend/routes/generate_code.py b/backend/routes/generate_code.py index 578b79a..9a1b2c0 100644 --- a/backend/routes/generate_code.py +++ b/backend/routes/generate_code.py @@ -286,6 +286,13 @@ async def stream_code(websocket: WebSocket): ) raise Exception("No Anthropic key") + # Do not allow non-subscribers to use Claude + if payment_method != PaymentMethod.SUBSCRIPTION: + await throw_error( + "Please subscribe to a paid plan to use the Claude models" + ) + raise Exception("Not subscribed to a paid plan for Claude") + completion = await stream_claude_response( prompt_messages, # type: ignore api_key=ANTHROPIC_API_KEY,