feat(06-02): move processors and platforms to kb_workflow.pipeline
- Move all processor implementations (parse, download, asr, analyze, archive) - Move all platform adapters (base, bilibili, zhihu) - Update all internal imports to use kb_workflow.pipeline prefix Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,21 @@
|
||||
"""Platform Adapters for KB Pipeline
|
||||
|
||||
This module provides platform-specific adapters for content extraction
|
||||
from various platforms (Bilibili, Zhihu, etc.).
|
||||
|
||||
Each adapter implements the PlatformAdapter interface:
|
||||
- get_content_type: Identify content type from URL
|
||||
- fetch_metadata: Fetch content metadata
|
||||
- fetch_content: Fetch content body in Markdown format
|
||||
- get_output_filename: Generate output filename
|
||||
"""
|
||||
|
||||
from kb_workflow.pipeline.platforms.base import PlatformAdapter
|
||||
from kb_workflow.pipeline.platforms.bilibili import BilibiliAdapter
|
||||
from kb_workflow.pipeline.platforms.zhihu import ZhihuAdapter
|
||||
|
||||
__all__ = [
|
||||
"PlatformAdapter",
|
||||
"BilibiliAdapter",
|
||||
"ZhihuAdapter",
|
||||
]
|
||||
@@ -0,0 +1,122 @@
|
||||
"""Platform Adapter Base Class
|
||||
|
||||
Provides an abstract base class for platform-specific content adapters.
|
||||
Each platform (Bilibili, Zhihu, etc.) should implement a concrete adapter
|
||||
that handles metadata extraction and content fetching.
|
||||
|
||||
Purpose: Define a unified interface for multi-platform content extraction
|
||||
"""
|
||||
|
||||
from abc import ABC, abstractmethod
|
||||
from typing import Dict, Any
|
||||
|
||||
|
||||
class PlatformAdapter(ABC):
|
||||
"""Abstract base class for platform-specific content adapters
|
||||
|
||||
A platform adapter encapsulates the logic for:
|
||||
- Identifying content type from URL (video, article, answer, etc.)
|
||||
- Fetching metadata for a given content ID
|
||||
- Fetching content body in Markdown format
|
||||
- Generating output filenames
|
||||
|
||||
Subclasses must implement all abstract methods to provide
|
||||
platform-specific functionality.
|
||||
|
||||
Example:
|
||||
class BilibiliAdapter(PlatformAdapter):
|
||||
def get_content_type(self, url: str) -> str:
|
||||
if '/video/' in url:
|
||||
return 'video'
|
||||
elif '/read/' in url:
|
||||
return 'article'
|
||||
return 'unknown'
|
||||
|
||||
async def fetch_metadata(self, content_id: str, content_type: str) -> Dict[str, Any]:
|
||||
# Platform-specific implementation
|
||||
...
|
||||
"""
|
||||
|
||||
@abstractmethod
|
||||
def get_content_type(self, url: str) -> str:
|
||||
"""Identify content type from URL
|
||||
|
||||
Args:
|
||||
url: The content URL to analyze
|
||||
|
||||
Returns:
|
||||
Content type string (e.g., "video", "article", "answer", "zhuanlan")
|
||||
|
||||
Example:
|
||||
adapter.get_content_type("https://www.bilibili.com/video/BV1xx...")
|
||||
-> "video"
|
||||
"""
|
||||
pass
|
||||
|
||||
@abstractmethod
|
||||
async def fetch_metadata(self, content_id: str, content_type: str) -> Dict[str, Any]:
|
||||
"""Fetch metadata for the given content
|
||||
|
||||
Args:
|
||||
content_id: Platform-specific content identifier (e.g., BV号, CV号)
|
||||
content_type: Type of content ("video", "article", etc.)
|
||||
|
||||
Returns:
|
||||
Dictionary containing metadata fields. Structure varies by platform
|
||||
and content type, but typically includes:
|
||||
- title: Content title
|
||||
- author: Author name
|
||||
- url: Original content URL
|
||||
- Platform-specific fields
|
||||
|
||||
Raises:
|
||||
NetworkError: If network request fails
|
||||
AuthError: If authentication fails or quota exceeded
|
||||
"""
|
||||
pass
|
||||
|
||||
@abstractmethod
|
||||
async def fetch_content(self, content_id: str, content_type: str) -> str:
|
||||
"""Fetch content body in Markdown format
|
||||
|
||||
Args:
|
||||
content_id: Platform-specific content identifier
|
||||
content_type: Type of content
|
||||
|
||||
Returns:
|
||||
Markdown-formatted content string with YAML frontmatter:
|
||||
---
|
||||
title: Article Title
|
||||
author: Author Name
|
||||
source_url: https://...
|
||||
platform: bilibili
|
||||
content_type: article
|
||||
fetched_at: 2024-01-15T10:30:00
|
||||
---
|
||||
|
||||
# Article Content...
|
||||
|
||||
Note:
|
||||
For video content, this may return an empty string or
|
||||
placeholder text, as video content extraction is handled
|
||||
separately (e.g., ASR transcription).
|
||||
"""
|
||||
pass
|
||||
|
||||
@abstractmethod
|
||||
def get_output_filename(self, content_id: str, content_type: str, suffix: str) -> str:
|
||||
"""Generate output filename for content
|
||||
|
||||
Args:
|
||||
content_id: Platform-specific content identifier
|
||||
content_type: Type of content
|
||||
suffix: Filename suffix (e.g., "metadata.json", "content.md")
|
||||
|
||||
Returns:
|
||||
Filename string without directory path
|
||||
|
||||
Example:
|
||||
adapter.get_output_filename("BV1xx...", "video", "metadata.json")
|
||||
-> "bilibili_BV1xx..._metadata.json"
|
||||
"""
|
||||
pass
|
||||
@@ -0,0 +1,485 @@
|
||||
"""Bilibili Platform Adapter
|
||||
|
||||
Implements the PlatformAdapter interface for Bilibili platform content.
|
||||
Supports video and article metadata extraction and content fetching.
|
||||
|
||||
Key features:
|
||||
- Video metadata extraction (BV号)
|
||||
- Article metadata extraction (CV号)
|
||||
- Video download using yt-dlp (for ASR/Whisper)
|
||||
- Automatic retry on network errors (3 attempts with exponential backoff)
|
||||
- State recording for failed steps
|
||||
- Fallback from API to Playwright for 403 errors (handled by BilibiliExtractor)
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import asyncio
|
||||
import logging
|
||||
import os
|
||||
import re
|
||||
import tempfile
|
||||
from pathlib import Path
|
||||
from typing import Dict, Any, Optional
|
||||
|
||||
from kb_workflow.pipeline.platforms.base import PlatformAdapter
|
||||
from kb_workflow.pipeline.errors import NetworkError, AuthError
|
||||
from kb_workflow.pipeline.retry import retry_on_network_error
|
||||
from kb_workflow.pipeline.state import WorkflowStateManager
|
||||
from kb_workflow.metadata.bilibili_extractor import BilibiliExtractor
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class BilibiliAdapter(PlatformAdapter):
|
||||
"""Bilibili platform adapter for video and article content
|
||||
|
||||
Implements the PlatformAdapter interface to provide:
|
||||
- Content type identification from URL
|
||||
- Video metadata extraction
|
||||
- Article metadata extraction
|
||||
- Article content fetching in Markdown format
|
||||
- Automatic error retry and state recording
|
||||
|
||||
Attributes:
|
||||
PLATFORM: Platform identifier string ("bilibili")
|
||||
_extractor: BilibiliExtractor instance for actual extraction
|
||||
_state_manager: Optional WorkflowStateManager for recording step status
|
||||
|
||||
Example:
|
||||
adapter = BilibiliAdapter()
|
||||
content_type = adapter.get_content_type("https://www.bilibili.com/video/BV1xx...")
|
||||
# -> "video"
|
||||
|
||||
metadata = await adapter.fetch_metadata("BV1xx...", "video")
|
||||
# -> {"metadata": {...}, "stats": {...}, ...}
|
||||
"""
|
||||
|
||||
PLATFORM = "bilibili"
|
||||
|
||||
# URL patterns for content type identification
|
||||
VIDEO_PATTERN = re.compile(r'bilibili\.com/video/(BV[a-zA-Z0-9]+)')
|
||||
ARTICLE_PATTERN = re.compile(r'bilibili\.com/read/(cv\d+)')
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
headless: bool = True,
|
||||
timeout: int = 30,
|
||||
state_manager: Optional[WorkflowStateManager] = None
|
||||
):
|
||||
"""Initialize BilibiliAdapter
|
||||
|
||||
Args:
|
||||
headless: Whether to use headless mode for Playwright (default True)
|
||||
timeout: Request timeout in seconds (default 30)
|
||||
state_manager: Optional WorkflowStateManager for recording step status
|
||||
"""
|
||||
self._extractor = BilibiliExtractor(headless=headless, timeout=timeout)
|
||||
self._state_manager = state_manager
|
||||
|
||||
def get_content_type(self, url: str) -> str:
|
||||
"""Identify content type from URL
|
||||
|
||||
Args:
|
||||
url: Bilibili URL to analyze
|
||||
|
||||
Returns:
|
||||
Content type string: "video", "article", or "unknown"
|
||||
|
||||
Example:
|
||||
adapter.get_content_type("https://www.bilibili.com/video/BV1xx...")
|
||||
-> "video"
|
||||
"""
|
||||
if self.VIDEO_PATTERN.search(url):
|
||||
return "video"
|
||||
if self.ARTICLE_PATTERN.search(url):
|
||||
return "article"
|
||||
return "unknown"
|
||||
|
||||
async def fetch_metadata(
|
||||
self,
|
||||
content_id: str,
|
||||
content_type: str
|
||||
) -> Dict[str, Any]:
|
||||
"""Fetch metadata for the given content
|
||||
|
||||
Args:
|
||||
content_id: BV号 for videos, CV号 for articles
|
||||
content_type: "video" or "article"
|
||||
|
||||
Returns:
|
||||
Dictionary containing metadata. Structure varies by content type:
|
||||
- Video: {metadata: {...}, stats: {...}, description: str}
|
||||
- Article: {cvid, title, author, publish_time, ...}
|
||||
|
||||
Raises:
|
||||
NetworkError: If network request fails after 3 retries
|
||||
AuthError: If authentication fails (403/quota exceeded)
|
||||
ValueError: If content_type is not supported
|
||||
"""
|
||||
try:
|
||||
return await self._fetch_metadata_with_retry(content_id, content_type)
|
||||
except (NetworkError, AuthError) as e:
|
||||
# Record failure to state manager only after all retries exhausted
|
||||
self._record_step_failed("metadata", str(e))
|
||||
raise
|
||||
except Exception as e:
|
||||
# Check if this is an API error that should be treated as network error
|
||||
if self._is_network_related_error(e):
|
||||
network_error = NetworkError(
|
||||
step_name="metadata",
|
||||
input_summary=f"content_id={content_id}, type={content_type}",
|
||||
cause=e
|
||||
)
|
||||
self._record_step_failed("metadata", str(network_error))
|
||||
raise network_error
|
||||
raise
|
||||
|
||||
@retry_on_network_error
|
||||
async def _fetch_metadata_with_retry(
|
||||
self,
|
||||
content_id: str,
|
||||
content_type: str
|
||||
) -> Dict[str, Any]:
|
||||
"""Internal method with retry logic for metadata fetching
|
||||
|
||||
Args:
|
||||
content_id: BV号 for videos, CV号 for articles
|
||||
content_type: "video" or "article"
|
||||
|
||||
Returns:
|
||||
Dictionary containing metadata
|
||||
|
||||
Raises:
|
||||
NetworkError: If network request fails
|
||||
AuthError: If authentication fails
|
||||
ValueError: If content_type is not supported
|
||||
"""
|
||||
if content_type == "video":
|
||||
# Call async method directly to avoid nested asyncio.run()
|
||||
return await self._extractor._extract_video_async(content_id)
|
||||
elif content_type == "article":
|
||||
# Call async method directly to avoid nested asyncio.run()
|
||||
return await self._extractor._extract_article_async(content_id)
|
||||
else:
|
||||
raise ValueError(f"Unsupported content type: {content_type}")
|
||||
|
||||
def _is_network_related_error(self, error: Exception) -> bool:
|
||||
"""Check if an error is network-related and should trigger retry
|
||||
|
||||
Args:
|
||||
error: Exception to check
|
||||
|
||||
Returns:
|
||||
True if error is network-related
|
||||
"""
|
||||
error_str = str(error).lower()
|
||||
network_indicators = [
|
||||
'timeout', 'connection', 'network', '503', '502', '500',
|
||||
'connection refused', 'connection reset', 'eof',
|
||||
]
|
||||
return any(indicator in error_str for indicator in network_indicators)
|
||||
|
||||
async def fetch_content(
|
||||
self,
|
||||
content_id: str,
|
||||
content_type: str
|
||||
) -> str:
|
||||
"""Fetch content body in Markdown format
|
||||
|
||||
For videos, this returns an empty string as video content
|
||||
extraction is handled separately (ASR transcription).
|
||||
|
||||
For articles, this fetches the article content and formats
|
||||
it with YAML frontmatter.
|
||||
|
||||
Args:
|
||||
content_id: CV号 for articles
|
||||
content_type: "video" or "article"
|
||||
|
||||
Returns:
|
||||
Markdown-formatted content string with YAML frontmatter,
|
||||
or empty string for videos
|
||||
"""
|
||||
if content_type == "video":
|
||||
# Video content is handled by ASR in Phase 3
|
||||
return ""
|
||||
|
||||
if content_type == "article":
|
||||
# Fetch article metadata first
|
||||
metadata = await self.fetch_metadata(content_id, "article")
|
||||
|
||||
# Fetch article content (requires additional implementation)
|
||||
content = await self._fetch_article_content(content_id)
|
||||
|
||||
# Format with YAML frontmatter
|
||||
return self._format_content_markdown(metadata, content)
|
||||
|
||||
raise ValueError(f"Unsupported content type: {content_type}")
|
||||
|
||||
async def _fetch_article_content(self, cvid: str) -> str:
|
||||
"""Fetch article body content
|
||||
|
||||
Args:
|
||||
cvid: Article CV号
|
||||
|
||||
Returns:
|
||||
Raw article content string
|
||||
|
||||
Note:
|
||||
This method currently returns a placeholder.
|
||||
Full implementation would use additional API calls
|
||||
or Playwright to fetch the article body.
|
||||
"""
|
||||
# TODO: Implement full article content fetching
|
||||
# This would require additional API endpoints or Playwright
|
||||
logger.warning(f"Article content fetching not fully implemented for {cvid}")
|
||||
return ""
|
||||
|
||||
def _format_content_markdown(
|
||||
self,
|
||||
metadata: Dict[str, Any],
|
||||
content: str,
|
||||
content_type: str = "article"
|
||||
) -> str:
|
||||
"""Format content with YAML frontmatter
|
||||
|
||||
Args:
|
||||
metadata: Article metadata dictionary
|
||||
content: Article body content
|
||||
content_type: Content type string
|
||||
|
||||
Returns:
|
||||
Markdown string with YAML frontmatter
|
||||
"""
|
||||
import yaml
|
||||
from datetime import datetime
|
||||
|
||||
frontmatter = {
|
||||
"title": metadata.get("title", ""),
|
||||
"author": metadata.get("author", ""),
|
||||
"source_url": metadata.get("url", ""),
|
||||
"platform": self.PLATFORM,
|
||||
"content_type": content_type,
|
||||
"fetched_at": datetime.now().isoformat(),
|
||||
}
|
||||
|
||||
# Add additional metadata if available
|
||||
if "cvid" in metadata:
|
||||
frontmatter["cvid"] = metadata["cvid"]
|
||||
if "mid" in metadata:
|
||||
frontmatter["author_id"] = metadata["mid"]
|
||||
if "word_count" in metadata:
|
||||
frontmatter["word_count"] = metadata["word_count"]
|
||||
|
||||
yaml_str = yaml.dump(frontmatter, allow_unicode=True, sort_keys=False)
|
||||
return f"---\n{yaml_str}---\n\n{content}"
|
||||
|
||||
def get_output_filename(
|
||||
self,
|
||||
content_id: str,
|
||||
content_type: str,
|
||||
suffix: str
|
||||
) -> str:
|
||||
"""Generate output filename for content
|
||||
|
||||
Args:
|
||||
content_id: BV号 or CV号
|
||||
content_type: "video" or "article"
|
||||
suffix: Filename suffix (e.g., "metadata.json", "content.md")
|
||||
|
||||
Returns:
|
||||
Filename string in format "bilibili_{content_id}_{suffix}"
|
||||
|
||||
Example:
|
||||
adapter.get_output_filename("BV1xx...", "video", "metadata.json")
|
||||
-> "bilibili_BV1xx..._metadata.json"
|
||||
"""
|
||||
return f"{self.PLATFORM}_{content_id}_{suffix}"
|
||||
|
||||
def _record_step_failed(self, step_name: str, error_message: str) -> None:
|
||||
"""Record step failure to state manager
|
||||
|
||||
Args:
|
||||
step_name: Name of the failed step
|
||||
error_message: Error message to record
|
||||
"""
|
||||
if self._state_manager is not None:
|
||||
self._state_manager.mark_step_failed(step_name, error_message)
|
||||
logger.warning(f"Step '{step_name}' failed: {error_message}")
|
||||
|
||||
async def download_video(
|
||||
self,
|
||||
bvid: str,
|
||||
output_dir: Path,
|
||||
quality: int = 32
|
||||
) -> Dict[str, Any]:
|
||||
"""Download Bilibili video using yt-dlp.
|
||||
|
||||
Downloads only audio (sufficient for Whisper ASR) to save bandwidth.
|
||||
|
||||
Args:
|
||||
bvid: Bilibili video ID (BV号)
|
||||
output_dir: Directory to save the video
|
||||
quality: Video quality (16=360P, 32=480P, 64=720P, 80=1080P)
|
||||
|
||||
Returns:
|
||||
Dictionary with:
|
||||
- success: bool
|
||||
- filepath: str or None
|
||||
- message: str
|
||||
"""
|
||||
result = {"success": False, "filepath": None, "message": ""}
|
||||
|
||||
try:
|
||||
from kb_workflow.pipeline.config import get_settings
|
||||
|
||||
url = f"https://www.bilibili.com/video/{bvid}"
|
||||
|
||||
# Get cookie from settings or environment
|
||||
settings = get_settings()
|
||||
cookie = settings.bilibili_cookie.get_secret_value() or os.environ.get('BILIBILI_COOKIE', '')
|
||||
|
||||
# Create cookie file if needed
|
||||
cookie_file = None
|
||||
if cookie:
|
||||
cookie_file = self._create_cookie_file(cookie)
|
||||
|
||||
try:
|
||||
# Get video title for filename
|
||||
title = await self._get_video_title(url, cookie_file)
|
||||
safe_title = self._sanitize_filename(title, bvid)
|
||||
|
||||
# Build yt-dlp command - download best audio only for ASR
|
||||
# This saves bandwidth since Whisper only needs audio
|
||||
output_template = str(output_dir / f"{safe_title}.%(ext)s")
|
||||
|
||||
# Use best audio format for ASR
|
||||
cmd = [
|
||||
"yt-dlp",
|
||||
"-f", "bestaudio/best",
|
||||
"-o", output_template,
|
||||
"--no-download-archive",
|
||||
"-q",
|
||||
url
|
||||
]
|
||||
|
||||
if cookie_file:
|
||||
cmd.extend(["--cookies", cookie_file])
|
||||
|
||||
logger.info(f"Downloading video: {bvid}")
|
||||
proc = await asyncio.create_subprocess_exec(
|
||||
*cmd,
|
||||
stdout=asyncio.subprocess.PIPE,
|
||||
stderr=asyncio.subprocess.PIPE
|
||||
)
|
||||
stdout, stderr = await proc.communicate()
|
||||
|
||||
# Find downloaded file
|
||||
downloaded_files = list(output_dir.glob(f"{safe_title}.*"))
|
||||
video_extensions = {'.mp4', '.mkv', '.webm', '.mp3', '.m4a', '.opus', '.wav'}
|
||||
downloaded_files = [f for f in downloaded_files if f.suffix.lower() in video_extensions]
|
||||
|
||||
if downloaded_files:
|
||||
video_file = downloaded_files[0]
|
||||
result["success"] = True
|
||||
result["filepath"] = str(video_file)
|
||||
result["message"] = f"Downloaded: {video_file.name}"
|
||||
logger.info(f"Video downloaded: {video_file}")
|
||||
else:
|
||||
# Try to find any new media file in output_dir
|
||||
all_media = list(output_dir.glob("*.*"))
|
||||
all_media = [f for f in all_media if f.suffix.lower() in video_extensions]
|
||||
if all_media:
|
||||
# Get the most recently modified file
|
||||
video_file = max(all_media, key=lambda f: f.stat().st_mtime)
|
||||
result["success"] = True
|
||||
result["filepath"] = str(video_file)
|
||||
result["message"] = f"Downloaded: {video_file.name}"
|
||||
logger.info(f"Video downloaded (fallback): {video_file}")
|
||||
else:
|
||||
result["message"] = "No downloaded file found"
|
||||
logger.warning(f"No downloaded file found for {bvid}")
|
||||
|
||||
finally:
|
||||
# Clean up cookie file
|
||||
if cookie_file and os.path.exists(cookie_file):
|
||||
os.remove(cookie_file)
|
||||
|
||||
except Exception as e:
|
||||
result["message"] = f"Download error: {str(e)}"
|
||||
logger.error(f"Video download failed for {bvid}: {e}")
|
||||
|
||||
return result
|
||||
|
||||
async def _get_video_title(self, url: str, cookie_file: Optional[str] = None) -> str:
|
||||
"""Get video title using yt-dlp.
|
||||
|
||||
Args:
|
||||
url: Bilibili video URL
|
||||
cookie_file: Optional path to cookie file
|
||||
|
||||
Returns:
|
||||
Video title or fallback name
|
||||
"""
|
||||
try:
|
||||
cmd = ["yt-dlp", "--get-title", url]
|
||||
if cookie_file:
|
||||
cmd.extend(["--cookies", cookie_file])
|
||||
|
||||
proc = await asyncio.create_subprocess_exec(
|
||||
*cmd,
|
||||
stdout=asyncio.subprocess.PIPE,
|
||||
stderr=asyncio.subprocess.PIPE
|
||||
)
|
||||
stdout, stderr = await proc.communicate()
|
||||
|
||||
title = stdout.decode('utf-8').strip()
|
||||
if title and 'error' not in title.lower():
|
||||
return title
|
||||
except Exception as e:
|
||||
logger.debug(f"Failed to get video title: {e}")
|
||||
|
||||
return "video"
|
||||
|
||||
def _sanitize_filename(self, title: str, bvid: str) -> str:
|
||||
"""Generate standardized filename for video download.
|
||||
|
||||
Args:
|
||||
title: Video title (unused, kept for API compatibility)
|
||||
bvid: Video BV号
|
||||
|
||||
Returns:
|
||||
Standardized filename in format "bilibili_{bvid}"
|
||||
"""
|
||||
# Use standardized naming convention: platform_bvid
|
||||
return f"bilibili_{bvid}"
|
||||
|
||||
def _create_cookie_file(self, cookie: str) -> str:
|
||||
"""Create Netscape format cookie file for yt-dlp.
|
||||
|
||||
Args:
|
||||
cookie: Cookie string (e.g., "SESSDATA=xxx; bili_jct=xxx")
|
||||
|
||||
Returns:
|
||||
Path to temporary cookie file
|
||||
"""
|
||||
# Parse cookie string
|
||||
cookies = {}
|
||||
for item in cookie.split(';'):
|
||||
item = item.strip()
|
||||
if '=' in item:
|
||||
key, value = item.split('=', 1)
|
||||
cookies[key.strip()] = value.strip()
|
||||
|
||||
# Create Netscape format
|
||||
cookie_content = "# Netscape HTTP Cookie File\n"
|
||||
for name, value in cookies.items():
|
||||
cookie_content += f".bilibili.com\tTRUE\t/\tFALSE\t0\t{name}\t{value}\n"
|
||||
|
||||
# Write to temp file
|
||||
fd, path = tempfile.mkstemp(suffix='.txt')
|
||||
with os.fdopen(fd, 'w') as f:
|
||||
f.write(cookie_content)
|
||||
|
||||
return path
|
||||
@@ -0,0 +1,263 @@
|
||||
"""Zhihu Platform Adapter
|
||||
|
||||
Provides content extraction for Zhihu platform:
|
||||
- Answer extraction from question/answer URLs
|
||||
- Zhuanlan (article) extraction from zhuanlan URLs
|
||||
- Cookie-based authentication
|
||||
- Login page detection for expired cookies
|
||||
"""
|
||||
|
||||
from typing import Dict, Any, Optional
|
||||
from datetime import datetime
|
||||
import re
|
||||
import asyncio
|
||||
|
||||
from kb_workflow.pipeline.platforms.base import PlatformAdapter
|
||||
from kb_workflow.pipeline.errors import AuthError
|
||||
from kb_workflow.pipeline.config import get_settings
|
||||
from kb_workflow.metadata.zhihu_extractor import ZhihuPlaywrightExtractor
|
||||
|
||||
|
||||
class ZhihuAdapter(PlatformAdapter):
|
||||
"""Zhihu platform adapter for answer and zhuanlan content extraction.
|
||||
|
||||
Uses ZhihuPlaywrightExtractor (from kb-zhihu-metadata) for content extraction.
|
||||
Requires KB_ZHIHU_COOKIE environment variable for authentication.
|
||||
|
||||
Example:
|
||||
adapter = ZhihuAdapter()
|
||||
content_type = adapter.get_content_type("https://www.zhihu.com/question/123/answer/456")
|
||||
# -> "answer"
|
||||
|
||||
metadata = await adapter.fetch_metadata("456", "answer")
|
||||
content = await adapter.fetch_content("456", "answer")
|
||||
"""
|
||||
|
||||
PLATFORM = "zhihu"
|
||||
|
||||
# URL patterns for content type detection
|
||||
ANSWER_PATTERN = re.compile(r'zhihu\.com/question/\d+/answer/(\d+)')
|
||||
ZHUANLAN_PATTERN = re.compile(r'zhuanlan\.zhihu\.com/p/([\w\d]+)')
|
||||
QUESTION_PATTERN = re.compile(r'zhihu\.com/question/(\d+)(?!/answer)')
|
||||
|
||||
def __init__(self, headless: bool = True, timeout: int = 30):
|
||||
"""Initialize ZhihuAdapter.
|
||||
|
||||
Args:
|
||||
headless: Run browser in headless mode (default: True)
|
||||
timeout: Page load timeout in seconds (default: 30)
|
||||
"""
|
||||
self.headless = headless
|
||||
self.timeout = timeout
|
||||
self._cookie: Optional[str] = None
|
||||
self._extractor = None
|
||||
|
||||
@property
|
||||
def cookie(self) -> str:
|
||||
"""Get Zhihu cookie from settings.
|
||||
|
||||
Returns:
|
||||
Cookie string from KB_ZHIHU_COOKIE environment variable.
|
||||
Returns empty string if not configured.
|
||||
"""
|
||||
if self._cookie is None:
|
||||
settings = get_settings()
|
||||
self._cookie = settings.zhihu_cookie.get_secret_value()
|
||||
return self._cookie
|
||||
|
||||
def get_content_type(self, url: str) -> str:
|
||||
"""Identify content type from URL.
|
||||
|
||||
Args:
|
||||
url: The content URL to analyze
|
||||
|
||||
Returns:
|
||||
Content type: 'answer', 'zhuanlan', or 'unknown'
|
||||
"""
|
||||
if self.ANSWER_PATTERN.search(url):
|
||||
return "answer"
|
||||
elif self.ZHUANLAN_PATTERN.search(url):
|
||||
return "zhuanlan"
|
||||
return "unknown"
|
||||
|
||||
def _get_extractor(self):
|
||||
"""Get or create ZhihuPlaywrightExtractor instance.
|
||||
|
||||
Returns:
|
||||
ZhihuPlaywrightExtractor instance
|
||||
"""
|
||||
if self._extractor is None:
|
||||
self._extractor = ZhihuPlaywrightExtractor(
|
||||
headless=self.headless,
|
||||
timeout=self.timeout,
|
||||
cookie=self.cookie
|
||||
)
|
||||
return self._extractor
|
||||
|
||||
def _extract_with_playwright(self, url: str) -> Optional[Dict[str, Any]]:
|
||||
"""Extract content using Playwright-based extractor.
|
||||
|
||||
Args:
|
||||
url: Full URL to extract content from
|
||||
|
||||
Returns:
|
||||
Extraction result dictionary or None if extraction failed
|
||||
"""
|
||||
extractor = self._get_extractor()
|
||||
return extractor.extract(url)
|
||||
|
||||
def _detect_login_page(self, result: Dict[str, Any]) -> bool:
|
||||
"""Detect if the result indicates a login page redirect.
|
||||
|
||||
This happens when the cookie has expired or is invalid.
|
||||
|
||||
Args:
|
||||
result: Extraction result dictionary
|
||||
|
||||
Returns:
|
||||
True if login page is detected, False otherwise
|
||||
"""
|
||||
if not result:
|
||||
return False
|
||||
|
||||
title = result.get("title", "")
|
||||
content = result.get("content", "")
|
||||
content_markdown = result.get("content_markdown", "")
|
||||
|
||||
# Check for login-related indicators
|
||||
login_indicators = ["登录", "登陆", "sign in", "login"]
|
||||
title_lower = title.lower()
|
||||
|
||||
# If title contains login keywords
|
||||
if any(indicator in title_lower for indicator in login_indicators):
|
||||
return True
|
||||
|
||||
# If content is suspiciously short (login page has minimal content)
|
||||
total_content = content + content_markdown
|
||||
if len(total_content.strip()) < 100:
|
||||
return True
|
||||
|
||||
return False
|
||||
|
||||
async def fetch_metadata(self, content_id: str, content_type: str, source_url: str = None) -> Dict[str, Any]:
|
||||
"""Fetch metadata for the content.
|
||||
|
||||
Args:
|
||||
content_id: Answer ID or article ID
|
||||
content_type: 'answer' or 'zhuanlan'
|
||||
source_url: Original source URL (optional, used for answers to get question ID)
|
||||
|
||||
Returns:
|
||||
Dictionary containing metadata (title, author, publish_time, etc.)
|
||||
|
||||
Raises:
|
||||
AuthError: If cookie is missing or expired
|
||||
"""
|
||||
# Check for missing cookie
|
||||
if not self.cookie:
|
||||
raise AuthError(
|
||||
step_name="zhihu_metadata",
|
||||
input_summary=f"content_id={content_id}, type={content_type}",
|
||||
auth_type="zhihu_cookie",
|
||||
)
|
||||
|
||||
# Build URL based on content type
|
||||
if content_type == "answer":
|
||||
# For answers, try to use source_url first, then fallback to constructed URL
|
||||
if source_url:
|
||||
url = source_url
|
||||
else:
|
||||
# Fallback: use answer ID only (may not work for all cases)
|
||||
url = f"https://www.zhihu.com/question/0/answer/{content_id}"
|
||||
elif content_type == "zhuanlan":
|
||||
url = f"https://zhuanlan.zhihu.com/p/{content_id}"
|
||||
else:
|
||||
raise ValueError(f"Unsupported content type: {content_type}")
|
||||
|
||||
# Extract content in a separate thread to avoid blocking asyncio
|
||||
result = await asyncio.to_thread(self._extract_with_playwright, url)
|
||||
|
||||
# Check for login page
|
||||
if self._detect_login_page(result):
|
||||
raise AuthError(
|
||||
step_name="zhihu_metadata",
|
||||
input_summary=f"content_id={content_id}, type={content_type}",
|
||||
auth_type="zhihu_cookie",
|
||||
cause=Exception("Cookie may have expired, please update KB_ZHIHU_COOKIE")
|
||||
)
|
||||
|
||||
if not result:
|
||||
raise AuthError(
|
||||
step_name="zhihu_metadata",
|
||||
input_summary=f"content_id={content_id}, type={content_type}",
|
||||
auth_type="zhihu_cookie",
|
||||
cause=Exception("Failed to extract content, check if content exists")
|
||||
)
|
||||
|
||||
return result
|
||||
|
||||
async def fetch_content(self, content_id: str, content_type: str, source_url: str = None) -> str:
|
||||
"""Fetch content in Markdown format.
|
||||
|
||||
Args:
|
||||
content_id: Answer ID or article ID
|
||||
content_type: 'answer' or 'zhuanlan'
|
||||
source_url: Original source URL (optional, used for answers)
|
||||
|
||||
Returns:
|
||||
Markdown-formatted content with YAML frontmatter
|
||||
|
||||
Raises:
|
||||
AuthError: If cookie is missing or expired
|
||||
"""
|
||||
# Get metadata (which also validates auth)
|
||||
metadata = await self.fetch_metadata(content_id, content_type, source_url=source_url)
|
||||
|
||||
# Get content markdown
|
||||
content_markdown = metadata.get("content_markdown", "") or metadata.get("content", "")
|
||||
|
||||
# Format with YAML frontmatter
|
||||
return self._format_with_frontmatter(metadata, content_markdown, content_type)
|
||||
|
||||
def _format_with_frontmatter(
|
||||
self,
|
||||
metadata: Dict[str, Any],
|
||||
content: str,
|
||||
content_type: str
|
||||
) -> str:
|
||||
"""Format content with YAML frontmatter.
|
||||
|
||||
Args:
|
||||
metadata: Content metadata dictionary
|
||||
content: Markdown content
|
||||
content_type: 'answer' or 'zhuanlan'
|
||||
|
||||
Returns:
|
||||
Formatted string with YAML frontmatter and content
|
||||
"""
|
||||
frontmatter_lines = [
|
||||
"---",
|
||||
f"title: {metadata.get('title', '')}",
|
||||
f"author: {metadata.get('author', '')}",
|
||||
f"source_url: {metadata.get('url', '')}",
|
||||
f"platform: {self.PLATFORM}",
|
||||
f"content_type: {content_type}",
|
||||
f"fetched_at: {datetime.now().isoformat()}",
|
||||
"---",
|
||||
"",
|
||||
]
|
||||
|
||||
return "\n".join(frontmatter_lines) + content
|
||||
|
||||
def get_output_filename(self, content_id: str, content_type: str, suffix: str) -> str:
|
||||
"""Generate output filename for the content.
|
||||
|
||||
Args:
|
||||
content_id: Answer ID or article ID
|
||||
content_type: 'answer' or 'zhuanlan'
|
||||
suffix: File suffix (e.g., 'metadata.json', 'content.md')
|
||||
|
||||
Returns:
|
||||
Filename string (e.g., 'zhihu_789012_metadata.json')
|
||||
"""
|
||||
return f"zhihu_{content_id}_{suffix}"
|
||||
@@ -0,0 +1,26 @@
|
||||
"""Processors module for KB Pipeline
|
||||
|
||||
This module contains processors for various content transformation tasks:
|
||||
- BaseProcessor: Abstract base class for all processors
|
||||
- ParseProcessor: URL parsing for various platforms
|
||||
- DownloadProcessor: Metadata and content downloading
|
||||
- ASRProcessor: Audio-to-text conversion (subtitle download + Whisper.cpp)
|
||||
- AnalyzeProcessor: AI-powered content analysis using DeepAnalyzer
|
||||
- ArchiveProcessor: Copy analysis reports to archive directory
|
||||
"""
|
||||
|
||||
from kb_workflow.pipeline.processors.base import BaseProcessor
|
||||
from kb_workflow.pipeline.processors.parse import ParseProcessor
|
||||
from kb_workflow.pipeline.processors.download import DownloadProcessor
|
||||
from kb_workflow.pipeline.processors.asr import ASRProcessor
|
||||
from kb_workflow.pipeline.processors.analyze import AnalyzeProcessor
|
||||
from kb_workflow.pipeline.processors.archive import ArchiveProcessor
|
||||
|
||||
__all__ = [
|
||||
"BaseProcessor",
|
||||
"ParseProcessor",
|
||||
"DownloadProcessor",
|
||||
"ASRProcessor",
|
||||
"AnalyzeProcessor",
|
||||
"ArchiveProcessor",
|
||||
]
|
||||
@@ -0,0 +1,325 @@
|
||||
"""AnalyzeProcessor - AI-powered content analysis
|
||||
|
||||
Unified processor for generating AI analysis reports from content.
|
||||
Wraps existing DeepAnalyzer for integration into the pipeline.
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
import logging
|
||||
import os
|
||||
import sys
|
||||
from pathlib import Path
|
||||
from typing import Optional, Dict, Any
|
||||
|
||||
# Add kb-deep-analyzer to path for import
|
||||
_kb_deep_analyzer_path = Path(__file__).parent.parent.parent / "kb-deep-analyzer"
|
||||
if str(_kb_deep_analyzer_path) not in sys.path:
|
||||
sys.path.insert(0, str(_kb_deep_analyzer_path))
|
||||
|
||||
from kb_workflow.pipeline.config import get_settings
|
||||
from kb_workflow.pipeline.schemas import AnalyzeInput, AnalyzeOutput
|
||||
from kb_workflow.pipeline.state import WorkflowStateManager
|
||||
from kb_workflow.pipeline.processors.base import BaseProcessor
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class AnalyzeProcessor(BaseProcessor):
|
||||
"""Analyze 处理器 - AI 深度内容分析
|
||||
|
||||
处理流程:
|
||||
1. 检查步骤是否已完成(幂等执行)
|
||||
2. 加载元数据和内容文件
|
||||
3. 初始化 AI 客户端
|
||||
4. 调用 DeepAnalyzer 生成分析报告
|
||||
5. 生成 frontmatter 并保存 Markdown 文件
|
||||
6. 记录处理状态到 workflow_state.json
|
||||
|
||||
Attributes:
|
||||
STEP_NAME: 步骤名称,用于状态记录
|
||||
settings: KB 配置
|
||||
state_manager: 状态管理器(可选)
|
||||
"""
|
||||
|
||||
STEP_NAME = "analyze"
|
||||
|
||||
async def process(self, input_data: AnalyzeInput) -> AnalyzeOutput:
|
||||
"""执行分析处理
|
||||
|
||||
策略:
|
||||
1. 检查步骤是否已完成(跳过已完成)
|
||||
2. 加载元数据和内容
|
||||
3. 调用 DeepAnalyzer
|
||||
4. 保存分析报告
|
||||
5. 记录状态
|
||||
|
||||
Args:
|
||||
input_data: Analyze 输入数据
|
||||
|
||||
Returns:
|
||||
Analyze 处理结果
|
||||
|
||||
Raises:
|
||||
Exception: 处理失败时抛出异常
|
||||
"""
|
||||
try:
|
||||
# Step 1: 检查是否已完成(幂等执行)
|
||||
if self.state_manager:
|
||||
state = self.state_manager.load()
|
||||
if state and state.is_step_completed(self.STEP_NAME):
|
||||
logger.info(f"Step {self.STEP_NAME} already completed, skipping")
|
||||
# 返回已有结果
|
||||
existing_file = Path(input_data.output_file)
|
||||
if existing_file.exists():
|
||||
return AnalyzeOutput(
|
||||
success=True,
|
||||
analysis_file=str(existing_file),
|
||||
summary="Skipped - already completed",
|
||||
main_topics=[],
|
||||
key_insights=[],
|
||||
tags=[],
|
||||
stats={"skipped": True}
|
||||
)
|
||||
|
||||
# Step 2: 加载元数据和内容
|
||||
metadata = self._load_json(input_data.metadata_file)
|
||||
content = self._load_json(input_data.content_file)
|
||||
|
||||
# 处理 Whisper 字幕格式 - 提取完整文本
|
||||
if content and 'transcription' in content:
|
||||
# Whisper JSON 格式,提取文本
|
||||
text_parts = []
|
||||
for item in content.get('transcription', []):
|
||||
text_parts.append(item.get('text', ''))
|
||||
content = {'text': ' '.join(text_parts)}
|
||||
logger.info(f"Extracted {len(content['text'])} chars from Whisper transcription")
|
||||
|
||||
# 对于文章类型,内容可能是纯文本文件
|
||||
if not content and input_data.content_type in ("article", "answer", "zhuanlan"):
|
||||
content_path = Path(input_data.content_file)
|
||||
if content_path.exists():
|
||||
text = content_path.read_text(encoding="utf-8")
|
||||
# 去除 YAML frontmatter(如果存在)
|
||||
text = self._strip_frontmatter(text)
|
||||
content = {"text": text}
|
||||
|
||||
# Step 3: 调用 DeepAnalyzer
|
||||
result = await self._run_analysis(
|
||||
metadata=metadata,
|
||||
content=content,
|
||||
content_type=input_data.content_type
|
||||
)
|
||||
|
||||
# Step 4: 生成并保存分析报告
|
||||
output_path = Path(input_data.output_file)
|
||||
output_path.parent.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
self._write_analysis_report(
|
||||
output_path=output_path,
|
||||
metadata=metadata,
|
||||
result=result,
|
||||
content_type=input_data.content_type
|
||||
)
|
||||
|
||||
# Step 5: 记录成功状态
|
||||
if self.state_manager:
|
||||
state = self.state_manager.load()
|
||||
if state:
|
||||
summary = f"topics={len(result.main_topics)}, insights={len(result.key_insights)}"
|
||||
state.mark_step_completed(self.STEP_NAME, summary)
|
||||
self.state_manager.save(state)
|
||||
|
||||
return AnalyzeOutput(
|
||||
success=True,
|
||||
analysis_file=str(output_path),
|
||||
summary=result.summary,
|
||||
main_topics=result.main_topics,
|
||||
key_insights=result.key_insights,
|
||||
tags=result.recommended_tags,
|
||||
stats={
|
||||
"topics_count": len(result.main_topics),
|
||||
"insights_count": len(result.key_insights),
|
||||
"tags_count": len(result.recommended_tags)
|
||||
}
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Analyze failed: {e}")
|
||||
|
||||
# 记录失败状态
|
||||
if self.state_manager:
|
||||
state = self.state_manager.load()
|
||||
if state:
|
||||
state.mark_step_failed(self.STEP_NAME, str(e))
|
||||
self.state_manager.save(state)
|
||||
|
||||
return AnalyzeOutput(
|
||||
success=False,
|
||||
error_message=str(e),
|
||||
analysis_file="",
|
||||
summary="",
|
||||
main_topics=[],
|
||||
key_insights=[],
|
||||
tags=[],
|
||||
stats={}
|
||||
)
|
||||
|
||||
def _load_json(self, file_path: str) -> Dict[str, Any]:
|
||||
"""加载 JSON 文件
|
||||
|
||||
Args:
|
||||
file_path: JSON 文件路径
|
||||
|
||||
Returns:
|
||||
解析后的字典,如果文件不存在或解析失败返回空字典
|
||||
"""
|
||||
if not file_path:
|
||||
return {}
|
||||
|
||||
path = Path(file_path)
|
||||
if not path.exists():
|
||||
logger.warning(f"File not found: {file_path}")
|
||||
return {}
|
||||
|
||||
try:
|
||||
with open(path, 'r', encoding='utf-8') as f:
|
||||
return json.load(f)
|
||||
except json.JSONDecodeError as e:
|
||||
logger.error(f"Failed to parse JSON: {file_path}, error: {e}")
|
||||
return {}
|
||||
|
||||
async def _run_analysis(self, metadata: Dict, content: Dict, content_type: str):
|
||||
"""调用 DeepAnalyzer 执行分析
|
||||
|
||||
Args:
|
||||
metadata: 元数据字典
|
||||
content: 内容字典
|
||||
content_type: 内容类型
|
||||
|
||||
Returns:
|
||||
AnalysisResult 对象
|
||||
"""
|
||||
# Lazy import to avoid dependency issues
|
||||
from deep_analyzer import DeepAnalyzer, SimpleAIClient
|
||||
|
||||
# 初始化 AI 客户端
|
||||
ai_client = self._create_ai_client()
|
||||
|
||||
# 创建分析器实例
|
||||
analyzer = DeepAnalyzer(ai_client=ai_client)
|
||||
|
||||
# 执行分析
|
||||
return analyzer.analyze(metadata, content, content_type)
|
||||
|
||||
def _create_ai_client(self):
|
||||
"""创建 AI 客户端
|
||||
|
||||
Returns:
|
||||
SimpleAIClient 实例
|
||||
"""
|
||||
from deep_analyzer import SimpleAIClient
|
||||
|
||||
# 从环境变量获取配置
|
||||
api_key = os.environ.get("AI_API_KEY", "")
|
||||
provider = os.environ.get("AI_PROVIDER", "")
|
||||
model = os.environ.get("AI_MODEL", "glm-5")
|
||||
base_url = os.environ.get("AI_BASE_URL", "")
|
||||
|
||||
return SimpleAIClient(
|
||||
api_key=api_key,
|
||||
provider=provider,
|
||||
model=model,
|
||||
base_url=base_url
|
||||
)
|
||||
|
||||
def _normalize_metadata(self, metadata: Dict) -> Dict:
|
||||
"""标准化元数据结构(扁平化嵌套结构)
|
||||
|
||||
Args:
|
||||
metadata: 原始元数据字典,可能是嵌套结构
|
||||
|
||||
Returns:
|
||||
扁平化的元数据字典
|
||||
"""
|
||||
if not metadata:
|
||||
return {}
|
||||
|
||||
# 如果已经是扁平结构(顶层有 title)
|
||||
if 'title' in metadata:
|
||||
return metadata
|
||||
|
||||
# 处理嵌套结构 {metadata: {...}, stats: {...}}
|
||||
if 'metadata' in metadata:
|
||||
return metadata['metadata'].copy()
|
||||
|
||||
return metadata
|
||||
|
||||
def _strip_frontmatter(self, text: str) -> str:
|
||||
"""去除 Markdown 文件中的 YAML frontmatter
|
||||
|
||||
Args:
|
||||
text: 可能包含 frontmatter 的文本
|
||||
|
||||
Returns:
|
||||
去除 frontmatter 后的文本
|
||||
"""
|
||||
if not text:
|
||||
return text
|
||||
|
||||
# 检查是否以 --- 开头
|
||||
if not text.startswith('---'):
|
||||
return text
|
||||
|
||||
# 找到第二个 ---
|
||||
lines = text.split('\n')
|
||||
end_index = -1
|
||||
for i in range(1, len(lines)):
|
||||
if lines[i].strip() == '---':
|
||||
end_index = i
|
||||
break
|
||||
|
||||
if end_index == -1:
|
||||
return text
|
||||
|
||||
# 返回 frontmatter 之后的内容(跳过空行)
|
||||
remaining = '\n'.join(lines[end_index + 1:])
|
||||
return remaining.lstrip('\n')
|
||||
|
||||
def _write_analysis_report(
|
||||
self,
|
||||
output_path: Path,
|
||||
metadata: Dict,
|
||||
result,
|
||||
content_type: str
|
||||
) -> None:
|
||||
"""写入分析报告 Markdown 文件
|
||||
|
||||
Args:
|
||||
output_path: 输出文件路径
|
||||
metadata: 元数据字典
|
||||
result: AnalysisResult 对象
|
||||
content_type: 内容类型
|
||||
"""
|
||||
from deep_analyzer import generate_frontmatter
|
||||
|
||||
# 标准化元数据(扁平化嵌套结构)
|
||||
normalized_metadata = self._normalize_metadata(metadata)
|
||||
|
||||
# 生成 frontmatter
|
||||
frontmatter = generate_frontmatter(normalized_metadata, result, content_type)
|
||||
|
||||
# 获取完整分析内容
|
||||
full_analysis = result.full_analysis
|
||||
|
||||
# 文章/回答类型:追加原文
|
||||
if content_type in ("article", "answer", "zhuanlan") and result.original_text:
|
||||
full_analysis = full_analysis.rstrip() + "\n\n## 原文\n\n" + result.original_text
|
||||
|
||||
# 写入文件
|
||||
with open(output_path, 'w', encoding='utf-8') as f:
|
||||
f.write(frontmatter + "\n\n")
|
||||
f.write(full_analysis)
|
||||
|
||||
logger.info(f"Analysis report saved to: {output_path}")
|
||||
@@ -0,0 +1,170 @@
|
||||
"""ArchiveProcessor - Copy analysis reports to archive directory
|
||||
|
||||
Final step in pipeline that copies the generated analysis.md
|
||||
to the archive directory organized by platform.
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import logging
|
||||
import shutil
|
||||
from datetime import datetime
|
||||
from pathlib import Path
|
||||
from typing import Optional
|
||||
|
||||
from kb_workflow.pipeline.config import get_settings
|
||||
from kb_workflow.pipeline.schemas import ArchiveInput, ArchiveOutput
|
||||
from kb_workflow.pipeline.state import WorkflowStateManager
|
||||
from kb_workflow.pipeline.processors.base import BaseProcessor
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class ArchiveProcessor(BaseProcessor):
|
||||
"""Archive 处理器 - 归档分析报告
|
||||
|
||||
处理流程:
|
||||
1. 检查步骤是否已完成(幂等执行)
|
||||
2. 检查归档目录是否存在相同 content_id 的文件
|
||||
3. 如果存在重复,询问用户是否覆盖
|
||||
4. 复制 md 文件到归档目录
|
||||
5. 记录处理状态到 workflow_state.json
|
||||
|
||||
归档目录结构:
|
||||
{KB_ARCHIVE_PATH}/{platform}/{YYYYMMDD}_{content_id}_analysis.md
|
||||
"""
|
||||
|
||||
STEP_NAME = "archive"
|
||||
|
||||
async def process(self, input_data: ArchiveInput) -> ArchiveOutput:
|
||||
"""执行归档处理
|
||||
|
||||
Args:
|
||||
input_data: Archive 输入数据
|
||||
|
||||
Returns:
|
||||
Archive 处理结果
|
||||
"""
|
||||
try:
|
||||
# Step 1: 检查是否已完成(幂等执行)
|
||||
if self.state_manager:
|
||||
state = self.state_manager.load()
|
||||
if state and state.is_step_completed(self.STEP_NAME):
|
||||
logger.info(f"Step {self.STEP_NAME} already completed, skipping")
|
||||
return ArchiveOutput(
|
||||
success=True,
|
||||
skipped=True,
|
||||
platform=input_data.platform,
|
||||
archive_file=""
|
||||
)
|
||||
|
||||
# Step 2: 准备归档路径
|
||||
archive_base = Path(self.settings.archive_path)
|
||||
platform_dir = archive_base / input_data.platform
|
||||
platform_dir.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
# 文件命名: {YYYYMMDD}_{content_id}_analysis.md
|
||||
timestamp = datetime.now().strftime("%Y%m%d")
|
||||
archive_filename = f"{timestamp}_{input_data.content_id}_analysis.md"
|
||||
archive_path = platform_dir / archive_filename
|
||||
|
||||
# Step 3: 检查重复
|
||||
existing_file = self._find_existing_archive(
|
||||
platform_dir, input_data.content_id
|
||||
)
|
||||
if existing_file:
|
||||
should_overwrite = self._ask_overwrite(existing_file)
|
||||
if not should_overwrite:
|
||||
logger.info(f"User chose not to overwrite, skipping archive")
|
||||
return ArchiveOutput(
|
||||
success=True,
|
||||
skipped=True,
|
||||
platform=input_data.platform,
|
||||
archive_file=str(existing_file)
|
||||
)
|
||||
# 删除旧文件
|
||||
existing_file.unlink()
|
||||
logger.info(f"Removed existing archive: {existing_file}")
|
||||
|
||||
# Step 4: 复制文件
|
||||
source_path = Path(input_data.analysis_file)
|
||||
if not source_path.exists():
|
||||
raise FileNotFoundError(f"Analysis file not found: {source_path}")
|
||||
|
||||
shutil.copy2(source_path, archive_path)
|
||||
logger.info(f"Archived: {source_path} -> {archive_path}")
|
||||
|
||||
# Step 5: 记录成功状态
|
||||
if self.state_manager:
|
||||
state = self.state_manager.load()
|
||||
if state:
|
||||
state.mark_step_completed(
|
||||
self.STEP_NAME,
|
||||
f"archived to {archive_path}"
|
||||
)
|
||||
self.state_manager.save(state)
|
||||
|
||||
return ArchiveOutput(
|
||||
success=True,
|
||||
skipped=False,
|
||||
platform=input_data.platform,
|
||||
archive_file=str(archive_path)
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Archive failed: {e}")
|
||||
|
||||
if self.state_manager:
|
||||
state = self.state_manager.load()
|
||||
if state:
|
||||
state.mark_step_failed(self.STEP_NAME, str(e))
|
||||
self.state_manager.save(state)
|
||||
|
||||
return ArchiveOutput(
|
||||
success=False,
|
||||
error_message=str(e),
|
||||
platform=input_data.platform,
|
||||
archive_file=""
|
||||
)
|
||||
|
||||
def _find_existing_archive(self, platform_dir: Path, content_id: str) -> Optional[Path]:
|
||||
"""查找已存在的归档文件
|
||||
|
||||
使用 glob 模式 `*_{content_id}_analysis.md` 搜索
|
||||
|
||||
Args:
|
||||
platform_dir: 平台目录路径
|
||||
content_id: 内容 ID
|
||||
|
||||
Returns:
|
||||
已存在的文件路径,如果不存在返回 None
|
||||
"""
|
||||
pattern = f"*_{content_id}_analysis.md"
|
||||
matches = list(platform_dir.glob(pattern))
|
||||
return matches[0] if matches else None
|
||||
|
||||
def _ask_overwrite(self, existing_file: Path) -> bool:
|
||||
"""询问用户是否覆盖已存在的文件
|
||||
|
||||
使用 rich.prompt.Confirm 模式(与 Pipeline._ask_rerun_step 相同)
|
||||
|
||||
Args:
|
||||
existing_file: 已存在的文件路径
|
||||
|
||||
Returns:
|
||||
True 如果用户选择覆盖
|
||||
"""
|
||||
try:
|
||||
from rich.console import Console
|
||||
from rich.prompt import Confirm
|
||||
|
||||
console = Console()
|
||||
return Confirm.ask(
|
||||
f"Archive already exists: {existing_file.name}\nOverwrite?",
|
||||
default=False
|
||||
)
|
||||
except ImportError:
|
||||
response = input(
|
||||
f"Archive already exists: {existing_file.name}\nOverwrite? [y/N] "
|
||||
)
|
||||
return response.lower() in ('y', 'yes')
|
||||
@@ -0,0 +1,685 @@
|
||||
"""ASRProcessor - Audio-to-text conversion
|
||||
|
||||
Unified processor for B站官方字幕下载 and Whisper.cpp 语音识别.
|
||||
Follows the pipeline pattern established in Phase 1/2.
|
||||
|
||||
Strategy:
|
||||
1. Try official subtitle first (highest quality, no compute cost)
|
||||
2. Fall back to Whisper.cpp if no subtitle available
|
||||
3. Auto-enable speaker diarization if HF_TOKEN is configured
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import asyncio
|
||||
import json
|
||||
import logging
|
||||
import os
|
||||
import subprocess
|
||||
from pathlib import Path
|
||||
from typing import Optional, Dict, Any, List
|
||||
|
||||
import aiohttp
|
||||
|
||||
from kb_workflow.pipeline.config import get_settings
|
||||
from kb_workflow.pipeline.schemas import ASRInput, ASROutput
|
||||
from kb_workflow.pipeline.state import WorkflowStateManager
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class WhisperNotFoundError(RuntimeError):
|
||||
"""Whisper.cpp 路径未找到"""
|
||||
pass
|
||||
|
||||
|
||||
def get_whisper_cpp_path() -> str:
|
||||
"""获取 whisper.cpp 安装路径
|
||||
|
||||
检查环境变量 WHISPER_CPP_PATH 或默认路径 ~/whisper.cpp
|
||||
|
||||
Returns:
|
||||
whisper.cpp 目录路径
|
||||
|
||||
Raises:
|
||||
WhisperNotFoundError: 如果路径不存在
|
||||
"""
|
||||
# 检查环境变量
|
||||
path = os.environ.get('WHISPER_CPP_PATH')
|
||||
if path and os.path.exists(path):
|
||||
return os.path.expanduser(path)
|
||||
|
||||
# 检查默认路径
|
||||
default = os.path.expanduser('~/whisper.cpp')
|
||||
if os.path.exists(default):
|
||||
return default
|
||||
|
||||
raise WhisperNotFoundError(
|
||||
"未找到 whisper.cpp。请设置 WHISPER_CPP_PATH 环境变量或安装到 ~/whisper.cpp"
|
||||
)
|
||||
|
||||
|
||||
def convert_to_wav(input_path: str, output_path: Optional[str] = None) -> str:
|
||||
"""使用 ffmpeg 将音频/视频转换为 16kHz WAV 格式
|
||||
|
||||
Args:
|
||||
input_path: 输入文件路径
|
||||
output_path: 输出文件路径(可选,默认为 input_path_audio.wav)
|
||||
|
||||
Returns:
|
||||
输出 WAV 文件路径
|
||||
"""
|
||||
if output_path is None:
|
||||
base = os.path.splitext(input_path)[0]
|
||||
output_path = f"{base}_audio.wav"
|
||||
|
||||
cmd = [
|
||||
'ffmpeg', '-y', '-i', input_path,
|
||||
'-ar', '16000', # 16kHz 采样率
|
||||
'-ac', '1', # 单声道
|
||||
'-c:a', 'pcm_s16le', # 16-bit PCM
|
||||
output_path
|
||||
]
|
||||
|
||||
subprocess.run(cmd, capture_output=True, check=True)
|
||||
return output_path
|
||||
|
||||
|
||||
class ASRProcessor:
|
||||
"""ASR 处理器 - 统一字幕下载和语音识别
|
||||
|
||||
处理流程:
|
||||
1. 如果是 B站视频,尝试下载官方字幕
|
||||
2. 无官方字幕则调用 Whisper.cpp
|
||||
3. HF_TOKEN 配置时自动启用说话者分离
|
||||
4. 记录处理状态到 workflow_state.json
|
||||
|
||||
Attributes:
|
||||
STEP_NAME: 步骤名称,用于状态记录
|
||||
settings: KB 配置
|
||||
state_manager: 状态管理器(可选)
|
||||
"""
|
||||
|
||||
STEP_NAME = "asr"
|
||||
|
||||
def __init__(self, state_manager: Optional[WorkflowStateManager] = None):
|
||||
"""初始化 ASR 处理器
|
||||
|
||||
Args:
|
||||
state_manager: 状态管理器(可选,用于记录处理状态)
|
||||
"""
|
||||
self.settings = get_settings()
|
||||
self.state_manager = state_manager
|
||||
|
||||
async def process(self, input_data: ASRInput) -> ASROutput:
|
||||
"""执行 ASR 处理
|
||||
|
||||
策略:
|
||||
0. 先检查是否已有字幕文件(幂等性)
|
||||
1. 先尝试官方字幕(仅 B站)
|
||||
2. 无官方字幕则调用 Whisper
|
||||
3. 记录状态
|
||||
|
||||
Args:
|
||||
input_data: ASR 输入数据
|
||||
|
||||
Returns:
|
||||
ASR 处理结果
|
||||
|
||||
Raises:
|
||||
Exception: 处理失败时抛出异常
|
||||
"""
|
||||
result = None
|
||||
|
||||
try:
|
||||
# Step 0: 检查是否已有字幕文件(幂等性)
|
||||
existing_result = self._check_existing_subtitle_files(input_data)
|
||||
if existing_result:
|
||||
logger.info(f"Found existing subtitle files, skipping processing")
|
||||
result = existing_result
|
||||
else:
|
||||
# Step 1: 尝试官方字幕(仅 B站)
|
||||
if input_data.platform == "bilibili":
|
||||
result = await self._try_official_subtitle(input_data)
|
||||
|
||||
# Step 2: 无官方字幕则调用 Whisper
|
||||
if result is None:
|
||||
result = await self._run_whisper(input_data)
|
||||
|
||||
# Step 3: 记录成功状态
|
||||
if self.state_manager:
|
||||
state = self.state_manager.load()
|
||||
if state:
|
||||
summary = f"tool={result.tool}, segments={len(result.segments)}"
|
||||
if result.speakers:
|
||||
summary += f", speakers={len(result.speakers)}"
|
||||
state.mark_step_completed(self.STEP_NAME, summary)
|
||||
self.state_manager.save(state)
|
||||
|
||||
return result
|
||||
|
||||
except Exception as e:
|
||||
# 记录失败状态(只在所有重试都失败后)
|
||||
if self.state_manager:
|
||||
state = self.state_manager.load()
|
||||
if state:
|
||||
state.mark_step_failed(self.STEP_NAME, str(e))
|
||||
self.state_manager.save(state)
|
||||
raise
|
||||
|
||||
def _check_existing_subtitle_files(self, input_data: ASRInput) -> Optional[ASROutput]:
|
||||
"""检查是否已存在字幕文件(幂等性检查)
|
||||
|
||||
Args:
|
||||
input_data: ASR 输入数据
|
||||
|
||||
Returns:
|
||||
如果找到现有字幕文件返回 ASROutput,否则返回 None
|
||||
"""
|
||||
output_dir = Path(input_data.content_dir)
|
||||
|
||||
# 尝试两种命名格式:新格式(带 _subtitle 后缀)和旧格式
|
||||
base_names = [
|
||||
f"{input_data.platform}_{input_data.content_id}_subtitle",
|
||||
f"{input_data.platform}_{input_data.content_id}",
|
||||
]
|
||||
|
||||
for base_name in base_names:
|
||||
json_path = output_dir / f"{base_name}.json"
|
||||
srt_path = output_dir / f"{base_name}.srt"
|
||||
txt_path = output_dir / f"{base_name}.txt"
|
||||
|
||||
if json_path.exists():
|
||||
# 读取现有字幕数据
|
||||
segments = []
|
||||
text = ""
|
||||
duration = 0.0
|
||||
|
||||
try:
|
||||
with open(json_path, 'r', encoding='utf-8') as f:
|
||||
data = json.load(f)
|
||||
|
||||
# 解析字幕格式
|
||||
if 'transcription' in data:
|
||||
# Whisper 格式
|
||||
for i, item in enumerate(data['transcription']):
|
||||
offsets = item.get('offsets', {})
|
||||
if isinstance(offsets, dict):
|
||||
# 新格式:{"from": 0, "to": 5320}
|
||||
start = offsets.get('from', 0) / 1000.0
|
||||
end = offsets.get('to', 0) / 1000.0
|
||||
else:
|
||||
# 旧格式:[0, 5320]
|
||||
start = offsets[0] / 1000.0 if len(offsets) > 0 else 0
|
||||
end = offsets[1] / 1000.0 if len(offsets) > 1 else 0
|
||||
segments.append({
|
||||
'id': i,
|
||||
'start': start,
|
||||
'end': end,
|
||||
'text': item.get('text', '')
|
||||
})
|
||||
elif 'body' in data:
|
||||
# B站官方字幕格式
|
||||
for i, item in enumerate(data['body']):
|
||||
segments.append({
|
||||
'id': i,
|
||||
'start': item.get('from', 0),
|
||||
'end': item.get('to', 0),
|
||||
'text': item.get('content', '')
|
||||
})
|
||||
|
||||
text = ' '.join(s['text'] for s in segments)
|
||||
if segments:
|
||||
duration = segments[-1]['end']
|
||||
|
||||
output_files = {'json': str(json_path)}
|
||||
if srt_path.exists():
|
||||
output_files['srt'] = str(srt_path)
|
||||
if txt_path.exists():
|
||||
output_files['txt'] = str(txt_path)
|
||||
|
||||
return ASROutput(
|
||||
success=True,
|
||||
tool="existing",
|
||||
has_subtitle=True,
|
||||
output_files=output_files,
|
||||
segments=segments,
|
||||
text=text,
|
||||
duration_seconds=duration
|
||||
)
|
||||
except Exception as e:
|
||||
logger.warning(f"Failed to read existing subtitle file: {e}")
|
||||
continue
|
||||
|
||||
return None
|
||||
|
||||
async def _try_official_subtitle(self, input_data: ASRInput) -> Optional[ASROutput]:
|
||||
"""尝试下载 B站官方字幕
|
||||
|
||||
Args:
|
||||
input_data: ASR 输入数据
|
||||
|
||||
Returns:
|
||||
如果找到字幕返回 ASROutput,否则返回 None
|
||||
|
||||
Note:
|
||||
无字幕返回 None(不是抛异常),让流程继续到 Whisper
|
||||
"""
|
||||
# 使用 content_dir 作为输出目录(而不是从 video_path 推断)
|
||||
output_dir = Path(input_data.content_dir)
|
||||
# 使用 content_id 作为基础文件名
|
||||
base_name = f"{input_data.platform}_{input_data.content_id}_subtitle"
|
||||
|
||||
# 输出文件路径
|
||||
txt_path = output_dir / f"{base_name}.txt"
|
||||
srt_path = output_dir / f"{base_name}.srt"
|
||||
json_path = output_dir / f"{base_name}.json"
|
||||
|
||||
async with aiohttp.ClientSession() as session:
|
||||
try:
|
||||
# Step 1: 获取 CID
|
||||
bvid = input_data.content_id
|
||||
pagelist_url = f"https://api.bilibili.com/x/player/pagelist?bvid={bvid}"
|
||||
async with session.get(pagelist_url) as resp:
|
||||
pagelist_data = await resp.json()
|
||||
|
||||
pages = pagelist_data.get('data', [])
|
||||
if not pages:
|
||||
logger.info(f"No pagelist data for {bvid}")
|
||||
return None
|
||||
|
||||
cid = pages[0].get('cid')
|
||||
if not cid:
|
||||
logger.info(f"No cid found for {bvid}")
|
||||
return None
|
||||
|
||||
# Step 2: 获取字幕列表
|
||||
player_url = f"https://api.bilibili.com/x/player/v2?bvid={bvid}&cid={cid}"
|
||||
async with session.get(player_url) as resp:
|
||||
player_data = await resp.json()
|
||||
|
||||
subtitle_info = player_data.get('data', {}).get('subtitle', {})
|
||||
subtitles = subtitle_info.get('list', [])
|
||||
|
||||
if not subtitles:
|
||||
logger.info(f"No official subtitle for {bvid}")
|
||||
return None
|
||||
|
||||
# Step 3: 下载第一个字幕(优先 AI 生成的中文字幕)
|
||||
subtitle_url = None
|
||||
for sub in subtitles:
|
||||
if sub.get('lan', '').startswith('zh'):
|
||||
subtitle_url = sub.get('subtitle_url')
|
||||
break
|
||||
|
||||
if not subtitle_url and subtitles:
|
||||
subtitle_url = subtitles[0].get('subtitle_url')
|
||||
|
||||
if not subtitle_url:
|
||||
return None
|
||||
|
||||
# 补全 URL
|
||||
if subtitle_url.startswith('//'):
|
||||
subtitle_url = f"https:{subtitle_url}"
|
||||
|
||||
async with session.get(subtitle_url) as resp:
|
||||
subtitle_content = await resp.json()
|
||||
|
||||
# Step 4: 转换为输出格式
|
||||
segments = []
|
||||
text_parts = []
|
||||
body = subtitle_content.get('body', [])
|
||||
|
||||
for i, item in enumerate(body):
|
||||
start = item.get('from', 0)
|
||||
end = item.get('to', 0)
|
||||
content = item.get('content', '')
|
||||
|
||||
segments.append({
|
||||
'id': i,
|
||||
'start': start,
|
||||
'end': end,
|
||||
'text': content
|
||||
})
|
||||
text_parts.append(content)
|
||||
|
||||
full_text = ' '.join(text_parts)
|
||||
duration = segments[-1]['end'] if segments else 0
|
||||
|
||||
# Step 5: 写入文件
|
||||
# TXT 格式
|
||||
txt_path.write_text(full_text, encoding='utf-8')
|
||||
|
||||
# SRT 格式
|
||||
srt_content = self._convert_to_srt(segments)
|
||||
srt_path.write_text(srt_content, encoding='utf-8')
|
||||
|
||||
# JSON 格式
|
||||
json_data = {
|
||||
'success': True,
|
||||
'tool': 'subtitle',
|
||||
'segments': segments,
|
||||
'text': full_text,
|
||||
'duration_seconds': duration
|
||||
}
|
||||
json_path.write_text(json.dumps(json_data, ensure_ascii=False, indent=2), encoding='utf-8')
|
||||
|
||||
return ASROutput(
|
||||
success=True,
|
||||
tool="subtitle",
|
||||
has_subtitle=True,
|
||||
output_files={
|
||||
'txt': str(txt_path),
|
||||
'srt': str(srt_path),
|
||||
'json': str(json_path)
|
||||
},
|
||||
segments=segments,
|
||||
text=full_text,
|
||||
duration_seconds=duration
|
||||
)
|
||||
|
||||
except aiohttp.ClientError as e:
|
||||
logger.warning(f"Failed to fetch subtitle: {e}")
|
||||
return None
|
||||
|
||||
def _convert_to_srt(self, segments: List[Dict]) -> str:
|
||||
"""将字幕分段转换为 SRT 格式
|
||||
|
||||
Args:
|
||||
segments: 字幕分段列表,每个包含 start, end, text
|
||||
|
||||
Returns:
|
||||
SRT 格式字符串
|
||||
"""
|
||||
lines = []
|
||||
for i, seg in enumerate(segments, 1):
|
||||
start_time = self._seconds_to_srt_time(seg['start'])
|
||||
end_time = self._seconds_to_srt_time(seg['end'])
|
||||
lines.append(f"{i}")
|
||||
lines.append(f"{start_time} --> {end_time}")
|
||||
lines.append(seg['text'])
|
||||
lines.append("") # 空行
|
||||
|
||||
return '\n'.join(lines)
|
||||
|
||||
def _seconds_to_srt_time(self, seconds: float) -> str:
|
||||
"""将秒数转换为 SRT 时间格式 (HH:MM:SS,mmm)"""
|
||||
hours = int(seconds // 3600)
|
||||
minutes = int((seconds % 3600) // 60)
|
||||
secs = int(seconds % 60)
|
||||
millis = int((seconds % 1) * 1000)
|
||||
return f"{hours:02d}:{minutes:02d}:{secs:02d},{millis:03d}"
|
||||
|
||||
async def _run_whisper(self, input_data: ASRInput) -> ASROutput:
|
||||
"""执行 Whisper 语音识别
|
||||
|
||||
根据 HF_TOKEN 配置自动选择普通模式或说话者分离模式。
|
||||
|
||||
Args:
|
||||
input_data: ASR 输入数据
|
||||
|
||||
Returns:
|
||||
ASR 处理结果
|
||||
"""
|
||||
hf_token = self.settings.hf_token.get_secret_value()
|
||||
if hf_token:
|
||||
return await self._run_whisper_with_diarization(input_data)
|
||||
else:
|
||||
return await self._run_whisper_basic(input_data)
|
||||
|
||||
async def _run_whisper_basic(self, input_data: ASRInput) -> ASROutput:
|
||||
"""执行普通 Whisper.cpp 语音识别
|
||||
|
||||
Args:
|
||||
input_data: ASR 输入数据
|
||||
|
||||
Returns:
|
||||
ASR 处理结果
|
||||
|
||||
Raises:
|
||||
RuntimeError: Whisper.cpp 路径不存在或执行失败,或 video_path 为空
|
||||
"""
|
||||
# 检查 video_path 是否有效
|
||||
if not input_data.video_path:
|
||||
raise RuntimeError(
|
||||
"video_path 为空,无法执行 Whisper 语音识别。"
|
||||
"请确保视频文件已下载(对于没有官方字幕的视频,需要先下载视频文件)。"
|
||||
)
|
||||
|
||||
video_path = Path(input_data.video_path)
|
||||
output_dir = Path(input_data.content_dir)
|
||||
base_name = f"{input_data.platform}_{input_data.content_id}_whisper"
|
||||
|
||||
# 获取 Whisper.cpp 路径
|
||||
try:
|
||||
whisper_path = get_whisper_cpp_path()
|
||||
except WhisperNotFoundError as e:
|
||||
raise RuntimeError(str(e))
|
||||
|
||||
# 模型路径
|
||||
model_path = os.path.join(whisper_path, 'models', 'ggml-base.bin')
|
||||
if not os.path.exists(model_path):
|
||||
raise RuntimeError(f"Whisper model not found: {model_path}")
|
||||
|
||||
# 转换音频格式
|
||||
audio_path = convert_to_wav(str(video_path))
|
||||
temp_audio_created = audio_path != str(video_path)
|
||||
|
||||
try:
|
||||
# 构建 Whisper 命令
|
||||
output_base = os.path.join(output_dir, base_name)
|
||||
cmd = [
|
||||
os.path.join(whisper_path, 'whisper-cli'),
|
||||
'-m', model_path,
|
||||
'-f', audio_path,
|
||||
'-of', output_base,
|
||||
'-l', 'zh',
|
||||
'--output-txt', '--output-srt', '--output-json',
|
||||
]
|
||||
|
||||
# 执行 Whisper
|
||||
result = subprocess.run(cmd, capture_output=True, text=True, timeout=3600)
|
||||
|
||||
if result.returncode != 0:
|
||||
raise RuntimeError(f"Whisper failed: {result.stderr}")
|
||||
|
||||
# 解析输出
|
||||
return self._parse_whisper_output(output_dir, base_name)
|
||||
|
||||
finally:
|
||||
# 清理临时音频文件
|
||||
if temp_audio_created and os.path.exists(audio_path):
|
||||
try:
|
||||
os.remove(audio_path)
|
||||
except Exception:
|
||||
pass # 静默失败,不阻塞主流程
|
||||
|
||||
def _parse_whisper_output(self, output_dir: Path, base_name: str) -> ASROutput:
|
||||
"""解析 Whisper 输出文件
|
||||
|
||||
Args:
|
||||
output_dir: 输出目录
|
||||
base_name: 文件基础名
|
||||
|
||||
Returns:
|
||||
ASR 处理结果
|
||||
"""
|
||||
txt_path = output_dir / f"{base_name}.txt"
|
||||
srt_path = output_dir / f"{base_name}.srt"
|
||||
json_path = output_dir / f"{base_name}.json"
|
||||
|
||||
# 读取 JSON 输出
|
||||
segments = []
|
||||
text = ""
|
||||
duration = 0.0
|
||||
|
||||
if json_path.exists():
|
||||
with open(json_path, 'r', encoding='utf-8') as f:
|
||||
data = json.load(f)
|
||||
# Whisper.cpp JSON 格式
|
||||
if 'transcription' in data:
|
||||
for i, item in enumerate(data['transcription']):
|
||||
offsets = item.get('offsets', {})
|
||||
if isinstance(offsets, dict):
|
||||
# 新格式:{"from": 0, "to": 5320}
|
||||
start = offsets.get('from', 0) / 1000.0
|
||||
end = offsets.get('to', 0) / 1000.0
|
||||
else:
|
||||
# 旧格式:[0, 5320]
|
||||
start = offsets[0] / 1000.0 if len(offsets) > 0 else 0
|
||||
end = offsets[1] / 1000.0 if len(offsets) > 1 else 0
|
||||
segments.append({
|
||||
'id': i,
|
||||
'start': start,
|
||||
'end': end,
|
||||
'text': item.get('text', '')
|
||||
})
|
||||
text = ' '.join(s['text'] for s in segments)
|
||||
if segments:
|
||||
duration = segments[-1]['end']
|
||||
|
||||
# 读取文本
|
||||
if txt_path.exists():
|
||||
text = txt_path.read_text(encoding='utf-8').strip()
|
||||
|
||||
return ASROutput(
|
||||
success=True,
|
||||
tool="whisper",
|
||||
has_subtitle=False,
|
||||
output_files={
|
||||
'txt': str(txt_path),
|
||||
'srt': str(srt_path),
|
||||
'json': str(json_path)
|
||||
},
|
||||
segments=segments,
|
||||
text=text,
|
||||
duration_seconds=duration
|
||||
)
|
||||
|
||||
async def _run_whisper_with_diarization(self, input_data: ASRInput) -> ASROutput:
|
||||
"""执行带说话者分离的 Whisper
|
||||
|
||||
Args:
|
||||
input_data: ASR 输入数据
|
||||
|
||||
Returns:
|
||||
ASR 处理结果,包含说话者信息
|
||||
"""
|
||||
# 先运行普通 Whisper
|
||||
basic_result = await self._run_whisper_basic(input_data)
|
||||
|
||||
try:
|
||||
# 运行说话者分离
|
||||
diarization_result = run_diarization(input_data.video_path)
|
||||
|
||||
# 合并结果
|
||||
merged_segments = assign_speakers_to_segments(
|
||||
basic_result.segments,
|
||||
diarization_result
|
||||
)
|
||||
|
||||
# 提取说话者列表
|
||||
speakers = list(set(
|
||||
seg.get('speaker') for seg in merged_segments if seg.get('speaker')
|
||||
))
|
||||
|
||||
return ASROutput(
|
||||
success=True,
|
||||
tool="whisper-diarization",
|
||||
has_subtitle=False,
|
||||
output_files=basic_result.output_files,
|
||||
segments=merged_segments,
|
||||
text=basic_result.text,
|
||||
duration_seconds=basic_result.duration_seconds,
|
||||
speakers=sorted(speakers)
|
||||
)
|
||||
|
||||
except ImportError:
|
||||
logger.warning("pyannote.audio not available, falling back to basic whisper")
|
||||
return basic_result
|
||||
except Exception as e:
|
||||
logger.warning(f"Diarization failed: {e}, falling back to basic whisper")
|
||||
return basic_result
|
||||
|
||||
|
||||
def run_diarization(audio_path: str, num_speakers: Optional[int] = None) -> Dict[str, List[Dict]]:
|
||||
"""运行 pyannote.audio 说话者分离
|
||||
|
||||
Args:
|
||||
audio_path: 音频文件路径
|
||||
num_speakers: 预期说话者数量(可选)
|
||||
|
||||
Returns:
|
||||
说话者时间线字典 {speaker_name: [{start, end}, ...]}
|
||||
|
||||
Raises:
|
||||
ImportError: 如果 pyannote.audio 未安装
|
||||
RuntimeError: 如果 HF_TOKEN 未配置
|
||||
"""
|
||||
hf_token = os.environ.get('HF_TOKEN')
|
||||
if not hf_token:
|
||||
raise RuntimeError("需要设置 HF_TOKEN 环境变量以使用说话者分离功能")
|
||||
|
||||
from huggingface_hub import login
|
||||
from pyannote.audio import Pipeline
|
||||
import torch
|
||||
|
||||
login(token=hf_token)
|
||||
|
||||
pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization-3.1")
|
||||
|
||||
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
||||
pipeline.to(device)
|
||||
|
||||
diarization = pipeline(audio_path, num_speakers=num_speakers)
|
||||
|
||||
# 解析结果
|
||||
speakers = {}
|
||||
for turn, _, speaker in diarization.itertracks(yield_label=True):
|
||||
if speaker not in speakers:
|
||||
speakers[speaker] = []
|
||||
speakers[speaker].append({
|
||||
"start": round(turn.start, 2),
|
||||
"end": round(turn.end, 2)
|
||||
})
|
||||
|
||||
return speakers
|
||||
|
||||
|
||||
def assign_speakers_to_segments(
|
||||
segments: List[Dict],
|
||||
diarization: Dict[str, List[Dict]]
|
||||
) -> List[Dict]:
|
||||
"""为每个段落分配说话者
|
||||
|
||||
根据时间戳匹配说话者分离结果到 Whisper 段落。
|
||||
|
||||
Args:
|
||||
segments: Whisper 段落列表
|
||||
diarization: 说话者分离结果
|
||||
|
||||
Returns:
|
||||
带有 speaker 字段的段落列表
|
||||
"""
|
||||
result = []
|
||||
|
||||
for seg in segments:
|
||||
seg_start = seg.get('start', 0)
|
||||
seg_end = seg.get('end', 0)
|
||||
seg_mid = (seg_start + seg_end) / 2
|
||||
|
||||
# 找到包含中点时间的说话者
|
||||
assigned_speaker = None
|
||||
for speaker, turns in diarization.items():
|
||||
for turn in turns:
|
||||
if turn['start'] <= seg_mid <= turn['end']:
|
||||
assigned_speaker = speaker
|
||||
break
|
||||
if assigned_speaker:
|
||||
break
|
||||
|
||||
new_seg = dict(seg)
|
||||
new_seg['speaker'] = assigned_speaker
|
||||
result.append(new_seg)
|
||||
|
||||
return result
|
||||
@@ -0,0 +1,55 @@
|
||||
"""BaseProcessor - Abstract base class for all pipeline processors
|
||||
|
||||
Provides a unified interface that all processors (parse, download, asr, analyze)
|
||||
must follow, enabling consistent state management and error handling.
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
from abc import ABC, abstractmethod
|
||||
from typing import Optional, TYPE_CHECKING
|
||||
|
||||
from kb_workflow.pipeline.config import get_settings
|
||||
from kb_workflow.pipeline.schemas import StepInput, StepOutput
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from kb_workflow.pipeline.state import WorkflowStateManager
|
||||
|
||||
|
||||
class BaseProcessor(ABC):
|
||||
"""Abstract base class for all pipeline processors.
|
||||
|
||||
All processors (ParseProcessor, DownloadProcessor, ASRProcessor, AnalyzeProcessor)
|
||||
must inherit from this class and implement the process() method.
|
||||
|
||||
Attributes:
|
||||
STEP_NAME: Class attribute identifying the processor step name.
|
||||
settings: KBSettings instance from get_settings().
|
||||
state_manager: Optional WorkflowStateManager for state recording.
|
||||
"""
|
||||
|
||||
STEP_NAME: str
|
||||
|
||||
def __init__(self, state_manager: Optional[WorkflowStateManager] = None):
|
||||
"""Initialize the processor.
|
||||
|
||||
Args:
|
||||
state_manager: Optional WorkflowStateManager for recording step state.
|
||||
"""
|
||||
self.settings = get_settings()
|
||||
self.state_manager = state_manager
|
||||
|
||||
@abstractmethod
|
||||
async def process(self, input_data: StepInput) -> StepOutput:
|
||||
"""Process the input data and return the result.
|
||||
|
||||
Args:
|
||||
input_data: Input data for the processor step.
|
||||
|
||||
Returns:
|
||||
StepOutput with processing results.
|
||||
|
||||
Raises:
|
||||
Exception: If processing fails.
|
||||
"""
|
||||
pass
|
||||
@@ -0,0 +1,338 @@
|
||||
"""DownloadProcessor - Content download processor
|
||||
|
||||
Handles metadata extraction and content downloading for all platforms.
|
||||
Works with PlatformAdapter implementations (BilibiliAdapter, ZhihuAdapter).
|
||||
|
||||
Purpose: Unify metadata and content fetching into a single processor
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
import logging
|
||||
from pathlib import Path
|
||||
from typing import Optional, TYPE_CHECKING, Dict, Any
|
||||
|
||||
from kb_workflow.pipeline.schemas import DownloadInput, DownloadOutput
|
||||
from kb_workflow.pipeline.processors.base import BaseProcessor
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from kb_workflow.pipeline.state import WorkflowStateManager
|
||||
from kb_workflow.pipeline.platforms.base import PlatformAdapter
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class DownloadProcessor(BaseProcessor):
|
||||
"""Content download processor.
|
||||
|
||||
Fetches metadata and content from platforms using PlatformAdapter.
|
||||
Supports video, article, answer, and zhuanlan content types.
|
||||
|
||||
Attributes:
|
||||
STEP_NAME: "download" - identifies this processor step
|
||||
"""
|
||||
|
||||
STEP_NAME = "download"
|
||||
|
||||
async def process(self, input_data: DownloadInput) -> DownloadOutput:
|
||||
"""Process the download request.
|
||||
|
||||
Args:
|
||||
input_data: DownloadInput with content_id, platform, content_dir, content_type
|
||||
|
||||
Returns:
|
||||
DownloadOutput with metadata_file, content_file, title, author, etc.
|
||||
"""
|
||||
content_dir = Path(input_data.content_dir)
|
||||
content_dir.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
# Check if step already completed (idempotency)
|
||||
if self.state_manager:
|
||||
state = self.state_manager.load()
|
||||
if state and state.is_step_completed(self.STEP_NAME):
|
||||
logger.info(f"Step {self.STEP_NAME} already completed, skipping")
|
||||
# Return a success output based on existing files
|
||||
return self._get_existing_output(input_data, content_dir)
|
||||
|
||||
# Get appropriate adapter
|
||||
try:
|
||||
adapter = self._get_adapter(input_data.platform)
|
||||
except ValueError as e:
|
||||
return self._record_failure(str(e))
|
||||
|
||||
try:
|
||||
# Fetch metadata (pass source_url for zhihu answers)
|
||||
if input_data.platform == "zhihu":
|
||||
metadata = await adapter.fetch_metadata(
|
||||
input_data.content_id,
|
||||
input_data.content_type,
|
||||
source_url=input_data.source_url
|
||||
)
|
||||
else:
|
||||
metadata = await adapter.fetch_metadata(
|
||||
input_data.content_id,
|
||||
input_data.content_type
|
||||
)
|
||||
|
||||
# Save metadata
|
||||
metadata_filename = adapter.get_output_filename(
|
||||
input_data.content_id,
|
||||
input_data.content_type,
|
||||
"metadata.json"
|
||||
)
|
||||
metadata_path = content_dir / metadata_filename
|
||||
self._save_metadata(metadata, metadata_path)
|
||||
|
||||
# Extract title and author from metadata
|
||||
title = self._extract_title(metadata, input_data.platform)
|
||||
author = self._extract_author(metadata, input_data.platform)
|
||||
duration = self._extract_duration(metadata, input_data.content_type)
|
||||
|
||||
# Fetch content (for text-based content)
|
||||
content_file = ""
|
||||
if input_data.content_type in ["article", "answer", "zhuanlan"]:
|
||||
if input_data.platform == "zhihu":
|
||||
content = await adapter.fetch_content(
|
||||
input_data.content_id,
|
||||
input_data.content_type,
|
||||
source_url=input_data.source_url
|
||||
)
|
||||
else:
|
||||
content = await adapter.fetch_content(
|
||||
input_data.content_id,
|
||||
input_data.content_type
|
||||
)
|
||||
content_filename = adapter.get_output_filename(
|
||||
input_data.content_id,
|
||||
input_data.content_type,
|
||||
"content.md"
|
||||
)
|
||||
content_path = content_dir / content_filename
|
||||
self._save_content(content, content_path)
|
||||
content_file = str(content_path)
|
||||
|
||||
# Download video for ASR (Whisper needs video file)
|
||||
if input_data.content_type == "video" and input_data.platform == "bilibili":
|
||||
download_result = await adapter.download_video(
|
||||
input_data.content_id,
|
||||
content_dir
|
||||
)
|
||||
if download_result["success"]:
|
||||
content_file = download_result["filepath"]
|
||||
logger.info(f"Video downloaded for ASR: {content_file}")
|
||||
else:
|
||||
logger.warning(f"Video download failed: {download_result['message']}")
|
||||
|
||||
# Record success
|
||||
result = DownloadOutput(
|
||||
success=True,
|
||||
metadata_file=str(metadata_path),
|
||||
content_file=content_file,
|
||||
title=title,
|
||||
author=author,
|
||||
duration=duration
|
||||
)
|
||||
|
||||
if self.state_manager:
|
||||
state = self.state_manager.load()
|
||||
if state:
|
||||
summary = f"title={title}, type={input_data.content_type}"
|
||||
state.mark_step_completed(self.STEP_NAME, summary)
|
||||
self.state_manager.save(state)
|
||||
|
||||
return result
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Download failed: {e}")
|
||||
return self._record_failure(str(e))
|
||||
|
||||
def _get_adapter(self, platform: str) -> PlatformAdapter:
|
||||
"""Get the appropriate PlatformAdapter for the platform.
|
||||
|
||||
Args:
|
||||
platform: Platform name ("bilibili" or "zhihu")
|
||||
|
||||
Returns:
|
||||
PlatformAdapter instance
|
||||
|
||||
Raises:
|
||||
ValueError: If platform is not supported
|
||||
"""
|
||||
if platform == "bilibili":
|
||||
from kb_workflow.pipeline.platforms.bilibili import BilibiliAdapter
|
||||
return BilibiliAdapter(state_manager=self.state_manager)
|
||||
elif platform == "zhihu":
|
||||
from kb_workflow.pipeline.platforms.zhihu import ZhihuAdapter
|
||||
return ZhihuAdapter()
|
||||
else:
|
||||
raise ValueError(f"Unsupported platform: {platform}")
|
||||
|
||||
def _save_metadata(self, metadata: Dict[str, Any], path: Path) -> None:
|
||||
"""Save metadata to JSON file.
|
||||
|
||||
Args:
|
||||
metadata: Metadata dictionary
|
||||
path: Output file path
|
||||
"""
|
||||
with open(path, 'w', encoding='utf-8') as f:
|
||||
json.dump(metadata, f, ensure_ascii=False, indent=2)
|
||||
logger.info(f"Saved metadata to {path}")
|
||||
|
||||
def _save_content(self, content: str, path: Path) -> None:
|
||||
"""Save content to Markdown file.
|
||||
|
||||
Args:
|
||||
content: Markdown content string
|
||||
path: Output file path
|
||||
"""
|
||||
with open(path, 'w', encoding='utf-8') as f:
|
||||
f.write(content)
|
||||
logger.info(f"Saved content to {path}")
|
||||
|
||||
def _extract_title(self, metadata: Dict[str, Any], platform: str) -> str:
|
||||
"""Extract title from metadata.
|
||||
|
||||
Args:
|
||||
metadata: Metadata dictionary
|
||||
platform: Platform name
|
||||
|
||||
Returns:
|
||||
Title string
|
||||
"""
|
||||
if platform == "bilibili":
|
||||
# Bilibili metadata structure: {metadata: {title: ...}, ...}
|
||||
if "metadata" in metadata:
|
||||
return metadata["metadata"].get("title", "")
|
||||
return metadata.get("title", "")
|
||||
else:
|
||||
return metadata.get("title", "")
|
||||
|
||||
def _extract_author(self, metadata: Dict[str, Any], platform: str) -> str:
|
||||
"""Extract author from metadata.
|
||||
|
||||
Args:
|
||||
metadata: Metadata dictionary
|
||||
platform: Platform name
|
||||
|
||||
Returns:
|
||||
Author string
|
||||
"""
|
||||
if platform == "bilibili":
|
||||
# Bilibili metadata structure: {metadata: {mid: ...}, ...}
|
||||
if "metadata" in metadata:
|
||||
return str(metadata["metadata"].get("mid", ""))
|
||||
return str(metadata.get("mid", ""))
|
||||
else:
|
||||
return metadata.get("author", "")
|
||||
|
||||
def _extract_duration(self, metadata: Dict[str, Any], content_type: str) -> Optional[int]:
|
||||
"""Extract duration from metadata for video content.
|
||||
|
||||
Args:
|
||||
metadata: Metadata dictionary
|
||||
content_type: Content type
|
||||
|
||||
Returns:
|
||||
Duration in seconds for videos, None otherwise
|
||||
"""
|
||||
if content_type != "video":
|
||||
return None
|
||||
|
||||
# Bilibili video duration
|
||||
if "metadata" in metadata and "duration" in metadata["metadata"]:
|
||||
duration = metadata["metadata"]["duration"]
|
||||
# Handle formatted duration string (e.g., "12:34" or "01:23:45")
|
||||
if isinstance(duration, str):
|
||||
if not duration.strip():
|
||||
return None
|
||||
parts = duration.split(':')
|
||||
try:
|
||||
if len(parts) == 2:
|
||||
return int(parts[0]) * 60 + int(parts[1])
|
||||
elif len(parts) == 3:
|
||||
return int(parts[0]) * 3600 + int(parts[1]) * 60 + int(parts[2])
|
||||
except (ValueError, IndexError):
|
||||
return None
|
||||
elif isinstance(duration, (int, float)):
|
||||
return int(duration)
|
||||
|
||||
return None
|
||||
|
||||
def _record_failure(self, error_message: str) -> DownloadOutput:
|
||||
"""Record failure to state manager and return failure output.
|
||||
|
||||
Args:
|
||||
error_message: Error message
|
||||
|
||||
Returns:
|
||||
DownloadOutput with success=False
|
||||
"""
|
||||
if self.state_manager:
|
||||
state = self.state_manager.load()
|
||||
if state:
|
||||
state.mark_step_failed(self.STEP_NAME, error_message)
|
||||
self.state_manager.save(state)
|
||||
|
||||
return DownloadOutput(
|
||||
success=False,
|
||||
error_message=error_message,
|
||||
metadata_file="",
|
||||
content_file="",
|
||||
title="",
|
||||
author=""
|
||||
)
|
||||
|
||||
def _get_existing_output(
|
||||
self,
|
||||
input_data: DownloadInput,
|
||||
content_dir: Path
|
||||
) -> DownloadOutput:
|
||||
"""Get output from existing files when step was already completed.
|
||||
|
||||
Args:
|
||||
input_data: DownloadInput
|
||||
content_dir: Content directory path
|
||||
|
||||
Returns:
|
||||
DownloadOutput based on existing files
|
||||
"""
|
||||
adapter = self._get_adapter(input_data.platform)
|
||||
|
||||
metadata_filename = adapter.get_output_filename(
|
||||
input_data.content_id,
|
||||
input_data.content_type,
|
||||
"metadata.json"
|
||||
)
|
||||
metadata_path = content_dir / metadata_filename
|
||||
|
||||
# Try to read metadata for title/author
|
||||
title = ""
|
||||
author = ""
|
||||
duration = None
|
||||
if metadata_path.exists():
|
||||
try:
|
||||
with open(metadata_path, 'r', encoding='utf-8') as f:
|
||||
metadata = json.load(f)
|
||||
title = self._extract_title(metadata, input_data.platform)
|
||||
author = self._extract_author(metadata, input_data.platform)
|
||||
duration = self._extract_duration(metadata, input_data.content_type)
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
content_filename = adapter.get_output_filename(
|
||||
input_data.content_id,
|
||||
input_data.content_type,
|
||||
"content.md"
|
||||
)
|
||||
content_path = content_dir / content_filename
|
||||
content_file = str(content_path) if content_path.exists() else ""
|
||||
|
||||
return DownloadOutput(
|
||||
success=True,
|
||||
metadata_file=str(metadata_path),
|
||||
content_file=content_file,
|
||||
title=title,
|
||||
author=author,
|
||||
duration=duration
|
||||
)
|
||||
@@ -0,0 +1,83 @@
|
||||
"""ParseProcessor - URL parsing processor
|
||||
|
||||
Parses URLs from various platforms (Bilibili, Zhihu) and extracts
|
||||
platform, content type, and content ID information.
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
from typing import Optional, TYPE_CHECKING
|
||||
|
||||
from kb_workflow.pipeline.config import get_settings
|
||||
from kb_workflow.pipeline.link_parser import LinkParser
|
||||
from kb_workflow.pipeline.schemas import ParseInput, ParseOutput
|
||||
from kb_workflow.pipeline.processors.base import BaseProcessor
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from kb_workflow.pipeline.state import WorkflowStateManager
|
||||
|
||||
|
||||
class ParseProcessor(BaseProcessor):
|
||||
"""URL parsing processor.
|
||||
|
||||
Parses URLs from Bilibili and Zhihu platforms, extracting
|
||||
platform, content type, and content ID.
|
||||
|
||||
Attributes:
|
||||
STEP_NAME: "parse" - identifies this processor step
|
||||
"""
|
||||
|
||||
STEP_NAME = "parse"
|
||||
|
||||
async def process(self, input_data: ParseInput) -> ParseOutput:
|
||||
"""Parse the URL and extract platform information.
|
||||
|
||||
Args:
|
||||
input_data: ParseInput with URL to parse
|
||||
|
||||
Returns:
|
||||
ParseOutput with parsed information or error message
|
||||
"""
|
||||
parser = LinkParser()
|
||||
parsed = parser.parse(input_data.url)
|
||||
|
||||
if parsed is None:
|
||||
# Unknown URL format
|
||||
result = ParseOutput(
|
||||
success=False,
|
||||
error_message="Unsupported URL format",
|
||||
platform="",
|
||||
content_type="",
|
||||
content_id="",
|
||||
resolved_url="",
|
||||
original_url=input_data.url
|
||||
)
|
||||
|
||||
# Record failure state
|
||||
if self.state_manager:
|
||||
state = self.state_manager.load()
|
||||
if state:
|
||||
state.mark_step_failed(self.STEP_NAME, result.error_message)
|
||||
self.state_manager.save(state)
|
||||
|
||||
return result
|
||||
|
||||
# Successful parse
|
||||
result = ParseOutput(
|
||||
success=True,
|
||||
platform=parsed.platform,
|
||||
content_type=parsed.content_type,
|
||||
content_id=parsed.content_id,
|
||||
resolved_url=parsed.resolved_url,
|
||||
original_url=parsed.original_url
|
||||
)
|
||||
|
||||
# Record success state
|
||||
if self.state_manager:
|
||||
state = self.state_manager.load()
|
||||
if state:
|
||||
summary = f"platform={parsed.platform}, type={parsed.content_type}, id={parsed.content_id}"
|
||||
state.mark_step_completed(self.STEP_NAME, summary)
|
||||
self.state_manager.save(state)
|
||||
|
||||
return result
|
||||
Reference in New Issue
Block a user