LangChain入门：14.LLMChain：最简单的链的使用

news2024/11/19 3:37:44

摘要

本文将介绍LangChain库中LLMChain工具的使用方法。LLMChain将提示模板、语言模型（LLM）和输出解析器整合在一起，形成一个连贯的处理链，简化了与语言模型的交互过程。我们将探讨LLMChain的技术特点、应用场景以及它解决的问题，并提供详细的代码示例。

技术介绍

LLMChain是LangChain库中的一项功能强大的工具，它提供了一个便捷的方式来与语言模型进行交互。通过LLMChain，用户可以轻松地构建提示模板、调用语言模型进行推理，并解析输出结果，而无需手动处理繁琐的过程。

应用场景

LLMChain适用于许多自然语言处理任务，包括文本生成、情感分析、文本分类等。它可以用于构建智能对话系统、自动文本摘要生成器、情感识别引擎等应用程序。此外，LLMChain还可以用于快速原型开发和实验，以及用于教育和研究目的。

解决的问题

LLMChain解决了与语言模型交互过程中的诸多问题，包括：

简化交互流程：通过整合提示模板、语言模型和输出解析器，LLMChain简化了与语言模型的交互过程，使用户可以更轻松地使用语言模型进行推理。
自动化处理：LLMChain提供了多种调用方法，可以根据需求自动处理输入数据和解析输出结果，减少了用户的手动干预。
提高效率：LLMChain的灵活性和高效性使其成为处理自然语言处理任务的理想工具，可以大大提高工作效率和准确性。

完整示例代码

from langchain.prompts import PromptTemplate
from langchain_openai import ChatOpenAI
from langchain.chains import LLMChain

# 导入所需库和模块
from langchain.prompts import PromptTemplate
from langchain_openai import ChatOpenAI
from langchain.chains import LLMChain

# 使用Pydantic创建数据格式
from pydantic import BaseModel, Field
from typing import List

class Flower(BaseModel):
    name: str = Field(description="name of a flower")
    colors: List[str] = Field(description="the colors of this flower")

# 创建提示模板实例
template = "{flower}的花语是？"
llm = ChatOpenAI(
    openai_api_key='# 替换为你的API密钥',
    base_url='https://api.chatanywhere.tech/v1',
    model='gpt-3.5-turbo',
    temperature=0,
)
prompt = PromptTemplate.from_template(template)

# 初始化LLMChain
llm_chain = LLMChain(
    llm=llm,
    prompt=prompt
)

# 调用LLMChain并获取结果
result = llm_chain.invoke("玫瑰")
print(result)

多种调用方法

通过run方法：

input_list = [
    { 'flower': "玫瑰", 'season': "夏季" },
    { 'flower': "康乃馨", 'season': "冬季" },
    { 'flower': "郁金香", 'season': "春季" },
]
print(llm_chain.run(input_list))

通过predict方法：

print(llm_chain.predict(flower="玫瑰", season="夏季"))

通过apply方法：

input_list = [
    { 'flower': "玫瑰", 'season': "夏季" },
    { 'flower': "康乃馨", 'season': "冬季" },
    { 'flower': "郁金香", 'season': "春季" },
]
print(llm_chain.apply(input_list))

通过generate方法：

input_list = [
    { 'flower': "玫瑰", 'season': "夏季" },
    { 'flower': "康乃馨", 'season': "冬季" },
    { 'flower': "郁金香", 'season': "春季" },
]
print(llm_chain.generate(input_list))