【LangChain系列 11】Prompt模版——拼装组合

原文地址：【LangChain系列 11】Prompt模版——拼装组合

本文速读：

• 多prompt模版组合

• 单prompt模版拼装

在平常业务开发中，我们常常需要把一些公共模块提取出来作为一个独立的部分，然后将业务中去将这些模块进行组合。在LLM应用开发中，我们也会需要采用这种思想，比如将一些公共的promt模版独立出来，这样prompt模版就可以更好地复用，减少不必要的代码，保持代码和逻辑的简洁。

LangChain对prompt模版的组合提供两种方式：

1. 针对多个prompt模版进行组合。

2. 将多个部分拼装成一个prompt模版。

01 多prompt模版组合

---

LangChain提供了PipelinePrompt来进行多prompt模版组合。一个PipelinePrompt包含两个部分：

• 最终的prompt模版：最终生成的prompt模版。

• 待组合的prompt模版：它是一个列表，列表里的每一项包含一个名字和一个prompt模版。

如下面代码所示，full_prompt就是最终的 prompt模版，input_prompts就是 待组合的prompt模版；将input_prompts中的prompt模版最终组合成了full_prompt。

from langchain.prompts.pipeline import PipelinePromptTemplate
from langchain.prompts.prompt import PromptTemplate

full_template = """{introduction}

{example}

{start}"""
full_prompt = PromptTemplate.from_template(full_template)

introduction_template = """You are impersonating {person}."""
introduction_prompt = PromptTemplate.from_template(introduction_template)

example_template = """Here's an example of an interaction: 

Q: {example_q}
A: {example_a}"""
example_prompt = PromptTemplate.from_template(example_template)

start_template = """Now, do this for real!

Q: {input}
A:"""
start_prompt = PromptTemplate.from_template(start_template)

input_prompts = [
    ("introduction", introduction_prompt),
    ("example", example_prompt),
    ("start", start_prompt)
]
pipeline_prompt = PipelinePromptTemplate(final_prompt=full_prompt, pipeline_prompts=input_prompts)
print(pipeline_prompt.input_variables)

输出结果：

['example_a', 'person', 'example_q', 'input']

执行下面代码：

print(pipeline_prompt.format(
    person="Elon Musk",
    example_q="What's your favorite car?",
    example_a="Tesla",
    input="What's your favorite social media site?"
))

输出结果：

    You are impersonating Elon Musk.
    Here's an example of an interaction: 

    Q: What's your favorite car?
    A: Tesla
    Now, do this for real!

    Q: What's your favorite social media site?
    A:

02 单prompt模版拼装

---

单prompt模版拼装是指将多个部分拼装成一个完整的prompt模版，一般来说是将字符串与prompt模版拼成一个新的prompt模版。下面主要介绍字符串prompt模版和对话prompt模版这两种模版的拼装，通过两个代码示例来介绍它们的用法。

字符串prompt模版

在下面代码中，将一个字符串prompt模版和两个字符串通过 + 拼装起来。

from langchain.prompts import PromptTemplate

prompt = (
    PromptTemplate.from_template("Tell me a joke about {topic}")
    + ", make it funny"
    + "\n\nand in {language}"
)
print(prompt)

输出结果：

PromptTemplate(input_variables=['language', 'topic'], output_parser=None, partial_variables={}, template='Tell me a joke about {topic}, make it funny\n\nand in {language}', template_format='f-string', validate_template=True)

执行代码：

print(prompt.format(topic="sports", language="spanish"))

输出结果：

'Tell me a joke about sports, make it funny\n\nand in spanish'

同样，我们可以在LLMChain中使用这个拼装的prompt。

from langchain.chat_models import ChatOpenAI
from langchain.chains import LLMChain

model = ChatOpenAI(openai_api_key="xxx")
chain = LLMChain(llm=model, prompt=prompt)
chain.run(topic="sports", language="spanish")

执行代码，输出结果：

'¿Por qué el futbolista llevaba un paraguas al partido?\n\nPorque pronosticaban lluvia de goles.'

对话prompt模版

在下面代码中，将对话prompt中的Message和字符串通过 + 进行拼装，形成一个新的prompt模版，不仅可以将Message进行拼装，而且可以将MessagePrompt进行拼装，不过先要将MessagePrompt中的变量进行赋值。

from langchain.prompts import ChatPromptTemplate, HumanMessagePromptTemplate
from langchain.schema import HumanMessage, AIMessage, SystemMessage

prompt = SystemMessage(content="You are a nice pirate")
new_prompt = (
    prompt
    + HumanMessage(content="hi")
    + AIMessage(content="what?")
    + "{input}"
)

print(new_prompt.format_messages(input="i said hi"))

输出结果：

[SystemMessage(content='You are a nice pirate', additional_kwargs={}),
   HumanMessage(content='hi', additional_kwargs={}, example=False),
   AIMessage(content='what?', additional_kwargs={}, example=False),
   HumanMessage(content='i said hi', additional_kwargs={}, example=False)]

同样地，可以在LLMChain中使用它：

from langchain.chat_models import ChatOpenAI
from langchain.chains import LLMChain

model = ChatOpenAI(openai_api_key="xxx")
chain = LLMChain(llm=model, prompt=new_prompt)
chain.run("i said hi")

执行代码，输出结果：

'Oh, hello! How can I assist you today?'

本文小结

本文主要介绍了prompt模版的拼装组合，既可以将多个prompt模版进行组合，也可以对单个prompt模版进行拼装。

 更多最新文章，请关注公众号：大白爱爬山