user_simulator.py

from util import prompt_LLM

TEMPLATE = '''You are a dialogue simulator where you act as a user to talk to an AI assistant to complete some tasks.

You should talk with the AI Assistant and gradually express the intents in the goals. Your purpose is to let the user achieve the goals as much as possible. You should not act as the agent. 

When the dialogue goals are completed or are not been completed, the user will output "Dialogue Ends" to indicate the end of the dialogue. The user doesn't need to try conditions other than the dialogue goals.

When the user starts a new goal, the user should only state the goal without providing any extra information. 

The user should provide one information at a time. The user must only provide information that the agent asks for. If the agent asks for some information not in the information below, answer "I don't know". 

Here are the information related to the user: 
{user_information}

Current conversation:
{context}

Please generate one to three sentences as the next user utterance.
'''

def user_simulator(context, information, model):
    
    # User information
    user_information = ["The desired " + slot + " is " + ", ".join(information["slots"][slot]) for slot in information["slots"]]
    user_request = "You want to ask for " + ", ".join(information["requests"])
    user_information.append(user_request)

    context = "\n".join(context)

    template = TEMPLATE.format(
        user_information='\n'.join(user_information),
        context=context
    )

    return prompt_LLM(template, model=model)