高效管理提示模板：LangChain中的部分格式化详解

nseejrukjhad

于 2024-10-03 18:35:15 发布

阅读量92

点赞数 2

文章标签： langchain java 服务器 python

本文链接：https://blog.csdn.net/nseejrukjhad/article/details/142694479

版权

高效管理提示模板：LangChain中的部分格式化详解

在使用LangChain进行提示模板管理时，部分格式化功能是一个强大的工具。本文将深入探讨如何在LangChain中使用部分格式化来提高代码的灵活性和可维护性。

引言

提示模板是构建复杂应用时的重要组成部分。部分格式化可以让我们在链路上部分绑定变量，简化模板的使用流程。本篇文章将详细介绍如何通过字符串和函数进行部分格式化。

主要内容

什么是部分格式化？

部分格式化类似于在函数中部分绑定参数。通过传入部分变量，生成一个新的模板，仅需剩余的变量即可使用。

LangChain中的部分格式化支持

LangChain支持两种方式的部分格式化：

使用字符串进行部分格式化。
使用返回字符串的函数进行部分格式化。

部分格式化字符串

在某些场景下，我们可能会先获取提示模板中部分变量的值。这时，可以先进行部分格式化，再传递到下一个环节。

from langchain_core.prompts import PromptTemplate

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 = PromptTemplate(
    template="Tell me a {adjective} joke about the day {date}",
    input_variables=["adjective", "date"],
)
partial_prompt = prompt.partial(date=_get_datetime)
print(partial_prompt.format(adjective="funny"))

初始化时使用函数：

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"))