datacontract-cli 1.2.4__tar.gz → 1.2.6__tar.gz

{datacontract_cli-1.2.4/datacontract_cli.egg-info → datacontract_cli-1.2.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datacontract-cli
-Version: 1.2.4
+Version: 1.2.6
 Summary: The datacontract CLI is an open source command-line tool for working with Data Contracts. It uses data contract YAML files to lint the data contract, connect to data sources and execute schema and quality tests, detect breaking changes, and export to different formats. The tool is written in Python. It can be used as a standalone CLI tool, in a CI/CD pipeline, or directly as a Python library.
 Author-email: Jochen Christ <jochen.christ@entropy-data.com>, Stefan Negele <stefan.negele@innoq.com>, Simon Harrer <simon.harrer@entropy-data.com>
 License-Expression: MIT
@@ -29,7 +29,7 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: typer<0.28,>=0.18.0
 Requires-Dist: click<9.0.0,>=8.1.0
-Requires-Dist: pydantic<2.14.0,>=2.8.2
+Requires-Dist: pydantic<2.15.0,>=2.8.2
 Requires-Dist: pydantic-settings<3.0.0,>=2.5.0
 Requires-Dist: pyyaml~=6.0.1
 Requires-Dist: ruamel.yaml<0.20,>=0.17
@@ -70,7 +70,7 @@ Requires-Dist: rich>=12.4.4; extra == "dataframe"
 Provides-Extra: databricks
 Requires-Dist: ibis-framework[databricks]<13.0.0,>=10.0.0; extra == "databricks"
 Requires-Dist: databricks-sql-connector<4.7.0,>=3.7.0; extra == "databricks"
-Requires-Dist: databricks-sdk<0.143.0; extra == "databricks"
+Requires-Dist: databricks-sdk<0.152.0; extra == "databricks"
 Requires-Dist: packaging<27.0,>=21.3; extra == "databricks"
 Provides-Extra: databricks-runtime
 Requires-Dist: datacontract-cli[databricks]; extra == "databricks-runtime"
@@ -101,7 +101,7 @@ Requires-Dist: datacontract-cli[s3]; extra == "gcs"
 Provides-Extra: azure
 Requires-Dist: datacontract-cli[duckdb]; extra == "azure"
 Requires-Dist: duckdb-extension-azure<1.6.0,>=1.5.0; extra == "azure"
-Requires-Dist: azure-identity<1.26,>=1.17.1; extra == "azure"
+Requires-Dist: azure-identity<1.27,>=1.17.1; extra == "azure"
 Requires-Dist: azure-storage-blob<12.31,>=12.24; extra == "azure"
 Provides-Extra: snowflake
 Requires-Dist: snowflake-connector-python[pandas]<4.9,>=3.6; extra == "snowflake"
@@ -161,12 +161,12 @@ Requires-Dist: pre-commit<4.7.0,>=3.7.1; extra == "dev"
 Requires-Dist: pytest; extra == "dev"
 Requires-Dist: pytest-cov; extra == "dev"
 Requires-Dist: pytest-xdist; extra == "dev"
-Requires-Dist: pymssql==2.4.2; extra == "dev"
+Requires-Dist: pymssql==2.4.4; extra == "dev"
 Requires-Dist: psycopg2-binary<3.0,>=2.9; extra == "dev"
 Requires-Dist: mysql-connector-python<26.8.0,>=8.0.30; extra == "dev"
 Requires-Dist: ruff==0.16.10; extra == "dev"
 Requires-Dist: testcontainers[kafka,minio,mssql,mysql,postgres]==4.15.0; extra == "dev"
-Requires-Dist: trino==0.339.0; extra == "dev"
+Requires-Dist: trino==0.340.0; extra == "dev"
 Dynamic: license-file
 
 # Data Contract CLI
@@ -183,6 +183,7 @@ Dynamic: license-file
   <a href="https://datacontract.com/slack" rel="nofollow"><img src="https://img.shields.io/badge/slack-join_chat-white.svg?logo=slack&amp;style=social" alt="Slack Status" data-canonical-src="https://img.shields.io/badge/slack-join_chat-white.svg?logo=slack&amp;style=social" style="max-width: 100%;"></a>
 </p>
 
+**Turn your schemas into data contracts and test your data against them, on Snowflake, Databricks, BigQuery, Kafka, Postgres & 15 more. Open source, built on the Open Data Contract Standard.**
 The `datacontract` CLI is an open-source command-line tool for working with [data contracts](https://datacontract.com).
 It natively supports the [Open Data Contract Standard](https://bitol-io.github.io/open-data-contract-standard/latest/) to lint data contracts, connect to data sources and execute schema and quality tests, and export to different formats. 
 The tool is written in Python. 

{datacontract_cli-1.2.4 → datacontract_cli-1.2.6}/README.md

@@ -12,6 +12,7 @@
   <a href="https://datacontract.com/slack" rel="nofollow"><img src="https://img.shields.io/badge/slack-join_chat-white.svg?logo=slack&amp;style=social" alt="Slack Status" data-canonical-src="https://img.shields.io/badge/slack-join_chat-white.svg?logo=slack&amp;style=social" style="max-width: 100%;"></a>
 </p>
 
+**Turn your schemas into data contracts and test your data against them, on Snowflake, Databricks, BigQuery, Kafka, Postgres & 15 more. Open source, built on the Open Data Contract Standard.**
 The `datacontract` CLI is an open-source command-line tool for working with [data contracts](https://datacontract.com).
 It natively supports the [Open Data Contract Standard](https://bitol-io.github.io/open-data-contract-standard/latest/) to lint data contracts, connect to data sources and execute schema and quality tests, and export to different formats. 
 The tool is written in Python. 

{datacontract_cli-1.2.4 → datacontract_cli-1.2.6}/datacontract/api.py

@@ -275,6 +275,8 @@ tenants. Send them over HTTPS only.
 See [Configuration](https://docs.datacontract.com/configuration) for the full list of options.
 """
 
+# The endpoints are plain `def`, not `async def`: they block on data sources, and FastAPI runs
+# plain functions in a thread pool, so one slow test does not stall every other request.
 app = FastAPI(
     docs_url="/",
     title="Data Contract CLI API",
@@ -763,7 +765,7 @@ handled. A contract that cannot be parsed is reported the same way, as a failed
     response_model_exclude_none=True,
     response_model_exclude_unset=True,
 )
-async def test(
+def test(
     request: Request,
     body: Annotated[
         str,
@@ -870,7 +872,7 @@ violation — not as an error status. A `schema` that is not an `http(s)` URL is
         },
     },
 )
-async def lint(
+def lint(
     body: Annotated[
         str,
         Body(
@@ -968,7 +970,7 @@ parsed is answered with `422`.
         },
     },
 )
-async def changelog_endpoint(
+def changelog_endpoint(
     body: ChangelogRequest,
     api_key: Annotated[str | None, Depends(api_key_header)] = None,
 ) -> ChangelogResponse:

{datacontract_cli-1.2.4 → datacontract_cli-1.2.6}/datacontract/command_edit.py

@@ -74,7 +74,29 @@ def _cli_version() -> str:
         return "unknown"
 
 
-def _generate_index_html(filename: str) -> str:
+# The editor's AI assistant is configured via these environment variables (e.g., in .env).
+# It is enabled when DATACONTRACT_EDITOR_AI_ENDPOINT is set.
+EDITOR_AI_ENV_VARS = {
+    "provider": "DATACONTRACT_EDITOR_AI_PROVIDER",
+    "endpoint": "DATACONTRACT_EDITOR_AI_ENDPOINT",
+    "apiKey": "DATACONTRACT_EDITOR_AI_API_KEY",
+    "model": "DATACONTRACT_EDITOR_AI_MODEL",
+    "authHeader": "DATACONTRACT_EDITOR_AI_AUTH_HEADER",
+}
+
+
+def editor_ai_config() -> dict | None:
+    """Return the editor's AI assistant config from the environment, or None if no endpoint is set."""
+    if not os.getenv(EDITOR_AI_ENV_VARS["endpoint"]):
+        return None
+    ai = {"enabled": True}
+    for key, env_var in EDITOR_AI_ENV_VARS.items():
+        if os.getenv(env_var):
+            ai[key] = os.getenv(env_var)
+    return ai
+
+
+def _generate_index_html(filename: str, ai_config: dict | None = None) -> str:
     """
     Generate the editor page: load the YAML from the local file API, write it back on
     save, and point the editor's test runner at this server's own /test endpoint.
@@ -82,8 +104,10 @@ def _generate_index_html(filename: str) -> str:
     The editor runs in EMBEDDED mode, since the file menu (New, Load Example, Open)
     makes no sense when the editor is bound to a single local file. The filename is
     shown in the header via titlePrefix, and Cancel reverts to the file on disk.
+    The AI assistant is configured if ai_config is given.
     """
     filename_js = json.dumps(filename)
+    ai_option = f"ai: {json.dumps(ai_config)}," if ai_config else ""
     file_api_path = f"/api/files/{quote(filename)}"
     return f"""<!doctype html>
 <html lang="en">
@@ -136,6 +160,7 @@ def _generate_index_html(filename: str) -> str:
               enabled: true,
               dataContractCliApiServerUrl: window.location.origin,
             }},
+            {ai_option}
             onCancel: async () => {{
               try {{
                 const response = await fetch('{file_api_path}');
@@ -213,7 +238,7 @@ def create_app(
 
     @edit_app.get("/", response_class=HTMLResponse, include_in_schema=False)
     def index():
-        return _generate_index_html(filename)
+        return _generate_index_html(filename, editor_ai_config())
 
     @edit_app.get(EDITOR_ASSETS_PATH + "/{asset_path:path}", include_in_schema=False)
     def editor_asset(asset_path: str):
@@ -337,6 +362,9 @@ def edit(
     the data contract tests locally against the servers defined in the data contract.
     Credentials for the data sources must be provided as environment variables, see
     https://docs.datacontract.com/testing
+    The editor's AI assistant is enabled by setting DATACONTRACT_EDITOR_AI_ENDPOINT, together with
+    DATACONTRACT_EDITOR_AI_API_KEY, DATACONTRACT_EDITOR_AI_MODEL, DATACONTRACT_EDITOR_AI_PROVIDER
+    (openai or anthropic), and DATACONTRACT_EDITOR_AI_AUTH_HEADER (bearer, api-key, or x-api-key).
     """
     enable_debug_logging(debug)
 

{datacontract_cli-1.2.4 → datacontract_cli-1.2.6}/datacontract/config/settings.py

@@ -13,6 +13,9 @@ logger = logging.getLogger(__name__)
 
 _ENV_PREFIX = "DATACONTRACT_"
 
+# Used by the drivers that would otherwise wait forever for an unresponsive data source.
+DEFAULT_CONNECT_TIMEOUT = 30
+
 # Deprecated option (field name) -> replacement (field name). Single source for
 # the accessor warnings, the Snowflake connect synonyms, and the generated docs.
 DEPRECATED_OPTIONS = {
@@ -112,6 +115,7 @@ class Config(BaseSettings):
     api_header_authorization: SecretStr | None = None
     custom_quality_checks: str | None = None
     max_errors: int | None = None
+    connect_timeout: int | None = None
 
     # athena (credentials come from the s3_* options)
     # overrides for the contract's servers block
@@ -530,6 +534,11 @@ class Config(BaseSettings):
     def get_max_errors(self) -> int | None:
         return self._int_option("max_errors")
 
+    # --- connect timeout ---
+    def get_connect_timeout(self) -> int:
+        """Seconds to wait for a data source to accept a connection, for drivers that wait forever by default."""
+        return self._int_option("connect_timeout") or DEFAULT_CONNECT_TIMEOUT
+
     # --- athena ---
     def get_athena_catalog(self, required: bool = False) -> str | None:
         return self._str_option("athena_catalog", required)

{datacontract_cli-1.2.4 → datacontract_cli-1.2.6}/datacontract/data_contract.py

@@ -32,6 +32,13 @@ from datacontract.model.run import Check, ResultEnum, Run
 logger = logging.getLogger(__name__)
 
 
+def _warning_checks(warnings: list[DataContractException]) -> list[Check]:
+    return [
+        Check(type=warning.type, result=warning.result, name=warning.name, reason=warning.reason, engine=warning.engine)
+        for warning in warnings
+    ]
+
+
 def _unrunnable_rule_checks(data_contract: OpenDataContractStandard) -> list[Check]:
     """Warnings for quality rules `datacontract test` would not be able to run."""
     checks = []
@@ -150,7 +157,7 @@ class DataContract:
         run = Run.create_run()
         try:
             run.log_info("Linting data contract")
-            data_contract, schema_version = resolve.resolve_data_contract_with_schema_version(
+            data_contract, schema_version, warnings = resolve.resolve_data_contract_with_lint_info(
                 self._data_contract_file,
                 self._data_contract_str,
                 self._data_contract,
@@ -171,6 +178,7 @@ class DataContract:
                     engine="datacontract-cli",
                 )
             )
+            run.checks.extend(_warning_checks(warnings))
             run.checks.extend(_unrunnable_rule_checks(data_contract))
             run.dataContractId = data_contract.id
             run.dataContractVersion = data_contract.version
@@ -221,7 +229,7 @@ class DataContract:
         try:
             run.log_info("Testing data contract")
             run.log_info(self._runtime_info())
-            data_contract = resolve.resolve_data_contract(
+            data_contract, _, warnings = resolve.resolve_data_contract_with_lint_info(
                 self._data_contract_file,
                 self._data_contract_str,
                 self._data_contract,
@@ -230,6 +238,7 @@ class DataContract:
                 config=self._config,
                 configured_host_only=self._untrusted_contract,
             )
+            run.checks.extend(_warning_checks(warnings))
 
             execute_data_contract_test(
                 data_contract,

datacontract_cli-1.2.6/datacontract/editor_assets/VERSION

@@ -0,0 +1 @@
+0.1.15