forked from github/plane
f9ee898d88
* chore: add workspace details and project details in page responses * fix: typo in workspace queryset
102 lines
3.5 KiB
Python
102 lines
3.5 KiB
Python
# Python imports
|
|
import requests
|
|
|
|
# Third party imports
|
|
from rest_framework.response import Response
|
|
from rest_framework import status
|
|
import openai
|
|
from sentry_sdk import capture_exception
|
|
|
|
# Django imports
|
|
from django.conf import settings
|
|
|
|
# Module imports
|
|
from .base import BaseAPIView
|
|
from plane.api.permissions import ProjectEntityPermission
|
|
from plane.db.models import Workspace, Project
|
|
from plane.api.serializers import ProjectLiteSerializer, WorkspaceLiteSerializer
|
|
|
|
|
|
class GPTIntegrationEndpoint(BaseAPIView):
|
|
permission_classes = [
|
|
ProjectEntityPermission,
|
|
]
|
|
|
|
def post(self, request, slug, project_id):
|
|
try:
|
|
if not settings.OPENAI_API_KEY or not settings.GPT_ENGINE:
|
|
return Response(
|
|
{"error": "OpenAI API key and engine is required"},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
|
|
count = 0
|
|
|
|
# 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,
|
|
)
|
|
count = response.json().get("count", 0)
|
|
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)
|
|
|
|
if not task:
|
|
return Response(
|
|
{"error": "Task is required"}, status=status.HTTP_400_BAD_REQUEST
|
|
)
|
|
|
|
final_text = task + "\n" + prompt
|
|
|
|
openai.api_key = settings.OPENAI_API_KEY
|
|
response = openai.Completion.create(
|
|
engine=settings.GPT_ENGINE,
|
|
prompt=final_text,
|
|
temperature=0.7,
|
|
max_tokens=1024,
|
|
)
|
|
|
|
workspace = Workspace.objects.get(slug=slug)
|
|
project = Project.objects.get(pk=project_id)
|
|
|
|
text = response.choices[0].text.strip()
|
|
text_html = text.replace("\n", "<br/>")
|
|
return Response(
|
|
{
|
|
"response": text,
|
|
"response_html": text_html,
|
|
"count": count,
|
|
"project_detail": ProjectLiteSerializer(project).data,
|
|
"workspace_detail": WorkspaceLiteSerializer(workspace).data,
|
|
},
|
|
status=status.HTTP_200_OK,
|
|
)
|
|
except (Workspace.DoesNotExist, Project.DoesNotExist) as e:
|
|
return Response(
|
|
{"error": "Workspace or Project Does not exist"},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
except Exception as e:
|
|
capture_exception(e)
|
|
return Response(
|
|
{"error": "Something went wrong please try again later"},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|