File tree Expand file tree Collapse file tree 2 files changed +6
-6
lines changed
Expand file tree Collapse file tree 2 files changed +6
-6
lines changed Original file line number Diff line number Diff line change 1- from openai import OpenAI
1+ import openai
22import json
33import os
44from datetime import date
55from sentence_transformers import SentenceTransformer
66import numpy as np
77from sklearn .metrics .pairwise import cosine_similarity
88
9- client = OpenAI ( )
9+ openai . api_key = os . getenv ( "OPENAI_API_KEY" )
1010
1111class 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
Original file line number Diff line number Diff line change @@ -7,4 +7,4 @@ numpy==1.22.0
77sentence-transformers == 2.1.0
88torch == 1.10.0
99huggingface_hub == 0.8.1
10- openai == 1.0 .0
10+ openai == 0.28 .0
You can’t perform that action at this time.
0 commit comments