第七届字节跳动青训营第一课 | LangChain 入门案例

178 阅读3分钟

LangChain 是一个专为构建基于大型语言模型(LLMs)应用而设计的框架。它不仅支持通过 API 调用如 ChatGPT 等语言模型,还提供了许多高级功能,帮助开发者创建更加智能和灵活的应用。以下是 LangChain 的几个主要特性:

  1. 大语言模型应用:LangChain 能够创建各种复杂的应用,比如文本生成、文档问答和聊天机器人。
  2. 数据感知: LangChain 允许语言模型连接到其他数据源,从而能够理解和利用更丰富的数据。
  3. 代理性: 这个框架使得语言模型能够与其环境进行互动,提升对环境的理解能力。

通过这些功能,LangChain 为开发者提供了一个强大的工具,帮助他们最大限度地释放大语言模型的潜力。 下面来介绍一下如何快速使用LangChain。

快速使用LangChain

应用 1:情人节玫瑰宣传语

  1. 需要安装三个包:首先,通过 pip 安装需要的包。如果你在一个新的虚拟环境中,运行以下命令:(在课程中的虚拟环境已经提前安装好了)
pip install langchain openai langchain-openai
  1. 设置OpenAi API Key:为了使用 OpenAI 的语言模型,你需要设置 API 密钥。在你的 Python 代码中添加以下内容:
import os
os.environ["OPENAI_API_KEY"] = '你的OpenAI Key'

在练中学中,提前设置好了API Key,直接使用即可。

llm = ChatOpenAI(model=os.environ.get("LLM_MODELEND"))
  1. 编写代码->生成宣传语:你可以使用以下代码来生成情人节红玫瑰的宣传语。将输入的提示信息修改为你想要的内容,模型将根据提示生成相应的文本。
from langchain_openai import OpenAI

llm = OpenAI(model_name="gpt-3.5-turbo-instruct", max_tokens=200)
text = llm.invoke("请给我写一句情人节红玫瑰的中文宣传语")
print(text)

点击运行。 下面是完整示例:

import os
from langchain_openai import ChatOpenAI

# llm = OpenAI(model_name="gpt-3.5-turbo-instruct",max_tokens=200)
llm = ChatOpenAI(model=os.environ.get("LLM_MODELEND"))

text = llm.predict("爱情是通往幸福的天堂")
print(text)

在运行后,你可能会得到如下回复:

爱情确实可以是通往幸福的天堂,但它也并非总是如此简单和直接。

通过修改提示,你可以生成不同风格的宣传语,灵活应对市场需求。

应用 2:海报文案生成器

在这个应用中,我们将创建一个可以根据图片生成文案的工具。你可以通过修改图片链接来生成不同的文案。

img_url = 'https://mir-s3-cdn-cf.behance.net/project_modules/hd/eec79e20058499.563190744f903.jpg'
agent.invoke(input=f"{img_url}\n请创作合适的中文推广文案")

在这里通过修改 img_url = ''中的图片链接生成不同的文案。我上传的是我的VX头像ahu.jpg,得到的回复如下:

Final Answer: 男女生站在教学楼前,一起享受校园的美好时光。

通过这两个示例,我们可以看到 LangChain 如何将语言处理与图像生成结合起来,帮助开发者创建更加智能和灵活的应用。你可以根据实际需求自由修改输入内容和图片链接,从而获得多样化的输出,充分发挥大语言模型的潜力。无论是营销宣传还是社交媒体内容,LangChain 都能提供强大的支持,帮助你在各种场景中实现创新。