LangChain Chain链的串行以及SimpleSequentialChain的使用

hehui0921

已于 2024-02-07 17:09:32 修改

阅读量219

点赞数 3

分类专栏： LangChain 文章标签： langchain java 前端

于 2024-02-07 11:18:31 首次发布

本文链接：https://blog.csdn.net/oHeHui1/article/details/136067065

版权

LangChain 专栏收录该内容

59 篇文章 4 订阅

订阅专栏

文章介绍了如何使用ChatGLM库中的不同链式结构，如SimpleSequentialChain和LLMChain，来创建对话式故事生成和文本计数功能，通过模板和解析器处理用户输入并获取相应输出。

摘要由CSDN通过智能技术生成

以下将chain_one 的结果输入到chain_two .

import ChatGLM
from langchain_core.output_parsers import StrOutputParser
from langchain_core.prompts import ChatPromptTemplate
from langchain.chains import SimpleSequentialChain

llm = ChatGLM.ChatGLM_LLM()

prompt1 = ChatPromptTemplate.from_template(
    "tell me a story about {what},no more than 100 words"
)
chain_one = prompt1 | llm |  StrOutputParser()
prompt2 = ChatPromptTemplate.from_template("what is this story talk about {expression},no more than 10 words")
chain_two = (
    {"expression": chain_one}
    | prompt2
    | llm
    | StrOutputParser()
)

print(chain_two.invoke({"what": "dog"}))

也可以使用 SimpleSequentialChain：


import ChatGLM
from langchain.chains import LLMChain
from langchain_core.output_parsers import StrOutputParser
from langchain_core.prompts import ChatPromptTemplate
from langchain.chains import SimpleSequentialChain
from operator import itemgetter
from langchain.chains import ConversationChain

llm = ChatGLM.ChatGLM_LLM()

prompt = ChatPromptTemplate.from_template("""
 give the count of the words in the sentence: {input}  .
current conversation:{history}
       
"""
    )


prompt1= ChatPromptTemplate.from_template("""
  give me answer about {input}, no more than 50 words
"""
    )

chain_one = LLMChain(
    llm=llm,
    prompt=prompt1,
    verbose=True
)
conversation = ConversationChain(llm=llm, prompt=prompt, verbose=True)

chain_3 =  SimpleSequentialChain(chains = [chain_one,conversation])
print(chain_3.invoke("who is Obama"))