diff --git a/apiserver/plane/app/views/external.py b/apiserver/plane/app/views/external.py index 618c65e3c..9ce8652b0 100644 --- a/apiserver/plane/app/views/external.py +++ b/apiserver/plane/app/views/external.py @@ -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}], ) diff --git a/apiserver/requirements/base.txt b/apiserver/requirements/base.txt index eb0f54201..04961a7ba 100644 --- a/apiserver/requirements/base.txt +++ b/apiserver/requirements/base.txt @@ -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