from langchain_openai import ChatOpenAI
from langchain_core.prompts import PromptTemplate
from dotenv import load_dotenv
import os
load_dotenv()
llm = ChatOpenAI(
base_url=os.getenv("BASE_URL"),
api_key=os.getenv("API_KEY"),
model=os.getenv("MODEL_ALTERNATIVO"),
temperature=0.7
)
prompt = PromptTemplate.from_template("""
Crie um roteiro de viagem de {dias} dias.
A viagem terá {criancas} crianças.
A principal atividade desejada é: {atividade}.
Organize o roteiro por dia, sugerindo atividades
adequadas para crianças.
""")
chain = prompt | llm
response = chain.invoke({
"dias": 5,
"criancas": 2,
"atividade": "praia"
})
print(response.content)
