ArduinoYuri's picture
Create config.py
a386724 verified
# config.py
# INSTRUÇÕES PARA CONVERSÃO DE CONSULTAS
instruction_str = (
"1. Converta a consulta para código Python executável usando Pandas.\n"
"2. A linha final do código deve ser uma expressão Python que possa ser chamada com a função `eval()`.\n"
"3. O código deve representar uma solução para a consulta.\n"
"4. IMPRIMA APENAS A EXPRESSÃO FINAL.\n"
"5. Não coloque a expressão entre aspas.\n"
"6. Evite atribuições (=) na linha final - prefira expressões que retornem valores.\n"
)
# PROMPTS
pandas_prompt_str = (
"Você está trabalhando com um dataframe do pandas em Python chamado `df`.\n"
"{colunas_detalhes}\n\n"
"Este é o resultado de `print(df.head())`:\n"
"{df_str}\n\n"
"Siga estas instruções:\n"
"{instruction_str}\n"
"Consulta: {query_str}\n\n"
"Expressão:"
)
RESPONSE_SYNTHESIS_PROMPT_STR = (
"Dada uma pergunta de entrada, atue como analista de dados e elabore uma resposta a partir dos resultados da consulta.\n"
"Responda de forma natural, sem introduções como 'A resposta é:' ou algo semelhante.\n"
"Consulta: {query_str}\n\n"
"Instruções do Pandas (opcional):\n{pandas_instructions}\n\n"
"Saída do Pandas: {pandas_output}\n\n"
"Resposta: "
"Ao final, exibir o código usado para gerar a resposta, no formato: O código utilizado foi {pandas_instructions}"
)