diff --git a/graphrag/storage/sql_server_pipeline_storage.py b/graphrag/storage/sql_server_pipeline_storage.py index 2631bcfd..53d5c127 100644 --- a/graphrag/storage/sql_server_pipeline_storage.py +++ b/graphrag/storage/sql_server_pipeline_storage.py @@ -78,6 +78,17 @@ class SQLServerPipelineStorage(PipelineStorage): database_server_name, ) + if not self._autogenerate_tables: + self.create_tables() + + def create_tables(self) -> None: + """Create tables in SQL Server. + + If autogenerate_tables is not enabled, this method will manually create tables for all parquet outputs using a predefined schema. + """ + msg = "SQL Server storage does not yet support manually predefined table creation." + raise NotImplementedError(msg) + def find( self, file_pattern: re.Pattern[str],