Update app.py
Browse files
app.py
CHANGED
@@ -12,7 +12,7 @@ import logging
|
|
12 |
|
13 |
load_dotenv()
|
14 |
|
15 |
-
CSV_FILE_PATH = "
|
16 |
SQL_DB_PATH = "data.db"
|
17 |
HUGGINGFACE_API_KEY = os.getenv("HUGGINGFACE_API_KEY")
|
18 |
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
|
@@ -100,7 +100,7 @@ def query_sql_agent(user_query):
|
|
100 |
query_cache[user_query] = greeting_response # Armazena saudação no cache
|
101 |
return greeting_response
|
102 |
|
103 |
-
column_data = pd.read_sql_query("SELECT * FROM
|
104 |
llama_instruction = query_with_llama(user_query, column_data)
|
105 |
if not llama_instruction:
|
106 |
return "Erro: O modelo Llama não conseguiu gerar uma instrução válida."
|
|
|
12 |
|
13 |
load_dotenv()
|
14 |
|
15 |
+
CSV_FILE_PATH = "tabela_tiago_formated.csv"
|
16 |
SQL_DB_PATH = "data.db"
|
17 |
HUGGINGFACE_API_KEY = os.getenv("HUGGINGFACE_API_KEY")
|
18 |
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
|
|
|
100 |
query_cache[user_query] = greeting_response # Armazena saudação no cache
|
101 |
return greeting_response
|
102 |
|
103 |
+
column_data = pd.read_sql_query("SELECT * FROM tabela_tiago_formated LIMIT 10", engine)
|
104 |
llama_instruction = query_with_llama(user_query, column_data)
|
105 |
if not llama_instruction:
|
106 |
return "Erro: O modelo Llama não conseguiu gerar uma instrução válida."
|