构建LangChain应用程序的示例代码：66、如何使用LangChain实现程序辅助语言模型(PAL)

程序辅助语言模型(PAL)链

实现程序辅助语言模型,如https://arxiv.org/pdf/2211.10435.pdf所述。

from langchain_experimental.pal_chain import PALChain
from langchain_openai import OpenAI# 创建OpenAI语言模型实例,设置温度为0,最大令牌数为512
llm = OpenAI(temperature=0, max_tokens=512)

数学提示

# 从数学提示创建PAL链,并启用详细输出
pal_chain = PALChain.from_math_prompt(llm, verbose=True)# 定义数学问题
question = "Jan有Marcia的三倍宠物数量。Marcia比Cindy多两只宠物。如果Cindy有四只宠物,三人总共有多少只宠物?"# 运行PAL链并获取结果
pal_chain.run(question)

输出:

[1m> Entering new PALChain chain...[0m
[32;1m[1;3mdef solution():"""Jan有Marcia的三倍宠物数量。Marcia比Cindy多两只宠物。如果Cindy有四只宠物,三人总共有多少只宠物?"""cindy_pets = 4marcia_pets = cindy_pets + 2jan_pets = marcia_pets * 3total_pets = cindy_pets + marcia_pets + jan_petsresult = total_petsreturn result[0m[1m> Finished chain.[0m'28'

彩色对象

# 从彩色对象提示创建PAL链,并启用详细输出
pal_chain = PALChain.from_colored_object_prompt(llm, verbose=True)# 定义彩色对象问题
question = "在桌子上,你看到两本蓝色小册子,两本紫色小册子,和两副黄色太阳镜。如果我从桌子上移除所有太阳镜,还剩下多少紫色物品?"# 运行PAL链并获取结果
pal_chain.run(question)

输出:

[1m> Entering new PALChain chain...[0m
[32;1m[1;3m# 将对象放入列表以记录顺序
objects = []
objects += [('booklet', 'blue')] * 2
objects += [('booklet', 'purple')] * 2
objects += [('sunglasses', 'yellow')] * 2# 移除所有太阳镜
objects = [object for object in objects if object[0] != 'sunglasses']# 计算紫色对象的数量
num_purple = len([object for object in objects if object[1] == 'purple'])
answer = num_purple[0m[1m> Finished PALChain chain.[0m'2'

中间步骤

您还可以使用中间步骤标志来返回生成答案的执行代码。

# 创建PAL链,启用详细输出和返回中间步骤
pal_chain = PALChain.from_colored_object_prompt(llm, verbose=True, return_intermediate_steps=True
)question = "在桌子上,你看到两本蓝色小册子,两本紫色小册子,和两副黄色太阳镜。如果我从桌子上移除所有太阳镜,还剩下多少紫色物品?"# 运行PAL链并获取结果
result = pal_chain({"question": question})# 获取中间步骤
result["intermediate_steps"]

输出:

"# 将对象放入列表以记录顺序\nobjects = []\nobjects += [('booklet', 'blue')] * 2\nobjects += [('booklet', 'purple')] * 2\nobjects += [('sunglasses', 'yellow')] * 2\n\n# 移除所有太阳镜\nobjects = [object for object in objects if object[0] != 'sunglasses']\n\n# 计算紫色对象的数量\nnum_purple = len([object for object in objects if object[1] == 'purple'])\nanswer = num_purple"