Chris4K commited on
Commit
0839158
1 Parent(s): 7f01be6

Update vector_store_retriever.py

Browse files
Files changed (1) hide show
  1. vector_store_retriever.py +2 -3
vector_store_retriever.py CHANGED
@@ -130,8 +130,7 @@ def predict(message, temperature=0.9, max_new_tokens=512, top_p=0.6, repetition_
130
 
131
  res_relevant_doc = [source.metadata['source'] for source in llm_response["source_documents"]]
132
  response = f"{res_result}" + "\n\n" + "[Answer Source Documents (Ctrl + Click!)] :" + "\n" + f" \n {res_relevant_doc}"
133
- print("response: =====> \n", response, "\n\n")
134
-
135
  tokens = response.split('\n')
136
  token_list = []
137
  for idx, token in enumerate(tokens):
@@ -157,7 +156,7 @@ def predict(message, temperature=0.9, max_new_tokens=512, top_p=0.6, repetition_
157
  yield partial_message
158
  time.sleep(0.01)
159
  else:
160
- print(f"[[워닝]] ==> The key 'text' does not exist or is None in this token entry: {token_entry}")
161
  pass
162
 
163
  except KeyError as e:
 
130
 
131
  res_relevant_doc = [source.metadata['source'] for source in llm_response["source_documents"]]
132
  response = f"{res_result}" + "\n\n" + "[Answer Source Documents (Ctrl + Click!)] :" + "\n" + f" \n {res_relevant_doc}"
133
+ print("response: =====>\n",response,"\n\n")
 
134
  tokens = response.split('\n')
135
  token_list = []
136
  for idx, token in enumerate(tokens):
 
156
  yield partial_message
157
  time.sleep(0.01)
158
  else:
159
+ print(f"Warning ==> The key 'text' does not exist or is None in this token entry: {token_entry}")
160
  pass
161
 
162
  except KeyError as e: