tests/test_prompts.py

from src.callbacks.retailgpt.graph import create_parser, create_llm_chain, get_context

def test_prompt():
    context = get_context()
    parser = create_parser()
    llm_context_chain = create_llm_chain(parser=parser)
    output = llm_context_chain(context)
    assert output is not None


def test_chain():
    params = {
        "temperature": 0,
        "model_name": "text-davinci-003",
        "max_tokens": 1000,
    }
    llm_context_chain = create_llm_chain(**params)