v0 litellm

apiserver/plane/api/views/gpt.py

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

apiserver/requirements/base.txt

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