# Definimos el prompt de forma clara y directa
prompt = (
"Tengo un dataframe llamado 'df' con las columnas 'años_experiencia_colaborador' y 'tiempo_entrega'. "
"Escribe el código Python con la biblioteca pandas para calcular la correlación entre las dos columnas. "
"Devuelve únicamente el bloque de código en formato Markdown, sin explicaciones."
)
# Invocamos al modelo
ai_msg = llm.invoke(prompt)
# Imprimimos la respuesta en una línea nueva
print(ai_msg.content)
import pandas as pd
correlacion = df['años_experiencia_colaborador'].corr(df['tiempo_entrega'])
print(correlacion)
