Spaces:
Runtime error
Runtime error
Commit ·
9a04841
1
Parent(s): 714fdb8
bug fix
Browse files- queryHelperManager.py +2 -1
queryHelperManager.py
CHANGED
|
@@ -50,7 +50,8 @@ class QueryHelper:
|
|
| 50 |
for col in selectedTablesAndCols[table]:
|
| 51 |
if col in prospectTablesAndColsText:
|
| 52 |
prospectTablesAndCols[table].append(col)
|
| 53 |
-
|
|
|
|
| 54 |
self.gptInstanceForQuery.setSystemPrompt(promptForQuery)
|
| 55 |
gptResponse = self.gptInstanceForQuery.getResponseForUserInput(userInput)
|
| 56 |
return gptResponse
|
|
|
|
| 50 |
for col in selectedTablesAndCols[table]:
|
| 51 |
if col in prospectTablesAndColsText:
|
| 52 |
prospectTablesAndCols[table].append(col)
|
| 53 |
+
print("tables and cols select by gpt", prospectTablesAndCols)
|
| 54 |
+
promptForQuery = self.getSystemPromptForQuery(prospectTablesAndCols)
|
| 55 |
self.gptInstanceForQuery.setSystemPrompt(promptForQuery)
|
| 56 |
gptResponse = self.gptInstanceForQuery.getResponseForUserInput(userInput)
|
| 57 |
return gptResponse
|