databao-context-engine 0.1.1__tar.gz → 0.1.2__tar.gz

{databao_context_engine-0.1.1 → databao_context_engine-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: databao-context-engine
-Version: 0.1.1
+Version: 0.1.2
 Summary: Add your description here
 Requires-Dist: click>=8.3.0
 Requires-Dist: duckdb>=1.4.3
@@ -11,13 +11,14 @@ Requires-Dist: clickhouse-connect>=0.10.0
 Requires-Dist: mcp>=1.23.3
 Requires-Dist: pyathena>=3.22.0
 Requires-Dist: snowflake-connector-python>=4.1.0
-Requires-Dist: mssql-python>=1.0.0
 Requires-Dist: pydantic>=2.12.4
 Requires-Dist: jinja2>=3.1.6
 Requires-Dist: asyncpg>=0.31.0
 Requires-Dist: asyncio>=4.0.0
 Requires-Dist: asyncpg-stubs>=0.31.1
+Requires-Dist: mssql-python>=1.0.0 ; extra == 'mssql'
 Requires-Python: >=3.12
+Provides-Extra: mssql
 Description-Content-Type: text/markdown
 
 [![official project](https://jb.gg/badges/official.svg)](https://confluence.jetbrains.com/display/ALL/JetBrains+on+GitHub)

{databao_context_engine-0.1.1 → databao_context_engine-0.1.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "databao-context-engine"
-version = "0.1.1"
+version = "0.1.2"
 description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.12"
@@ -14,7 +14,6 @@ dependencies = [
     "mcp>=1.23.3",
     "pyathena>=3.22.0",
     "snowflake-connector-python>=4.1.0",
-    "mssql-python>=1.0.0",
     "pydantic>=2.12.4",
     "jinja2>=3.1.6",
     "asyncpg>=0.31.0",
@@ -22,6 +21,11 @@ dependencies = [
     "asyncpg-stubs>=0.31.1",
 ]
 
+[project.optional-dependencies]
+mssql = [
+    "mssql-python>=1.0.0"
+]
+
 [build-system]
 requires = ["uv_build>=0.9.6,<0.10.0"]
 build-backend = "uv_build"
@@ -44,7 +48,7 @@ dce = "databao_context_engine.main:main"
 generate_configs_schemas = "databao_context_engine.generate_configs_schemas:main"
 
 [tool.uv]
-override-dependencies = ["urllib3>=2.6.3"]
+override-dependencies = ["urllib3>=2.6.3", "filelock>=3.20.3"]
 
 [tool.uv.build-backend]
 source-exclude = ["tests"]
@@ -55,3 +59,4 @@ ignore_missing_imports = true
 
 [tool.pytest]
 addopts = ["--import-mode=importlib"]
+strict = true

{databao_context_engine-0.1.1 → databao_context_engine-0.1.2}/src/databao_context_engine/plugins/plugin_loader.py

@@ -67,22 +67,30 @@ def _load_builtin_datasource_plugins() -> list[BuildDatasourcePlugin]:
     from databao_context_engine.plugins.athena_db_plugin import AthenaDbPlugin
     from databao_context_engine.plugins.clickhouse_db_plugin import ClickhouseDbPlugin
     from databao_context_engine.plugins.duckdb_db_plugin import DuckDbPlugin
-    from databao_context_engine.plugins.mssql_db_plugin import MSSQLDbPlugin
     from databao_context_engine.plugins.mysql_db_plugin import MySQLDbPlugin
     from databao_context_engine.plugins.parquet_plugin import ParquetPlugin
     from databao_context_engine.plugins.postgresql_db_plugin import PostgresqlDbPlugin
     from databao_context_engine.plugins.snowflake_db_plugin import SnowflakeDbPlugin
 
-    return [
+    # optional plugins are added to the python environment via extras
+    optional_plugins: list[BuildDatasourcePlugin] = []
+    try:
+        from databao_context_engine.plugins.mssql_db_plugin import MSSQLDbPlugin
+
+        optional_plugins = [MSSQLDbPlugin()]
+    except ImportError:
+        pass
+
+    required_plugins: list[BuildDatasourcePlugin] = [
         AthenaDbPlugin(),
         ClickhouseDbPlugin(),
         DuckDbPlugin(),
-        MSSQLDbPlugin(),
         MySQLDbPlugin(),
         PostgresqlDbPlugin(),
         SnowflakeDbPlugin(),
         ParquetPlugin(),
     ]
+    return required_plugins + optional_plugins
 
 
 def _load_external_plugins(exclude_file_plugins: bool = False) -> list[BuildPlugin]:

{databao_context_engine-0.1.1 → databao_context_engine-0.1.2}/src/databao_context_engine/project/datasource_discovery.py

@@ -74,7 +74,7 @@ def get_datasource_descriptors(project_dir: Path, datasource_ids: list[Datasourc
         raise ValueError(f"src directory does not exist in {project_dir}")
 
     datasources: list[DatasourceDescriptor] = []
-    for datasource_id in datasource_ids:
+    for datasource_id in sorted(datasource_ids, key=lambda id: str(id)):
         config_file_path = src.joinpath(datasource_id.relative_path_to_config_file())
         if not config_file_path.is_file():
             raise ValueError(f"Datasource config file not found: {config_file_path}")