Merge branch 'develop' of github.com:ishaan-jaff/plane into dev/litellm

This commit is contained in:
pablohashescobar 2024-02-28 18:43:42 +05:30
commit 581206d701
2 changed files with 5 additions and 5 deletions

View File

@ -3,7 +3,8 @@ import requests
import os import os
# Third party imports # Third party imports
from openai import OpenAI import litellm
from litellm import completion
from rest_framework.response import Response from rest_framework.response import Response
from rest_framework import status from rest_framework import status
@ -60,11 +61,9 @@ class GPTIntegrationEndpoint(BaseAPIView):
final_text = task + "\n" + prompt final_text = task + "\n" + prompt
client = OpenAI( litellm.api_key = OPENAI_API_KEY
api_key=OPENAI_API_KEY,
)
response = client.chat.completions.create( response = completion(
model=GPT_ENGINE, model=GPT_ENGINE,
messages=[{"role": "user", "content": final_text}], messages=[{"role": "user", "content": final_text}],
) )

View File

@ -20,6 +20,7 @@ django-redis==5.3.0
uvicorn==0.23.2 uvicorn==0.23.2
channels==4.0.0 channels==4.0.0
openai==1.2.4 openai==1.2.4
litellm==1.27.14
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