Remove entity list showing for debug.
Browse files- 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)
|