设计语言模型提示词
您可以使用字符串连接来构建语言模型提示词,但这很难组合功能并确保您的提示词保持在语言模型的上下文窗口内。为了克服这些限制,您可以使用 @vscode/prompt-tsx
库。
@vscode/prompt-tsx
库提供以下功能:
- 基于 TSX 的提示词渲染:使用 TSX 组件组合提示词,使其更具可读性和可维护性
- 基于优先级的剪枝:自动剪枝提示词中不重要的部分,以适应模型的上下文窗口
- 灵活的 Token 管理:使用
flexGrow
、flexReserve
和flexBasis
等属性来协作使用 Token 预算 - 工具集成:与 VS Code 的语言模型工具 API 集成
有关所有功能和详细使用说明的完整概述,请参阅 完整 README。
本文介绍了使用该库进行提示词设计的实际示例。这些示例的完整代码可以在 prompt-tsx 存储库中找到。
管理对话历史记录中的优先级
在提示词中包含对话历史记录很重要,因为它使用户能够对以前的消息提出后续问题。但是,您需要确保其优先级得到适当处理,因为历史记录会随着时间的推移而变得很大。我们发现最有意义的模式通常是按顺序确定优先级:
- 基本提示词指令
- 当前用户查询
- 最近几轮聊天历史
- 任何支持数据
- 尽可能多的剩余历史记录
因此,将历史记录分成两部分在提示词中,其中最近的提示词轮次优先于一般上下文信息。
在此库中,树中的每个 TSX 节点都有一个优先级,这在概念上类似于 zIndex,数字越高表示优先级越高。
步骤 1:定义 HistoryMessages 组件
要列出历史消息,请定义一个 HistoryMessages
组件。此示例提供了一个很好的起点,但如果您处理更复杂的数据类型,则可能需要对其进行扩展。
此示例使用 PrioritizedList
辅助组件,该组件自动为其每个子项分配升序或降序优先级。
import {
UserMessage,
AssistantMessage,
PromptElement,
BasePromptElementProps,
PrioritizedList,
} from '@vscode/prompt-tsx';
import { ChatContext, ChatRequestTurn, ChatResponseTurn, ChatResponseMarkdownPart } from 'vscode';
interface IHistoryMessagesProps extends BasePromptElementProps {
history: ChatContext['history'];
}
export class HistoryMessages extends PromptElement<IHistoryMessagesProps> {
render(): PromptPiece {
const history: (UserMessage | AssistantMessage)[] = [];
for (const turn of this.props.history) {
if (turn instanceof ChatRequestTurn) {
history.push(<UserMessage>{turn.prompt}</UserMessage>);
} else if (turn instanceof ChatResponseTurn) {
history.push(
<AssistantMessage name={turn.participant}>
{chatResponseToMarkdown(turn)}
</AssistantMessage>
);
}
}
return (
<PrioritizedList priority={0} descending={false}>
{history}
</PrioritizedList>
);
}
}
步骤 2:定义 Prompt 组件
接下来,定义一个 MyPrompt
组件,其中包含基本指令、用户查询和具有相应优先级的历史消息。优先级值在同级之间是局部的。请记住,您可能希望在触及提示词中的任何其他内容之前修剪历史记录中较旧的消息,因此您需要拆分两个 <HistoryMessages>
元素
import {
UserMessage,
PromptElement,
BasePromptElementProps,
} from '@vscode/prompt-tsx';
interface IMyPromptProps extends BasePromptElementProps {
history: ChatContext['history'];
userQuery: string;
}
export class MyPrompt extends PromptElement<IMyPromptProps> {
render() {
return (
<>
<UserMessage priority={100}>
Here are your base instructions. They have the highest priority because you want to make
sure they're always included!
</UserMessage>
{/* Older messages in the history have the lowest priority since they're less relevant */}
<HistoryMessages history={this.props.history.slice(0, -2)} priority={0} />
{/* The last 2 history messages are preferred over any workspace context you have below */}
<HistoryMessages history={this.props.history.slice(-2)} priority={80} />
{/* The user query is right behind the based instructions in priority */}
<UserMessage priority={90}>{this.props.userQuery}</UserMessage>
<UserMessage priority={70}>
With a slightly lower priority, you can include some contextual data about the workspace
or files here...
</UserMessage>
</>
);
}
}
现在,所有较旧的历史消息都在库尝试剪枝提示词中的其他元素之前被剪枝。
步骤 3:定义 History 组件
为了使使用更容易,定义一个 History
组件,该组件包装历史消息并使用 passPriority
属性充当直通容器。通过 passPriority
,其子项在优先级方面被视为包含元素的直接子项。
import { PromptElement, BasePromptElementProps } from '@vscode/prompt-tsx';
interface IHistoryProps extends BasePromptElementProps {
history: ChatContext['history'];
newer: number; // last 2 message priority values
older: number; // previous message priority values
passPriority: true; // require this prop be set!
}
export class History extends PromptElement<IHistoryProps> {
render(): PromptPiece {
return (
<>
<HistoryMessages history={this.props.history.slice(0, -2)} priority={this.props.older} />
<HistoryMessages history={this.props.history.slice(-2)} priority={this.props.newer} />
</>
);
}
}
现在,您可以使用和重用此单个元素来包含聊天历史记录
<History history={this.props.history} passPriority older={0} newer={80}/>
根据需要扩展文件内容
在此示例中,您希望在提示词中包含用户当前正在查看的所有文件的内容。这些文件可能很大,以至于包含所有文件会导致其文本被剪枝!此示例演示如何使用 flexGrow
属性来协作调整文件内容的大小以适应 Token 预算。
步骤 1:定义基本指令和用户查询
首先,定义一个包含基本指令的 UserMessage
组件。
<UserMessage priority={100}>Here are your base instructions.</UserMessage>
然后,通过使用 UserMessage
组件包含用户查询。此组件具有高优先级,以确保它紧跟在基本指令之后。
<UserMessage priority={90}>{this.props.userQuery}</UserMessage>
步骤 2:包含文件内容
现在,您可以使用 FileContext
组件包含文件内容。为其分配 flexGrow
值 1
,以确保它在基本指令、用户查询和历史记录之后渲染。
<FileContext priority={70} flexGrow={1} files={this.props.files} />
通过 flexGrow
值,元素在其传递给其 render()
和 prepare()
调用的 PromptSizing
对象中获得任何未使用的 Token 预算。您可以在 prompt-tsx 文档中阅读有关 flex 元素行为的更多信息。
步骤 3:包含历史记录
接下来,使用您之前创建的 History
组件包含历史消息。这有点棘手,因为您确实希望显示一些历史记录,但也希望文件内容占用大部分提示词。
因此,为 History
组件分配 flexGrow
值 2
,以确保它在所有其他元素(包括 <FileContext />
)之后渲染。但是,也设置 flexReserve
值 "/5"
,为历史记录保留总预算的 1/5。
<History
history={this.props.history}
passPriority
older={0}
newer={80}
flexGrow={2}
flexReserve="/5"
/>
步骤 3:组合提示词的所有元素
现在,将所有元素组合到 MyPrompt
组件中。
import {
UserMessage,
PromptElement,
BasePromptElementProps,
} from '@vscode/prompt-tsx';
import { History } from './history';
interface IFilesToInclude {
document: TextDocument;
line: number;
}
interface IMyPromptProps extends BasePromptElementProps {
history: ChatContext['history'];
userQuery: string;
files: IFilesToInclude[];
}
export class MyPrompt extends PromptElement<IMyPromptProps> {
render() {
return (
<>
<UserMessage priority={100}>Here are your base instructions.</UserMessage>
<History
history={this.props.history}
passPriority
older={0}
newer={80}
flexGrow={2}
flexReserve="/5"
/>
<UserMessage priority={90}>{this.props.userQuery}</UserMessage>
<FileContext priority={70} flexGrow={1} files={this.props.files} />
</>
);
}
}
步骤 4:定义 FileContext 组件
最后,定义一个 FileContext
组件,其中包含用户当前正在查看的文件的内容。由于您使用了 flexGrow
,因此您可以实现逻辑,通过使用 PromptSizing
中的信息,获取每个文件“有趣”行周围尽可能多的行。
为简洁起见,省略了 getExpandedFiles
的实现逻辑。您可以在 prompt-tsx 存储库中查看它。
import { PromptElement, BasePromptElementProps, PromptSizing, PromptPiece } from '@vscode/prompt-tsx';
class FileContext extends PromptElement<{ files: IFilesToInclude[] } & BasePromptElementProps> {
async render(_state: void, sizing: PromptSizing): Promise<PromptPiece> {
const files = await this.getExpandedFiles(sizing);
return <>{files.map(f => f.toString())}</>;
}
private async getExpandedFiles(sizing: PromptSizing) {
// Implementation details are summarized here.
// Refer to the repo for the complete implementation.
}
}
总结
在这些示例中,您创建了一个 MyPrompt
组件,其中包含基本指令、用户查询、历史消息和具有不同优先级的文件内容。您使用 flexGrow
协作调整文件内容的大小以适应 Token 预算。
通过遵循此模式,您可以确保始终包含提示词中最重要的部分,同时根据需要剪枝不重要的部分以适应模型的上下文窗口。有关 getExpandedFiles
方法和 FileContextTracker
类的完整实现细节,请参阅 prompt-tsx 存储库。