LangChain基本使用

  • 安装并配置
  • 实现第一个程序
  • 使用PromptTemplate+LLMChain生成多主题文案

一、安装

pip install langchain
pip install openai

二、API KEY 配置

想要调用大模型,我们需要使用大模型对应的API KEY才能进行大模型的使用,例如要使用DeepSeek,需要去官网获取API KEY,并且要设置环境变量

1.前往官网 DeepSeek | 深度求索,点击API开放平台

2.点击API keys,点击创建API KEY

 3.输入名称,点击创建

4.创建成功后点击复制并使用

5.可以使用配置环境变量来配置API KEY,也可以直接填写

OPENAI_API_KEY = "刚才复制的API KEY"
# 或者使用
os.environ["DeepSeek_API_KEY"] = "刚才复制的API KEY"

三、代码实现

# 导入需要的组件
from langchain_openai import ChatOpenAI
from langchain_core.output_parsers import StrOutputParser
from langchain_core.prompts import ChatPromptTemplate

OPENAI_API_KEY = ("你的API KEY")

# 模型初始化
model = ChatOpenAI(
    model="deepseek-chat",
    openai_api_key=OPENAI_API_KEY,
    base_url="https://api.deepseek.com/v1",  # 关键配置:指定 DeepSeek 的 API 端点
    streaming=False,  # 是否使用流式响应
    temperature=0.7  # 控制生成文本的随机性
)

# 创建处理链
prompt = ChatPromptTemplate.from_template("请根据下面的主题写一篇小红书营销的短文:{topic}")
output_parser = StrOutputParser()
chain = prompt | model | output_parser

# 调用链
result = chain.invoke({"topic": "康师傅绿茶"})
print(result)

运行结果如下:

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐