diff --git a/src/pai_rag/integrations/data_analysis/nl2sql_retriever.py b/src/pai_rag/integrations/data_analysis/nl2sql_retriever.py index b140d5b9..b3e30118 100644 --- 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)