coze-coding-dev-sdk 0.5.11__tar.gz → 0.5.13__tar.gz

{coze_coding_dev_sdk-0.5.11/coze_coding_dev_sdk.egg-info → coze_coding_dev_sdk-0.5.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: coze-coding-dev-sdk
-Version: 0.5.11
+Version: 0.5.13
 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>

{coze_coding_dev_sdk-0.5.11 → coze_coding_dev_sdk-0.5.13}/coze_coding_dev_sdk/cli/cli.py

@@ -3,6 +3,7 @@ import click
 from .chat import chat
 from .db import db
 from .embedding import embedding
+from .storage import storage
 from .image import image
 from .knowledge import knowledge
 from .search import search
@@ -32,6 +33,7 @@ main.add_command(chat)
 main.add_command(db)
 main.add_command(embedding)
 main.add_command(document)
+main.add_command(storage)
 
 
 if __name__ == "__main__":

coze_coding_dev_sdk-0.5.13/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.13/coze_coding_dev_sdk/cli/storage.py

@@ -0,0 +1,241 @@
+"""
+对象存储 CLI 命令
+"""
+
+import json
+import mimetypes
+import sys
+import tempfile
+import zipfile
+from concurrent.futures import ThreadPoolExecutor, as_completed
+from pathlib import Path, PurePosixPath
+from typing import Any, Dict, List, Optional, Tuple
+from uuid import uuid4
+
+import click
+
+
+def _normalize_prefix(prefix: str) -> str:
+    if not prefix:
+        return ""
+    prefix = prefix.strip("/")
+    return prefix + "/" if prefix else ""
+
+
+def _build_object_key(prefix: str, name: str, use_uuid: bool, sanitize_fn) -> str:
+    sanitized = sanitize_fn(name)
+    if use_uuid:
+        p = Path(sanitized)
+        uniq = uuid4().hex[:8]
+        if str(p.parent) != ".":
+            sanitized = str(PurePosixPath(p.parent / f"{p.stem}_{uniq}{p.suffix}"))
+        else:
+            sanitized = f"{p.stem}_{uniq}{p.suffix}"
+    return prefix + sanitized
+
+
+def _resolve_paths(paths: Tuple[str, ...]) -> List[Tuple[Path, str]]:
+    """解析目录为 (绝对路径, 相对key) 列表，保留目录名"""
+    results: List[Tuple[Path, str]] = []
+    seen: set = set()
+    for path_str in paths:
+        p = Path(path_str).resolve()
+        if not p.exists() or not p.is_dir():
+            continue
+        for f in sorted(p.rglob("*")):
+            if f.is_file() and f not in seen:
+                seen.add(f)
+                rel = f.relative_to(p.parent)
+                results.append((f, str(PurePosixPath(rel))))
+    return results
+
+
+def _zip_directory(dir_path: Path) -> Path:
+    """将目录打包为临时 zip 文件"""
+    zip_name = f"{dir_path.name}.zip"
+    tmp_dir = tempfile.mkdtemp()
+    zip_path = Path(tmp_dir) / zip_name
+    with zipfile.ZipFile(zip_path, "w", zipfile.ZIP_DEFLATED) as zf:
+        for f in sorted(dir_path.rglob("*")):
+            if f.is_file():
+                arcname = str(f.relative_to(dir_path.parent))
+                zf.write(f, arcname)
+    return zip_path
+
+
+def _upload_file_to_s3(s3_client, bucket: str, file_path: Path, object_key: str, threshold: int) -> int:
+    """上传单个文件到 S3，返回字节数"""
+    content_type = mimetypes.guess_type(str(file_path))[0] or "application/octet-stream"
+    file_size = file_path.stat().st_size
+
+    if file_size <= threshold:
+        s3_client.put_object(
+            Bucket=bucket,
+            Key=object_key,
+            Body=file_path.read_bytes(),
+            ContentType=content_type,
+        )
+    else:
+        from boto3.s3.transfer import TransferConfig
+        config = TransferConfig(
+            multipart_threshold=threshold,
+            multipart_chunksize=threshold,
+            max_concurrency=1,
+            use_threads=False,
+        )
+        with open(file_path, "rb") as f:
+            s3_client.upload_fileobj(
+                Fileobj=f,
+                Bucket=bucket,
+                Key=object_key,
+                ExtraArgs={"ContentType": content_type},
+                Config=config,
+            )
+    return file_size
+
+
+@click.group()
+def storage():
+    """Object storage commands."""
+    pass
+
+
+@storage.command()
+@click.argument("paths", nargs=-1, required=True, type=click.Path())
+@click.option("--prefix", "-p", default="", help="Target S3 key prefix (virtual directory)")
+@click.option("--uuid", "use_uuid", is_flag=True, help="Append UUID suffix to filenames to avoid collisions")
+@click.option("--expire", "-e", default=86400, type=int, help="Presigned URL expiration in seconds (default: 86400 = 24h)")
+@click.option("--threshold", "-t", default=5 * 1024 * 1024, type=int, help="Size threshold (bytes) for streaming upload (default: 5MB)")
+@click.option("--max-concurrency", "-c", default=4, type=int, help="Number of parallel upload threads (default: 4)")
+def upload(paths, prefix, use_uuid, expire, threshold, max_concurrency):
+    """Upload files and directories to object storage.
+
+    PATHS: One or more file or directory paths to upload.
+    Files are uploaded directly. Directories are uploaded recursively
+    AND also zipped as a single archive.
+    Outputs JSON with name-to-URL mappings to stdout.
+
+    \b
+    Examples:
+      coze-coding-ai storage upload ./data --prefix my-project/
+      coze-coding-ai storage upload file1.txt dir1/ dir2/ --prefix uploads/
+      coze-coding-ai storage upload report.csv --prefix reports/ --expire 3600
+    """
+    from coze_coding_dev_sdk.s3 import S3SyncStorage
+
+    # 1. 分离文件和目录
+    files: List[Path] = []
+    dirs: List[Path] = []
+
+    for path_str in paths:
+        p = Path(path_str).resolve()
+        if not p.exists():
+            print(f"Warning: {path_str} does not exist, skipping", file=sys.stderr)
+            continue
+        if p.is_file():
+            files.append(p)
+        elif p.is_dir():
+            dirs.append(p)
+        else:
+            print(f"Warning: {path_str} is not a file or directory, skipping", file=sys.stderr)
+
+    if not files and not dirs:
+        print(json.dumps({"_error": "No files or directories found to upload."}))
+        raise SystemExit(1)
+
+    # 2. 规范化 prefix
+    prefix = _normalize_prefix(prefix)
+
+    # 3. 初始化 S3
+    try:
+        storage_client = S3SyncStorage()
+        target_bucket = storage_client._resolve_bucket(None)
+    except Exception as e:
+        print(json.dumps({"_error": f"S3 configuration error: {e}"}))
+        raise SystemExit(1)
+
+    sanitize_fn = storage_client._sanitize_file_name
+
+    # 4. 构建上传任务
+    single_file_tasks: List[Tuple[Path, str, str]] = []
+    for f in files:
+        key = _build_object_key(prefix, f.name, use_uuid, sanitize_fn)
+        single_file_tasks.append((f, key, f.name))
+
+    dir_file_tasks: List[Tuple[Path, str, str]] = []
+    for d in dirs:
+        dir_files = _resolve_paths((str(d),))
+        for file_path, relative_key in dir_files:
+            key = _build_object_key(prefix, relative_key, use_uuid, sanitize_fn)
+            dir_file_tasks.append((file_path, key, relative_key))
+
+    zip_tasks: List[Tuple[Path, str, str]] = []
+    for d in dirs:
+        zip_name = f"{d.name}.zip"
+        key = _build_object_key(prefix, zip_name, use_uuid, sanitize_fn)
+        zip_tasks.append((d, key, f"{d.name}/"))
+
+    all_file_tasks = single_file_tasks + dir_file_tasks
+
+    # 5. 获取 boto3 client
+    try:
+        s3_client = storage_client._get_client()
+    except Exception as e:
+        print(json.dumps({"_error": f"Failed to initialize S3 client: {e}"}))
+        raise SystemExit(1)
+
+    result_map: Dict[str, str] = {}
+    failures: List[Tuple[str, str]] = []
+    single_file_names = {name for _, _, name in single_file_tasks}
+
+    # 6. 并发上传文件
+    def _do_upload(fp: Path, key: str, name: str):
+        try:
+            _upload_file_to_s3(s3_client, target_bucket, fp, key, threshold)
+            if name in single_file_names:
+                url = storage_client.generate_presigned_url(key=key, expire_time=expire)
+                return (name, url, None)
+            return (name, None, None)
+        except Exception as e:
+            return (name, None, str(e))
+
+    if all_file_tasks:
+        with ThreadPoolExecutor(max_workers=max_concurrency) as executor:
+            futures = {
+                executor.submit(_do_upload, fp, key, name): name
+                for fp, key, name in all_file_tasks
+            }
+            for future in as_completed(futures):
+                name, url, error = future.result()
+                if error:
+                    failures.append((name, error))
+                elif url:
+                    result_map[name] = url
+
+    # 7. 打包并上传 zip
+    for dir_path, object_key, display_name in zip_tasks:
+        tmp_zip_path: Optional[Path] = None
+        try:
+            tmp_zip_path = _zip_directory(dir_path)
+            _upload_file_to_s3(s3_client, target_bucket, tmp_zip_path, object_key, threshold)
+            url = storage_client.generate_presigned_url(key=object_key, expire_time=expire)
+            result_map[display_name] = url
+        except Exception as e:
+            failures.append((display_name, str(e)))
+        finally:
+            if tmp_zip_path and tmp_zip_path.exists():
+                try:
+                    tmp_zip_path.unlink()
+                    tmp_zip_path.parent.rmdir()
+                except OSError:
+                    pass
+
+    # 8. 输出 JSON
+    output: Dict[str, Any] = dict(result_map)
+    if failures:
+        output["_failures"] = [{"name": n, "error": e} for n, e in failures]
+
+    print(json.dumps(output, indent=2, ensure_ascii=False))
+
+    if failures:
+        raise SystemExit(1)

{coze_coding_dev_sdk-0.5.11 → coze_coding_dev_sdk-0.5.13}/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.11 → coze_coding_dev_sdk-0.5.13}/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.11 → coze_coding_dev_sdk-0.5.13}/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.11 → coze_coding_dev_sdk-0.5.13}/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.11 → coze_coding_dev_sdk-0.5.13/coze_coding_dev_sdk.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: coze-coding-dev-sdk
-Version: 0.5.11
+Version: 0.5.13
 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>

{coze_coding_dev_sdk-0.5.11 → coze_coding_dev_sdk-0.5.13}/coze_coding_dev_sdk.egg-info/SOURCES.txt

@@ -20,6 +20,7 @@ coze_coding_dev_sdk/cli/embedding.py
 coze_coding_dev_sdk/cli/image.py
 coze_coding_dev_sdk/cli/knowledge.py
 coze_coding_dev_sdk/cli/search.py
+coze_coding_dev_sdk/cli/storage.py
 coze_coding_dev_sdk/cli/supabase.py
 coze_coding_dev_sdk/cli/utils.py
 coze_coding_dev_sdk/cli/video.py

{coze_coding_dev_sdk-0.5.11 → coze_coding_dev_sdk-0.5.13}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "coze-coding-dev-sdk"
-version = "0.5.11"
+version = "0.5.13"
 description = "Coze Coding Dev SDK - 优雅的多功能 AI SDK，支持图片生成、视频生成、语音合成、语音识别、大语言模型、联网搜索和文本/多模态 Embedding。包含命令行工具 coze-coding-ai，支持 Context 上下文追踪"
 readme = "README.md"
 requires-python = ">=3.11"

coze_coding_dev_sdk-0.5.11/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)