59 lines
2.1 KiB
Python
59 lines
2.1 KiB
Python
import os
|
|
from dotenv import load_dotenv
|
|
from src.session import SessionState
|
|
from src.case_info import format_to_report
|
|
import ast
|
|
# import os
|
|
# from dotenv import load_dotenv
|
|
# from session import SessionState
|
|
# from case_info import format_to_report
|
|
class Consultation:
|
|
def __init__(self):
|
|
# initialize one patient and session state
|
|
self.session_map = {}
|
|
|
|
def init_session(self, session_id, case_data):
|
|
load_dotenv()
|
|
self.session_map[session_id] = SessionState(case_data)
|
|
|
|
def qa_chat(self, session_id, question=None):
|
|
session_state = self.session_map[session_id]
|
|
print("22222self.session_map",self.session_map)
|
|
|
|
if_overall = '总体评估' in session_state.option if session_state.option else False
|
|
session_state.process_query_task(question, overall=if_overall)
|
|
print("1111111session_state",vars(session_state))
|
|
cur_card = [session_state.info, session_state.option, session_state.agent_names[session_state.agent_order]]
|
|
print("3333cur_card",cur_card)
|
|
if not session_state.info or session_state.info['status'] == 'success':
|
|
# update recorded information and update agent's knowledge
|
|
session_state.update()
|
|
|
|
# choose agent and option
|
|
session_state.choose_agent_and_option_task()
|
|
if session_state.option == '无':
|
|
session_state.choose_agent_and_option_task()
|
|
|
|
elif session_state.info['status'] == 'need_clarification':
|
|
session_state.missing_info = session_state.info['missing_info']
|
|
|
|
# doctor asks the question
|
|
stream_state = session_state.doctor_state_task()
|
|
if session_state.if_end():
|
|
pass
|
|
|
|
self.session_map[session_id] = session_state
|
|
return cur_card, stream_state
|
|
|
|
def save_result(self, session_id):
|
|
return self.session_map[session_id].patient.recorded_info
|
|
|
|
def format_report(self, session_id):
|
|
return format_to_report(self.session_map[session_id].patient.recorded_info)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
print(os.getenv('CASE'))
|
|
consultation = Consultation()
|
|
consultation.qa_chat()
|