Skip to content

Commit 2eccc4e

Browse files
committed
Revert "fix openai call to use new api format"
This reverts commit 8d4ca04.
1 parent 020e1b3 commit 2eccc4e

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

app/rag_system.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,12 @@
1-
from openai import OpenAI
1+
import openai
22
import json
33
import os
44
from datetime import date
55
from sentence_transformers import SentenceTransformer
66
import numpy as np
77
from sklearn.metrics.pairwise import cosine_similarity
88

9-
client = OpenAI()
9+
openai.api_key = os.getenv("OPENAI_API_KEY")
1010

1111
class RAGSystem:
1212
def __init__(self, knowledge_base_path='knowledge_base.json'):
@@ -131,7 +131,7 @@ def answer_query_stream(self, query):
131131

132132
messages.append(system_message)
133133

134-
stream = client.chat.completions.create(
134+
stream = openai.ChatCompletion.create(
135135
model="gpt-4-turbo",
136136
messages=messages,
137137
temperature=0.5,
@@ -144,9 +144,9 @@ def answer_query_stream(self, query):
144144

145145
collected_messages = []
146146
for chunk in stream:
147-
if chunk.choices[0].finish_reason is not None:
147+
if chunk['choices'][0]['finish_reason'] is not None:
148148
break
149-
content = chunk.choices[0].delta.content or ''
149+
content = chunk['choices'][0]['delta'].get('content', '')
150150
collected_messages.append(content)
151151
yield content
152152

app/requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,4 +7,4 @@ numpy==1.22.0
77
sentence-transformers==2.1.0
88
torch==1.10.0
99
huggingface_hub==0.8.1
10-
openai==1.0.0
10+
openai==0.28.0

0 commit comments

Comments
 (0)