from uptrain import EvalLLM, ConversationNumberOfTurns
OPENAI_API_KEY = "sk-********************" # Insert your OpenAI key here
data = [{
'conversation' : [
{"role": "patient", "content": "Hello"},
{"role": "nurse", "content": "Hello, how can I help you"},
{"role": "patient", "content": "I am feeling very sick"},
{"role": "nurse", "content": "What are your symptoms"},
{"role": "patient", "content": "I have a headache"},
{"role": "nurse", "content": "Have you taken any medication"},
{"role": "patient", "content": "No, I haven't"},
{"role": "nurse", "content": "Please take some paracetamol"},
{"role": "patient", "content": "Thank you nurse"},
]
}]
eval_llm = EvalLLM(openai_api_key=OPENAI_API_KEY)
res = eval_llm.evaluate(
data=data,
checks=[ConversationNumberOfTurns(user_persona="patient", llm_persona="nurse")],
)