coze-coding-dev-sdk 0.5.10__tar.gz → 0.5.12__tar.gz

{coze_coding_dev_sdk-0.5.10/coze_coding_dev_sdk.egg-info → coze_coding_dev_sdk-0.5.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: coze-coding-dev-sdk
-Version: 0.5.10
+Version: 0.5.12
 Summary: Coze Coding Dev SDK - 优雅的多功能 AI SDK，支持图片生成、视频生成、语音合成、语音识别、大语言模型、联网搜索和文本/多模态 Embedding。包含命令行工具 coze-coding-ai，支持 Context 上下文追踪
 Author-email: Coze Coding Integration Team <support@coze.com>
 Maintainer-email: Coze Coding Integration Team <support@coze.com>
@@ -44,6 +44,10 @@ Requires-Dist: click>=8.0.0
 Requires-Dist: rich>=13.0.0
 Requires-Dist: supabase==2.27.3
 Requires-Dist: httpx==0.28.1
+Requires-Dist: markdown>=3.0.0
+Requires-Dist: python-docx>=1.0.0
+Requires-Dist: reportlab>=4.0.0
+Requires-Dist: cssutils>=2.0.0
 Provides-Extra: dev
 Requires-Dist: pytest>=7.0.0; extra == "dev"
 Requires-Dist: pytest-asyncio>=0.21.0; extra == "dev"

coze_coding_dev_sdk-0.5.12/coze_coding_dev_sdk/cli/db.py

@@ -0,0 +1,156 @@
+"""
+数据库 CLI 命令
+"""
+
+import json
+import os
+from dataclasses import asdict
+from typing import Optional
+
+import click
+from rich.console import Console
+
+# 默认路径
+DEFAULT_MODEL_OUTPUT = "src/storage/database/shared/model.py"
+DEFAULT_MODEL_IMPORT_PATH = "storage.database.shared.model"
+DEFAULT_MODEL_PATH = "src"
+
+console = Console()
+
+
+def _get_workspace_path() -> str:
+    """获取工作目录"""
+    return os.getenv("WORKSPACE_PATH", os.getcwd())
+
+
+@click.group()
+def db():
+    """Database management commands."""
+    pass
+
+
+@db.command()
+@click.argument("output_path", required=False, default=None)
+@click.option("--verbose", "-v", is_flag=True, help="Verbose output")
+def generate_models(output_path: str, verbose: bool):
+    """
+    Generate ORM models from database.
+
+    OUTPUT_PATH: Path to output model file (default: src/storage/database/shared/model.py)
+    """
+    from coze_coding_dev_sdk.database import generate_models as _generate_models
+
+    if output_path is None:
+        workspace = _get_workspace_path()
+        output_path = os.path.join(workspace, DEFAULT_MODEL_OUTPUT)
+
+    try:
+        _generate_models(output_path, verbose=verbose)
+        click.echo(f"Models generated at {output_path}")
+    except Exception as e:
+        click.echo(f"Error: {e}", err=True)
+        raise SystemExit(1)
+
+
+@db.command()
+@click.option(
+    "--model-import-path",
+    default=DEFAULT_MODEL_IMPORT_PATH,
+    help=f"Model import path (default: {DEFAULT_MODEL_IMPORT_PATH})",
+)
+@click.option(
+    "--model-path",
+    default=None,
+    help=f"Path to add to sys.path for model import (default: $WORKSPACE_PATH/{DEFAULT_MODEL_PATH})",
+)
+@click.option("--verbose", "-v", is_flag=True, help="Verbose output")
+def upgrade(model_import_path: str, model_path: str, verbose: bool):
+    """
+    Run database migrations.
+
+    Automatically generates migration and upgrades to head.
+    """
+    from coze_coding_dev_sdk.database import upgrade as _upgrade
+
+    if model_path is None:
+        workspace = _get_workspace_path()
+        model_path = os.path.join(workspace, DEFAULT_MODEL_PATH)
+
+    try:
+        _upgrade(
+            model_import_path=model_import_path,
+            model_path=model_path,
+            verbose=verbose,
+        )
+        click.echo("Database upgraded successfully")
+    except Exception as e:
+        click.echo(f"Error: {e}", err=True)
+        raise SystemExit(1)
+
+
+@db.command()
+@click.option(
+    "--env",
+    "-e",
+    type=click.Choice(["dev", "prod", "develop", "product"]),
+    default=None,
+    help="目标环境: dev/develop 或 prod/product，不传则使用 token 中的环境",
+)
+@click.option("--mock", is_flag=True, help="使用 mock 模式（测试运行，不会真正执行操作）")
+@click.option(
+    "--header",
+    "-H",
+    multiple=True,
+    help="自定义 HTTP 请求头 (格式: 'Key: Value' 或 'Key=Value'，可多次使用)",
+)
+@click.option("--verbose", "-v", is_flag=True, help="显示详细的 HTTP 请求和响应日志")
+def diagnose(env: Optional[str], mock: bool, header: tuple, verbose: bool):
+    """获取数据库运行状态诊断报告。
+
+    诊断内容包括：慢查询、连接状态、阻塞查询、长时间运行查询、数据库大小、表大小等。
+
+    \b
+    示例：
+      coze-coding-ai db diagnose
+      coze-coding-ai db diagnose --env dev
+      coze-coding-ai db diagnose --env prod --verbose
+    """
+    try:
+        from coze_coding_utils.runtime_ctx.context import new_context
+
+        from ..core.config import Config
+        from ..supabase import SupabaseClient
+        from .constants import RUN_MODE_HEADER, RUN_MODE_TEST
+        from .utils import parse_headers
+
+        # normalize env aliases
+        env_map = {"develop": "dev", "product": "prod"}
+        if env in env_map:
+            env = env_map[env]
+
+        config = Config()
+        ctx = None
+        custom_headers = parse_headers(header) or {}
+
+        if mock:
+            ctx = new_context(method="db.diagnose", headers=custom_headers)
+            custom_headers[RUN_MODE_HEADER] = RUN_MODE_TEST
+            console.print("[yellow]🧪 Mock 模式已启用（测试运行）[/yellow]")
+
+        client = SupabaseClient(config, ctx=ctx, custom_headers=custom_headers, verbose=verbose)
+        response = client.diagnose_database(env=env)
+
+        if response.code != 0:
+            msg = response.msg or ""
+            if "record not found" in msg.lower():
+                env_label = env or "dev/prod"
+                console.print(f"[red]✗ {env_label} 数据库还未创建，请先创建数据库后再进行诊断[/red]")
+            else:
+                console.print(f"[red]Error: {msg}[/red]")
+            return
+
+        console.print(json.dumps(asdict(response.report) if response.report else None, indent=2, ensure_ascii=False))
+
+    except Exception as e:
+        console.print(f"[red]✗ Error: {str(e)}[/red]")
+        raise click.Abort()

{coze_coding_dev_sdk-0.5.10 → coze_coding_dev_sdk-0.5.12}/coze_coding_dev_sdk/llm/client.py

@@ -109,11 +109,6 @@ class LLMClient:
                 - AIMessage: AI 回复，用于多轮对话
 
             model: 模型ID，默认 "doubao-seed-1-8-251228"
-                可选模型：
-                - "doubao-seed-1-8-251228": 最新模型，更高性能
-                - "doubao-seed-1-6-251015": 平衡性能
-                - "doubao-seed-1-6-flash-250615": 快速模型
-                - "doubao-seed-1-6-thinking-250715": 思考模型
 
             thinking: 思考模式，默认 "disabled"
                 - "enabled": 启用深度思考，适合复杂推理任务
@@ -227,11 +222,6 @@ class LLMClient:
                 - AIMessage: AI 回复，用于多轮对话
 
             model: 模型ID，默认 "doubao-seed-1-8-251228"
-                可选模型：
-                - "doubao-seed-1-8-251228": 最新模型，更高性能
-                - "doubao-seed-1-6-251015": 平衡性能
-                - "doubao-seed-1-6-flash-250615": 快速模型
-                - "doubao-seed-1-6-thinking-250715": 思考模型
 
             thinking: 思考模式，默认 "disabled"
                 - "enabled": 启用深度思考，适合复杂推理任务

{coze_coding_dev_sdk-0.5.10 → coze_coding_dev_sdk-0.5.12}/coze_coding_dev_sdk/supabase/__init__.py

@@ -1,11 +1,16 @@
 from .client import SupabaseClient
 from .models import (
     AuthEmailConfig,
+    BlockedQueryInfo,
     Bucket,
+    ConnectionStateCount,
+    ConnectionStats,
     CreateBucketResponse,
     DeleteBucketResponse,
     DeleteEdgeFunctionResponse,
     DeployEdgeFunctionResponse,
+    DiagnoseDatabaseResponse,
+    DiagnosticReport,
     EdgeFunction,
     FunctionFile,
     GetAuthConfigResponse,
@@ -13,6 +18,11 @@ from .models import (
     GetEdgeFunctionResponse,
     ListBucketsResponse,
     ListEdgeFunctionsResponse,
+    LongRunningQueryInfo,
+    PGDiagnosticMetrics,
+    QueryStatsInfo,
+    SupabaseDiagnosticMetrics,
+    TableSizeInfo,
     UpdateAuthConfigResponse,
     UpdateBucketResponse,
 )
@@ -34,4 +44,14 @@ __all__ = [
     "DeleteBucketResponse",
     "GetAuthConfigResponse",
     "UpdateAuthConfigResponse",
+    "QueryStatsInfo",
+    "ConnectionStateCount",
+    "ConnectionStats",
+    "BlockedQueryInfo",
+    "LongRunningQueryInfo",
+    "TableSizeInfo",
+    "SupabaseDiagnosticMetrics",
+    "PGDiagnosticMetrics",
+    "DiagnosticReport",
+    "DiagnoseDatabaseResponse",
 ]

{coze_coding_dev_sdk-0.5.10 → coze_coding_dev_sdk-0.5.12}/coze_coding_dev_sdk/supabase/client.py

@@ -6,11 +6,16 @@ from ..core.client import BaseClient
 from ..core.config import Config
 from .models import (
     AuthEmailConfig,
+    BlockedQueryInfo,
     Bucket,
+    ConnectionStateCount,
+    ConnectionStats,
     CreateBucketResponse,
     DeleteBucketResponse,
     DeleteEdgeFunctionResponse,
     DeployEdgeFunctionResponse,
+    DiagnoseDatabaseResponse,
+    DiagnosticReport,
     EdgeFunction,
     FunctionFile,
     GetAuthConfigResponse,
@@ -18,6 +23,11 @@ from .models import (
     GetEdgeFunctionResponse,
     ListBucketsResponse,
     ListEdgeFunctionsResponse,
+    LongRunningQueryInfo,
+    PGDiagnosticMetrics,
+    QueryStatsInfo,
+    SupabaseDiagnosticMetrics,
+    TableSizeInfo,
     UpdateAuthConfigResponse,
     UpdateBucketResponse,
 )
@@ -334,3 +344,71 @@ class SupabaseClient(BaseClient):
             code=response.get("code", 0),
             msg=response.get("msg", ""),
         )
+
+    def diagnose_database(
+        self,
+        env: Optional[str] = None,
+    ) -> DiagnoseDatabaseResponse:
+        payload: Dict = {}
+        if env is not None:
+            payload["env"] = env
+
+        response = self._request(
+            method="POST",
+            url=f"{self.base_url}/v1/supabase/database/diagnose",
+            json=payload,
+            headers=self._get_headers(),
+        )
+
+        code = response.get("code", 0)
+        msg = response.get("msg", "")
+
+        report = None
+        r = response.get("report")
+        if r:
+            pg_raw = r.get("pg_metrics") or {}
+            pg_metrics = PGDiagnosticMetrics(
+                top_queries_by_total_time=[
+                    QueryStatsInfo(**q) for q in (pg_raw.get("top_queries_by_total_time") or [])
+                ],
+                top_queries_by_frequency=[
+                    QueryStatsInfo(**q) for q in (pg_raw.get("top_queries_by_frequency") or [])
+                ],
+                slow_queries=[
+                    QueryStatsInfo(**q) for q in (pg_raw.get("slow_queries") or [])
+                ],
+                connection_stats=ConnectionStats(
+                    connections_by_state=[
+                        ConnectionStateCount(**c)
+                        for c in (pg_raw.get("connection_stats", {}).get("connections_by_state") or [])
+                    ],
+                    max_connections=pg_raw.get("connection_stats", {}).get("max_connections"),
+                    connection_usage_pct=pg_raw.get("connection_stats", {}).get("connection_usage_pct"),
+                    total_connections=pg_raw.get("connection_stats", {}).get("total_connections"),
+                ) if pg_raw.get("connection_stats") else None,
+                blocked_queries=[
+                    BlockedQueryInfo(**q) for q in (pg_raw.get("blocked_queries") or [])
+                ],
+                long_running_queries=[
+                    LongRunningQueryInfo(**q) for q in (pg_raw.get("long_running_queries") or [])
+                ],
+                database_size=pg_raw.get("database_size"),
+                table_sizes=[
+                    TableSizeInfo(**t) for t in (pg_raw.get("table_sizes") or [])
+                ],
+            )
+
+            supabase_metrics = None
+            sm = r.get("supabase_metrics")
+            if sm:
+                supabase_metrics = SupabaseDiagnosticMetrics(
+                    rate_limited_429_count=sm.get("rate_limited_429_count"),
+                )
+
+            report = DiagnosticReport(
+                database_type=r.get("database_type", 0),
+                pg_metrics=pg_metrics,
+                supabase_metrics=supabase_metrics,
+            )
+
+        return DiagnoseDatabaseResponse(report=report, code=code, msg=msg)

{coze_coding_dev_sdk-0.5.10 → coze_coding_dev_sdk-0.5.12}/coze_coding_dev_sdk/supabase/models.py

@@ -132,3 +132,83 @@ class UpdateAuthConfigResponse:
     success: bool = False
     code: int = 0
     msg: str = ""
+
+
+# Database Diagnostic
+
+
+@dataclass
+class QueryStatsInfo:
+    query: Optional[str] = None
+    calls: Optional[int] = None
+    avg_ms: Optional[float] = None
+    total_ms: Optional[float] = None
+    rows: Optional[int] = None
+
+
+@dataclass
+class ConnectionStateCount:
+    state: Optional[str] = None
+    count: Optional[int] = None
+
+
+@dataclass
+class ConnectionStats:
+    connections_by_state: Optional[List["ConnectionStateCount"]] = None
+    max_connections: Optional[int] = None
+    connection_usage_pct: Optional[float] = None
+    total_connections: Optional[int] = None
+
+
+@dataclass
+class BlockedQueryInfo:
+    blocked_pid: Optional[int] = None
+    blocked_query: Optional[str] = None
+    blocking_pid: Optional[int] = None
+    blocking_query: Optional[str] = None
+    blocked_duration: Optional[str] = None
+
+
+@dataclass
+class LongRunningQueryInfo:
+    pid: Optional[int] = None
+    duration: Optional[str] = None
+    state: Optional[str] = None
+    query: Optional[str] = None
+
+
+@dataclass
+class TableSizeInfo:
+    table_name: Optional[str] = None
+    total_size: Optional[str] = None
+
+
+@dataclass
+class SupabaseDiagnosticMetrics:
+    rate_limited_429_count: Optional[int] = None
+
+
+@dataclass
+class PGDiagnosticMetrics:
+    top_queries_by_total_time: Optional[List[QueryStatsInfo]] = None
+    top_queries_by_frequency: Optional[List[QueryStatsInfo]] = None
+    slow_queries: Optional[List[QueryStatsInfo]] = None
+    connection_stats: Optional[ConnectionStats] = None
+    blocked_queries: Optional[List[BlockedQueryInfo]] = None
+    long_running_queries: Optional[List[LongRunningQueryInfo]] = None
+    database_size: Optional[str] = None
+    table_sizes: Optional[List[TableSizeInfo]] = None
+
+
+@dataclass
+class DiagnosticReport:
+    database_type: int = 0
+    pg_metrics: Optional[PGDiagnosticMetrics] = None
+    supabase_metrics: Optional[SupabaseDiagnosticMetrics] = None
+
+
+@dataclass
+class DiagnoseDatabaseResponse:
+    report: Optional[DiagnosticReport] = None
+    code: int = 0
+    msg: str = ""

{coze_coding_dev_sdk-0.5.10 → coze_coding_dev_sdk-0.5.12/coze_coding_dev_sdk.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: coze-coding-dev-sdk
-Version: 0.5.10
+Version: 0.5.12
 Summary: Coze Coding Dev SDK - 优雅的多功能 AI SDK，支持图片生成、视频生成、语音合成、语音识别、大语言模型、联网搜索和文本/多模态 Embedding。包含命令行工具 coze-coding-ai，支持 Context 上下文追踪
 Author-email: Coze Coding Integration Team <support@coze.com>
 Maintainer-email: Coze Coding Integration Team <support@coze.com>
@@ -44,6 +44,10 @@ Requires-Dist: click>=8.0.0
 Requires-Dist: rich>=13.0.0
 Requires-Dist: supabase==2.27.3
 Requires-Dist: httpx==0.28.1
+Requires-Dist: markdown>=3.0.0
+Requires-Dist: python-docx>=1.0.0
+Requires-Dist: reportlab>=4.0.0
+Requires-Dist: cssutils>=2.0.0
 Provides-Extra: dev
 Requires-Dist: pytest>=7.0.0; extra == "dev"
 Requires-Dist: pytest-asyncio>=0.21.0; extra == "dev"

{coze_coding_dev_sdk-0.5.10 → coze_coding_dev_sdk-0.5.12}/coze_coding_dev_sdk.egg-info/requires.txt

@@ -17,6 +17,10 @@ click>=8.0.0
 rich>=13.0.0
 supabase==2.27.3
 httpx==0.28.1
+markdown>=3.0.0
+python-docx>=1.0.0
+reportlab>=4.0.0
+cssutils>=2.0.0
 
 [dev]
 pytest>=7.0.0

{coze_coding_dev_sdk-0.5.10 → coze_coding_dev_sdk-0.5.12}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "coze-coding-dev-sdk"
-version = "0.5.10"
+version = "0.5.12"
 description = "Coze Coding Dev SDK - 优雅的多功能 AI SDK，支持图片生成、视频生成、语音合成、语音识别、大语言模型、联网搜索和文本/多模态 Embedding。包含命令行工具 coze-coding-ai，支持 Context 上下文追踪"
 readme = "README.md"
 requires-python = ">=3.11"
@@ -72,6 +72,11 @@ dependencies = [
     # Supabase
     "supabase==2.27.3",
     "httpx==0.28.1",
+    # Document generation
+    "markdown>=3.0.0",
+    "python-docx>=1.0.0",
+    "reportlab>=4.0.0",
+    "cssutils>=2.0.0",
 ]
 
 [project.urls]

coze_coding_dev_sdk-0.5.10/coze_coding_dev_sdk/cli/db.py

@@ -1,81 +0,0 @@
-"""
-数据库 CLI 命令
-"""
-
-import os
-import click
-
-# 默认路径
-DEFAULT_MODEL_OUTPUT = "src/storage/database/shared/model.py"
-DEFAULT_MODEL_IMPORT_PATH = "storage.database.shared.model"
-DEFAULT_MODEL_PATH = "src"
-
-
-def _get_workspace_path() -> str:
-    """获取工作目录"""
-    return os.getenv("WORKSPACE_PATH", os.getcwd())
-
-
-@click.group()
-def db():
-    """Database management commands."""
-    pass
-
-
-@db.command()
-@click.argument("output_path", required=False, default=None)
-@click.option("--verbose", "-v", is_flag=True, help="Verbose output")
-def generate_models(output_path: str, verbose: bool):
-    """
-    Generate ORM models from database.
-
-    OUTPUT_PATH: Path to output model file (default: src/storage/database/shared/model.py)
-    """
-    from coze_coding_dev_sdk.database import generate_models as _generate_models
-
-    if output_path is None:
-        workspace = _get_workspace_path()
-        output_path = os.path.join(workspace, DEFAULT_MODEL_OUTPUT)
-
-    try:
-        _generate_models(output_path, verbose=verbose)
-        click.echo(f"Models generated at {output_path}")
-    except Exception as e:
-        click.echo(f"Error: {e}", err=True)
-        raise SystemExit(1)
-
-
-@db.command()
-@click.option(
-    "--model-import-path",
-    default=DEFAULT_MODEL_IMPORT_PATH,
-    help=f"Model import path (default: {DEFAULT_MODEL_IMPORT_PATH})",
-)
-@click.option(
-    "--model-path",
-    default=None,
-    help=f"Path to add to sys.path for model import (default: $WORKSPACE_PATH/{DEFAULT_MODEL_PATH})",
-)
-@click.option("--verbose", "-v", is_flag=True, help="Verbose output")
-def upgrade(model_import_path: str, model_path: str, verbose: bool):
-    """
-    Run database migrations.
-
-    Automatically generates migration and upgrades to head.
-    """
-    from coze_coding_dev_sdk.database import upgrade as _upgrade
-
-    if model_path is None:
-        workspace = _get_workspace_path()
-        model_path = os.path.join(workspace, DEFAULT_MODEL_PATH)
-
-    try:
-        _upgrade(
-            model_import_path=model_import_path,
-            model_path=model_path,
-            verbose=verbose,
-        )
-        click.echo("Database upgraded successfully")
-    except Exception as e:
-        click.echo(f"Error: {e}", err=True)
-        raise SystemExit(1)