coze-coding-dev-sdk 0.5.2__tar.gz → 0.5.4__tar.gz

{coze_coding_dev_sdk-0.5.2/coze_coding_dev_sdk.egg-info → coze_coding_dev_sdk-0.5.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: coze-coding-dev-sdk
-Version: 0.5.2
+Version: 0.5.4
 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>
@@ -39,9 +39,8 @@ Requires-Dist: psycopg-pool>=3.0.0
 Requires-Dist: langgraph>=0.2.0
 Requires-Dist: langgraph-checkpoint-postgres>=3.0.0
 Requires-Dist: boto3>=1.26.0
-Provides-Extra: cli
-Requires-Dist: click>=8.0.0; extra == "cli"
-Requires-Dist: rich>=13.0.0; extra == "cli"
+Requires-Dist: click>=8.0.0
+Requires-Dist: rich>=13.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.2 → coze_coding_dev_sdk-0.5.4}/coze_coding_dev_sdk/__init__.py

@@ -17,8 +17,16 @@ from .image import (
     UsageInfo,
 )
 from .llm import LLMClient, LLMConfig
+from .knowledge import (
+    ChunkConfig,
+    KnowledgeChunk,
+    KnowledgeClient,
+    KnowledgeDocument,
+    KnowledgeInsertResponse,
+    KnowledgeSearchResponse,
+)
 from .search import ImageItem, SearchClient, WebItem
-from .video import VideoConfig, VideoGenerationClient, VideoGenerationTask
+from .video import VideoGenerationClient, VideoGenerationTask
 from .voice import ASRClient, ASRRequest, ASRResponse, TTSClient, TTSConfig, TTSRequest
 
 # Database
@@ -55,11 +63,17 @@ __all__ = [
     "ASRResponse",
     "LLMClient",
     "LLMConfig",
+    "KnowledgeClient",
+    "ChunkConfig",
+    "DataSourceType",
+    "KnowledgeSearchResponse",
+    "KnowledgeChunk",
+    "KnowledgeDocument",
+    "KnowledgeInsertResponse",
     "SearchClient",
     "WebItem",
     "ImageItem",
     "VideoGenerationClient",
-    "VideoConfig",
     "VideoGenerationTask",
     # Database
     "Base",

{coze_coding_dev_sdk-0.5.2 → coze_coding_dev_sdk-0.5.4}/coze_coding_dev_sdk/cli/cli.py

@@ -3,8 +3,10 @@ import click
 from .chat import chat
 from .db import db
 from .image import image
+from .knowledge import knowledge
 from .search import search
 from .video import video
+from .video_edit import video_edit
 from .voice import asr, tts
 
 
@@ -16,7 +18,9 @@ def main():
 
 
 main.add_command(video)
+main.add_command(video_edit)
 main.add_command(image)
+main.add_command(knowledge)
 main.add_command(search)
 main.add_command(tts)
 main.add_command(asr)

coze_coding_dev_sdk-0.5.4/coze_coding_dev_sdk/cli/knowledge.py

@@ -0,0 +1,195 @@
+import json
+from typing import Optional
+
+import click
+from coze_coding_utils.runtime_ctx.context import new_context
+from rich.console import Console
+from rich.table import Table
+
+from ..core.config import Config
+from ..knowledge import (
+    ChunkConfig,
+    DataSourceType,
+    KnowledgeClient,
+    KnowledgeDocument,
+)
+from .constants import RUN_MODE_HEADER, RUN_MODE_TEST
+
+console = Console()
+
+
+@click.group()
+def knowledge():
+    """Knowledge Base tools."""
+    pass
+
+
+@knowledge.command()
+@click.option("--query", "-q", required=True, help="Search query")
+@click.option("--dataset", "-d", multiple=True, help="Dataset names (tables) to search in")
+@click.option("--top-k", "-k", default=5, help="Number of results to return")
+@click.option("--min-score", "-m", default=0.0, help="Minimum similarity score")
+@click.option("--mock", is_flag=True, help="Use mock mode")
+@click.option(
+    "--header",
+    "-H",
+    multiple=True,
+    help="Custom HTTP headers (format: 'Key: Value')",
+)
+@click.option("--verbose", "-v", is_flag=True, help="Show verbose logs")
+def search(
+    query: str,
+    dataset: tuple,
+    top_k: int,
+    min_score: float,
+    mock: bool,
+    header: tuple,
+    verbose: bool,
+):
+    """Search for knowledge chunks."""
+    try:
+        from .utils import parse_headers
+
+        config = Config()
+        ctx = None
+        custom_headers = parse_headers(header) or {}
+
+        if mock:
+            ctx = new_context(method="knowledge.search", headers=custom_headers)
+            custom_headers[RUN_MODE_HEADER] = RUN_MODE_TEST
+            console.print("[yellow]🧪 Mock mode enabled[/yellow]")
+
+        client = KnowledgeClient(
+            config, ctx=ctx, custom_headers=custom_headers, verbose=verbose
+        )
+
+        dataset_list = list(dataset) if dataset else None
+
+        response = client.search(
+            query=query,
+            table_names=dataset_list,
+            top_k=top_k,
+            min_score=min_score,
+        )
+
+        if response.code != 0:
+            console.print(f"[red]Error {response.code}: {response.msg}[/red]")
+            return
+
+        table = Table(title="Search Results")
+        table.add_column("Score", style="cyan", no_wrap=True)
+        table.add_column("Content", style="green")
+        table.add_column("Chunk ID", style="dim")
+        table.add_column("Doc ID", style="dim")
+
+        for chunk in response.chunks:
+            table.add_row(
+                f"{chunk.score:.4f}",
+                chunk.content,
+                chunk.chunk_id or "",
+                chunk.doc_id or "",
+            )
+
+        console.print(table)
+
+    except Exception as e:
+        console.print(f"[red]✗ Error: {str(e)}[/red]")
+        raise click.Abort()
+
+
+@knowledge.command()
+@click.option("--dataset", "-d", required=True, help="Dataset name (table) to add to")
+@click.option("--content", "-c", multiple=True, help="Raw text content to add")
+@click.option("--url", "-u", multiple=True, help="Web URL to add")
+@click.option("--chunk-separator", default="\n", help="Chunk separator")
+@click.option("--max-tokens", default=500, help="Max tokens per chunk")
+@click.option("--remove-extra-spaces", is_flag=True, help="Normalize extra spaces")
+@click.option("--remove-urls-emails", is_flag=True, help="Strip URLs and emails")
+@click.option("--mock", is_flag=True, help="Use mock mode")
+@click.option(
+    "--header",
+    "-H",
+    multiple=True,
+    help="Custom HTTP headers (format: 'Key: Value')",
+)
+@click.option("--verbose", "-v", is_flag=True, help="Show verbose logs")
+def add(
+    dataset: str,
+    content: tuple,
+    url: tuple,
+    chunk_separator: str,
+    max_tokens: int,
+    remove_extra_spaces: bool,
+    remove_urls_emails: bool,
+    mock: bool,
+    header: tuple,
+    verbose: bool,
+):
+    """Add documents to knowledge base."""
+    try:
+        from .utils import parse_headers
+
+        config = Config()
+        ctx = None
+        custom_headers = parse_headers(header) or {}
+
+        if mock:
+            ctx = new_context(method="knowledge.add", headers=custom_headers)
+            custom_headers[RUN_MODE_HEADER] = RUN_MODE_TEST
+            console.print("[yellow]🧪 Mock mode enabled[/yellow]")
+
+        client = KnowledgeClient(
+            config, ctx=ctx, custom_headers=custom_headers, verbose=verbose
+        )
+
+        documents = []
+        for text in content:
+            documents.append(
+                KnowledgeDocument(
+                    source=DataSourceType.TEXT,
+                    raw_data=text,
+                )
+            )
+        for u in url:
+            documents.append(
+                KnowledgeDocument(
+                    source=DataSourceType.URL,
+                    url=u,
+                )
+            )
+
+        if not documents:
+            console.print("[red]No content or URL provided to add.[/red]")
+            return
+
+        chunk_config = ChunkConfig(
+            separator=chunk_separator,
+            max_tokens=max_tokens,
+            remove_extra_spaces=remove_extra_spaces,
+            remove_urls_emails=remove_urls_emails,
+        )
+
+        response = client.add_documents(
+            documents=documents,
+            table_name=dataset,
+            chunk_config=chunk_config,
+        )
+
+        if response.code != 0:
+            console.print(f"[red]Error {response.code}: {response.msg}[/red]")
+            return
+
+        table = Table(title="Add Documents Result")
+        table.add_column("Field", style="cyan", no_wrap=True)
+        table.add_column("Value", style="green")
+
+        table.add_row("Code", str(response.code))
+        table.add_row("Message", response.msg)
+        if response.doc_ids:
+            table.add_row("Doc IDs", ", ".join(response.doc_ids))
+
+        console.print(table)
+
+    except Exception as e:
+        console.print(f"[red]✗ Error: {str(e)}[/red]")
+        raise click.Abort()

{coze_coding_dev_sdk-0.5.2 → coze_coding_dev_sdk-0.5.4}/coze_coding_dev_sdk/cli/video.py

@@ -13,7 +13,6 @@ from ..video import (
     ImageURL,
     ImageURLContent,
     TextContent,
-    VideoConfig,
     VideoGenerationClient,
 )
 from .constants import RUN_MODE_HEADER, RUN_MODE_TEST
@@ -96,15 +95,6 @@ def video(
 
         ratio, resolution = parse_resolution(size)
 
-        video_config = VideoConfig(
-            resolution=resolution or "720p",
-            ratio=ratio or "16:9",
-            duration=duration or 5,
-            watermark=watermark,
-            seed=seed,
-            camerafixed=camerafixed,
-        )
-
         content_items = []
 
         if prompt:
@@ -147,9 +137,14 @@ def video(
                 video_url, response, last_frame_url = client.video_generation(
                     content_items=content_items,
                     model=model_name,
-                    config=video_config,
                     callback_url=callback_url,
                     return_last_frame=return_last_frame,
+                    resolution=resolution or "720p",
+                    ratio=ratio or "16:9",
+                    duration=duration or 5,
+                    watermark=watermark,
+                    seed=seed,
+                    camerafixed=camerafixed,
                 )
 
                 progress.update(