|
from langchain_openai import OpenAI |
|
from langchain.prompts import PromptTemplate |
|
from langchain.chains import LLMChain |
|
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." |
|
) |
|
|
|
chain = LLMChain(llm=llm, prompt=prompt_template_name) |
|
response = chain.invoke({"cuisine":"American"}) |
|
print(response) |
|
|