探索LangChain中的Prompt模板部分格式化技巧

最新推荐文章于 2024-10-03 22:46:37 发布

mmlihaio

最新推荐文章于 2024-10-03 22:46:37 发布

阅读量286

点赞数 4

文章标签： langchain prompt python

本文链接：https://blog.csdn.net/mmlihaio/article/details/142321772

版权

引言

在文本生成任务中，使用Prompt模板可以极大地简化生成过程，特别是在构建复杂的AI对话或生成场景时。而部分格式化Prompt模板则是一种增强灵活性的技巧，它允许你在不同时刻对模板进行初始化。这篇文章将深入探讨如何在LangChain中实现Prompt模板的部分格式化。

主要内容

部分格式化的动机

在使用Prompt模板时，通常需要同时提供多个参数。但是，如果在不同步骤中获取参数，可能不便于每次传递所有参数。部分格式化允许你先行绑定已知的参数，推迟未获取参数的绑定。

使用字符串进行部分格式化

示例场景

假设有一个模板需要两个变量foo和bar。当你在获取foo的值后，想要先行绑定这一值，而后在链条中传递部分格式化的模板，只需在之后补充bar的值。

from langchain_core.prompts import PromptTemplate

# 创建Prompt模板
prompt = PromptTemplate.from_template("{foo}{bar}")

# 部分格式化
partial_prompt = prompt.partial(foo="foo")

# 格式化完成
print(partial_prompt.format(bar="baz"))  # 输出: foobaz

# 也可以在初始化时定义部分变量
prompt = PromptTemplate(
    template="{foo}{bar}", input_variables=["bar"], partial_variables={"foo": "foo"}
)
print(prompt.format(bar="baz"))  # 输出: foobaz

使用函数进行部分格式化

另一种常见场景是通过函数获取动态值，比如当前日期和时间。在这种情况下，可以定义一个函数来生成所需的值，并将其绑定到模板中。

from datetime import datetime

# 定义一个返回当前日期时间的函数
def _get_datetime():
    now = datetime.now()
    return now.strftime("%m/%d/%Y, %H:%M:%S")

# 创建Prompt模板
prompt = PromptTemplate(
    template="Tell me a {adjective} joke about the day {date}",
    input_variables=["adjective"],
    partial_variables={"date": _get_datetime},
)

# 完成格式化
print(prompt.format(adjective="funny"))  # 输出: Tell me a funny joke about the day 04/21/2024, 19:43:57