v0 litellm

openedx · Sep 5, 2023 · d1b09f6 · d1b09f6
1 parent 2c39d13
commit d1b09f6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/requirements/base.in b/requirements/base.in
@@ -14,4 +14,5 @@ django-choices
 django-filter
 openedx-events
 openai
+litellm
 pandas
diff --git a/taxonomy/openai/client.py b/taxonomy/openai/client.py
@@ -1,10 +1,10 @@
 """openai client"""
 
-import openai
+import litellm
 
 from django.conf import settings
 
-openai.api_key = settings.OPENAI_API_KEY
+litellm.api_key = settings.OPENAI_API_KEY
 
 
 def chat_completion(prompt):
@@ -14,7 +14,7 @@ def chat_completion(prompt):
     Arguments:
         prompt (str): chatGPT prompt
     """
-    response = openai.ChatCompletion.create(
+    response = litellm.completion(
         model="gpt-3.5-turbo",
         messages=[
             {"role": "user", "content": prompt},
-Original file line number
+Diff line change
@@ Expand Up / @@ -14,4 +14,5 @@ django-choices @@
     django-filter
     openedx-events
     openai
+    litellm
     pandas