from llm_expect import llm_expect

@llm_expect(dataset="tests.jsonl")
def answer(prompt: str) -> str:
    responses = {"Capital of France": "Paris", "2+2": "4"}
    return responses.get(prompt, "Unknown")

if __name__ == "__main__":
    results = answer.run_eval()
    if not results["passed"]:
        raise SystemExit(1)