from langchain_core.prompts import PromptTemplate, FewShotPromptTemplate
# 准备示例数据
examples = [
    {"input": "Hello", "output": "你好"},
    {"input": "Thank you", "output": "谢谢"},
    {"input": "Good morning", "output": "早上好"},
]
# 定义示例的格式化模板
example_prompt = PromptTemplate.from_template(
    "{input} -> {output}"
)
# 创建 FewShotPromptTemplate
few_shot_prompt = FewShotPromptTemplate(
    examples=examples,
    example_prompt=example_prompt,
    prefix="请把以下英文翻译成中文：",
    suffix="{new_input} -> ",
    input_variables=["new_input"],
    example_separator="\n",
)
prompt_value = few_shot_prompt.invoke(
    {
        "new_input": "Good bye"
    }
)
print(prompt_value.to_string())
res = llm.invoke(prompt_value, config=config)
print(res.content)

from langchain_core.prompts import ChatPromptTemplate, FewShotChatMessagePromptTemplate
 
# 准备示例数据
examples = [
    {"input": "Hello", "output": "你好"},
    {"input": "Thank you", "output": "谢谢"},
    {"input": "Good morning", "output": "早上好"},
]
# 定义示例的格式化模板
example_prompt = ChatPromptTemplate.from_messages(
    [
        {"role": "user", "content": "{input}"},
        {"role": "ai", "content": "{output}"}
    ]
)
# 创建示例提示词模版
few_shot_prompt = FewShotChatMessagePromptTemplate(
    examples=examples,
    example_prompt=example_prompt
)
# 构建最终的聊天提示模板
chat_prompt = ChatPromptTemplate.from_messages(
    [
        {"role": "system", "content": "请把以下英文翻译成中文："},
        few_shot_prompt,
        {"role": "user", "content": "{new_input}"}
    ]
)
# 输入内容
prompt_value = chat_prompt.invoke(
    {
        "new_input": "Where are you from?",
    }
)
print(prompt_value.to_string())
res = llm.invoke(prompt_value, config=config)
print(res.content)