简单顺序链- - 将第一个链的输出作为第二个链的输入

from langchain.chains import LLMChain, SimpleSequentialChain #简单序列链
from langchain_community.llms.ollama import Ollama
from langchain_core.prompts import PromptTemplate

llm = Ollama(model="qwen:7b")

template = """您的工作是根据用户建议的区域制作一道经典菜肴。
%用户位置
{user_location}

AI回答:
"""

prompt_template = PromptTemplate(input_variables=["user_location"], template = template)

location_chain = LLMChain(llm=llm, prompt=prompt_template)

template2 = """给出一个简短的食谱,说明如何在家做这到菜
%菜谱
{user_meal}

AI回答
"""

prompt_template2 = PromptTemplate(input_variables=["user_meal"], template=template2)

meal_chain = LLMChain(llm = llm, prompt=prompt_template2)

#顺序链  就是将第一个链的输出做为第二个链的输入
overall_chain = SimpleSequentialChain(chains=[location_chain, meal_chain], verbose=True)

response = overall_chain.run("河北张家口")
print(response)

 

posted @ 2024-04-07 10:24  林**  阅读(7)  评论(0编辑  收藏  举报