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

View File

@ -20,6 +20,7 @@ django-redis==5.3.0
uvicorn==0.23.2
channels==4.0.0
openai==1.2.4
litellm==1.27.14
slack-sdk==3.21.3
celery==5.3.4
django_celery_beat==2.5.0