ryanrwatkins commited on
Commit
6ca1133
Β·
1 Parent(s): a17cb7a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -123,10 +123,9 @@ def submit_message(prompt, prompt_template, temperature, max_tokens, context_len
123
 
124
  # VectorDBQA.from_chain_type(llm=OpenAI(), chain_type="stuff", vectorstore=docsearch, return_source_documents=True)
125
  # https://colab.research.google.com/drive/1dzdNDZyofRB0f2KIB4gHXmIza7ehMX30?usp=sharing#scrollTo=b-ejDn_JfpWW
126
- history.append("test")
127
  history.append(prompt_msg)
128
- history.append(completion.message.to_dict())
129
- #history.append(completion.choices[0].message.to_dict())
130
  #history.append(completion["result"].choices[0].message.to_dict())
131
 
132
  state['total_tokens'] += completion['usage']['total_tokens']
 
123
 
124
  # VectorDBQA.from_chain_type(llm=OpenAI(), chain_type="stuff", vectorstore=docsearch, return_source_documents=True)
125
  # https://colab.research.google.com/drive/1dzdNDZyofRB0f2KIB4gHXmIza7ehMX30?usp=sharing#scrollTo=b-ejDn_JfpWW
 
126
  history.append(prompt_msg)
127
+ #history.append(completion.message.to_dict())
128
+ history.append(completion.choices[0].message.to_dict())
129
  #history.append(completion["result"].choices[0].message.to_dict())
130
 
131
  state['total_tokens'] += completion['usage']['total_tokens']