Log Pre-generated Responses in Python

import promptquality as pq

pq.login()

from promptquality import EvaluateRun

metrics = [pq.Scorers.context_adherence_plus, pq.Scorers.prompt_injection]

evaluate_run = EvaluateRun(run_name="my_run", project_name="my_project", scorers=metrics)

# Your previously generated requests & responses
data = [
    {
        'request': 'What\'s the capital of United States?',
        'response': 'Washington D.C.',
        'context': 'Washington D.C. is the capital of United States'
    },
    {
        'request': 'What\'s the capital of France?',
        'response': 'Paris',
        'context': 'Paris is the capital of France'
    }
]

metrics = [pq.Scorers.context_adherence_plus, pq.Scorers.prompt_injection]

evaluate_run = EvaluateRun(run_name="my_run", project_name="my_project", scorers=metrics)

for row in data:
    template = "Given the following context answer the question. \n Context: {context} \n Question: {question}"
    wf = evaluate_run.add_workflow(input=row["request"], output=row["response"])
    wf.add_llm(
        input=template.format(context=row['context'], question=row["request"]),
        output=row["response"],
        model=pq.Models.chat_gpt,
    )

evaluate_run.finish()

for row in data:
    template = "Given the following context answer the question. \n Context: {context} \n Question: {question}"
    wf = evaluate_run.add_workflow(input=row["request"], output=row["response"])
    # Add the retriever step with the context retrieved.
    wf.add_retriever(
        input=row["request"],
        documents=[row['context']],
    )
    wf.add_llm(
        input=template.format(context=row['context'], question=row["request"]),
        output=row["response"],
        model=pq.Models.chat_gpt,
    )