content-find-agent-new/content_agent/integrations/mock_platform.py

from __future__ import annotations

from typing import Any

from content_agent.integrations.douyin import RAW_AUTHOR_ID_KEY, RAW_CONTENT_ID_KEY


class MockPlatformClient:
    """Deterministic platform client used before real Douyin access exists."""

    def search(self, query: dict[str, Any]) -> list[dict[str, Any]]:
        search_query_id = query["search_query_id"]
        if search_query_id == "q_001":
            return [
                {
                    "content_discovery_id": "content_001",
                    "search_query_id": search_query_id,
                    "platform": "douyin",
                    "platform_content_id": "7390000000000000000",
                    "platform_content_format": "video",
                    "description": "对比分析类人物故事观察",
                    "platform_author_id": "MS4wLjABAAAA001",
                    "author_display_name": "人物故事观察",
                    "statistics": {"digg_count": 9000, "comment_count": 800, "share_count": 700},
                    "tags": ["#人物故事"],
                    "score": 72,
                    "age_50_plus_level": "medium",
                    "risk_level": "low",
                    "availability": "available",
                    "portrait_available": True,
                    "pattern_recall": "matched",
                    "category_or_element_binding": "matched",
                    "discovery_relation": "mock_pattern_matched",
                    "discovery_start_source": "pattern_itemset",
                    "previous_discovery_step": "search_query_direct",
                    "content_metadata_source": "mock_platform_search",
                    "platform_raw_payload": {
                        RAW_CONTENT_ID_KEY: "7390000000000000000",
                        "author": {RAW_AUTHOR_ID_KEY: "MS4wLjABAAAA001"},
                    },
                },
                {
                    "content_discovery_id": "content_099",
                    "search_query_id": search_query_id,
                    "platform": "douyin",
                    "platform_content_id": "7390000000000000099",
                    "platform_content_format": "video",
                    "description": "缺少画像的低可信发现内容",
                    "platform_author_id": "MS4wLjABAAAA099",
                    "author_display_name": "未知作者",
                    "statistics": {"digg_count": 12, "comment_count": 0, "share_count": 1},
                    "tags": [],
                    "score": None,
                    "age_50_plus_level": "missing",
                    "risk_level": "unknown",
                    "portrait_available": False,
                    "pattern_recall": "matched",
                    "category_or_element_binding": "matched",
                    "discovery_relation": "mock_pattern_matched",
                    "discovery_start_source": "pattern_itemset",
                    "previous_discovery_step": "search_query_direct",
                    "content_metadata_source": "mock_platform_search",
                    "platform_raw_payload": {
                        RAW_CONTENT_ID_KEY: "7390000000000000099",
                        "author": {RAW_AUTHOR_ID_KEY: "MS4wLjABAAAA099"},
                    },
                },
            ]
        return [
            {
                "content_discovery_id": "content_002",
                "search_query_id": search_query_id,
                "platform": "douyin",
                "platform_content_id": "7390000000000000001",
                "platform_content_format": "video",
                "description": "同主题延展发现内容",
                "platform_author_id": "MS4wLjABAAAA002",
                "author_display_name": "主题观察",
                "statistics": {"digg_count": 3000, "comment_count": 20, "share_count": 70},
                "tags": ["#基层治理"],
                "score": 55,
                "age_50_plus_level": "medium",
                "risk_level": "low",
                "availability": "available",
                "portrait_available": True,
                "pattern_recall": "matched",
                "category_or_element_binding": "matched",
                "discovery_relation": "mock_pattern_matched",
                "discovery_start_source": "pattern_itemset",
                "previous_discovery_step": "search_query_direct",
                "content_metadata_source": "mock_platform_search",
                "platform_raw_payload": {
                    RAW_CONTENT_ID_KEY: "7390000000000000001",
                    "author": {RAW_AUTHOR_ID_KEY: "MS4wLjABAAAA002"},
                },
            }
        ]