Langchain RunnableParallel

 https://medium.com/seaniap/%E5%BE%9E%E7%B0%A1%E5%96%AE%E5%88%B0%E8%A4%87%E9%9B%9C-%E8%A7%A3%E9%8E%96-lcel-%E6%B5%81%E7%A8%8B%E9%8F%88%E6%8A%80%E8%A1%93-1-2911dd0bce3f





# 引入語言模型
from langchain_openai import ChatOpenAI

# 引入PromptTemplate
from langchain_core.prompts import ChatPromptTemplate, PromptTemplate
from langchain_core.output_parsers import StrOutputParser
from langchain_core.runnables import RunnableParallel

# 建立 OpenAI 物件
chat_model = ChatOpenAI(model="gpt-3.5-turbo", api_key="<your_api_key_here>")

str_parser = StrOutputParser()

prompt1 = ChatPromptTemplate.from_template('{city} 位於哪個國家?')

chain1 = prompt1 | chat_model | str_parser

# 建立第二個 PromptTemplate
prompt2 = ChatPromptTemplate.from_template('在 {city} 的人都如何用餐?')

chain2 = prompt2 | chat_model | str_parser

country_and_style_chain = RunnableParallel(
country=chain1,
style=chain2
)

result = country_and_lang_chain.invoke({'city': '東京'})
print(result)

留言

此網誌的熱門文章