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

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

@@ -44,5 +44,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - Written research: new `download_zip_async` helper that downloads a group over a date range (split into calendar-month windows to fit the available-files endpoint limit) and safely extracts ZIP archives as each download completes, overlapping unzip with in-flight downloads
 - Group downloads: `run_group_download_async` accepts an `on_file_complete` async callback awaited per file on `completed`/`already_exists`
 - 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  
 
 

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

@@ -1,12 +1,12 @@
 Metadata-Version: 2.4
 Name: dataquery-sdk
-Version: 1.2.1
+Version: 1.2.2
 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/dataquery/dataquery-sdk
-Project-URL: Bug Tracker, https://github.com/dataquery/dataquery-sdk/issues
-Project-URL: Documentation, https://github.com/dataquery/dataquery-sdk/wiki
-Project-URL: Source Code, https://github.com/dataquery/dataquery-sdk
+Project-URL: Homepage, https://github.com/jpmorganchase/dataquery-sdk
+Project-URL: Bug Tracker, https://github.com/jpmorganchase/dataquery-sdk/issues
+Project-URL: Documentation, https://github.com/jpmorganchase/dataquery-sdk/wiki
+Project-URL: Source Code, https://github.com/jpmorganchase/dataquery-sdk
 Keywords: dataquery,data,api,economic,financial,download,async,oauth,rate-limiting,sdk
 Classifier: Development Status :: 5 - Production/Stable
 Classifier: Intended Audience :: Developers
@@ -49,6 +49,8 @@ Requires-Dist: mike>=1.1.0; extra == "docs"
 Requires-Dist: pymdown-extensions>=10.21.3; extra == "docs"
 Provides-Extra: pandas
 Requires-Dist: pandas>=2.0.0; extra == "pandas"
+Provides-Extra: mcp
+Requires-Dist: fastmcp>=2.14; extra == "mcp"
 Provides-Extra: all
 Requires-Dist: pytest>=7.0.0; extra == "all"
 Requires-Dist: pytest-asyncio>=0.21.0; extra == "all"
@@ -61,6 +63,7 @@ Requires-Dist: mkdocs-material>=9.0.0; extra == "all"
 Requires-Dist: mkdocstrings[python]>=0.24.0; extra == "all"
 Requires-Dist: mike>=1.1.0; extra == "all"
 Requires-Dist: pandas>=2.0.0; extra == "all"
+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"

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

@@ -11,10 +11,10 @@ Quick Start:
     ...     groups = await dq.list_groups_async()
     ...     print(f"Found {len(groups)} groups")
 
-For more information, visit: https://github.com/dataquery/dataquery-sdk
+For more information, visit: https://github.com/jpmorganchase/dataquery-sdk
 """
 
-__version__ = "1.2.1"
+__version__ = "1.2.2"
 __author__ = "DATAQUERY SDK Team"
 __email__ = "dataquery_support@jpmorgan.com"
 __license__ = "MIT"

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

@@ -4,6 +4,7 @@ import argparse
 import asyncio
 import json
 import sys
+from collections.abc import AsyncGenerator
 from pathlib import Path
 from typing import Any, Dict, List, Optional
 
@@ -290,6 +291,20 @@ def create_parser() -> argparse.ArgumentParser:
 
     subparsers.add_parser("heartbeat", help="Check if DataQuery is running")
 
+    p_connect = subparsers.add_parser(
+        "mcp-connect",
+        help="Connect a local stdio MCP client to a remote MCP server (AuthE OAuth)",
+        description=(
+            "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>"
+        ),
+        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)")
+
     return parser
 
 
@@ -716,6 +731,56 @@ def cmd_function_help(args: argparse.Namespace) -> int:
     return 0
 
 
+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.
+    """
+    try:
+        import httpx
+        from fastmcp import FastMCP
+        from fastmcp.client.transports import StreamableHttpTransport
+    except ImportError:
+        print(
+            "The MCP bridge requires the 'mcp' extra. Install it with:\n"
+            "    pip install 'dataquery-sdk[mcp]'\n"
+            "or run it directly with:\n"
+            "    uvx --from 'dataquery-sdk[mcp]' dataquery mcp-connect --url <MCP_URL>",
+            file=sys.stderr,
+        )
+        return 1
+
+    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)
+    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>"
+            if not header:
+                raise DataQueryError(
+                    "Could not obtain an OAuth token \u2014 check DATAQUERY_CLIENT_ID, "
+                    "DATAQUERY_CLIENT_SECRET, DATAQUERY_OAUTH_TOKEN_URL and "
+                    "DATAQUERY_OAUTH_AUD."
+                )
+            request.headers["Authorization"] = header
+            yield request
+
+    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
+
+
 def main_sync(ns: argparse.Namespace) -> int:
     if ns.command == "config":
         if ns.config_command == "show":
@@ -745,6 +810,7 @@ _ASYNC_COMMANDS = {
     "expression-timeseries": cmd_expression_timeseries,
     "grid-data": cmd_grid_data,
     "heartbeat": cmd_heartbeat,
+    "mcp-connect": cmd_mcp_connect,
 }
 
 

{dataquery_sdk-1.2.1 → dataquery_sdk-1.2.2}/dataquery_sdk.egg-info/PKG-INFO

@@ -1,12 +1,12 @@
 Metadata-Version: 2.4
 Name: dataquery-sdk
-Version: 1.2.1
+Version: 1.2.2
 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/dataquery/dataquery-sdk
-Project-URL: Bug Tracker, https://github.com/dataquery/dataquery-sdk/issues
-Project-URL: Documentation, https://github.com/dataquery/dataquery-sdk/wiki
-Project-URL: Source Code, https://github.com/dataquery/dataquery-sdk
+Project-URL: Homepage, https://github.com/jpmorganchase/dataquery-sdk
+Project-URL: Bug Tracker, https://github.com/jpmorganchase/dataquery-sdk/issues
+Project-URL: Documentation, https://github.com/jpmorganchase/dataquery-sdk/wiki
+Project-URL: Source Code, https://github.com/jpmorganchase/dataquery-sdk
 Keywords: dataquery,data,api,economic,financial,download,async,oauth,rate-limiting,sdk
 Classifier: Development Status :: 5 - Production/Stable
 Classifier: Intended Audience :: Developers
@@ -49,6 +49,8 @@ Requires-Dist: mike>=1.1.0; extra == "docs"
 Requires-Dist: pymdown-extensions>=10.21.3; extra == "docs"
 Provides-Extra: pandas
 Requires-Dist: pandas>=2.0.0; extra == "pandas"
+Provides-Extra: mcp
+Requires-Dist: fastmcp>=2.14; extra == "mcp"
 Provides-Extra: all
 Requires-Dist: pytest>=7.0.0; extra == "all"
 Requires-Dist: pytest-asyncio>=0.21.0; extra == "all"
@@ -61,6 +63,7 @@ Requires-Dist: mkdocs-material>=9.0.0; extra == "all"
 Requires-Dist: mkdocstrings[python]>=0.24.0; extra == "all"
 Requires-Dist: mike>=1.1.0; extra == "all"
 Requires-Dist: pandas>=2.0.0; extra == "all"
+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"

{dataquery_sdk-1.2.1 → dataquery_sdk-1.2.2}/dataquery_sdk.egg-info/requires.txt

@@ -16,6 +16,7 @@ mkdocs-material>=9.0.0
 mkdocstrings[python]>=0.24.0
 mike>=1.1.0
 pandas>=2.0.0
+fastmcp>=2.14
 pip-audit>=2.7.0
 build>=1.3.0
 urllib3>=2.7.0
@@ -40,5 +41,8 @@ mkdocstrings[python]>=0.24.0
 mike>=1.1.0
 pymdown-extensions>=10.21.3
 
+[mcp]
+fastmcp>=2.14
+
 [pandas]
 pandas>=2.0.0

{dataquery_sdk-1.2.1 → dataquery_sdk-1.2.2}/pyproject.toml

@@ -48,10 +48,10 @@ keywords = [
 ]
 
 [project.urls]
-Homepage = "https://github.com/dataquery/dataquery-sdk"
-"Bug Tracker" = "https://github.com/dataquery/dataquery-sdk/issues"
-Documentation = "https://github.com/dataquery/dataquery-sdk/wiki"
-"Source Code" = "https://github.com/dataquery/dataquery-sdk"
+Homepage = "https://github.com/jpmorganchase/dataquery-sdk"
+"Bug Tracker" = "https://github.com/jpmorganchase/dataquery-sdk/issues"
+Documentation = "https://github.com/jpmorganchase/dataquery-sdk/wiki"
+"Source Code" = "https://github.com/jpmorganchase/dataquery-sdk"
 
 [project.optional-dependencies]
 dev = [
@@ -76,6 +76,9 @@ docs = [
 pandas = [
     "pandas>=2.0.0",
 ]
+mcp = [
+    "fastmcp>=2.14",
+]
 all = [
     "pytest>=7.0.0",
     "pytest-asyncio>=0.21.0",
@@ -88,6 +91,7 @@ all = [
     "mkdocstrings[python]>=0.24.0",
     "mike>=1.1.0",
     "pandas>=2.0.0",
+    "fastmcp>=2.14",
     "pip-audit>=2.7.0",
     "build>=1.3.0",
     "urllib3>=2.7.0",
@@ -150,6 +154,8 @@ module = [
     "structlog.*",
     "dotenv.*",
     "pytest.*",
+    "fastmcp.*",
+    "httpx.*",
 ]
 ignore_missing_imports = true