zerone-cli 0.1.1__tar.gz → 0.1.2__tar.gz

{zerone_cli-0.1.1 → zerone_cli-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: zerone-cli
-Version: 0.1.1
+Version: 0.1.2
 Summary: Zerone 私募股权数据 CLI 工具
 Author-email: Zerone <tech@zerone.com.cn>
 License: MIT

zerone_cli-0.1.2/cli/__init__.py

@@ -0,0 +1 @@
+__version__ = "1.0.0"

zerone_cli-0.1.2/cli/client.py

@@ -0,0 +1,176 @@
+from typing import Optional
+
+import httpx
+
+from cli.config import config_manager
+
+
+class MCPClient:
+
+    def __init__(self):
+        self.api_key = config_manager.get_api_key()
+        self.server_url = config_manager.get_server_url()
+        self.client = httpx.Client(timeout=30.0)
+
+    def _request(
+        self,
+        method: str,
+        path: str,
+        json_data: Optional[dict] = None,
+        params: Optional[dict] = None,
+    ) -> dict:
+        url = f"{self.server_url}{path}"
+        headers = {
+            "Authorization": f"Bearer {self.api_key}",
+            "Content-Type": "application/json",
+        }
+
+        try:
+            response = self.client.request(
+                method=method,
+                url=url,
+                headers=headers,
+                json=json_data,
+                params=params,
+            )
+            response.raise_for_status()
+            data = response.json()
+            if not data.get("success", True):
+                error = data.get("error", {})
+                raise ValueError(f"[{error.get('code', 'UNKNOWN')}] {error.get('message', '未知错误')}")
+            return data
+        except httpx.HTTPStatusError as e:
+            try:
+                error_data = e.response.json()
+                detail = error_data.get("detail", "")
+                if detail:
+                    raise ValueError(f"[HTTP {e.response.status_code}] {detail}")
+            except ValueError:
+                raise
+            except Exception:
+                pass
+            raise ValueError(f"请求失败: {e.response.status_code} - {e.response.text}")
+        except httpx.ConnectError:
+            raise ValueError(f"无法连接到 MCP Server，请检查 server_url 配置：{self.server_url}")
+        except httpx.TimeoutException:
+            raise ValueError("请求超时，请检查网络连接或稍后重试")
+
+    def entity_search(self, keyword: str) -> dict:
+        return self._request(
+            "POST",
+            "/rest-api/tool/entity_search",
+            json_data={"keyword": keyword},
+        )
+
+    def entity_profile(
+        self,
+        entity_name: str,
+        entity_type: Optional[str] = None,
+    ) -> dict:
+        return self._request(
+            "POST",
+            "/rest-api/tool/entity_profile",
+            json_data={
+                "entity_name": entity_name,
+                "entity_type": entity_type or "",
+            },
+        )
+
+    def entity_financial_list(
+        self,
+        entity_name: str,
+        entity_type: Optional[str] = None,
+        page: int = 1,
+        page_size: int = 20,
+    ) -> dict:
+        return self._request(
+            "POST",
+            "/rest-api/tool/entity_financial_list",
+            json_data={
+                "entity_name": entity_name,
+                "entity_type": entity_type or "",
+                "page": page,
+                "page_size": page_size,
+            },
+        )
+
+    def entity_managed_funds_list(
+        self,
+        entity_name: str,
+        entity_type: Optional[str] = None,
+        page: int = 1,
+        page_size: int = 20,
+    ) -> dict:
+        return self._request(
+            "POST",
+            "/rest-api/tool/entity_managed_funds_list",
+            json_data={
+                "entity_name": entity_name,
+                "entity_type": entity_type or "",
+                "page": page,
+                "page_size": page_size,
+            },
+        )
+
+    def entity_exit_list(
+        self,
+        entity_name: str,
+        entity_type: Optional[str] = None,
+        exit_type: Optional[str] = None,
+        page: int = 1,
+        page_size: int = 20,
+    ) -> dict:
+        return self._request(
+            "POST",
+            "/rest-api/tool/entity_exit_list",
+            json_data={
+                "entity_name": entity_name,
+                "entity_type": entity_type or "",
+                "exit_type": exit_type or "",
+                "page": page,
+                "page_size": page_size,
+            },
+        )
+
+    def entity_invest_list(
+        self,
+        entity_name: str,
+        entity_type: Optional[str] = None,
+        invest_type: Optional[str] = None,
+        page: int = 1,
+        page_size: int = 20,
+    ) -> dict:
+        return self._request(
+            "POST",
+            "/rest-api/tool/entity_invest_list",
+            json_data={
+                "entity_name": entity_name,
+                "entity_type": entity_type or "",
+                "invest_type": invest_type or "",
+                "page": page,
+                "page_size": page_size,
+            },
+        )
+
+    def model_financing_probability(
+        self,
+        entity_name: str,
+        entity_type: Optional[str] = None,
+    ) -> dict:
+        return self._request(
+            "POST",
+            "/rest-api/tool/model_financing_probability",
+            json_data={
+                "entity_name": entity_name,
+                "entity_type": entity_type or "",
+            },
+        )
+
+    def close(self):
+        self.client.close()
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self.close()

zerone_cli-0.1.2/cli/commands/__init__.py

@@ -0,0 +1,19 @@
+from cli.commands import config_cmd
+from cli.commands.entity_search import entity_search
+from cli.commands.entity_profile import entity_profile
+from cli.commands.entity_financial_list import entity_financial_list
+from cli.commands.entity_managed_funds_list import entity_managed_funds_list
+from cli.commands.entity_exit_list import entity_exit_list
+from cli.commands.entity_invest_list import entity_invest_list
+from cli.commands.model_financing_probability import model_financing_probability
+
+__all__ = [
+    "config_cmd",
+    "entity_search",
+    "entity_profile",
+    "entity_financial_list",
+    "entity_managed_funds_list",
+    "entity_exit_list",
+    "entity_invest_list",
+    "model_financing_probability",
+]

zerone_cli-0.1.2/cli/commands/basic_info.py

@@ -0,0 +1,77 @@
+"""
+查询基本信息命令模块
+"""
+
+import asyncio
+from typing import Optional
+
+import typer
+
+from cli.config import config_manager
+from cli.utils.output import print_result
+from mcp_server.tools.executor import ToolExecutor
+from mcp_server.router.entity_router import EntityRouter
+from mcp_server.router.type_router import TypeRouter
+from mcp_server.services.entity_service import EntityService
+from mcp_server.utils.field_mapper import FieldMapper
+from mcp_server.openapi_client.client import OpenAPIClient
+
+
+async def execute(entity_name: str, entity_type: Optional[str] = None) -> dict:
+    """
+    执行查询基本信息
+
+    Args:
+        entity_name: 实体名称
+        entity_type: 实体类型 (gp/lp/fund/company/preipo/list/neeq)
+
+    Returns:
+        查询结果字典
+    """
+    app_id, app_key = config_manager.get_credentials()
+    config = config_manager.load()
+
+    client = OpenAPIClient(base_url=config.base_url)
+    entity_service = EntityService()
+    executor = ToolExecutor(
+        entity_router=EntityRouter(entity_service),
+        type_router=TypeRouter(),
+        field_mapper=FieldMapper(),
+        openapi_client=client,
+    )
+
+    try:
+        result = await executor.execute_get_basic_info(
+            entity_name=entity_name,
+            entity_type=entity_type,
+            app_id=app_id,
+            app_key=app_key,
+        )
+        return result
+    finally:
+        await client.close()
+
+
+def get_basic_info(
+    entity_name: str = typer.Argument(..., help="实体名称，如'字节跳动'"),
+    entity_type: Optional[str] = typer.Option(
+        None, "--type", "-t", help="实体类型：gp/lp/fund/company/preipo/list/neeq"
+    ),
+    output: str = typer.Option(
+        "table", "--output", "-o", help="输出格式：table/json/csv"
+    ),
+):
+    """
+    查询实体基本信息
+
+    示例：
+        zerone-cli get-basic-info "字节跳动"
+        zerone-cli get-basic-info "红杉资本" --type gp
+        zerone-cli get-basic-info "字节跳动" --output json
+    """
+    try:
+        result = asyncio.run(execute(entity_name, entity_type))
+        print_result(result, output)
+    except ValueError as e:
+        typer.echo(f"[red]错误: {e}[/red]", err=True)
+        raise typer.Exit(1)

zerone_cli-0.1.2/cli/commands/config_cmd.py

@@ -0,0 +1,47 @@
+import typer
+
+from cli.config import config_manager
+
+app = typer.Typer(help="管理 CLI 配置")
+
+
+@app.command("show")
+def show_config():
+    """显示当前配置"""
+    config = config_manager.load()
+    typer.echo("=== 当前配置 ===")
+    typer.echo(f"api_key: {'*' * 10 if config.api_key else '(未设置)'}")
+    typer.echo(f"server_url: {config.server_url}")
+    typer.echo(f"output_format: {config.output_format}")
+
+
+@app.command("set-api-key")
+def set_api_key(api_key: str = typer.Argument(..., help="API Key（从平台获取）")):
+    """设置 API Key"""
+    config = config_manager.load()
+    config.api_key = api_key
+    config_manager.save(config)
+    typer.echo("✓ API Key 已设置")
+
+
+@app.command("set-server-url")
+def set_server_url(url: str = typer.Argument(..., help="MCP Server 地址，如 http://localhost:8080")):
+    """设置 MCP Server 地址"""
+    config = config_manager.load()
+    config.server_url = url
+    config_manager.save(config)
+    typer.echo(f"✓ Server URL 已设置: {url}")
+
+
+@app.command("set-output-format")
+def set_output_format(
+    fmt: str = typer.Argument(..., help="输出格式 (table/json/csv)")
+):
+    """设置默认输出格式"""
+    if fmt not in ("table", "json", "csv"):
+        typer.echo("✗ 无效的输出格式，请选择: table, json, csv")
+        raise typer.Exit(1)
+    config = config_manager.load()
+    config.output_format = fmt
+    config_manager.save(config)
+    typer.echo(f"✓ Output format 已设置: {fmt}")

zerone_cli-0.1.2/cli/commands/entity_exit_list.py

@@ -0,0 +1,47 @@
+from typing import Optional
+
+import typer
+
+from cli.client import MCPClient
+from cli.utils.output import print_result
+
+
+def entity_exit_list(
+    entity_name: str = typer.Argument(..., help="实体名称，如'高瓴资本'"),
+    entity_type: Optional[str] = typer.Option(
+        None, "--type", "-t",
+        help="实体类型：GP/LP/FUND"
+    ),
+    exit_type: Optional[str] = typer.Option(
+        None, "--exit-type", "-e",
+        help="退出类型：COMPANY/FUND"
+    ),
+    page: int = typer.Option(1, "--page", "-p", help="页码，从 1 开始"),
+    page_size: int = typer.Option(20, "--page-size", "-s", help="每页条数，默认 20"),
+    output: str = typer.Option("table", "--output", "-o", help="输出格式：table/json/csv"),
+):
+    """
+    查询退出列表（分页）
+
+    支持 GP/LP/FUND 类型。
+
+    示例：
+        zerone entity-exit-list "高瓴资本" --type GP
+        zerone entity-exit-list "高瓴资本" --type GP --exit-type COMPANY
+    """
+    try:
+        with MCPClient() as client:
+            result = client.entity_exit_list(
+                entity_name=entity_name,
+                entity_type=entity_type,
+                exit_type=exit_type,
+                page=page,
+                page_size=page_size,
+            )
+            print_result(result, output)
+    except ValueError as e:
+        typer.echo(f"错误: {e}", err=True)
+        raise typer.Exit(1)
+    except Exception as e:
+        typer.echo(f"请求失败: {e}", err=True)
+        raise typer.Exit(1)

zerone_cli-0.1.2/cli/commands/entity_financial_list.py

@@ -0,0 +1,42 @@
+from typing import Optional
+
+import typer
+
+from cli.client import MCPClient
+from cli.utils.output import print_result
+
+
+def entity_financial_list(
+    entity_name: str = typer.Argument(..., help="实体名称，如'美团'"),
+    entity_type: Optional[str] = typer.Option(
+        None, "--type", "-t",
+        help="实体类型：LISTED/PREIPO/NEEQ"
+    ),
+    page: int = typer.Option(1, "--page", "-p", help="页码，从 1 开始"),
+    page_size: int = typer.Option(20, "--page-size", "-s", help="每页条数，默认 20"),
+    output: str = typer.Option("table", "--output", "-o", help="输出格式：table/json/csv"),
+):
+    """
+    查询实体财务数据列表（分页）
+
+    仅支持 LISTED/PREIPO/NEEQ 类型。
+
+    示例：
+        zerone entity-financial-list "美团" --type LISTED
+        zerone entity-financial-list "美团" --type LISTED --page 2
+    """
+    try:
+        with MCPClient() as client:
+            result = client.entity_financial_list(
+                entity_name=entity_name,
+                entity_type=entity_type,
+                page=page,
+                page_size=page_size,
+            )
+            print_result(result, output)
+    except ValueError as e:
+        typer.echo(f"错误: {e}", err=True)
+        raise typer.Exit(1)
+    except Exception as e:
+        typer.echo(f"请求失败: {e}", err=True)
+        raise typer.Exit(1)

zerone_cli-0.1.2/cli/commands/entity_invest_list.py

@@ -0,0 +1,47 @@
+from typing import Optional
+
+import typer
+
+from cli.client import MCPClient
+from cli.utils.output import print_result
+
+
+def entity_invest_list(
+    entity_name: str = typer.Argument(..., help="实体名称，如'高瓴资本'"),
+    entity_type: Optional[str] = typer.Option(
+        None, "--type", "-t",
+        help="实体类型：GP/LP/FUND"
+    ),
+    invest_type: Optional[str] = typer.Option(
+        None, "--invest-type", "-i",
+        help="投资类型：COMPANY/FUND/OTHER"
+    ),
+    page: int = typer.Option(1, "--page", "-p", help="页码，从 1 开始"),
+    page_size: int = typer.Option(20, "--page-size", "-s", help="每页条数，默认 20"),
+    output: str = typer.Option("table", "--output", "-o", help="输出格式：table/json/csv"),
+):
+    """
+    查询投资列表（分页）
+
+    支持 GP/LP/FUND 类型。
+
+    示例：
+        zerone entity-invest-list "高瓴资本" --type GP
+        zerone entity-invest-list "高瓴资本" --type GP --invest-type COMPANY
+    """
+    try:
+        with MCPClient() as client:
+            result = client.entity_invest_list(
+                entity_name=entity_name,
+                entity_type=entity_type,
+                invest_type=invest_type,
+                page=page,
+                page_size=page_size,
+            )
+            print_result(result, output)
+    except ValueError as e:
+        typer.echo(f"错误: {e}", err=True)
+        raise typer.Exit(1)
+    except Exception as e:
+        typer.echo(f"请求失败: {e}", err=True)
+        raise typer.Exit(1)

zerone_cli-0.1.2/cli/commands/entity_managed_funds_list.py

@@ -0,0 +1,42 @@
+from typing import Optional
+
+import typer
+
+from cli.client import MCPClient
+from cli.utils.output import print_result
+
+
+def entity_managed_funds_list(
+    entity_name: str = typer.Argument(..., help="实体名称，如'红杉资本'"),
+    entity_type: Optional[str] = typer.Option(
+        None, "--type", "-t",
+        help="实体类型：GP/FUND_MANAGER"
+    ),
+    page: int = typer.Option(1, "--page", "-p", help="页码，从 1 开始"),
+    page_size: int = typer.Option(20, "--page-size", "-s", help="每页条数，默认 20"),
+    output: str = typer.Option("table", "--output", "-o", help="输出格式：table/json/csv"),
+):
+    """
+    查询管理基金列表（分页）
+
+    支持 GP 和 FUND_MANAGER 类型。
+
+    示例：
+        zerone entity-managed-funds-list "红杉资本" --type GP
+        zerone entity-managed-funds-list "红杉资本" --type GP --page 2
+    """
+    try:
+        with MCPClient() as client:
+            result = client.entity_managed_funds_list(
+                entity_name=entity_name,
+                entity_type=entity_type,
+                page=page,
+                page_size=page_size,
+            )
+            print_result(result, output)
+    except ValueError as e:
+        typer.echo(f"错误: {e}", err=True)
+        raise typer.Exit(1)
+    except Exception as e:
+        typer.echo(f"请求失败: {e}", err=True)
+        raise typer.Exit(1)

zerone_cli-0.1.2/cli/commands/entity_profile.py

@@ -0,0 +1,40 @@
+from typing import Optional
+
+import typer
+
+from cli.client import MCPClient
+from cli.utils.output import print_result
+
+
+def entity_profile(
+    entity_name: str = typer.Argument(..., help="实体名称，如'字节跳动'"),
+    entity_type: Optional[str] = typer.Option(
+        None, "--type", "-t",
+        help="实体类型：GP/FUND/LP/COMPANY/PREIPO/LISTED/NEEQ/FUND_MANAGER"
+    ),
+    output: str = typer.Option("table", "--output", "-o", help="输出格式：table/json/csv"),
+):
+    """
+    查询实体综合画像
+
+    返回基本信息、标签画像、概览数据、股东列表、融资历史、关联GP列表等。
+    支持全部 8 种实体类型：GP/FUND/LP/COMPANY/FUND_MANAGER/PREIPO/LISTED/NEEQ。
+
+    示例：
+        zerone entity-profile "字节跳动"
+        zerone entity-profile "红杉资本" --type GP
+        zerone entity-profile "字节跳动" --output json
+    """
+    try:
+        with MCPClient() as client:
+            result = client.entity_profile(
+                entity_name=entity_name,
+                entity_type=entity_type,
+            )
+            print_result(result, output)
+    except ValueError as e:
+        typer.echo(f"错误: {e}", err=True)
+        raise typer.Exit(1)
+    except Exception as e:
+        typer.echo(f"请求失败: {e}", err=True)
+        raise typer.Exit(1)

zerone_cli-0.1.2/cli/commands/entity_search.py

@@ -0,0 +1,31 @@
+from typing import Optional
+
+import typer
+
+from cli.client import MCPClient
+from cli.utils.output import print_result
+
+
+def entity_search(
+    keyword: str = typer.Argument(..., help="搜索关键词"),
+    output: str = typer.Option("table", "--output", "-o", help="输出格式：table/json/csv"),
+):
+    """
+    实体模糊搜索
+
+    根据关键词搜索匹配的实体，返回候选实体列表。
+
+    示例：
+        zerone entity-search "字节跳动"
+        zerone entity-search "红杉" --output json
+    """
+    try:
+        with MCPClient() as client:
+            result = client.entity_search(keyword=keyword)
+            print_result(result, output)
+    except ValueError as e:
+        typer.echo(f"错误: {e}", err=True)
+        raise typer.Exit(1)
+    except Exception as e:
+        typer.echo(f"请求失败: {e}", err=True)
+        raise typer.Exit(1)

zerone_cli-0.1.2/cli/commands/financing.py

@@ -0,0 +1,84 @@
+"""
+查询融资概率命令模块
+"""
+
+import asyncio
+from typing import Optional
+
+import typer
+
+from cli.config import config_manager
+from cli.utils.output import print_result
+from mcp_server.tools.executor import ToolExecutor
+from mcp_server.router.entity_router import EntityRouter
+from mcp_server.router.type_router import TypeRouter
+from mcp_server.services.entity_service import EntityService
+from mcp_server.utils.field_mapper import FieldMapper
+from mcp_server.openapi_client.client import OpenAPIClient
+
+
+async def execute(
+    entity_name: str,
+    entity_type: Optional[str] = None,
+) -> dict:
+    """
+    执行查询融资概率
+
+    注意：此命令仅支持项目公司类型（company/preipo/list/neeq）
+
+    Args:
+        entity_name: 实体名称
+        entity_type: 实体类型
+
+    Returns:
+        查询结果字典
+    """
+    app_id, app_key = config_manager.get_credentials()
+    config = config_manager.load()
+
+    client = OpenAPIClient(base_url=config.base_url)
+    entity_service = EntityService()
+    executor = ToolExecutor(
+        entity_router=EntityRouter(entity_service),
+        type_router=TypeRouter(),
+        field_mapper=FieldMapper(),
+        openapi_client=client,
+    )
+
+    try:
+        result = await executor.execute_get_financing_probability(
+            entity_name=entity_name,
+            entity_type=entity_type,
+            app_id=app_id,
+            app_key=app_key,
+        )
+        return result
+    finally:
+        await client.close()
+
+
+def get_financing_probability(
+    entity_name: str = typer.Argument(..., help="实体名称，如'字节跳动'"),
+    entity_type: Optional[str] = typer.Option(
+        None, "--type", "-t", help="实体类型：company/preipo/list/neeq"
+    ),
+    output: str = typer.Option(
+        "table", "--output", "-o", help="输出格式：table/json/csv"
+    ),
+):
+    """
+    查询公司融资概率（2年）
+
+    注意：此命令仅支持项目公司类型（company/preipo/list/neeq）
+
+    示例：
+        zerone-cli get-financing-probability "蔚来汽车"
+        zerone-cli get-financing-probability "字节跳动" --type company
+        zerone-cli get-financing-probability "蔚来汽车" --output json
+    """
+    try:
+        result = asyncio.run(execute(entity_name, entity_type))
+        print_result(result, output)
+    except ValueError as e:
+        typer.echo(f"[red]错误: {e}[/red]", err=True)
+        raise typer.Exit(1)

zerone_cli-0.1.2/cli/commands/model_financing_probability.py

@@ -0,0 +1,39 @@
+from typing import Optional
+
+import typer
+
+from cli.client import MCPClient
+from cli.utils.output import print_result
+
+
+def model_financing_probability(
+    entity_name: str = typer.Argument(..., help="实体名称，如'蔚来汽车'"),
+    entity_type: Optional[str] = typer.Option(
+        None, "--type", "-t",
+        help="实体类型：COMPANY/PREIPO/LISTED/NEEQ"
+    ),
+    output: str = typer.Option("table", "--output", "-o", help="输出格式：table/json/csv"),
+):
+    """
+    查询项目公司融资概率预测模型
+
+    仅支持 COMPANY/PREIPO/LISTED/NEEQ 类型。
+    返回融资概率（2年）和模型指标数据。
+
+    示例：
+        zerone model-financing-probability "蔚来汽车"
+        zerone model-financing-probability "蔚来汽车" --type COMPANY
+    """
+    try:
+        with MCPClient() as client:
+            result = client.model_financing_probability(
+                entity_name=entity_name,
+                entity_type=entity_type,
+            )
+            print_result(result, output)
+    except ValueError as e:
+        typer.echo(f"错误: {e}", err=True)
+        raise typer.Exit(1)
+    except Exception as e:
+        typer.echo(f"请求失败: {e}", err=True)
+        raise typer.Exit(1)

zerone_cli-0.1.2/cli/commands/shareholders.py

@@ -0,0 +1,88 @@
+"""
+查询股东信息命令模块
+"""
+
+import asyncio
+from typing import Optional
+
+import typer
+
+from cli.config import config_manager
+from cli.utils.output import print_result
+from mcp_server.tools.executor import ToolExecutor
+from mcp_server.router.entity_router import EntityRouter
+from mcp_server.router.type_router import TypeRouter
+from mcp_server.services.entity_service import EntityService
+from mcp_server.utils.field_mapper import FieldMapper
+from mcp_server.openapi_client.client import OpenAPIClient
+
+
+async def execute(
+    entity_name: str,
+    entity_type: Optional[str] = None,
+    page: int = 1,
+    page_size: int = 20,
+) -> dict:
+    """
+    执行查询股东信息
+
+    Args:
+        entity_name: 实体名称
+        entity_type: 实体类型
+        page: 页码，从 1 开始
+        page_size: 每页数量
+
+    Returns:
+        查询结果字典
+    """
+    app_id, app_key = config_manager.get_credentials()
+    config = config_manager.load()
+
+    client = OpenAPIClient(base_url=config.base_url)
+    entity_service = EntityService()
+    executor = ToolExecutor(
+        entity_router=EntityRouter(entity_service),
+        type_router=TypeRouter(),
+        field_mapper=FieldMapper(),
+        openapi_client=client,
+    )
+
+    try:
+        result = await executor.execute_get_shareholders(
+            entity_name=entity_name,
+            entity_type=entity_type,
+            page=page,
+            page_size=page_size,
+            app_id=app_id,
+            app_key=app_key,
+        )
+        return result
+    finally:
+        await client.close()
+
+
+def get_shareholders(
+    entity_name: str = typer.Argument(..., help="实体名称，如'字节跳动'"),
+    entity_type: Optional[str] = typer.Option(
+        None, "--type", "-t", help="实体类型：gp/lp/fund/company/preipo/list/neeq"
+    ),
+    page: int = typer.Option(1, "--page", "-p", help="页码"),
+    page_size: int = typer.Option(20, "--page-size", "-s", help="每页数量"),
+    output: str = typer.Option(
+        "table", "--output", "-o", help="输出格式：table/json/csv"
+    ),
+):
+    """
+    查询实体股东信息（支持分页）
+
+    示例：
+        zerone-cli get-shareholders "美团"
+        zerone-cli get-shareholders "高瓴资本" --type gp --page 1 --page-size 10
+        zerone-cli get-shareholders "美团" --output csv
+    """
+    try:
+        result = asyncio.run(execute(entity_name, entity_type, page, page_size))
+        print_result(result, output)
+    except ValueError as e:
+        typer.echo(f"[red]错误: {e}[/red]", err=True)
+        raise typer.Exit(1)

zerone_cli-0.1.2/cli/config.py

@@ -0,0 +1,69 @@
+import os
+import yaml
+from pathlib import Path
+from typing import Optional
+
+from pydantic import BaseModel
+
+SERVER_URL_PROD = "https://ai.zerone.com.cn/mcp"
+SERVER_URL_DEV = "http://ai-dev.zerone.com.cn/mcp"
+
+_env = os.getenv("ZERONE_ENV", "prod").lower()
+DEFAULT_SERVER_URL = SERVER_URL_DEV if _env == "dev" else SERVER_URL_PROD
+
+
+class CLIConfig(BaseModel):
+    api_key: Optional[str] = None
+    server_url: str = DEFAULT_SERVER_URL
+    output_format: str = "table"
+
+
+class ConfigManager:
+
+    CONFIG_DIR = Path.home() / ".zerone"
+    CONFIG_FILE = CONFIG_DIR / "config.yaml"
+
+    def __init__(self):
+        self._config: Optional[CLIConfig] = None
+
+    def _ensure_config_dir(self):
+        self.CONFIG_DIR.mkdir(parents=True, exist_ok=True)
+
+    def load(self) -> CLIConfig:
+        if self._config:
+            return self._config
+
+        if self.CONFIG_FILE.exists():
+            with open(self.CONFIG_FILE, "r", encoding="utf-8") as f:
+                data = yaml.safe_load(f) or {}
+        else:
+            data = {}
+
+        data["api_key"] = os.getenv("ZERONE_API_KEY", data.get("api_key"))
+        data["server_url"] = os.getenv("ZERONE_SERVER_URL", data.get("server_url", DEFAULT_SERVER_URL))
+
+        self._config = CLIConfig(**data)
+        return self._config
+
+    def save(self, config: CLIConfig):
+        self._ensure_config_dir()
+        self._config = config
+        with open(self.CONFIG_FILE, "w", encoding="utf-8") as f:
+            yaml.dump(config.model_dump(), f, allow_unicode=True, default_flow_style=False)
+
+    def get_api_key(self) -> str:
+        config = self.load()
+        if not config.api_key:
+            raise ValueError(
+                "未配置 API Key。\n"
+                "请运行：zerone config set-api-key <your-api-key>\n"
+                "或设置环境变量：ZERONE_API_KEY"
+            )
+        return config.api_key
+
+    def get_server_url(self) -> str:
+        config = self.load()
+        return config.server_url
+
+
+config_manager = ConfigManager()

zerone_cli-0.1.2/cli/main.py

@@ -0,0 +1,29 @@
+import typer
+
+from cli.commands import config_cmd
+from cli.commands.entity_search import entity_search
+from cli.commands.entity_profile import entity_profile
+from cli.commands.entity_financial_list import entity_financial_list
+from cli.commands.entity_managed_funds_list import entity_managed_funds_list
+from cli.commands.entity_exit_list import entity_exit_list
+from cli.commands.entity_invest_list import entity_invest_list
+from cli.commands.model_financing_probability import model_financing_probability
+
+app = typer.Typer(
+    name="zerone",
+    help="Zerone 私募股权数据 CLI 工具 - MCP HTTP 客户端",
+    no_args_is_help=True,
+)
+
+app.add_typer(config_cmd.app, name="config")
+
+app.command("entity-search")(entity_search)
+app.command("entity-profile")(entity_profile)
+app.command("entity-financial-list")(entity_financial_list)
+app.command("entity-managed-funds-list")(entity_managed_funds_list)
+app.command("entity-exit-list")(entity_exit_list)
+app.command("entity-invest-list")(entity_invest_list)
+app.command("model-financing-probability")(model_financing_probability)
+
+if __name__ == "__main__":
+    app()

zerone_cli-0.1.2/cli/utils/__init__.py

@@ -0,0 +1,3 @@
+from .output import print_result, print_error
+
+__all__ = ["print_result", "print_error"]

zerone_cli-0.1.2/cli/utils/output.py

@@ -0,0 +1,97 @@
+"""
+CLI 输出格式化模块
+
+支持 table、json、csv 三种输出格式。
+"""
+
+import csv
+import io
+import json
+from typing import Any
+
+from rich.console import Console
+from rich.table import Table
+
+console = Console()
+
+
+def print_result(result: dict, output_format: str = "table"):
+    """
+    主入口函数，根据格式输出结果
+
+    Args:
+        result: 包含 data、metadata 的结果字典
+        output_format: 输出格式 (table/json/csv)
+    """
+    if "error" in result:
+        print_error(result)
+        return
+
+    data = result.get("data", {})
+    metadata = result.get("metadata", {})
+
+    if output_format == "json":
+        print_json(data)
+    elif output_format == "csv":
+        print_csv(data)
+    else:
+        print_table(data, metadata)
+
+
+def print_table(data: Any, metadata: dict):
+    """表格格式输出"""
+    if isinstance(data, list):
+        if not data:
+            console.print("[yellow]没有查询到数据[/yellow]")
+            return
+        keys = list(data[0].keys()) if data else []
+        table = Table(show_header=True, header_style="bold cyan", box=None)
+        for key in keys:
+            table.add_column(key, style="white")
+        for row in data:
+            table.add_row(*[str(row.get(k, "")) for k in keys])
+        console.print(table)
+    elif isinstance(data, dict):
+        table = Table(show_header=True, header_style="bold cyan", box=None)
+        table.add_column("字段", style="cyan")
+        table.add_column("值", style="white")
+        for key, value in data.items():
+            table.add_row(key, str(value) if value is not None else "")
+        console.print(table)
+    else:
+        console.print(str(data))
+
+    entity_name = metadata.get("entityName", "")
+    entity_type = metadata.get("entityType", "")
+    if entity_name:
+        console.print(f"\n[dim]实体名称: {entity_name} | 类型: {entity_type}[/dim]")
+
+
+def print_json(data: Any):
+    """JSON 格式输出"""
+    console.print(json.dumps(data, ensure_ascii=False, indent=2))
+
+
+def print_csv(data: Any):
+    """CSV 格式输出"""
+    output = io.StringIO()
+    if isinstance(data, list):
+        if not data:
+            return
+        writer = csv.DictWriter(output, fieldnames=data[0].keys())
+        writer.writeheader()
+        writer.writerows(data)
+    elif isinstance(data, dict):
+        writer = csv.DictWriter(output, fieldnames=data.keys())
+        writer.writeheader()
+        writer.writerow(data)
+    else:
+        output.write(str(data))
+    console.print(output.getvalue())
+
+
+def print_error(error: dict):
+    """错误信息输出"""
+    code = error.get("code", error.get("error_code", "UNKNOWN"))
+    message = error.get("message", error.get("msg", "未知错误"))
+    console.print(f"[bold red]错误 {code}:[/bold red] {message}")

{zerone_cli-0.1.1 → zerone_cli-0.1.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "zerone-cli"
-version = "0.1.1"
+version = "0.1.2"
 description = "Zerone 私募股权数据 CLI 工具"
 readme = "README.md"
 license = { text = "MIT" }
@@ -42,8 +42,9 @@ requires = ["setuptools>=61.0", "wheel"]
 build-backend = "setuptools.build_meta"
 
 [tool.setuptools.packages.find]
-where = [".."]
-include = ["cli*"]
+where = ["."]
+include = ["cli", "cli.commands", "cli.utils"]
+namespaces = false
 
 # 环境说明：
 # prod（默认）: ZERONE_ENV=prod  -> https://ai.zerone.com.cn/mcp

zerone_cli-0.1.2/zerone_cli.egg-info/PKG-INFO

@@ -0,0 +1,110 @@
+Metadata-Version: 2.4
+Name: zerone-cli
+Version: 0.1.2
+Summary: Zerone 私募股权数据 CLI 工具
+Author-email: Zerone <tech@zerone.com.cn>
+License: MIT
+Project-URL: Homepage, https://ai.zerone.com.cn
+Project-URL: Documentation, https://ai.zerone.com.cn
+Keywords: cli,private-equity,finance,mcp,zerone
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Operating System :: OS Independent
+Classifier: Environment :: Console
+Classifier: Intended Audience :: Financial and Insurance Industry
+Classifier: Topic :: Office/Business :: Financial
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+Requires-Dist: typer>=0.9.0
+Requires-Dist: httpx>=0.24.0
+Requires-Dist: pydantic>=2.0.0
+Requires-Dist: pyyaml>=6.0
+Requires-Dist: rich>=13.0.0
+Requires-Dist: pandas>=2.0.0
+
+# Zerone CLI
+
+Zerone 私募股权数据命令行工具，通过 API Key 直接查询 Zerone 数据平台。
+
+## 安装
+
+```bash
+pip install zerone-cli
+```
+
+## 快速开始
+
+### 1. 配置 API Key
+
+```bash
+zerone config set-api-key YOUR_API_KEY
+```
+
+> 如何获取 API Key？请访问 [https://ai.zerone.com.cn](https://ai.zerone.com.cn) 注册账号后申请。
+
+### 2. 查看帮助
+
+```bash
+zerone --help
+```
+
+### 3. 开始查询
+
+```bash
+# 搜索实体
+zerone entity-search "字节跳动"
+
+# 查询实体综合画像
+zerone entity-profile "中国石油"
+
+# 查询财务数据列表
+zerone entity-financial-list "中国石油"
+
+# 查询管理基金列表
+zerone entity-managed-funds-list "深创投"
+
+# 查询退出列表
+zerone entity-exit-list "深创投"
+
+# 查询投资列表
+zerone entity-invest-list "深创投"
+
+# 查询融资概率预测
+zerone model-financing-probability "字节跳动"
+```
+
+## 命令列表
+
+| 命令 | 说明 |
+|------|------|
+| `zerone config set-api-key` | 设置 API Key |
+| `zerone config set-server-url` | 设置服务器地址（可选） |
+| `zerone config show` | 查看当前配置 |
+| `zerone entity-search` | 实体模糊搜索 |
+| `zerone entity-profile` | 查询实体综合画像 |
+| `zerone entity-financial-list` | 查询实体财务数据列表（分页） |
+| `zerone entity-managed-funds-list` | 查询管理基金列表（分页） |
+| `zerone entity-exit-list` | 查询退出列表（分页） |
+| `zerone entity-invest-list` | 查询投资列表（分页） |
+| `zerone model-financing-probability` | 查询融资概率预测模型 |
+
+## 配置说明
+
+配置文件保存在 `~/.zerone/config.yaml`，支持以下环境变量覆盖：
+
+| 环境变量 | 说明 |
+|----------|------|
+| `ZERONE_API_KEY` | API Key |
+| `ZERONE_SERVER_URL` | 服务器地址 |
+
+## 系统要求
+
+- Python 3.9+
+- 有效的 Zerone API Key
+
+## 许可证
+
+MIT

zerone_cli-0.1.2/zerone_cli.egg-info/SOURCES.txt

@@ -0,0 +1,27 @@
+README.md
+pyproject.toml
+cli/__init__.py
+cli/client.py
+cli/config.py
+cli/main.py
+cli/commands/__init__.py
+cli/commands/basic_info.py
+cli/commands/config_cmd.py
+cli/commands/entity_exit_list.py
+cli/commands/entity_financial_list.py
+cli/commands/entity_invest_list.py
+cli/commands/entity_managed_funds_list.py
+cli/commands/entity_profile.py
+cli/commands/entity_search.py
+cli/commands/financing.py
+cli/commands/model_financing_probability.py
+cli/commands/shareholders.py
+cli/utils/__init__.py
+cli/utils/errors.py
+cli/utils/output.py
+zerone_cli.egg-info/PKG-INFO
+zerone_cli.egg-info/SOURCES.txt
+zerone_cli.egg-info/dependency_links.txt
+zerone_cli.egg-info/entry_points.txt
+zerone_cli.egg-info/requires.txt
+zerone_cli.egg-info/top_level.txt

zerone_cli-0.1.2/zerone_cli.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

zerone_cli-0.1.2/zerone_cli.egg-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+zerone = cli.main:app
+zerone-cli = cli.main:app

zerone_cli-0.1.2/zerone_cli.egg-info/requires.txt

@@ -0,0 +1,6 @@
+typer>=0.9.0
+httpx>=0.24.0
+pydantic>=2.0.0
+pyyaml>=6.0
+rich>=13.0.0
+pandas>=2.0.0

zerone_cli-0.1.2/zerone_cli.egg-info/top_level.txt

@@ -0,0 +1 @@
+cli