Update app.py
Browse files
app.py
CHANGED
@@ -128,12 +128,19 @@ sql_agent = create_sql_agent(llm, db=db, agent_type="openai-tools", verbose=True
|
|
128 |
|
129 |
def generate_initial_context(db_sample):
|
130 |
return (
|
131 |
-
f"Você é um assistente que gera queries SQL objetivas e eficientes. Sempre inclua LIMIT
|
132 |
f"Exemplos do banco de dados:\n{db_sample.head().to_string(index=False)}\n\n"
|
133 |
-
"\n***IMPORTANTE***: Detecte automaticamente o idioma da pergunta do usuário e responda sempre no mesmo idioma
|
134 |
-
|
135 |
-
|
136 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
137 |
"\nPergunta: <pergunta do usuário>\n"
|
138 |
"\nOpção de Query SQL:\n<query SQL>"
|
139 |
"\nIdioma: <idioma>"
|
|
|
128 |
|
129 |
def generate_initial_context(db_sample):
|
130 |
return (
|
131 |
+
f"Você é um assistente que gera queries SQL objetivas e eficientes. Sempre inclua LIMIT 20 nas queries. Aqui está o banco de dados:\n\n"
|
132 |
f"Exemplos do banco de dados:\n{db_sample.head().to_string(index=False)}\n\n"
|
133 |
+
"\n***IMPORTANTE***: Detecte automaticamente o idioma da pergunta do usuário e responda sempre no mesmo idioma."
|
134 |
+
"\nEsta base contém os SKUs (produtos) que foram promocionados por meio de TABLOIDE OU PROMOCAO OU ANUNCIO.\n"
|
135 |
+
"Cada linha representa um SKU OU PRODUTO único PRESENTE NO TABLOIDE OU PROMOCAO OU ANUNCIO, incluindo sua descrição completa, os veículos OU MIDIAS de promoção utilizados e o respectivo período em que a promoção ocorreu.\n"
|
136 |
+
|
137 |
+
"\nInformações imporatantes:\n"
|
138 |
+
"- Use `LIKE '%<palavras-chave>%'` para buscas em colunas de texto.\n"
|
139 |
+
"- Quando o usuário mencionar uma categoria, procure nas colunas: `CATEGORIA_PRODUTO_SKU`.\n"
|
140 |
+
"- Se o usuário se referir a Nestle, o jeito correto de se escrever é Nestle sem acento e não Nestlé.\n"
|
141 |
+
"- Você está usando um banco de dados SQLite.\n"
|
142 |
+
|
143 |
+
"\nRetorne apenas a pergunta e a query SQL mais eficiente para entregar ao agent SQL do LangChain para gerar uma resposta para a pergunta. O formato deve ser:\n"
|
144 |
"\nPergunta: <pergunta do usuário>\n"
|
145 |
"\nOpção de Query SQL:\n<query SQL>"
|
146 |
"\nIdioma: <idioma>"
|