Skip to content

Commit eef2b5e

Browse files
committed
fix logic for mssql db creation
1 parent b1432db commit eef2b5e

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

vector_db/mssql_provider.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -134,9 +134,14 @@ def _ensure_database_exists(self) -> None:
134134
f"IF DB_ID('{database}') IS NULL CREATE DATABASE [{database}]"
135135
)
136136
cursor.close()
137-
except Exception as e:
137+
except pyodbc.ProgrammingError as e:
138+
if "1801" in str(e):
139+
logger.info("Database %s already exists, continuing", database)
140+
return
138141
logger.exception("Failed to ensure database '%s' exists", database)
139-
raise RuntimeError(f"Failed to ensure database '{database}' exists: {e}")
142+
raise RuntimeError(
143+
f"Failed to ensure database '{database}' exists: {e}"
144+
) from e
140145

141146
def add_documents(self, docs: List[Document]) -> None:
142147
"""

0 commit comments

Comments
 (0)