from langchain_core.output_parsers.openai_tools import JsonOutputKeyToolsParser
# 1. Configuramos el parser apuntando al nombre exacto de nuestra función/herramienta
parser = JsonOutputKeyToolsParser(key_name="ejecutar_codigo_python", first_tool_only=True)
# 2. Armamos la cadena usando LCEL (LangChain Expression Language)
cadena = llm_con_herramienta | parser
# 3. Invocamos la cadena con el prompt
respuesta = cadena.invoke("""
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 el script de Python únicamente.
""")
# 4. Accedemos a la clave correcta que generó el modelo ('codigo')
print("Código extraído automáticamente por la cadena:")
print(respuesta["codigo"])
Resultado:
Código extraído automáticamente por la cadena:
import pandas as pd
print(df['años_experiencia_colaborador'].corr(df['tiempo_entrega']))
