"""This is the ideal user flow
"""
from evals.test_utils import assert_llm_output, TestEvalCase, tags


def generate_llm_output(text_input: str):
    return text_input


def test_llm_output(self):
    input = "What is the customer success phone line?"
    expected_output = "Our customer success phone line is 1200-231-231."
    output = generate_llm_output(input)
    assert_llm_output(output, expected_output, metric="entailment")

class TestLLM(TestEvalCase):
    @tags(tags=["customer success"])
    def test_llm_output(self):
        input = "What is the customer success phone line?"
        expected_output = "Our customer success phone line is 1200-231-231."
        output = generate_llm_output(input)
        assert_llm_output(output, expected_output, metric="entailment")
