binarybeach/bb-plane-fork
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: add API limit for AI assistance (#592)

Browse source
This commit is contained in:
pablohashescobar 2023-03-30 01:23:56 +05:30 • committed by GitHub
parent 5c1607f39b
commit d7ed237f78
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 38 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

26
apiserver/plane/api/views/gpt.py
View file

@ -1,3 +1,6 @@
# Python imports
import requests
# Third party imports
from rest_framework.response import Response
from rest_framework import status
@ -25,6 +28,28 @@ class GPTIntegrationEndpoint(BaseAPIView):
status=status.HTTP_400_BAD_REQUEST,
)
# If logger is enabled check for request limit
if settings.LOGGER_BASE_URL:
try:
headers = {
"Content-Type": "application/json",
}
response = requests.post(
settings.LOGGER_BASE_URL,
json={"user_id": str(request.user.id)},
headers=headers,
)
if not response.json().get("success", False):
return Response(
{
"error": "You have surpassed the monthly limit for AI assistance"
},
status=status.HTTP_429_TOO_MANY_REQUESTS,
)
except Exception as e:
capture_exception(e)
prompt = request.data.get("prompt", False)
task = request.data.get("task", False)
@ -50,6 +75,7 @@ class GPTIntegrationEndpoint(BaseAPIView):
status=status.HTTP_200_OK,
)
except Exception as e:
print(e)
capture_exception(e)
return Response(
{"error": "Something went wrong please try again later"},