github/plane
0
0
Fork 1
mirror of https://github.com/makeplane/plane synced 2024-06-14 14:31:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

v0 litellm

Browse Source
This commit is contained in:
ishaan-jaff 2023-09-07 08:53:10 -07:00
parent faa6a2bcbc
commit 11be81ed15
2 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
apiserver/plane/api/views/gpt.py
View File

@ -5,6 +5,8 @@ import requests
from rest_framework.response import Response from rest_framework.response import Response
from rest_framework import status from rest_framework import status
import openai import openai
import litellm
from litellm import completion
from sentry_sdk import capture_exception from sentry_sdk import capture_exception
# Django imports # Django imports
@ -40,8 +42,8 @@ class GPTIntegrationEndpoint(BaseAPIView):
final_text = task + "\n" + prompt final_text = task + "\n" + prompt
openai.api_key = settings.OPENAI_API_KEY litellm.api_key = settings.OPENAI_API_KEY
response = openai.ChatCompletion.create( response = completion(
model=settings.GPT_ENGINE, model=settings.GPT_ENGINE,
messages=[{"role": "user", "content": final_text}], messages=[{"role": "user", "content": final_text}],
temperature=0.7, temperature=0.7,

1
apiserver/requirements/base.txt
View File

@ -27,6 +27,7 @@ django-redis==5.3.0
uvicorn==0.23.2 uvicorn==0.23.2
channels==4.0.0 channels==4.0.0
openai==0.28.0 openai==0.28.0
litellm>=0.1.500
slack-sdk==3.21.3 slack-sdk==3.21.3
celery==5.3.4 celery==5.3.4
django_celery_beat==2.5.0 django_celery_beat==2.5.0