2025-09-14 langchain学习-08 chain的用法

import os
from langchain.prompts import ChatPromptTemplate
from langchain_community.chat_models import ChatZhipuAI
from langchain_core.output_parsers import StrOutputParser
from dotenv import load_dotenv

load_dotenv()

#定义模型
model=ChatZhipuAI(
  api_key=os.getenv("ZHIPU_API_KEY"),
  model='glm-3-turbo'
)

#定义输出解析器
output_parser=StrOutputParser()

#提示词
chat_template=ChatPromptTemplate.from_template("根据用户输入的主题{topic}写一首七言绝句")

#链
chain=chat_template | model | output_parser

result=chain.invoke(input="太阳")
print(result)

import os
from langchain.prompts import ChatPromptTemplate
from langchain_community.chat_models import ChatZhipuAI
from langchain_core.output_parsers import StrOutputParser
from langchain_core.runnables import RunnableParallel
from dotenv import load_dotenv


load_dotenv()

#定义模型
model=ChatZhipuAI(
  api_key=os.getenv("ZHIPU_API_KEY"),
  model='glm-3-turbo'
)

#定义输出解析器
output_parser=StrOutputParser()

#提示词1
pome_template=ChatPromptTemplate.from_template("根据用户输入的主题{topic}写一首七言绝句")

#链1
pome_chain=pome_template | model | output_parser

#提示词2
code_template=ChatPromptTemplate.from_template("根据用户输入的主题{topic}写一段代码")

#链2
code_chain=code_template | model | output_parser

combine_chain=RunnableParallel(
  pome=pome_chain,
  code=code_chain
)

result=combine_chain.invoke("水滴")