| | from langchain_openai import OpenAI |
| | from langchain.prompts import PromptTemplate |
| | from langchain.chains import LLMChain |
| | from langchain.chains import SimpleSequentialChain |
| | from secret_key import openapi_key |
| |
|
| | import os |
| | os.environ['OPENAI_API_KEY'] = openapi_key |
| |
|
| | llm = OpenAI(temperature=0.6) |
| |
|
| | prompt_template_name = PromptTemplate( |
| | input_variables = ['cuisine'], |
| | template = "I want to open a restaurant for {cuisine} food. Suggest a fancy name for this." |
| | ) |
| |
|
| | name_chain = LLMChain(llm=llm, prompt=prompt_template_name) |
| |
|
| | prompt_template_items = PromptTemplate( |
| | input_variables = ['restaurant_name'], |
| | template = "Suggest some menu items for {restaurant_name}. Return it as a comma separated list. " |
| | ) |
| |
|
| | food_items_chain = LLMChain(llm=llm, prompt=prompt_template_items) |
| |
|
| | chain = SimpleSequentialChain(chains = [name_chain, food_items_chain]) |
| |
|
| | response = chain.invoke("American") |
| | print(response) |
| | |