1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
from langchain_core.prompts import FewShotPromptTemplate,PromptTemplate
from langchain_google_genai import ChatGoogleGenerativeAI

example_prompt = PromptTemplate.from_template(
    "單字:{word}, 相反詞:{anti-word}")
example_data = [
    {"word":"大", "anti-word":"小"},
    {"word":"上", "anti-word":"下"},
]
few_shot_template = FewShotPromptTemplate(
    example_prompt = example_prompt,
    examples = example_data,
    prefix = "請給出單字的相反詞，例如:",
    suffix = "根據範例告訴我，{input_word} 的相反詞是？",
    input_variables= ["input_word"],
)
prompt_text = few_shot_template.invoke(input={"input_word":"長"}).to_string()
llm = ChatGoogleGenerativeAI(model="gemini-3-flash-preview", temperature=0)
res = llm.invoke(input=prompt_text)
print(res.text)