def get_prompt()

in prompts/stories/filter_openhermes.py [0:0]

• 10 lines of code
• 3 McCabe index (conditional complexity)

def get_prompt(x):
    conversations = x["conversations"]
    prompt = ""
    for i in range(len(conversations)):
        if conversations[i]["from"] == "human":
            prompt += conversations[i]["value"] + "\n"
            assert conversations[i+1]["from"] == "gpt", f"role is {conversations[i+1]['from']} not 'gpt'!"
            prompt += conversations[i+1]["value"] 
            break
    return {"prompt": prompt}