dataquery-sdk 1.2.2__tar.gz → 1.2.4__tar.gz

{dataquery_sdk-1.2.2 → dataquery_sdk-1.2.4}/CHANGELOG.md

@@ -46,5 +46,12 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - Extraction is Zip Slip-guarded, skips current-day archives, and surfaces failures via `extraction_errors` (downgrading overall status to `partial`); date windows with no available files no longer mark a multi-window run as `partial`
 ## [1.2.2] - 2026-07-18
 - MCP: new `mcp-connect` CLI command  
-
-
+## [1.2.3] - 2026-07-23
+- Pagination: new client-driven `get_next_page_async(page)` helper — read `next_link` off any paged response and hand the page back to fetch the next one (manual counterpart to the SDK-driven `iter_pages`); next-page links are resolved against the surface the page came from and never sent off-host
+- Pagination models: `items`, `page-size`, and `info` fields, plus the `next_link` property and `get_self_link()` accessor, moved onto the shared `Paginated` mixin; `FileList` is now paginated; list fields default to empty instead of being required so partial/empty envelopes parse cleanly
+- New `APIResponseError` raised when a 2xx response carries an `errors`/`error` envelope (e.g. `498 Unrecognized Page Token`); an `info` `204` "no content" envelope now yields an empty page so pagination stops cleanly, while any other unrecognized body fails loudly
+- Exports: `Paginated`, `Link`, `APIResponseError` are now part of the public API
+## [1.2.4] - 2026-08-12
+- MCP: `mcp-connect` accepts `--client-id`, `--client-secret` and `--bearer-token` and exports them into the `DATAQUERY_*` environment of the process, so the MCP bridge and the SDK share one credential set instead of each needing its own setup
+ 
+ 

{dataquery_sdk-1.2.2 → dataquery_sdk-1.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dataquery-sdk
-Version: 1.2.2
+Version: 1.2.4
 Summary: Python SDK for DATAQUERY Data API - Query, download, and check availability of economic data files
 Author-email: DATAQUERY SDK Team <dataquery_support@jpmorgan.com>
 Project-URL: Homepage, https://github.com/jpmorganchase/dataquery-sdk
@@ -25,7 +25,7 @@ Classifier: Topic :: Internet :: WWW/HTTP :: Dynamic Content
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: aiohttp<4.0.0,>=3.10.11
+Requires-Dist: aiohttp<4.0.0,>=3.14.3
 Requires-Dist: pydantic<3.0.0,>=2.0.0
 Requires-Dist: structlog>=23.0.0
 Requires-Dist: python-dotenv>=1.0.0
@@ -46,7 +46,7 @@ Requires-Dist: mkdocs>=1.5.0; extra == "docs"
 Requires-Dist: mkdocs-material>=9.0.0; extra == "docs"
 Requires-Dist: mkdocstrings[python]>=0.24.0; extra == "docs"
 Requires-Dist: mike>=1.1.0; extra == "docs"
-Requires-Dist: pymdown-extensions>=10.21.3; extra == "docs"
+Requires-Dist: pymdown-extensions>=11.0.0; extra == "docs"
 Provides-Extra: pandas
 Requires-Dist: pandas>=2.0.0; extra == "pandas"
 Provides-Extra: mcp
@@ -67,7 +67,7 @@ Requires-Dist: fastmcp>=2.14; extra == "all"
 Requires-Dist: pip-audit>=2.7.0; extra == "all"
 Requires-Dist: build>=1.3.0; extra == "all"
 Requires-Dist: urllib3>=2.7.0; extra == "all"
-Requires-Dist: pymdown-extensions>=10.21.3; extra == "all"
+Requires-Dist: pymdown-extensions>=11.0.0; extra == "all"
 Dynamic: license-file
 
 # DataQuery SDK

{dataquery_sdk-1.2.2 → dataquery_sdk-1.2.4}/dataquery/__init__.py

@@ -1,20 +1,6 @@
-"""
-DATAQUERY SDK - Python SDK for DATAQUERY Data API
+"""DATAQUERY SDK - Python SDK for DATAQUERY Data API."""
 
-A high-performance Python SDK for the DATAQUERY Data API, providing seamless access
-to economic data files with advanced features like querying, downloading, availability
-checking, rate limiting, retry logic, connection pool monitoring, and comprehensive logging.
-
-Quick Start:
-    >>> from dataquery import DataQuery
-    >>> async with DataQuery() as dq:
-    ...     groups = await dq.list_groups_async()
-    ...     print(f"Found {len(groups)} groups")
-
-For more information, visit: https://github.com/jpmorganchase/dataquery-sdk
-"""
-
-__version__ = "1.2.2"
+__version__ = "1.2.4"
 __author__ = "DATAQUERY SDK Team"
 __email__ = "dataquery_support@jpmorgan.com"
 __license__ = "MIT"
@@ -34,6 +20,7 @@ from .function_registry import (
 from .sse.client import SSEClient, SSEEvent
 from .sse.subscriber import NotificationDownloadManager
 from .types.exceptions import (
+    APIResponseError,
     AuthenticationError,
     AvailabilityError,
     ConfigurationError,
@@ -45,6 +32,7 @@ from .types.exceptions import (
     GroupNotFoundError,
     NetworkError,
     NotFoundError,
+    PaginationError,
     RateLimitError,
     ValidationError,
     WorkflowError,
@@ -62,6 +50,8 @@ from .types.models import (
     FileList,
     Group,
     GroupList,
+    Link,
+    Paginated,
 )
 from .utils import download_zip_async
 
@@ -71,6 +61,8 @@ __all__ = [
     "ClientConfig",
     "Group",
     "GroupList",
+    "Paginated",
+    "Link",
     "FileInfo",
     "FileList",
     "AvailabilityInfo",
@@ -81,9 +73,11 @@ __all__ = [
     "DownloadProgress",
     "DateRange",
     "DataQueryError",
+    "APIResponseError",
     "AuthenticationError",
     "ValidationError",
     "NotFoundError",
+    "PaginationError",
     "RateLimitError",
     "NetworkError",
     "ConfigurationError",

{dataquery_sdk-1.2.2 → dataquery_sdk-1.2.4}/dataquery/cli.py

@@ -3,6 +3,7 @@
 import argparse
 import asyncio
 import json
+import os
 import sys
 from collections.abc import AsyncGenerator
 from pathlib import Path
@@ -11,8 +12,6 @@ from typing import Any, Dict, List, Optional
 from dataquery import DataQuery
 from dataquery.types.exceptions import DataQueryError
 
-# ── Output helpers (legacy-CLI "summary + --- JSON ---" format) ────────────
-
 
 def _to_dict(payload: Any) -> Dict[str, Any]:
     """Normalize a Pydantic model or dict into a plain dict for JSON dump."""
@@ -82,6 +81,64 @@ def _print_error(message: str, *, suggestion: Optional[str] = None) -> None:
     print(json.dumps(envelope, indent=2))
 
 
+_MCP_CREDENTIAL_FLAGS = (
+    ("client_id", "CLIENT_ID"),
+    ("client_secret", "CLIENT_SECRET"),
+    ("bearer_token", "BEARER_TOKEN"),
+)
+
+# Endpoint settings saved alongside the credentials, so a later DataQuery()
+# mints tokens against the same environment instead of the PROD defaults.
+_MCP_SAVED_SETTINGS = ("BASE_URL", "OAUTH_TOKEN_URL", "OAUTH_AUD")
+
+
+def _export_mcp_credentials(args: argparse.Namespace) -> List[str]:
+    """Export ``mcp-connect`` credential flags into the ``DATAQUERY_*`` process env.
+
+    Flags win over anything already exported, so the rest of this process —
+    the token manager here, and any SDK use later on — resolves them the usual
+    way with no separate environment setup. Returns the unprefixed keys that
+    are actually set afterwards, in save order.
+    """
+    from dataquery.config import EnvConfig
+
+    for attr, env_key in _MCP_CREDENTIAL_FLAGS:
+        value = getattr(args, attr, None)
+        if value:
+            os.environ[f"{EnvConfig.PREFIX}{env_key}"] = value
+
+    keys = [env_key for _, env_key in _MCP_CREDENTIAL_FLAGS] + list(_MCP_SAVED_SETTINGS)
+    # os.environ rather than get_env_var: model defaults must not be persisted
+    # as if the user had chosen them.
+    return [key for key in keys if os.environ.get(f"{EnvConfig.PREFIX}{key}")]
+
+
+def _save_mcp_credentials(keys: List[str]) -> None:
+    """Persist the resolved credentials to the user-level ``.env``.
+
+    Best effort: a failure here is reported but never takes the MCP connection
+    down with it. Messages go to stderr — stdout is the JSON-RPC channel.
+    """
+    from dataquery.config import EnvConfig
+
+    if not any(key in keys for key in ("CLIENT_ID", "CLIENT_SECRET", "BEARER_TOKEN")):
+        print(
+            "--save-credentials: nothing to save; pass --client-id/--client-secret "
+            "or set DATAQUERY_CLIENT_ID/DATAQUERY_CLIENT_SECRET.",
+            file=sys.stderr,
+        )
+        return
+
+    values = {key: os.environ.get(f"{EnvConfig.PREFIX}{key}") for key in keys}
+    try:
+        env_file = EnvConfig.save_user_env(values)
+    except OSError as exc:
+        print(f"--save-credentials: could not write credentials: {exc}", file=sys.stderr)
+        return
+    saved = ", ".join(f"{EnvConfig.PREFIX}{key}" for key in keys)
+    print(f"Saved {saved} to {env_file} (owner-only)", file=sys.stderr)
+
+
 def create_parser() -> argparse.ArgumentParser:
     """Create the top-level CLI parser with subcommands."""
     parser = argparse.ArgumentParser(description="Command Line Interface for the DataQuery SDK")
@@ -204,8 +261,6 @@ def create_parser() -> argparse.ArgumentParser:
     p_fn.add_argument("--list", action="store_true", help="List all available functions")
     p_fn.add_argument("--json", action="store_true", help="Output raw JSON")
 
-    # ── DataQuery API v2 endpoints (skill-facing surface) ────────────────
-
     def _ts_args(p: argparse.ArgumentParser) -> None:
         p.add_argument("--data", choices=["REFERENCE_DATA", "NO_REFERENCE_DATA", "ALL"], default=None)
         p.add_argument("--start-date", help="YYYYMMDD or TODAY-Nx (x=D/W/M/Y). Default: TODAY-1D")
@@ -298,12 +353,35 @@ def create_parser() -> argparse.ArgumentParser:
             "Bridge a desktop MCP client (stdio) to a remote streamable-HTTP MCP\n"
             "server, authenticating with an OAuth client-credentials (AuthE) token\n"
             "minted from the DATAQUERY_* environment. Point your MCP client's\n"
-            "`command` at:  dataquery mcp-connect --url <MCP_URL>"
+            "`command` at:  dataquery mcp-connect --url <MCP_URL>\n"
+            "\n"
+            "Credentials passed as flags are exported into the DATAQUERY_*\n"
+            "environment of this process; add --save-credentials to also write\n"
+            "them to ~/.dataquery/.env, which every later SDK call picks up so\n"
+            "no environment has to be set up again."
         ),
         formatter_class=argparse.RawDescriptionHelpFormatter,
     )
     p_connect.add_argument("--url", required=True, help="Remote MCP endpoint URL")
     p_connect.add_argument("--name", default="dataquery-mcp", help="Proxy server name (default: dataquery-mcp)")
+    p_connect.add_argument("--client-id", default=None, help="OAuth client ID (exported as DATAQUERY_CLIENT_ID)")
+    p_connect.add_argument(
+        "--client-secret",
+        default=None,
+        help="OAuth client secret (exported as DATAQUERY_CLIENT_SECRET); visible in the process list, "
+        "so prefer DATAQUERY_CLIENT_SECRET or --env-file on shared machines",
+    )
+    p_connect.add_argument(
+        "--bearer-token",
+        default=None,
+        help="Bearer token to use instead of OAuth credentials (exported as DATAQUERY_BEARER_TOKEN)",
+    )
+    p_connect.add_argument(
+        "--save-credentials",
+        action="store_true",
+        help="Also save the resolved credentials to ~/.dataquery/.env (owner-only) so later SDK use "
+        "needs no environment variables",
+    )
 
     return parser
 
@@ -518,9 +596,6 @@ async def cmd_search(args: argparse.Namespace) -> int:
     return 0
 
 
-# ── DataQuery API v2 command handlers ────────────────────────────────────
-
-
 async def cmd_groups_search(args: argparse.Namespace) -> int:
     async with DataQuery(args.env_file) as dq:
         items = await dq.search_groups_async(args.keywords, page=args.page)
@@ -732,14 +807,7 @@ def cmd_function_help(args: argparse.Namespace) -> int:
 
 
 async def cmd_mcp_connect(args: argparse.Namespace) -> int:
-    """Bridge a desktop MCP client (stdio) to a remote streamable-HTTP MCP server.
-
-    Mints an OAuth client-credentials (AuthE) token with the SDK's own
-    TokenManager from the ``DATAQUERY_*`` environment and attaches a fresh bearer
-    token to every upstream request. Launched by an MCP client as its stdio
-    ``command``; stdout carries only the MCP JSON-RPC stream, so all diagnostics
-    go to stderr.
-    """
+    """Bridge a desktop MCP client (stdio) to a remote streamable-HTTP MCP server."""
     try:
         import httpx
         from fastmcp import FastMCP
@@ -757,14 +825,24 @@ async def cmd_mcp_connect(args: argparse.Namespace) -> int:
     from dataquery.config import EnvConfig
     from dataquery.transport.auth import TokenManager
 
-    config = EnvConfig.create_client_config(env_file=Path(args.env_file) if getattr(args, "env_file", None) else None)
+    if getattr(args, "env_file", None):
+        EnvConfig.load_env_file(Path(args.env_file))
+    # Load saved credentials before exporting, so they too land in the process
+    # env and a re-launch with --save-credentials rewrites them instead of
+    # reporting nothing to save. Loaded last, so it never wins over the above.
+    EnvConfig.load_user_env_file()
+    savable = _export_mcp_credentials(args)
+    if getattr(args, "save_credentials", False):
+        _save_mcp_credentials(savable)
+
+    config = EnvConfig.create_client_config()
     token_manager = TokenManager(config)
 
     class _AutheAuth(httpx.Auth):
         """Stamp a fresh AuthE bearer token (from the SDK TokenManager) per request."""
 
         async def async_auth_flow(self, request: httpx.Request) -> AsyncGenerator[httpx.Request, httpx.Response]:
-            header = await token_manager.get_valid_token()  # "Bearer <jwt>"
+            header = await token_manager.get_valid_token()
             if not header:
                 raise DataQueryError(
                     "Could not obtain an OAuth token \u2014 check DATAQUERY_CLIENT_ID, "
@@ -776,7 +854,6 @@ async def cmd_mcp_connect(args: argparse.Namespace) -> int:
 
     transport = StreamableHttpTransport(args.url, auth=_AutheAuth())
     proxy = FastMCP.as_proxy(transport, name=args.name)
-    # stdout is reserved for the MCP JSON-RPC stream; keep the banner off it.
     await proxy.run_async(transport="stdio", show_banner=False)
     return 0
 

{dataquery_sdk-1.2.2 → dataquery_sdk-1.2.4}/dataquery/config/env.py

@@ -1,21 +1,11 @@
-"""
-Environment-based configuration for the DATAQUERY SDK.
-
-Loads settings from environment variables (or a ``.env`` file) into a
-:class:`ClientConfig` Pydantic model. Field declarations live on
-``ClientConfig`` — this module only handles the env-var → field projection,
-type coercion, and cross-field validation.
-
-Single source of truth: :data:`ClientConfig.model_fields`. The
-``DEFAULTS`` table here is derived from the model at import time so the two
-cannot drift.
-"""
+"""Environment-based configuration for the DATAQUERY SDK."""
 
 from __future__ import annotations
 
 import os
+import re
 from pathlib import Path
-from typing import Any, Dict, Optional, Union, get_args, get_origin
+from typing import Any, Dict, List, Mapping, Optional, Union, get_args, get_origin
 
 from dotenv import load_dotenv
 from pydantic_core import PydanticUndefined
@@ -41,6 +31,37 @@ _SENSITIVE_FIELDS = frozenset(
     }
 )
 
+# User-level config: credentials saved once (e.g. by `dataquery mcp-connect
+# --save-credentials`) and reused by every later SDK call in any process.
+_USER_CONFIG_DIR_ENV = "DATAQUERY_CONFIG_DIR"
+_USER_CONFIG_DIR_DEFAULT = "~/.dataquery"
+_USER_ENV_FILENAME = ".env"
+
+_USER_ENV_HEADER = [
+    "# DataQuery SDK credentials.",
+    "# Written by `dataquery mcp-connect --save-credentials`; loaded as a",
+    "# fallback only, so shell env vars and a local .env still win.",
+    "# Values are read literally: ${VAR} is not expanded here.",
+    "",
+]
+
+_ENV_LINE_RE = re.compile(r"^\s*(?:export\s+)?([A-Za-z_][A-Za-z0-9_]*)\s*=")
+
+# Characters common in ids, secrets and URLs that need no .env quoting.
+_UNQUOTED_EXTRA_CHARS = "._-:/@+~"
+
+
+def _format_env_value(value: str) -> str:
+    """Render a value for a ``.env`` line, quoting only when it needs it."""
+    if value and all(ch.isalnum() or ch in _UNQUOTED_EXTRA_CHARS for ch in value):
+        return value
+    if "'" not in value and "\n" not in value and "\r" not in value:
+        return f"'{value}'"
+    # Escapes keep the entry on a single physical line, which the line-based
+    # merge in ``save_user_env`` depends on.
+    escaped = value.replace("\\", "\\\\").replace('"', '\\"').replace("\n", "\\n").replace("\r", "\\r")
+    return f'"{escaped}"'
+
 
 def _env_name_for(field_name: str) -> str:
     """Project a Pydantic field name to its env-var key (no PREFIX)."""
@@ -57,12 +78,7 @@ def _unwrap_optional(annotation: Any) -> Any:
 
 
 def _build_defaults() -> Dict[str, Optional[str]]:
-    """Compute the ``EnvConfig.DEFAULTS`` table from the model.
-
-    Booleans become ``"true"``/``"false"`` (lowercase, matching the
-    tokens :meth:`EnvConfig.get_bool` accepts). Numeric / string defaults
-    are stringified directly. ``None`` defaults stay ``None``.
-    """
+    """Compute the ``EnvConfig.DEFAULTS`` table from the model."""
     defaults: Dict[str, Optional[str]] = {}
     for field_name, field in ClientConfig.model_fields.items():
         env_key = _env_name_for(field_name)
@@ -79,12 +95,7 @@ def _build_defaults() -> Dict[str, Optional[str]]:
 
 
 class EnvConfig:
-    """Environment-based configuration loader for the DataQuery SDK.
-
-    All methods are class methods — this class is a namespace, never
-    instantiated. Field definitions, types, and defaults all originate from
-    :class:`dataquery.types.models.ClientConfig`.
-    """
+    """Environment-based configuration loader for the DataQuery SDK."""
 
     PREFIX = "DATAQUERY_"
     DEFAULTS: Dict[str, Optional[str]] = _build_defaults()
@@ -97,6 +108,77 @@ class EnvConfig:
         if env_file.exists():
             load_dotenv(env_file)
 
+    @classmethod
+    def user_config_dir(cls) -> Path:
+        """Directory holding user-level SDK config (``$DATAQUERY_CONFIG_DIR`` or ``~/.dataquery``)."""
+        raw = os.getenv(_USER_CONFIG_DIR_ENV) or _USER_CONFIG_DIR_DEFAULT
+        return Path(raw).expanduser()
+
+    @classmethod
+    def user_env_file(cls) -> Path:
+        """Path of the user-level ``.env`` holding saved credentials."""
+        return cls.user_config_dir() / _USER_ENV_FILENAME
+
+    @classmethod
+    def load_user_env_file(cls) -> bool:
+        """Load the user-level ``.env`` as a fallback; never overrides what is already set."""
+        env_file = cls.user_env_file()
+        if not env_file.exists():
+            return False
+        # interpolate=False: values here are literal secrets, and dotenv's
+        # ${VAR} expansion runs after unquoting, so no quoting style escapes it.
+        load_dotenv(env_file, override=False, interpolate=False)
+        return True
+
+    @classmethod
+    def save_user_env(cls, values: Mapping[str, Optional[str]]) -> Path:
+        """Merge ``DATAQUERY_<key>=<value>`` pairs into the user-level ``.env``.
+
+        Keys are unprefixed (``CLIENT_ID``) and empty values are skipped. Lines
+        for a key being saved are rewritten in place, anything else already in
+        the file is preserved, and the file is (re)written owner-readable only.
+        """
+        updates: Dict[str, str] = {}
+        for key, value in values.items():
+            if value:
+                updates[f"{cls.PREFIX}{key}"] = value
+
+        env_file = cls.user_env_file()
+        env_file.parent.mkdir(parents=True, exist_ok=True)
+        try:
+            env_file.parent.chmod(0o700)
+        except OSError:  # pragma: no cover - non-POSIX filesystem
+            pass
+
+        lines: List[str] = []
+        written: set[str] = set()
+        if env_file.exists():
+            for line in env_file.read_text().splitlines():
+                match = _ENV_LINE_RE.match(line)
+                line_key = match.group(1) if match else None
+                if line_key is None or line_key not in updates:
+                    lines.append(line)
+                elif line_key not in written:
+                    lines.append(f"{line_key}={_format_env_value(updates[line_key])}")
+                    written.add(line_key)
+                # else: drop a stale duplicate of a key just rewritten above
+        else:
+            lines.extend(_USER_ENV_HEADER)
+
+        for key, value in updates.items():
+            if key not in written:
+                lines.append(f"{key}={_format_env_value(value)}")
+
+        # O_CREAT mode only applies to new files, so chmod covers the rest.
+        fd = os.open(env_file, os.O_WRONLY | os.O_CREAT | os.O_TRUNC, 0o600)
+        with os.fdopen(fd, "w") as handle:
+            handle.write("\n".join(lines) + "\n")
+        try:
+            env_file.chmod(0o600)
+        except OSError:  # pragma: no cover - non-POSIX filesystem
+            pass
+        return env_file
+
     @classmethod
     def get_env_var(cls, key: str, default: Optional[str] = None) -> Optional[str]:
         """Read ``$DATAQUERY_<key>`` with the model default as the fallback."""
@@ -146,17 +228,18 @@ class EnvConfig:
         config_data: Optional[Dict[str, Any]] = None,
         env_file: Optional[Path] = None,
     ) -> ClientConfig:
-        """Build a :class:`ClientConfig` from env vars or an explicit dict.
-
-        When ``config_data`` is provided the env-var path is skipped entirely
-        — the dict is passed straight to :class:`ClientConfig`.
-        """
+        """Build a :class:`ClientConfig` from env vars or an explicit dict."""
         if config_data is not None:
             return ClientConfig(**config_data)
 
         if env_file is not None:
             cls.load_env_file(env_file)
 
+        # Lowest precedence, loaded last so it fills only what is still unset:
+        # credentials saved by `mcp-connect --save-credentials`, which let a
+        # later DataQuery() authenticate with no environment of its own.
+        cls.load_user_env_file()
+
         if not cls.get_env_var("BASE_URL"):
             raise ConfigurationError(f"{cls.PREFIX}BASE_URL environment variable is required")
 
@@ -275,11 +358,7 @@ class EnvConfig:
 
     @classmethod
     def create_env_template(cls, output_path: Optional[Path] = None) -> Path:
-        """Write a ``.env`` template listing every supported variable.
-
-        Lines are auto-generated from :data:`ClientConfig.model_fields` so a
-        new field shows up here automatically (no separate maintenance).
-        """
+        """Write a ``.env`` template listing every supported variable."""
         if output_path is None:
             output_path = Path(".env.template")
         if not isinstance(output_path, Path):

{dataquery_sdk-1.2.2 → dataquery_sdk-1.2.4}/dataquery/config/logging.py

@@ -1,9 +1,4 @@
-"""
-Enhanced logging configuration for the DATAQUERY SDK.
-
-Provides structured logging, performance metrics, request/response logging,
-and configurable log levels and formats.
-"""
+"""Enhanced logging configuration for the DATAQUERY SDK."""
 
 import logging
 from dataclasses import dataclass, field
@@ -353,21 +348,7 @@ def create_logging_config(
     enable_request_logging: bool = False,
     enable_performance_logging: bool = True,
 ) -> LoggingConfig:
-    """
-    Create logging configuration.
-
-    Args:
-        level: Log level
-        format: Log format
-        enable_console: Whether to enable console logging
-        enable_file: Whether to enable file logging
-        log_file: Path to log file
-        enable_request_logging: Whether to log HTTP requests/responses
-        enable_performance_logging: Whether to log performance metrics
-
-    Returns:
-        Logging configuration
-    """
+    """Create logging configuration."""
     return LoggingConfig(
         level=level,
         format=format,
@@ -380,13 +361,5 @@ def create_logging_config(
 
 
 def create_logging_manager(config: LoggingConfig) -> LoggingManager:
-    """
-    Create a logging manager with the specified configuration.
-
-    Args:
-        config: Logging configuration
-
-    Returns:
-        Configured logging manager
-    """
+    """Create a logging manager with the specified configuration."""
     return LoggingManager(config)

{dataquery_sdk-1.2.2 → dataquery_sdk-1.2.4}/dataquery/constants/__init__.py

@@ -1,16 +1,4 @@
-"""Internal tunable constants used across the SDK.
-
-Centralizing these values here avoids drift between code paths that must agree
-(e.g. the single-stream and parallel downloaders share thresholds) and gives a
-single place to adjust tunables. Pydantic model field defaults and validation
-bounds are intentionally NOT hoisted here — those are part of the public
-schema.
-
-Constants are organized into topical submodules; this ``__init__`` re-exports
-them so existing call sites using ``from . import constants as C`` keep
-working without change. New code may also import from a specific submodule,
-e.g. ``from dataquery.constants.api import API_GROUPS``.
-"""
+"""Internal tunable constants used across the SDK."""
 
 from __future__ import annotations
 

{dataquery_sdk-1.2.2 → dataquery_sdk-1.2.4}/dataquery/constants/api.py

@@ -1,8 +1,4 @@
-"""API endpoint paths.
-
-Paths are relative to the main API base URL (``_build_api_url``) or the files
-API base URL (``_build_files_api_url``) unless otherwise noted.
-"""
+"""API endpoint paths."""
 
 from __future__ import annotations
 

{dataquery_sdk-1.2.2 → dataquery_sdk-1.2.4}/dataquery/constants/download.py

@@ -26,15 +26,6 @@ TEMP_SUFFIX = ".part"
 MBPS_TO_BYTES_PER_SECOND = 125_000
 
 
-# Default size, in days, of each chunk when splitting a date range so that
-# the available-files endpoint is queried over a smaller window per call.
-# Used by ``split_date_range_into_chunks``. The endpoint caps each call at
-# one calendar month; smaller windows just keep responses lighter. (The old
-# group-level limits, e.g. 14 days for RESEARCH_EQUITY_ALL, were lifted
-# server-side.)
 DEFAULT_WRITTEN_RESEARCH_CHUNK_DAYS: int = 7
 
-# Error string set on the OperationReport when the available-files endpoint
-# returns nothing for a date window. Chunked workflows match on it to tell
-# a quiet window apart from a real failure.
 NO_FILES_FOUND_ERROR: str = "No available files found for date range"