import os
# LLM ラッパーをインポート
from langchain.llms import OpenAI

# `PromptTemplate` をインポート
from langchain.prompts import PromptTemplate
# from langchain.memory import ConversationBufferMemory

api_key = os.getenv("OPENAI_API_KEY")


# # LLM ラッパーを初期化
# llm = OpenAI(openai_api_key=api_key, temperature=0.7)

# # LLM に渡す入力テキスト
# text = "カラフルな靴下を作る会社の社名として、何かいいものはないですか？日本語の社名でお願いします。"

# # LLM から予測を受け取って表示
# prediction = llm(text)
# print(prediction.strip())


# プロンプトテンプレートの作成
prompt = PromptTemplate(
    input_variables=["product"],
    template="{product}を作る会社の社名として、何かいいものはないですか？日本語の社名でお願いします。",
)

# テンプレートからプロンプトを作成
prompt = prompt.format(product="カラフルな靴下")
print(prompt)