From 159e0f0d77655cb8f54db6a6b5b9fe8ee807b4a3 Mon Sep 17 00:00:00 2001 From: akmhmgc Date: Thu, 30 Mar 2023 08:39:25 +0900 Subject: [PATCH] chain --- main.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index fa06231..775559c 100644 --- a/main.py +++ b/main.py @@ -2,6 +2,7 @@ import os from langchain.llms import OpenAI from langchain.prompts import PromptTemplate +from langchain.chains import LLMChain prompt = PromptTemplate( input_variables=["product"], @@ -16,5 +17,11 @@ llm = OpenAI(temperature=0.9, openai_api_key=openai_api_key) -text = "What would be a good company name for a company that makes colorful socks?" -print(llm(prompt.format(product="colorful socks"))) +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'))