如何使用LangChain中的自定义示例选择器

akhfuiigabv

于 2024-10-03 16:38:52 发布

阅读量86

点赞数 2

文章标签： langchain python

本文链接：https://blog.csdn.net/akhfuiigabv/article/details/142693293

版权

# 如何使用LangChain中的自定义示例选择器

在处理大量示例时，选择合适的示例加入到提示中是非常重要的。LangChain提供了一种灵活的方式，即通过自定义示例选择器来完成这一任务。这篇文章将带你详细了解如何创建一个自定义的示例选择器。

## 引言

LangChain库中的示例选择器类用于选择要包括在提示中的示例。本文的目的是帮助你理解如何创建自己的示例选择器，并在代码中灵活应用。

## 主要内容

### 示例选择器概念

`BaseExampleSelector`是LangChain中所有示例选择器的基类。它定义了两个抽象方法：

- `select_examples`: 根据输入变量选择示例。
- `add_example`: 添加新的示例到存储中。

### 自定义示例选择器

我们将创建一个选择器，该选择器根据输入和示例中单词的长度来选择合适的示例。

```python
from langchain_core.example_selectors.base import BaseExampleSelector

class CustomExampleSelector(BaseExampleSelector):
    def __init__(self, examples):
        self.examples = examples

    def add_example(self, example):
        self.examples.append(example)

    def select_examples(self, input_variables):
        new_word = input_variables["input"]
        new_word_length = len(new_word)
        best_match = None
        smallest_diff = float("inf")

        for example in self.examples:
            current_diff = abs(len(example["input"]) - new_word_length)
            if current_diff < smallest_diff:
                smallest_diff = current_diff
                best_match = example

        return [best_match]

使用示例选择器

创建一个示例选择器实例，并选择最合适的示例：

examples = [
    {"input": "hi", "output": "ciao"},
    {"input": "bye", "output": "arrivederci"},
    {"input": "soccer", "output": "calcio"},
]

example_selector = CustomExampleSelector(examples)
selected_examples = example_selector.select_examples({"input": "okay"})
print(selected_examples)

在提示中使用

使用选定的示例来生成完整的翻译提示：

from langchain_core.prompts.few_shot import FewShotPromptTemplate
from langchain_core.prompts.prompt import PromptTemplate

example_prompt = PromptTemplate.from_template("Input: {input} -> Output: {output}")

prompt = FewShotPromptTemplate(
    example_selector=example_selector,
    example_prompt=example_prompt,
    suffix="Input: {input} -> Output:",
    prefix="Translate the following words from English to Italian:",
    input_variables=["input"],
)

print(prompt.format(input="word"))