Merge branch 'main' into support-auth

Author: jerryshao

.github/workflows/ci.yml

@@ -1,3 +1,17 @@
+# Copyright 2026 Datastrato, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
 name: CI
 
 on:
@@ -26,6 +40,9 @@ jobs:
       - name: Install dependencies
         run: uv sync --all-extras
 
+      - name: Check license headers
+        run: uv run python scripts/check-license.py
+
       - name: Run ruff check
         run: uv run ruff check src tests
 

CONTRIBUTING.md

@@ -0,0 +1,33 @@
+# Contributing to ADP Python SDK
+
+Thank you for your interest in contributing to the ADP Python SDK!
+
+## How to Contribute
+
+1. Fork the repository and create your branch from `main`.
+2. Make your changes and ensure any new code is covered by tests.
+3. Submit a pull request with a clear description of the change.
+
+## License
+
+By contributing, you agree that your contributions will be licensed under the [Apache License 2.0](LICENSE).
+
+## License Header
+
+Every source file must include the following header:
+
+```
+# Copyright 2026 Datastrato, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+```

LICENSE

@@ -1,3 +1,19 @@
+Copyright 2026 Datastrato, Inc.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+
+---
+
                                  Apache License
                            Version 2.0, January 2004
                         http://www.apache.org/licenses/

NOTICE

@@ -0,0 +1,2 @@
+ADP Python SDK
+Copyright 2026 Datastrato, Inc.

scripts/check-license.py

@@ -0,0 +1,75 @@
+# Copyright 2026 Datastrato, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Check that all source files contain the required license header."""
+
+import os
+import sys
+
+COPYRIGHT_MARKER = "Copyright 2026 Datastrato, Inc."
+
+SCAN_RULES = [
+    {"dirs": ["src", "tests"], "ext": ".py"},
+    {"dirs": [os.path.join(".github", "workflows")], "ext": ".yml"},
+]
+
+SKIP_DIRS = {".venv", "dist", "__pycache__", ".mypy_cache", ".pytest_cache", ".ruff_cache"}
+SKIP_FILES = {os.path.join("src", "adp_sdk", "py.typed")}
+
+
+def should_skip_dir(dirname: str) -> bool:
+    return dirname in SKIP_DIRS
+
+
+def collect_files() -> list[str]:
+    files: list[str] = []
+    for rule in SCAN_RULES:
+        for base_dir in rule["dirs"]:
+            for root, dirs, filenames in os.walk(base_dir):
+                dirs[:] = [d for d in dirs if not should_skip_dir(d)]
+                for filename in filenames:
+                    if filename.endswith(rule["ext"]):
+                        filepath = os.path.join(root, filename)
+                        if filepath not in SKIP_FILES:
+                            files.append(filepath)
+    return sorted(files)
+
+
+def check_header(filepath: str) -> bool:
+    with open(filepath, encoding="utf-8") as f:
+        content = f.read()
+    return COPYRIGHT_MARKER in content
+
+
+def main() -> int:
+    files = collect_files()
+    missing: list[str] = []
+
+    for filepath in files:
+        if not check_header(filepath):
+            missing.append(filepath)
+
+    if missing:
+        print("Files missing license header:")
+        for f in missing:
+            print(f"  {f}")
+        print(f"\n{len(missing)} file(s) missing the required license header.")
+        return 1
+
+    print(f"All {len(files)} file(s) have the required license header.")
+    return 0
+
+
+if __name__ == "__main__":
+    sys.exit(main())

src/adp_sdk/__init__.py

@@ -1,3 +1,17 @@
+# Copyright 2026 Datastrato, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
 """ADP SDK - Python SDK for the Agentic Data Protocol."""
 
 from collections.abc import AsyncIterator

src/adp_sdk/client/__init__.py

@@ -1,3 +1,17 @@
+# Copyright 2026 Datastrato, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
 """ADP client session."""
 
 from .session import ClientSession, basic_auth

src/adp_sdk/client/session.py

@@ -1,3 +1,17 @@
+# Copyright 2026 Datastrato, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
 """High-level ADP client session.
 
 Manages the lifecycle of a connection to an ADP server, including

src/adp_sdk/shared/__init__.py

@@ -1,3 +1,17 @@
+# Copyright 2026 Datastrato, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
 """Shared utilities for the ADP SDK."""
 
 from .exceptions import (

src/adp_sdk/shared/exceptions.py

@@ -1,3 +1,17 @@
+# Copyright 2026 Datastrato, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
 """ADP-specific exceptions."""
 
 from typing import Any