Add SQL Server support

.github/wordlist.txt

@@ -121,3 +121,6 @@ pytest
 Radix
 Zod
 Dependabot
+pymssql
+sqlserver
+SQLServerLoader

api/core/schema_loader.py

@@ -14,6 +14,7 @@
 from api.loaders.postgres_loader import PostgresLoader
 from api.loaders.mysql_loader import MySQLLoader
 from api.loaders.snowflake_loader import SnowflakeLoader
+from api.loaders.sqlserver_loader import SQLServerLoader
 
 # Use the same delimiter as in the JavaScript frontend for streaming chunks
 MESSAGE_DELIMITER = "|||FALKORDB_MESSAGE_BOUNDARY|||"
@@ -48,6 +49,9 @@ def _step_detect_db_type(steps_counter: int, url: str) -> tuple[type[BaseLoader]
     elif url.startswith("snowflake://"):
         db_type = "snowflake"
         loader = SnowflakeLoader
+    elif url.startswith("sqlserver://"):
+        db_type = "sqlserver"
+        loader = SQLServerLoader
     else:
         raise InvalidArgumentError("Invalid database URL format")
 

api/core/text2sql.py

@@ -21,6 +21,7 @@
 from api.loaders.postgres_loader import PostgresLoader
 from api.loaders.mysql_loader import MySQLLoader
 from api.loaders.snowflake_loader import SnowflakeLoader
+from api.loaders.sqlserver_loader import SQLServerLoader
 from api.memory.graphiti_tool import MemoryTool
 from api.sql_utils import SQLIdentifierQuoter, DatabaseSpecificQuoter
 
@@ -87,6 +88,8 @@ def get_database_type_and_loader(db_url: str):
         return 'mysql', MySQLLoader
     if db_url_lower.startswith('snowflake://'):
         return 'snowflake', SnowflakeLoader
+    if db_url_lower.startswith('sqlserver://'):
+        return 'sqlserver', SQLServerLoader
 
     # Default to PostgresLoader for backward compatibility
     return 'postgresql', PostgresLoader