From 529cb2e30c02777973fe127339907a69a1c10839 Mon Sep 17 00:00:00 2001 From: Eugene Yurtsev Date: Sun, 30 Jul 2023 22:39:14 -0400 Subject: [PATCH] Update doc-string in few shot template (#8474) Partial update of doc-string, need to update other instances in documentation --- libs/langchain/langchain/prompts/few_shot.py | 28 +++++++++----------- 1 file changed, 12 insertions(+), 16 deletions(-) diff --git a/libs/langchain/langchain/prompts/few_shot.py b/libs/langchain/langchain/prompts/few_shot.py index 220b236a138..7a28f10066b 100644 --- a/libs/langchain/langchain/prompts/few_shot.py +++ b/libs/langchain/langchain/prompts/few_shot.py @@ -190,12 +190,9 @@ class FewShotChatMessagePromptTemplate( .. code-block:: python - from langchain.schema import SystemMessage from langchain.prompts import ( FewShotChatMessagePromptTemplate, - HumanMessagePromptTemplate, - SystemMessagePromptTemplate, - AIMessagePromptTemplate + ChatPromptTemplate ) examples = [ @@ -203,24 +200,23 @@ class FewShotChatMessagePromptTemplate( {"input": "2+3", "output": "5"}, ] + example_prompt = ChatPromptTemplate.from_messages( + [('human', '{input}'), ('ai', '{output}')] + ) + few_shot_prompt = FewShotChatMessagePromptTemplate( examples=examples, # This is a prompt template used to format each individual example. - example_prompt=( - HumanMessagePromptTemplate.from_template("{input}") - + AIMessagePromptTemplate.from_template("{output}") - ), + example_prompt=example_prompt, ) - - final_prompt = ( - SystemMessagePromptTemplate.from_template( - "You are a helpful AI Assistant" - ) - + few_shot_prompt - + HumanMessagePromptTemplate.from_template("{input}") + final_prompt = ChatPromptTemplate.from_messages( + [ + ('system', 'You are a helpful AI Assistant'), + few_shot_prompt, + ('human', '{input}'), + ] ) - final_prompt.format(input="What is 4+4?") Prompt template with dynamically selected examples: