fix openai call to use new api format

Author: commit111

app/rag_system.py

@@ -1,12 +1,12 @@
-import openai
+from openai import OpenAI
 import json
 import os
 from datetime import date
 from sentence_transformers import SentenceTransformer
 import numpy as np
 from sklearn.metrics.pairwise import cosine_similarity
 
-openai.api_key = os.getenv("OPENAI_API_KEY")
+client = OpenAI()
 
 class RAGSystem:
     def __init__(self, knowledge_base_path='knowledge_base.json'):
@@ -132,7 +132,7 @@ def answer_query_stream(self, query):
 
             messages.append(system_message)
 
-            stream = openai.ChatCompletion.create(
+            stream = client.chat.completions.create(
                 model="gpt-4-turbo",
                 messages=messages,
                 temperature=0.5,
@@ -145,9 +145,9 @@ def answer_query_stream(self, query):
 
             collected_messages = []
             for chunk in stream:
-                if chunk['choices'][0]['finish_reason'] is not None:
+                if chunk.choices[0].finish_reason is not None:
                     break
-                content = chunk['choices'][0]['delta'].get('content', '')
+                content = chunk.choices[0].delta.content or ''
                 collected_messages.append(content)
                 yield content
 

app/requirements.txt

@@ -7,4 +7,4 @@ numpy==1.22.0
 sentence-transformers==2.1.0
 torch==1.10.0
 huggingface_hub==0.8.1
-openai==0.28.0
+openai==1.0.0