from langchain.prompts import PromptTemplate
from langchain.llms import OpenAI
from langchain.chains import LLMChain

from dotenv import load_dotenv, find_dotenv
_ = load_dotenv(find_dotenv()) # read local .env file
#openai.api_key  = os.environ['OPENAI_API_KEY']

llm_basic = OpenAI(verbose=True,
                   temperature=0.0
                  )

# Para usar o LLMChain, primeiro crie um modelo de prompt:
expand_prompt = PromptTemplate(
    input_variables=["technology"],
    template="""Escreva um parágrafo explicando a seguinte tecnologia: {technology}. \
                A sua resposta deverá conter como máximo 30 palavras.""",
)

# Agora podemos criar uma cadeia muito simples que receberá a entrada do usuário,
# formatará o prompt com ela e depois a enviará ao LLM.
expand_chain = LLMChain(llm=llm_basic, prompt=expand_prompt)

print(expand_chain.run("Linguagem Python"))