File size: 354 Bytes
5bd6c7f
 
 
 
 
 
 
 
 
d02575f
5bd6c7f
1
2
3
4
5
6
7
8
9
10
11
from langchain.prompts import PromptTemplate
from langchain.llms import OpenAI
from langchain.chains import LLMChain

llm = OpenAI(temperature=0.9)
prompt = PromptTemplate(
    input_variables=["product"],
    template="What is a good name for a company that makes {product}?"
)
chain = LLMChain(llm=llm, prompt=prompt)
print(chain.run("colorful socks"))