执行多个chain

因为他是链式的，所以他也可以按顺序依次去执行多个 chain

  1. from langchain.llms import OpenAI
  2. from langchain.chains import LLMChain
  3. from langchain.prompts import PromptTemplate
  4. from langchain.chains import SimpleSequentialChain
  6. # location 链
  7. llm = OpenAI(temperature=1)
  8. template = """Your job is to come up with a classic dish from the area that the users suggests.
  9. % USER LOCATION
  10. {user_location}
  12. YOUR RESPONSE:
  13. """
  14. prompt_template = PromptTemplate(input_variables=["user_location"], template=template)
  15. location_chain = LLMChain(llm=llm, prompt=prompt_template)
  17. # meal 链
  18. template = """Given a meal, give a short and simple recipe on how to make that dish at home.
  19. % MEAL
  20. {user_meal}
  22. YOUR RESPONSE:
  23. """
  24. prompt_template = PromptTemplate(input_variables=["user_meal"], template=template)
  25. meal_chain = LLMChain(llm=llm, prompt=prompt_template)
  27. # 通过 SimpleSequentialChain 串联起来，第一个答案会被替换第二个中的user_meal，然后再进行询问
  28. overall_chain = SimpleSequentialChain(chains=[location_chain, meal_chain], verbose=True)
  29. review = overall_chain.run("Rome")

image-20230406000133339