From 11be81ed15168750593dcc8b84770b3b8824d877 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Thu, 7 Sep 2023 08:53:10 -0700 Subject: [PATCH] v0 litellm --- apiserver/plane/api/views/gpt.py | 6 ++++-- apiserver/requirements/base.txt | 1 + 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/apiserver/plane/api/views/gpt.py b/apiserver/plane/api/views/gpt.py index 63c3f4f18..c5fb99c9d 100644 --- a/apiserver/plane/api/views/gpt.py +++ b/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, diff --git a/apiserver/requirements/base.txt b/apiserver/requirements/base.txt index 969ab3c89..6ed167099 100644 --- a/apiserver/requirements/base.txt +++ b/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