openboost-cli 2.0.0__tar.gz → 2.0.1__tar.gz

{openboost_cli-2.0.0/src/openboost_cli.egg-info → openboost_cli-2.0.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openboost-cli
-Version: 2.0.0
+Version: 2.0.1
 Summary: Unified CLI and Python client for OpenBoost TikTok, Patent, and Amazon MCP servers.
 Author: OpenBoost CLI
 License-Expression: MIT
@@ -56,7 +56,7 @@ openboost config set --secret-key "your-secret-key"
 
 ```bash
 openboost config set --mcp amazon --secret-key "your-secret-key"
-openboost config set --mcp amazon --base-url "http://mcp.microdata-inc.com/mcp-servers/proboost-amazon-mcp"
+openboost config set --mcp amazon --base-url "http://mcp.microdata-inc.com/mcp-servers/openboost-all-mcp/sse"
 ```
 
 默认配置文件位于：

{openboost_cli-2.0.0 → openboost_cli-2.0.1}/README.md

@@ -34,7 +34,7 @@ openboost config set --secret-key "your-secret-key"
 
 ```bash
 openboost config set --mcp amazon --secret-key "your-secret-key"
-openboost config set --mcp amazon --base-url "http://mcp.microdata-inc.com/mcp-servers/proboost-amazon-mcp"
+openboost config set --mcp amazon --base-url "http://mcp.microdata-inc.com/mcp-servers/openboost-all-mcp/sse"
 ```
 
 默认配置文件位于：

{openboost_cli-2.0.0 → openboost_cli-2.0.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "openboost-cli"
-version = "2.0.0"
+version = "2.0.1"
 description = "Unified CLI and Python client for OpenBoost TikTok, Patent, and Amazon MCP servers."
 readme = "README.md"
 requires-python = ">=3.10"

{openboost_cli-2.0.0 → openboost_cli-2.0.1}/src/openboost_cli/__init__.py

@@ -1,12 +1,12 @@
 """Unified OpenBoost MCP CLI and Python client."""
 
-__version__ = "2.0.0"
+__version__ = "2.0.1"
 
-DEFAULT_BASE_URLS = {
-    "tiktok": "http://mcp.microdata-inc.com/mcp-servers/proboost-tiktok-mcp",
-    "patent": "http://mcp.microdata-inc.com/mcp-servers/proboost-patent-mcp",
-    "amazon": "http://mcp.microdata-inc.com/mcp-servers/proboost-amazon-mcp",
-}
+DEFAULT_BASE_URLS = {
+    "tiktok": "http://mcp.microdata-inc.com/mcp-servers/openboost-all-mcp/sse",
+    "patent": "http://mcp.microdata-inc.com/mcp-servers/openboost-all-mcp/sse",
+    "amazon": "http://mcp.microdata-inc.com/mcp-servers/openboost-all-mcp/sse",
+}
 
 MCP_LABELS = {
     "tiktok": "TikTok",

{openboost_cli-2.0.0 → openboost_cli-2.0.1}/src/openboost_cli/cli/main.py

@@ -71,7 +71,7 @@ amazon_app.add_typer(amazon_hot_app, name="hot")
 _state: dict[str, str | None] = {"secret_key": None, "base_url": None}
 
 KeyOpt = Annotated[Optional[str], typer.Option("--key", "-k", help="secret-key")]
-BaseUrlOpt = Annotated[Optional[str], typer.Option("--base-url", "-u", help="MCP Streamable HTTP URL")]
+BaseUrlOpt = Annotated[Optional[str], typer.Option("--base-url", "-u", help="MCP SSE URL")]
 McpArg = Annotated[str, typer.Argument(help="MCP 名称: tiktok / patent / amazon")]
 CountryOpt = Annotated[str, typer.Option("--country-region", "--country", help="国家/地区，如 美国")]
 CurrentOpt = Annotated[Optional[int], typer.Option("--current", help="当前页码")]

{openboost_cli-2.0.0 → openboost_cli-2.0.1}/src/openboost_cli/mcp_client.py

@@ -4,11 +4,9 @@ import asyncio
 from datetime import timedelta
 from typing import Any
 
-import httpx
-from mcp import ClientSession
-from mcp import types as mtypes
-from mcp.client.streamable_http import streamable_http_client
-from mcp.shared._httpx_utils import create_mcp_http_client
+from mcp import ClientSession
+from mcp import types as mtypes
+from mcp.client.sse import sse_client
 
 from openboost_cli.config import effective_secret_and_base, normalize_mcp_name
 from openboost_cli.parse_result import decode_embedded_payload
@@ -31,19 +29,22 @@ async def invoke_tool_async(
     *,
     connect_timeout_sec: float = 30,
     read_timeout_sec: float = 300,
-) -> mtypes.CallToolResult:
-    headers = {"secret-key": secret_key}
-    timeout = httpx.Timeout(connect_timeout_sec, read=read_timeout_sec, write=30, pool=30)
-
-    async with create_mcp_http_client(headers=headers, timeout=timeout) as client:
-        async with streamable_http_client(base_url, http_client=client) as (read, write, _get_id):
-            async with ClientSession(read, write) as session:
-                await session.initialize()
-                return await session.call_tool(
-                    tool_name,
-                    arguments,
-                    read_timeout_seconds=timedelta(seconds=read_timeout_sec),
-                )
+) -> mtypes.CallToolResult:
+    headers = {"secret-key": secret_key}
+
+    async with sse_client(
+        base_url,
+        headers=headers,
+        timeout=connect_timeout_sec,
+        sse_read_timeout=read_timeout_sec,
+    ) as (read, write):
+        async with ClientSession(read, write) as session:
+            await session.initialize()
+            return await session.call_tool(
+                tool_name,
+                arguments,
+                read_timeout_seconds=timedelta(seconds=read_timeout_sec),
+            )
 
 
 class McpClient:

{openboost_cli-2.0.0 → openboost_cli-2.0.1/src/openboost_cli.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openboost-cli
-Version: 2.0.0
+Version: 2.0.1
 Summary: Unified CLI and Python client for OpenBoost TikTok, Patent, and Amazon MCP servers.
 Author: OpenBoost CLI
 License-Expression: MIT
@@ -56,7 +56,7 @@ openboost config set --secret-key "your-secret-key"
 
 ```bash
 openboost config set --mcp amazon --secret-key "your-secret-key"
-openboost config set --mcp amazon --base-url "http://mcp.microdata-inc.com/mcp-servers/proboost-amazon-mcp"
+openboost config set --mcp amazon --base-url "http://mcp.microdata-inc.com/mcp-servers/openboost-all-mcp/sse"
 ```
 
 默认配置文件位于：

{openboost_cli-2.0.0 → openboost_cli-2.0.1}/tests/test_cli.py

@@ -29,7 +29,7 @@ def test_version() -> None:
     result = runner.invoke(cli.app, ["version"])
 
     assert result.exit_code == 0
-    assert "2.0.0" in result.output
+    assert "2.0.1" in result.output
 
 
 def test_list_reports_all_bundled_tools() -> None:

{openboost_cli-2.0.0 → openboost_cli-2.0.1}/tests/test_config.py

@@ -4,10 +4,16 @@ import json
 from pathlib import Path
 
 from openboost_cli import DEFAULT_BASE_URLS
-from openboost_cli.config import config_path, effective_secret_and_base, load_config, mask_secret, update_config
-
-
-def test_config_uses_only_openboost_names(monkeypatch, tmp_path: Path) -> None:
+from openboost_cli.config import config_path, effective_secret_and_base, load_config, mask_secret, update_config
+
+
+def test_default_mcp_urls_use_unified_openboost_sse_endpoint() -> None:
+    expected = "http://mcp.microdata-inc.com/mcp-servers/openboost-all-mcp/sse"
+
+    assert set(DEFAULT_BASE_URLS.values()) == {expected}
+
+
+def test_config_uses_only_openboost_names(monkeypatch, tmp_path: Path) -> None:
     path = tmp_path / "openboost.json"
     monkeypatch.setenv("OPENBOOST_CONFIG", str(path))
     monkeypatch.setenv("PROBOOST_SECRET_KEY", "ignored-old-secret")

{openboost_cli-2.0.0 → openboost_cli-2.0.1}/tests/test_mcp_client.py

@@ -1,8 +1,7 @@
 from __future__ import annotations
 
 import asyncio
-from contextlib import asynccontextmanager
-from datetime import timedelta
+from datetime import timedelta
 from types import SimpleNamespace
 
 import pytest
@@ -62,21 +61,30 @@ def test_python_client_raises_mcp_error(monkeypatch: pytest.MonkeyPatch) -> None
         client.call_tool("patent_query_count", {"query_text": "test"})
 
 
-def test_invoke_tool_uses_existing_mcp_protocol(monkeypatch: pytest.MonkeyPatch) -> None:
-    observed = {}
-    http_client = object()
-
-    @asynccontextmanager
-    async def fake_create_mcp_http_client(*, headers, timeout):
-        observed["headers"] = headers
-        observed["timeout"] = timeout
-        yield http_client
-
-    @asynccontextmanager
-    async def fake_streamable_http_client(base_url, *, http_client):
-        observed["base_url"] = base_url
-        observed["http_client"] = http_client
-        yield "read", "write", lambda: None
+def test_invoke_tool_uses_sse_protocol(monkeypatch: pytest.MonkeyPatch) -> None:
+    observed = {}
+
+    class FakeSseClient:
+        def __init__(self, base_url, *, headers, timeout, sse_read_timeout):
+            observed["base_url"] = base_url
+            observed["headers"] = headers
+            observed["timeout"] = timeout
+            observed["sse_read_timeout"] = sse_read_timeout
+
+        async def __aenter__(self):
+            return "read", "write"
+
+        async def __aexit__(self, exc_type, exc, tb):
+            return False
+
+    def fake_sse_client(base_url, *, headers, timeout, sse_read_timeout):
+        observed["base_url"] = base_url
+        return FakeSseClient(
+            base_url,
+            headers=headers,
+            timeout=timeout,
+            sse_read_timeout=sse_read_timeout,
+        )
 
     class FakeSession:
         def __init__(self, read, write):
@@ -95,9 +103,8 @@ def test_invoke_tool_uses_existing_mcp_protocol(monkeypatch: pytest.MonkeyPatch)
             observed["call"] = (name, arguments, read_timeout_seconds)
             return text_result('{"ok": true}')
 
-    monkeypatch.setattr(mcp_client, "create_mcp_http_client", fake_create_mcp_http_client)
-    monkeypatch.setattr(mcp_client, "streamable_http_client", fake_streamable_http_client)
-    monkeypatch.setattr(mcp_client, "ClientSession", FakeSession)
+    monkeypatch.setattr(mcp_client, "sse_client", fake_sse_client)
+    monkeypatch.setattr(mcp_client, "ClientSession", FakeSession)
 
     result = asyncio.run(
         mcp_client.invoke_tool_async(
@@ -110,11 +117,12 @@ def test_invoke_tool_uses_existing_mcp_protocol(monkeypatch: pytest.MonkeyPatch)
         )
     )
 
-    assert result.isError is False
-    assert observed["headers"] == {"secret-key": "test-secret"}
-    assert observed["base_url"] == "http://mcp.example/server"
-    assert observed["http_client"] is http_client
-    assert observed["initialized"] is True
+    assert result.isError is False
+    assert observed["headers"] == {"secret-key": "test-secret"}
+    assert observed["base_url"] == "http://mcp.example/server"
+    assert observed["timeout"] == 12
+    assert observed["sse_read_timeout"] == 34
+    assert observed["initialized"] is True
     assert observed["call"] == (
         "existing_tool_name",
         {"existingField": "value"},