探索自定义输出解析器：从基础到进阶_自定义解析器-优快云博客

# 探索自定义输出解析器：从基础到进阶

在处理自然语言处理任务时，通常需要解析和格式化模型的输出。这篇文章将介绍如何创建自定义输出解析器，以便将模型输出结构化为所需的格式。我们将探讨两种主要的方法：通过 `RunnableLambda` 或 `RunnableGenerator` 实现，和通过继承基础解析类实现。

## 1. 引言

输出解析器在处理生成模型输出时非常有用，能够将模型返回的不规则文本转化为更具意义的结构化数据。本文将逐步介绍如何实现自定义解析器，提供代码示例，并讨论实现过程中可能遇到的挑战和解决方案。

## 2. 主要内容

### 使用 `RunnableLambda` 和 `RunnableGenerator`

推荐使用 `RunnableLambda` 和 `RunnableGenerator` 来解析输出。以下是一个反转字符串大小写的简单解析器示例：

```python
from typing import Iterable
from langchain_anthropic.chat_models import ChatAnthropic
from langchain_core.messages import AIMessage, AIMessageChunk

model = ChatAnthropic(model_name="claude-2.1")

def parse(ai_message: AIMessage) -> str:
    """Parse the AI message."""
    return ai_message.content.swapcase()

# 使用 `|` 语法自动将 `parse` 转换为 `RunnableLambda`
chain = model | parse
chain.invoke("hello")  # 输出 'hELLO!'

# 如果需要流式处理，使用 `RunnableGenerator`
from langchain_core.runnables import RunnableGenerator

def streaming_parse(chunks: Iterable[AIMessageChunk]) -> Iterable[str]:
    for chunk in chunks:
        yield chunk.content.swapcase()

streaming_parse = RunnableGenerator(streaming_parse)

chain = model | streaming_parse
# 确认流式处理是否正常
for chunk in chain.stream("tell me about yourself in one sentence"):
    print(chunk, end="|", flush=True)

继承解析基础类

另一种方法是从基础解析类继承。虽然这种方法写的代码更多，但带来了更多的可定制性。例如，一个布尔值的解析器：

from langchain_core.exceptions import OutputParserException
from langchain_core.output_parsers import BaseOutputParser

class BooleanOutputParser(BaseOutputParser[bool]):
    """Custom boolean parser."""
    
    true_val: str = "YES"
    false_val: str = "NO"

    def parse(self, text: str) -> bool:
        cleaned_text = text.strip().upper()
        if cleaned_text not in (self.true_val.upper(), self.false_val.upper()):
            raise OutputParserException(
                f"BooleanOutputParser expected output value to either be "
                f"{self.true_val} or {self.false_val} (case-insensitive). "
                f"Received {cleaned_text}."
            )
        return cleaned_text == self.true_val.upper()

3. 代码示例

下面是一个使用自定义解析器的完整示例：

parser = BooleanOutputParser()
result = parser.invoke("YES")
print(result)  # 输出 True

try:
    parser.invoke("MEOW")
except Exception as e:
    print(f"Triggered an exception of type: {type(e)}")

# 更新参数化
parser = BooleanOutputParser(true_val="OKAY")
print(parser.invoke("OKAY"))  # 输出 True