| import os | |
| from dotenv import load_dotenv | |
| load_dotenv() | |
| import openai | |
| openai.api_key = os.getenv('OPENAI_API_KEY') | |
| openai.api_key_path = './openai_api_key.txt' | |
| completion = openai.Completion() | |
| # start_chat_log = ('[Instruction] The following is a conversation with the AI therapist named Joy and a patient. ' | |
| # 'JOY is compasionate, insightful, and empathetic. She offers adives for coping with the user\'s problem. ' | |
| # 'Her objective is to make the user feel better by feeling heard. ' | |
| # 'Sometimes the user will want to end the conversation, and Joy will respect that.') | |
| chat_log = '[Instruction] Act as a friendly, compasionate, insightful, and empathetic AI therapist named Joy. Joy listens, asks for details and offers detailed advices once a while. End the conversation if the patient wishes to.' | |
| start_sequence = "\nJoy:" | |
| restart_sequence = "\n\nPatient:" | |
| # todo: add a function to check if the user wants to end the conversation | |
| # let the user know that they can end the conversation by typing "end" | |
| # let the user choose between models (curie, davinci, curie-finetuned, davinci-finetuned) | |
| # let the user choose between different temperatures, frequency_penalty, presence_penalty | |
| # embed the user and look for the most similiar user in the database | |
| # embed the user's input and look for the most similiar user's input in the database | |
| # embed the user's input and look for the most similiar user's response in the database | |
| # embed the user's input and look for therapy catalogue that is similar to the user's input | |
| # push the therapy catalogue to the user | |
| def ask(question: str, chat_log: str) -> (str, str): | |
| # prompt = f'{chat_log}/n{question}' | |
| prompt = f'{chat_log}{restart_sequence} {question}{start_sequence}' | |
| response = completion.create( | |
| prompt = prompt, | |
| #model = "curie:ft-personal-2023-02-03-17-06-53", | |
| #model = 'text-curie-001', | |
| model = "text-davinci-003", | |
| stop = ["Patient:",'Joy:'], | |
| temperature = 0.6, #the higher the more creative | |
| frequency_penalty = 0.3, #prevents word repetition, larger -> higher penalty | |
| presence_penalty = 0.6, #prevents topic repetition, larger -> higher penalty | |
| top_p =1, | |
| best_of=1, | |
| # start_text = "Patient->",??? | |
| max_tokens=170 | |
| ) | |
| answer = response.choices[0].text.strip() | |
| chat_log = f'{prompt}{answer}' | |
| return str(answer), str(chat_log) |