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


# api_key = os.getenv("OPENAI_API_KEY")
api_key = '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'


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

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

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