mtyrrell commited on
Commit
b88b103
·
verified ·
1 Parent(s): 6e9a62d

Update appStore/rag.py

Browse files
Files changed (1) hide show
  1. appStore/rag.py +1 -1
appStore/rag.py CHANGED
@@ -218,7 +218,7 @@ def run_query(context, label):
218
  # iterate through the streamed output
219
  report = []
220
  res_box = st.empty()
221
- for chunk in client.chat_completion(messages, max_tokens=10, stream=True)::
222
  # extract the object containing the text (totally different structure when streaming)
223
  chunk_message = chunk['choices'][0]['delta']
224
  # test to make sure there is text in the object (some don't have)
 
218
  # iterate through the streamed output
219
  report = []
220
  res_box = st.empty()
221
+ for chunk in client.chat_completion(messages, stream=True)::
222
  # extract the object containing the text (totally different structure when streaming)
223
  chunk_message = chunk['choices'][0]['delta']
224
  # test to make sure there is text in the object (some don't have)