coze-coding-dev-sdk 0.4.1__tar.gz → 0.4.2__tar.gz

{coze_coding_dev_sdk-0.4.1/coze_coding_dev_sdk.egg-info → coze_coding_dev_sdk-0.4.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: coze-coding-dev-sdk
-Version: 0.4.1
+Version: 0.4.2
 Summary: Coze Coding Dev SDK - 优雅的多功能 AI SDK，支持图片生成、视频生成、语音合成、语音识别、大语言模型和联网搜索。包含命令行工具 coze-coding-ai，支持 Context 上下文追踪
 Author-email: Coze Coding Integration Team <support@coze.com>
 Maintainer-email: Coze Coding Integration Team <support@coze.com>
@@ -513,6 +513,10 @@ coze-coding-ai video \
 
 # 查询任务状态
 coze-coding-ai video-status <task-id>
+
+# Mock 模式（测试运行，不消耗配额）
+coze-coding-ai video -p "Test video" --mock --poll
+coze-coding-ai video-status <task-id> --mock
 ```
 
 ### Chat 对话

{coze_coding_dev_sdk-0.4.1 → coze_coding_dev_sdk-0.4.2}/README.md

@@ -470,6 +470,10 @@ coze-coding-ai video \
 
 # 查询任务状态
 coze-coding-ai video-status <task-id>
+
+# Mock 模式（测试运行，不消耗配额）
+coze-coding-ai video -p "Test video" --mock --poll
+coze-coding-ai video-status <task-id> --mock
 ```
 
 ### Chat 对话

coze_coding_dev_sdk-0.4.2/coze_coding_dev_sdk/cli/constants.py

@@ -0,0 +1,7 @@
+"""
+CLI 常量定义
+"""
+
+RUN_MODE_HEADER = "x-run-mode"
+RUN_MODE_TEST = "test_run"
+RUN_MODE_PRODUCTION = "production"

{coze_coding_dev_sdk-0.4.1 → coze_coding_dev_sdk-0.4.2}/coze_coding_dev_sdk/cli/image.py

@@ -1,11 +1,15 @@
-import os
 import base64
+import os
+import time
 from typing import Optional
+
 import click
+from coze_coding_utils.runtime_ctx.context import Context
 from rich.console import Console
 
-from ..image import ImageGenerationClient
 from ..core.config import Config
+from ..image import ImageGenerationClient
+from .constants import RUN_MODE_HEADER, RUN_MODE_TEST
 
 console = Console()
 
@@ -13,41 +17,70 @@ console = Console()
 def validate_and_normalize_size(size: str) -> str:
     if size in ["2K", "4K"]:
         return size
-    
+
     try:
-        width, height = size.split('x')
+        width, height = size.split("x")
         w, h = int(width), int(height)
-        
+
         if 2560 <= w <= 4096 and 1440 <= h <= 4096:
             return size
         else:
-            console.print(f"[yellow]Warning: Size {size} is out of range [2560x1440, 4096x4096], using default 2K[/yellow]")
+            console.print(
+                f"[yellow]Warning: Size {size} is out of range [2560x1440, 4096x4096], using default 2K[/yellow]"
+            )
             return "2K"
     except (ValueError, AttributeError):
-        console.print(f"[yellow]Warning: Invalid size format {size}, using default 2K[/yellow]")
+        console.print(
+            f"[yellow]Warning: Invalid size format {size}, using default 2K[/yellow]"
+        )
         return "2K"
 
 
 @click.command()
 @click.option("--prompt", "-p", required=True, help="Text description of the image")
-@click.option("--output", "-o", required=True, type=click.Path(), help="Output file path")
-@click.option("--size", "-s", default="2K", help="Image size (2K, 4K, or WIDTHxHEIGHT in range [2560x1440, 4096x4096])")
-@click.option("--image", "-i", help="Reference image URL or path (for image-to-image generation)")
-@click.option("--images", multiple=True, help="Multiple reference image URLs or paths (can be used multiple times)")
-def image(prompt: str, output: str, size: str, image: Optional[str], images: tuple):
+@click.option(
+    "--output", "-o", required=True, type=click.Path(), help="Output file path"
+)
+@click.option(
+    "--size",
+    "-s",
+    default="2K",
+    help="Image size (2K, 4K, or WIDTHxHEIGHT in range [2560x1440, 4096x4096])",
+)
+@click.option(
+    "--image", "-i", help="Reference image URL or path (for image-to-image generation)"
+)
+@click.option(
+    "--images",
+    multiple=True,
+    help="Multiple reference image URLs or paths (can be used multiple times)",
+)
+@click.option("--mock", is_flag=True, help="使用 mock 模式（测试运行）")
+def image(
+    prompt: str, output: str, size: str, image: Optional[str], images: tuple, mock: bool
+):
     """Generate image using AI."""
     try:
         config = Config()
-        client = ImageGenerationClient(config)
-        
+
+        ctx = None
+        if mock:
+            ctx = Context(
+                request_id=f"mock-req-{int(time.time())}",
+                headers={RUN_MODE_HEADER: RUN_MODE_TEST},
+            )
+            console.print("[yellow]🧪 Mock 模式已启用（测试运行）[/yellow]")
+
+        client = ImageGenerationClient(config, ctx=ctx)
+
         reference_images = None
         if image:
             reference_images = image
         elif images:
             reference_images = list(images)
-        
+
         validated_size = validate_and_normalize_size(size)
-        
+
         console.print(f"[bold cyan]Generating image...[/bold cyan]")
         console.print(f"Prompt: [yellow]{prompt}[/yellow]")
         console.print(f"Size: [yellow]{validated_size}[/yellow]")
@@ -55,48 +88,51 @@ def image(prompt: str, output: str, size: str, image: Optional[str], images: tup
             if isinstance(reference_images, str):
                 console.print(f"Reference image: [blue]{reference_images}[/blue]")
             else:
-                console.print(f"Reference images: [blue]{len(reference_images)} images[/blue]")
-        
+                console.print(
+                    f"Reference images: [blue]{len(reference_images)} images[/blue]"
+                )
+
         response = client.generate(
             prompt=prompt,
             size=validated_size,
             image=reference_images,
-            response_format="b64_json"
+            response_format="b64_json",
         )
-        
+
         if not response.data or len(response.data) == 0:
             raise ValueError("No image data returned")
-        
+
         image_data = response.data[0]
-        
+
         if image_data.error:
-            error_msg = image_data.error.get('message', 'Unknown error')
+            error_msg = image_data.error.get("message", "Unknown error")
             raise Exception(f"Image generation failed: {error_msg}")
-        
+
         if image_data.b64_json:
             image_bytes = base64.b64decode(image_data.b64_json)
-            
+
             os.makedirs(os.path.dirname(os.path.abspath(output)), exist_ok=True)
-            
-            with open(output, 'wb') as f:
+
+            with open(output, "wb") as f:
                 f.write(image_bytes)
-            
+
             console.print(f"[green]✓[/green] Image saved to: [bold]{output}[/bold]")
         elif image_data.url:
             import requests
+
             img_response = requests.get(image_data.url)
             img_response.raise_for_status()
-            
+
             os.makedirs(os.path.dirname(os.path.abspath(output)), exist_ok=True)
-            
-            with open(output, 'wb') as f:
+
+            with open(output, "wb") as f:
                 f.write(img_response.content)
-            
+
             console.print(f"[green]✓[/green] Image saved to: [bold]{output}[/bold]")
             console.print(f"Original URL: [blue]{image_data.url}[/blue]")
         else:
             raise ValueError("No image data (b64_json or url) in response")
-    
+
     except Exception as e:
         console.print(f"[red]✗ Error: {str(e)}[/red]")
         raise click.Abort()

{coze_coding_dev_sdk-0.4.1 → coze_coding_dev_sdk-0.4.2}/coze_coding_dev_sdk/cli/video.py

@@ -1,15 +1,18 @@
-import time
 import json
 import os
+import time
 from typing import Optional
+
 import click
+from coze_coding_utils.runtime_ctx.context import Context
 from rich.console import Console
 from rich.progress import Progress, SpinnerColumn, TextColumn
 from rich.table import Table
 
-from ..video import VideoGenerationClient, VideoConfig
 from ..core.config import Config
 from ..core.exceptions import APIError
+from ..video import VideoConfig, VideoGenerationClient
+from .constants import RUN_MODE_HEADER, RUN_MODE_TEST
 
 console = Console()
 
@@ -17,10 +20,10 @@ console = Console()
 def parse_resolution(size: Optional[str]) -> tuple[Optional[str], Optional[str]]:
     if not size:
         return None, None
-    
-    width, height = size.split('x')
+
+    width, height = size.split("x")
     w, h = int(width), int(height)
-    
+
     if w == h:
         return "1:1", "720p" if w <= 1024 else "1080p"
     elif w > h:
@@ -32,16 +35,25 @@ def parse_resolution(size: Optional[str]) -> tuple[Optional[str], Optional[str]]
 @click.command()
 @click.option("--prompt", "-p", help="Text description of the video")
 @click.option("--image-url", "-i", help="Image URL (single or comma-separated pair)")
-@click.option("--quality", "-q", type=click.Choice(["speed", "quality"]), default="speed", help="Output mode")
+@click.option(
+    "--quality",
+    "-q",
+    type=click.Choice(["speed", "quality"]),
+    default="speed",
+    help="Output mode",
+)
 @click.option("--with-audio", is_flag=True, help="Generate AI audio effects")
 @click.option("--size", "-s", help="Video resolution (e.g., 1920x1080)")
 @click.option("--fps", type=int, help="Frame rate (30 or 60)")
 @click.option("--duration", "-d", type=int, help="Duration in seconds (5 or 10)")
 @click.option("--model", "-m", help="Model name to use")
 @click.option("--poll", is_flag=True, help="Auto-poll until task completes")
-@click.option("--poll-interval", type=int, default=5, help="Polling interval in seconds")
+@click.option(
+    "--poll-interval", type=int, default=5, help="Polling interval in seconds"
+)
 @click.option("--max-polls", type=int, default=60, help="Maximum poll attempts")
 @click.option("--output", "-o", type=click.Path(), help="Output file path (JSON)")
+@click.option("--mock", is_flag=True, help="使用 mock 模式（测试运行）")
 def video(
     prompt: Optional[str],
     image_url: Optional[str],
@@ -54,38 +66,48 @@ def video(
     poll: bool,
     poll_interval: int,
     max_polls: int,
-    output: Optional[str]
+    output: Optional[str],
+    mock: bool,
 ):
     """Generate video using AI."""
     try:
         config = Config()
-        client = VideoGenerationClient(config)
-        
+
+        ctx = None
+        if mock:
+            ctx = Context(
+                request_id=f"mock-req-{int(time.time())}",
+                headers={RUN_MODE_HEADER: RUN_MODE_TEST},
+            )
+            console.print("[yellow]🧪 Mock 模式已启用（测试运行）[/yellow]")
+
+        client = VideoGenerationClient(config, ctx=ctx)
+
         image_urls = None
         if image_url:
             image_urls = [url.strip() for url in image_url.split(",")]
-        
+
         ratio, resolution = parse_resolution(size)
-        
+
         video_config = VideoConfig(
             resolution=resolution or "720p",
             ratio=ratio or "16:9",
-            duration=duration or 5
+            duration=duration or 5,
         )
-        
+
         console.print("[bold cyan]Creating video generation task...[/bold cyan]")
-        
+
         model_name = model or "doubao-seedance-1-0-pro-250528"
-        
+
         if image_urls:
             if len(image_urls) == 1:
                 task_id = client._create_task(
                     model=model_name,
                     content=[
                         {"type": "text", "text": prompt or ""},
-                        {"type": "image_url", "image_url": {"url": image_urls[0]}}
+                        {"type": "image_url", "image_url": {"url": image_urls[0]}},
                     ],
-                    config=video_config
+                    config=video_config,
                 )
             elif len(image_urls) == 2:
                 task_id = client._create_task(
@@ -93,78 +115,85 @@ def video(
                     content=[
                         {"type": "text", "text": prompt or ""},
                         {"type": "image_url", "image_url": {"url": image_urls[0]}},
-                        {"type": "image_url", "image_url": {"url": image_urls[1]}}
+                        {"type": "image_url", "image_url": {"url": image_urls[1]}},
                     ],
-                    config=video_config
+                    config=video_config,
                 )
             else:
                 raise ValueError("Only 1 or 2 images are supported")
         else:
             if not prompt:
                 raise ValueError("Either --prompt or --image-url must be provided")
-            
+
             task_id = client._create_task(
                 model=model_name,
                 content=[{"type": "text", "text": prompt}],
-                config=video_config
+                config=video_config,
             )
-        
+
         console.print(f"[green]✓[/green] Task created: [bold]{task_id}[/bold]")
-        
+
         result = {"id": task_id, "status": "processing"}
-        
+
         if poll:
             with Progress(
                 SpinnerColumn(),
                 TextColumn("[progress.description]{task.description}"),
-                console=console
+                console=console,
             ) as progress:
                 task = progress.add_task(
-                    f"[cyan]Generating video (Task ID: {task_id})...",
-                    total=None
+                    f"[cyan]Generating video (Task ID: {task_id})...", total=None
                 )
-                
+
                 max_wait_time = poll_interval * max_polls
                 start_time = time.time()
-                
+
                 while time.time() - start_time < max_wait_time:
                     task_result = client._get_task_status(task_id)
-                    
+
                     if task_result.status == "completed":
-                        progress.update(task, description="[green]✓ Video generation completed!")
+                        progress.update(
+                            task, description="[green]✓ Video generation completed!"
+                        )
                         result = task_result.model_dump()
                         break
                     elif task_result.status == "failed":
-                        progress.update(task, description=f"[red]✗ Video generation failed")
+                        progress.update(
+                            task, description=f"[red]✗ Video generation failed"
+                        )
                         result = task_result.model_dump()
                         break
-                    
+
                     time.sleep(poll_interval)
                 else:
-                    raise TimeoutError(f"Task did not complete within {max_wait_time} seconds")
-            
+                    raise TimeoutError(
+                        f"Task did not complete within {max_wait_time} seconds"
+                    )
+
             table = Table(title="Video Generation Result")
             table.add_column("Field", style="cyan")
             table.add_column("Value", style="green")
-            
+
             table.add_row("Task ID", result.get("id", ""))
             table.add_row("Status", result.get("status", ""))
-            
+
             if result.get("video_url"):
                 table.add_row("Video URL", result.get("video_url"))
             if result.get("error_message"):
                 table.add_row("Error", result.get("error_message"))
-            
+
             console.print(table)
         else:
-            console.print(f"\n[yellow]Use the following command to check status:[/yellow]")
+            console.print(
+                f"\n[yellow]Use the following command to check status:[/yellow]"
+            )
             console.print(f"coze-coding-ai video-status {task_id}")
-        
+
         if output:
             with open(output, "w") as f:
                 json.dump(result, f, indent=2)
             console.print(f"\n[green]✓[/green] Result saved to: {output}")
-    
+
     except Exception as e:
         console.print(f"[red]✗ Error: {str(e)}[/red]")
         raise click.Abort()
@@ -173,34 +202,44 @@ def video(
 @click.command()
 @click.argument("task_id")
 @click.option("--output", "-o", type=click.Path(), help="Output file path (JSON)")
-def video_status(task_id: str, output: Optional[str]):
+@click.option("--mock", is_flag=True, help="使用 mock 模式（测试运行）")
+def video_status(task_id: str, output: Optional[str], mock: bool):
     """Check video generation task status."""
     try:
         config = Config()
-        client = VideoGenerationClient(config)
-        
+
+        ctx = None
+        if mock:
+            ctx = Context(
+                request_id=f"mock-req-{int(time.time())}",
+                headers={RUN_MODE_HEADER: RUN_MODE_TEST},
+            )
+            console.print("[yellow]🧪 Mock 模式已启用（测试运行）[/yellow]")
+
+        client = VideoGenerationClient(config, ctx=ctx)
+
         task_result = client._get_task_status(task_id)
         result = task_result.model_dump()
-        
+
         table = Table(title=f"Task Status: {task_id}")
         table.add_column("Field", style="cyan")
         table.add_column("Value", style="green")
-        
+
         table.add_row("Task ID", result.get("id", ""))
         table.add_row("Status", result.get("status", ""))
-        
+
         if result.get("video_url"):
             table.add_row("Video URL", result.get("video_url"))
         if result.get("error_message"):
             table.add_row("Error", result.get("error_message"))
-        
+
         console.print(table)
-        
+
         if output:
             with open(output, "w") as f:
                 json.dump(result, f, indent=2)
             console.print(f"\n[green]✓[/green] Result saved to: {output}")
-    
+
     except Exception as e:
         console.print(f"[red]✗ Error: {str(e)}[/red]")
         raise click.Abort()