HengJay commited on
Commit
211ed9a
·
1 Parent(s): 82d0f3a

Remove entity list showing for debug.

Browse files
Files changed (1) hide show
  1. SNOMED-CT_Assistant.py +0 -2
SNOMED-CT_Assistant.py CHANGED
@@ -122,11 +122,9 @@ def chat_input(prompt, med_text):
122
  model=model_tag, response_format={ "type": "json_object" }, messages=st.session_state.messages, temperature=0.5)
123
  msg = entity_identification_response.choices[0].message.content
124
  entity_list = json.loads(msg)["identified_entity"]
125
- print("entity list: ", entity_list)
126
  st.session_state.messages.append({"role": "assistant", "content": msg})
127
  st.chat_message("assistant").write(msg)
128
  for entity in entity_list:
129
- print("entity: ", entity)
130
  results = query_chroma_db(entity, 10)
131
  results_dict = get_dict_from_chroma_results(results)
132
  results_table = entity_mapping_result_to_table(entity, results_dict)
 
122
  model=model_tag, response_format={ "type": "json_object" }, messages=st.session_state.messages, temperature=0.5)
123
  msg = entity_identification_response.choices[0].message.content
124
  entity_list = json.loads(msg)["identified_entity"]
 
125
  st.session_state.messages.append({"role": "assistant", "content": msg})
126
  st.chat_message("assistant").write(msg)
127
  for entity in entity_list:
 
128
  results = query_chroma_db(entity, 10)
129
  results_dict = get_dict_from_chroma_results(results)
130
  results_table = entity_mapping_result_to_table(entity, results_dict)