udpate sql parse

aigc-apps · Dec 10, 2024 · 7ffe6f4 · 7ffe6f4
1 parent 581cf8b
commit 7ffe6f4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/pai_rag/integrations/data_analysis/nl2sql_retriever.py b/src/pai_rag/integrations/data_analysis/nl2sql_retriever.py
@@ -253,6 +253,7 @@ def parse_response_to_sql(self, response: str, query_bundle: QueryBundle) -> str
             response = response[:sql_query_end].rstrip().replace("```", "")
         return response.strip().replace("```", "").lstrip("sql")
 
+
 def get_sql_info(sql_config: SqlAnalysisConfig):
     if isinstance(sql_config, SqliteAnalysisConfig):
         db_path = os.path.join(sql_config.db_path, sql_config.database)