Create config.py

config.py

@@ -0,0 +1,33 @@
+# 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}"
+)