from langchain import PromptTemplate, FewShotPromptTemplate
examples = [
    {
        "question": "戴帽卫衣可以穿了吗？",
        "answer":
            """
    可以的，颜色需要符合上衣标准要求。
    """
    },
    {
        "question": "所有的上衣下摆都需要塞进裤子里吗？",
        "answer":
            """
T恤、衬衫等材质较薄的上衣下摆需要塞进裤子里，太厚不能塞进裤子里的上衣，需要整理整齐。
"""
    },
    {
        "question": "臻选门店的白衬衫和酒坊的灰衬衫还可以穿吗？",
        "answer":
            """
可以，这些服装符合新的着装标准。
"""
    },
#     {
#         "question": "运动鞋可以穿吗？",
#         "answer":
#             """
# 材质为皮革、橡胶、反绒皮等材质的运动鞋可以穿，主色需为黑色、深灰色、深蓝色、深棕色。
# """
#     }, {
#         "question": "能不能不戴帽子？",
#         "answer":
#             """
# 由于法规要求，门店伙伴必须佩戴帽子工作。
# """
#     },
#     {
#         "question": "伙伴上班可以戴手表吗？这样可以方便看时间~",
#         "answer":
#             """
# 不可以，佩戴手表会影响我们的洗手流程，根据食品安全要求，洗手中需要搓洗前臂（在腕与肘之间），戴着手表是不方便做手部清洁。
# """
#     }, {
#         "question": "女生可以穿连衣裙吗？",
#         "answer":
#             """
# 可以，只要符合我们的颜色要求，也需要是纯色的。衣领不能时过宽或者是V字领，且裙子要及膝或者过膝。
# """
#     },
#     {
#         "question": "条纹和格子衬衫可以穿吗？",
#         "answer":
#             """
# 不可以，在新的着装标准里，衣服需要符合纯色要求，除了一些制造商的小logo外。
# """
#     }, {
#         "question": "门店伙伴的穿着是否需要店经理统一？",
#         "answer":
#             """
# 伙伴可以与店经理讨论门店服装风格，具体以店经理要求为主。
# """
#     },
#     {
#         "question": "穿衬衫时，衣领部分的扣子是不是要扣到最高？",
#         "answer":
#             """
# 不用，大家可以搭配喜欢的方巾、领带或者领结，找到适合自己的风格。
# """
#     }, {
#         "question": "为社么不可以穿紧身裤？",
#         "answer":
#             """
# 如果被热水烫伤穿着紧身裁剪的裤子会影响伤口处理，造成再次伤害。为了伙伴的安全考虑，不要在上班时穿着任何面料的紧身裤。
# """
#     },
    {
        "question": "酒坊门店Bar Mixato的伙伴也是按照新的着装标准吗？",
        "answer":
            """
可以按照新的穿搭，也可以继续穿之前的服装。旗舰店伙伴请和店经理先确认。
"""
    },
    {
        "question": "背带裤可以穿吗？",
        "answer":
            """
可以的，颜色需要符合裤装标准要求。
"""
    },
    {
        "question": "上衣的标准是什么？",
        "answer":
            """
1.伙伴可穿着T恤、戴帽卫衣、带领衬衫、立领衬衫、POLO、长袖POLO、夹克外套、羊毛衫、牛仔衬衫、套领衫等。
2.伙伴的上衣应是纯色的，如果伙伴穿着打底衣物（如套领衫、打底衫），那么颜色应符合可选颜色范围。可在套领衫或衬衫外面穿一件毛衣或夹克，颜色也应符合上衣标准。（考虑到异物风险，请勿选择长绒、松散的、容易掉落的毛衣材质，如：兔毛、马海毛、貂绒/仿貂绒、羊驼毛等）
3.T恤、衬衫等材质较薄的上衣下摆需塞进裤子里。无法塞进裤子里的上衣，请确保看起来整齐、不拖沓。
4.颜色要求：黑色、灰色系列（深灰、中灰和浅灰）、白色、深蓝色、棕色系和卡其色，都是可选的。
"""
    }, {
        "question": "下装的标准是什么？",
        "answer":
            """
1.伙伴可以穿着长裤或及膝短裤，也可以穿裙子（包括连衣裙），但需要是纯色并且长度及膝或过膝。伙伴不应穿着颜色不均匀的牛仔裤，宽松下垂、破洞或者做旧效果的牛仔裤也不能穿。出于安全考虑，伙伴也不应穿着皮裤、瑜伽裤、弹力纤维裤和紧身裤（包括黑色连裤袜）。
2.颜色要求：卡其色、深蓝色、深灰色、黑色。
"""
    }
]

EXAMPLE_PROMPT = PromptTemplate(input_variables=["question", "answer"], template="Question: {question}\n{answer}")
FEW_SHOT_PROMPT_TEMPLATE = FewShotPromptTemplate(
    examples=examples,
    example_prompt=EXAMPLE_PROMPT,
    suffix="Question: {input}",
    input_variables=["input"]
)


def get_few_shot_template() -> FewShotPromptTemplate:
    return FEW_SHOT_PROMPT_TEMPLATE