rwayz commited on
Commit
0441e1c
·
1 Parent(s): 5c9e296

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -5
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 10 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.\n"
134
- #"Essa base de dados representa o sellout de 2025, janeiro, fevereiro e março até dia 11, de uma farmácia.\n"
135
- #"Cada linha representa a venda de um SKU em uma determinada data.\n"
136
- "\nRetorne apenas a pergunta e a query SQL mais eficiente para entregar ao agent SQL do LangChain para gerar uma resposta. O formato deve ser:\n"
 
 
 
 
 
 
 
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>"