Update utils.py

utils.py

@@ -141,4 +141,10 @@ def addSchemaToTableInSQL(sqlQuery, schemaName, tablesList):
     pattern = re.compile(rf'(?<![a-zA-Z0-9_]){re.escape(table)}(?![a-zA-Z0-9_])', re.IGNORECASE)
     replacement = f'{schemaName}.{table}'
     sqlQuery = re.sub(pattern, replacement, sqlQuery)
-  return sqlQuery
+  return sqlQuery
+
+def preProcessGptQueryReponse(gptResponse, metadataLayout: MetaDataLayout):
+   schemaName = metadataLayout.schemaName
+   tablesList = metadataLayout.getAllTablesCols().keys()
+   gptResponse = addSchemaToTableInSQL(gptResponse, schemaName=schemaName, tablesList=tablesList)
+   return gptResponse