图解大模型 : 生成式AI 原理与实战

176

｜

第

7

章

和

{title}

占位符的新提示词模板：

# 使用故事梗概和标题创建一个链式架构来生成角色描述

template = """<s><|user|>

Describe the main character of a story about {summary} with the title {title}.

Use only two sentences.<|end|>

<|assistant|>"""

character_prompt = PromptTemplate

(

template=template

,

input_variables=

[

"summary"

,

"title"

]

)

character = LLMChain

(

llm=llm

,

prompt=character_prompt

,

output_key="character"

)

虽然目前我们可以手动利用

character

变量生成角色描述，但它最终将被整合到自动化链

式架构中运行。

现在，我们创建最终组件，该组件将整合故事梗概、标题和角色描述，以生成剧情概要：

# 使用故事梗概、标题和角色描述创建一个链式架构来生成故事

template = """<s><|user|>

Create a story about {summary} with the title {title}. The main character is:

{character}. Only return the story and it cannot be longer ...