[fix] allow setting openai base url

USTC-iCourse · Sep 11, 2024 · 1ad6321 · 1ad6321
1 parent 6aafd92
commit 1ad6321
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/app/views/ai/summarize_course.py b/app/views/ai/summarize_course.py
@@ -15,9 +15,15 @@
 
 
 def get_chatgpt_completion(system_prompt, user_prompt):
-    client = OpenAI(
-        api_key=app.config['OPENAI_API_KEY'],
-    )
+    if 'OPENAI_BASE_URL' in app.config:
+        client = OpenAI(
+            base_url=app.config['OPENAI_BASE_URL'],
+            api_key=app.config['OPENAI_API_KEY'],
+        )
+    else:
+        client = OpenAI(
+            api_key=app.config['OPENAI_API_KEY'],
+        )
     response = client.chat.completions.create(
         model="gpt-4o",
         messages=[