redfox-python-sdk 0.2.2__py3-none-any.whl → 0.3.0__py3-none-any.whl

redfox/client.py

@@ -4,7 +4,7 @@ import os
 import time
 import random
 import logging
-from typing import Optional, Dict, Any
+from typing import Optional, Dict, Any, Union, IO, Tuple
 
 import httpx
 
@@ -17,10 +17,20 @@ from .endpoints.wechat import WechatAPI
 from .endpoints.bilibili import BilibiliAPI
 from .endpoints.toutiao import ToutiaoAPI
 from .endpoints.tiktok import TikTokAPI
+from .endpoints.kuaishou import KuaishouAPI
+from .endpoints.wechat_channels import WechatChannelsAPI
+from .endpoints.youtube import YouTubeAPI
+from .endpoints.twitter import TwitterAPI
+from .endpoints.instagram import InstagramAPI
+from .endpoints.dongchedi import DongchediAPI
+from .endpoints.yiche import YicheAPI
+from .endpoints.autohome import AutohomeAPI
+from .endpoints.hotspot import HotspotAPI
 from .endpoints.gpt_image import GPTImageAPI
 from .endpoints.doubao_image import DoubaoImageAPI
 from .endpoints.doubao_video import DoubaoVideoAPI
 from .endpoints.ai_search import AISearchAPI
+from .endpoints.tools import ToolsAPI
 
 logger = logging.getLogger("redfox")
 
@@ -104,6 +114,27 @@ class _RequestMixin:
             return RedFoxAPIError("网络连接失败，请检查网络")
         return RedFoxAPIError(f"请求异常: {str(exc)}")
 
+    @staticmethod
+    def _read_file(file, field_name: str = "file") -> Tuple[str, bytes]:
+        """读取上传文件内容，返回 (文件名, 字节内容)
+
+        支持：文件路径、(filename, bytes) 元组、文件对象、bytes
+        """
+        if isinstance(file, tuple):
+            return file[0], file[1]
+        if isinstance(file, (str, os.PathLike)):
+            path = os.fspath(file)
+            with open(path, "rb") as f:
+                return os.path.basename(path), f.read()
+        if isinstance(file, bytes):
+            return "upload.bin", file
+        if hasattr(file, "read"):
+            content = file.read()
+            name = getattr(file, "name", None)
+            name = os.path.basename(name) if name else "upload.bin"
+            return name, content
+        raise TypeError(f"不支持的文件类型: {type(file)}")
+
 
 class RedFoxClient(_RequestMixin):
     """
@@ -168,7 +199,6 @@ class RedFoxClient(_RequestMixin):
         self._client = httpx.Client(
             timeout=httpx.Timeout(timeout),
             headers={
-                "Content-Type": "application/json",
                 "REDFOX_API_KEY": self.api_key,
             },
         )
@@ -180,10 +210,20 @@ class RedFoxClient(_RequestMixin):
         self.bilibili = BilibiliAPI(self)
         self.toutiao = ToutiaoAPI(self)
         self.tiktok = TikTokAPI(self)
+        self.kuaishou = KuaishouAPI(self)
+        self.wechat_channels = WechatChannelsAPI(self)
+        self.youtube = YouTubeAPI(self)
+        self.twitter = TwitterAPI(self)
+        self.instagram = InstagramAPI(self)
+        self.dongchedi = DongchediAPI(self)
+        self.yiche = YicheAPI(self)
+        self.autohome = AutohomeAPI(self)
+        self.hotspot = HotspotAPI(self)
         self.gpt_image = GPTImageAPI(self)
         self.doubao_image = DoubaoImageAPI(self)
         self.doubao_video = DoubaoVideoAPI(self)
         self.ai_search = AISearchAPI(self)
+        self.tools = ToolsAPI(self)
 
     def request(
         self,
@@ -253,6 +293,59 @@ class RedFoxClient(_RequestMixin):
         """发送 GET 请求"""
         return self.request("GET", path, params=params)
 
+    def upload(
+        self,
+        path: str,
+        file,
+        data: Optional[Dict[str, Any]] = None,
+        field_name: str = "file",
+    ) -> dict:
+        """
+        上传文件（multipart/form-data，含自动重试）
+
+        :param path: API 路径
+        :param file: 文件，支持文件路径 / (filename, bytes) 元组 / 文件对象 / bytes
+        :param data: 附加表单字段
+        :param field_name: 文件表单字段名，默认 "file"
+        :return: API 响应 data 字段内容
+        """
+        url = f"{self.base_url}{path}"
+        filename, content = self._read_file(file, field_name)
+        if data:
+            data = {k: v for k, v in data.items() if v is not None}
+        last_exception = None
+
+        for attempt in range(self.max_retries + 1):
+            try:
+                response = self._client.post(
+                    url,
+                    files={field_name: (filename, content)},
+                    data=data,
+                )
+
+                if response.status_code in RETRYABLE_STATUSES and attempt < self.max_retries:
+                    delay = _compute_delay(attempt, self.backoff_factor)
+                    logger.debug(
+                        "上传收到 %d，第 %d/%d 次重试，等待 %.1fs",
+                        response.status_code, attempt + 1, self.max_retries, delay,
+                    )
+                    time.sleep(delay)
+                    continue
+
+                return self._handle_response(response)
+
+            except RETRYABLE_EXCEPTIONS as exc:
+                last_exception = exc
+                if attempt < self.max_retries:
+                    time.sleep(_compute_delay(attempt, self.backoff_factor))
+                else:
+                    raise self._map_exception(exc)
+
+            except (RedFoxAPIError, RedFoxAuthError, RedFoxRateLimitError):
+                raise
+
+        raise self._map_exception(last_exception) if last_exception else RedFoxAPIError("上传失败")
+
     def close(self):
         """关闭 HTTP 连接"""
         self._client.close()
@@ -320,7 +413,6 @@ class AsyncRedFoxClient(_RequestMixin):
         self._client = httpx.AsyncClient(
             timeout=httpx.Timeout(timeout),
             headers={
-                "Content-Type": "application/json",
                 "REDFOX_API_KEY": self.api_key,
             },
         )
@@ -332,10 +424,20 @@ class AsyncRedFoxClient(_RequestMixin):
         self.bilibili = BilibiliAPI(self)
         self.toutiao = ToutiaoAPI(self)
         self.tiktok = TikTokAPI(self)
+        self.kuaishou = KuaishouAPI(self)
+        self.wechat_channels = WechatChannelsAPI(self)
+        self.youtube = YouTubeAPI(self)
+        self.twitter = TwitterAPI(self)
+        self.instagram = InstagramAPI(self)
+        self.dongchedi = DongchediAPI(self)
+        self.yiche = YicheAPI(self)
+        self.autohome = AutohomeAPI(self)
+        self.hotspot = HotspotAPI(self)
         self.gpt_image = GPTImageAPI(self)
         self.doubao_image = DoubaoImageAPI(self)
         self.doubao_video = DoubaoVideoAPI(self)
         self.ai_search = AISearchAPI(self)
+        self.tools = ToolsAPI(self)
 
     async def request(
         self,
@@ -401,6 +503,59 @@ class AsyncRedFoxClient(_RequestMixin):
         """异步发送 GET 请求"""
         return await self.request("GET", path, params=params)
 
+    async def upload(
+        self,
+        path: str,
+        file,
+        data: Optional[Dict[str, Any]] = None,
+        field_name: str = "file",
+    ) -> dict:
+        """
+        异步上传文件（multipart/form-data，含自动重试）
+
+        :param path: API 路径
+        :param file: 文件，支持文件路径 / (filename, bytes) 元组 / 文件对象 / bytes
+        :param data: 附加表单字段
+        :param field_name: 文件表单字段名，默认 "file"
+        :return: API 响应 data 字段内容
+        """
+        url = f"{self.base_url}{path}"
+        filename, content = self._read_file(file, field_name)
+        if data:
+            data = {k: v for k, v in data.items() if v is not None}
+        last_exception = None
+
+        for attempt in range(self.max_retries + 1):
+            try:
+                response = await self._client.post(
+                    url,
+                    files={field_name: (filename, content)},
+                    data=data,
+                )
+
+                if response.status_code in RETRYABLE_STATUSES and attempt < self.max_retries:
+                    delay = _compute_delay(attempt, self.backoff_factor)
+                    logger.debug(
+                        "上传收到 %d，第 %d/%d 次重试，等待 %.1fs",
+                        response.status_code, attempt + 1, self.max_retries, delay,
+                    )
+                    await self._async_sleep(delay)
+                    continue
+
+                return self._handle_response(response)
+
+            except RETRYABLE_EXCEPTIONS as exc:
+                last_exception = exc
+                if attempt < self.max_retries:
+                    await self._async_sleep(_compute_delay(attempt, self.backoff_factor))
+                else:
+                    raise self._map_exception(exc)
+
+            except (RedFoxAPIError, RedFoxAuthError, RedFoxRateLimitError):
+                raise
+
+        raise self._map_exception(last_exception) if last_exception else RedFoxAPIError("上传失败")
+
     async def _async_sleep(self, seconds: float):
         """异步延时"""
         import asyncio

redfox/endpoints/ai_search.py

@@ -7,7 +7,7 @@ class AISearchAPI:
     """
     AI 搜索工具 API 集合
 
-    支持 Kimi、豆包、Deepseek 三种 AI 搜索引擎。
+    支持 Kimi、豆包、Deepseek、元宝、千问、百度六种 AI 搜索引擎。
     每种引擎均为异步模式：先 submit 提交任务，再 result 查询结果。
     """
 
@@ -85,3 +85,75 @@ class AISearchAPI:
         return self._client.post(
             "/story/api/deepSearch/dsResult", data={"taskId": task_id}
         )
+
+    # ─── 元宝 ───────────────────────────────────────────────
+
+    def yuanbao_submit(self, inquiry_text: str) -> dict:
+        """
+        元宝纯文字搜索 - 提交任务
+
+        :param inquiry_text: 搜索文本（必填）
+        :return: 包含 taskId 的字典
+        """
+        return self._client.post(
+            "/story/api/deepSearch/ybSubmit", data={"inquiryText": inquiry_text}
+        )
+
+    def yuanbao_result(self, task_id: str) -> dict:
+        """
+        元宝纯文字搜索 - 查询任务结果
+
+        :param task_id: 任务 ID（由 yuanbao_submit 返回）
+        :return: 任务结果字典
+        """
+        return self._client.post(
+            "/story/api/deepSearch/ybResult", data={"taskId": task_id}
+        )
+
+    # ─── 千问 ───────────────────────────────────────────────
+
+    def qianwen_submit(self, inquiry_text: str) -> dict:
+        """
+        千问纯文字搜索 - 提交任务
+
+        :param inquiry_text: 搜索文本（必填）
+        :return: 包含 taskId 的字典
+        """
+        return self._client.post(
+            "/story/api/deepSearch/qwSubmit", data={"inquiryText": inquiry_text}
+        )
+
+    def qianwen_result(self, task_id: str) -> dict:
+        """
+        千问纯文字搜索 - 查询任务结果
+
+        :param task_id: 任务 ID（由 qianwen_submit 返回）
+        :return: 任务结果字典
+        """
+        return self._client.post(
+            "/story/api/deepSearch/qwResult", data={"taskId": task_id}
+        )
+
+    # ─── 百度 ───────────────────────────────────────────────
+
+    def baidu_submit(self, inquiry_text: str) -> dict:
+        """
+        百度纯文字搜索 - 提交任务
+
+        :param inquiry_text: 搜索文本（必填）
+        :return: 包含 taskId 的字典
+        """
+        return self._client.post(
+            "/story/api/deepSearch/bdSubmit", data={"inquiryText": inquiry_text}
+        )
+
+    def baidu_result(self, task_id: str) -> dict:
+        """
+        百度纯文字搜索 - 查询任务结果
+
+        :param task_id: 任务 ID（由 baidu_submit 返回）
+        :return: 任务结果字典
+        """
+        return self._client.post(
+            "/story/api/deepSearch/bdResult", data={"taskId": task_id}
+        )

redfox/endpoints/autohome.py

@@ -0,0 +1,77 @@
+"""汽车之家平台 API 端点"""
+
+from typing import Optional, Dict, Any
+
+
+class AutohomeAPI:
+    """
+    汽车之家平台 API 集合
+
+    包含汽车之家作品搜索、文章/视频详情、账号作品列表等接口。
+    """
+
+    def __init__(self, client):
+        self._client = client
+
+    def search_works(
+        self,
+        keyword: str,
+        offset: str = "0",
+        page: str = "1",
+        source_type: str = "video",
+    ) -> dict:
+        """
+        汽车之家关键词搜索作品
+
+        :param keyword: 搜索关键词（必填）
+        :param offset: 翻页偏移量，从 0 开始，翻页 +10
+        :param page: 页码，从 1 开始，翻页 +1
+        :param source_type: 内容类型：club / article / video
+        :return: 搜索结果字典
+        """
+        data: Dict[str, Any] = {
+            "keyword": keyword,
+            "offset": str(offset),
+            "page": str(page),
+            "sourceType": source_type,
+        }
+        return self._client.post("/story/api/autohome/searchWork", data=data)
+
+    def get_article(self, work_id: str, page: int = 0) -> dict:
+        """
+        汽车之家文章详情（车家号）
+
+        :param work_id: 作品 ID（必填）
+        :param page: 页码，默认 0
+        :return: 文章详情字典
+        """
+        return self._client.post(
+            "/story/api/autohome/articleDetail",
+            data={"workId": work_id, "page": page},
+        )
+
+    def get_video(self, video_id: str, video_type: int) -> dict:
+        """
+        汽车之家视频详情（原创账号+车家号）
+
+        :param video_id: 视频 ID（必填）
+        :param video_type: 视频类型：0=原创账号，4=车家号（必填）
+        :return: 视频详情字典
+        """
+        return self._client.post(
+            "/story/api/autohome/videoDetail",
+            data={"videoId": video_id, "videoType": video_type},
+        )
+
+    def get_user_works(self, author_id: str, page: int = 0) -> dict:
+        """
+        汽车之家作品列表（原创账号）
+
+        :param author_id: 作者 ID（必填）
+        :param page: 页码，默认 0
+        :return: 作品列表字典
+        """
+        return self._client.post(
+            "/story/api/autohome/workList",
+            data={"authorId": author_id, "page": page},
+        )

redfox/endpoints/bilibili.py

@@ -121,3 +121,38 @@ class BilibiliAPI:
         if order is not None:
             data["order"] = order
         return self._client.post("/story/api/bili/data/accountWorkList", data=data)
+
+    # ─── 音频 / 提文案 ─────────────────────────────────────
+
+    def get_audio(self, url: str) -> dict:
+        """
+        获取哔哩哔哩音频地址
+
+        :param url: 作品链接（必填）
+        :return: 音频地址字典
+        """
+        return self._client.post(
+            "/story/api/parseWork/audioExtract/bilibili", data={"url": url}
+        )
+
+    def transcript_submit(self, url: str) -> dict:
+        """
+        哔哩哔哩链接提文案 - 提交任务
+
+        :param url: 作品地址（必填）
+        :return: 包含 taskId 的字典
+        """
+        return self._client.post(
+            "/story/api/parseWork/bilibiliSubtitle/submit", data={"url": url}
+        )
+
+    def transcript_result(self, task_id: str) -> dict:
+        """
+        哔哩哔哩链接提文案 - 查询结果
+
+        :param task_id: 任务 ID（由 transcript_submit 返回）
+        :return: 文案提取结果字典
+        """
+        return self._client.post(
+            "/story/api/parseWork/bilibiliSubtitle/result", data={"taskId": task_id}
+        )

redfox/endpoints/dongchedi.py

@@ -0,0 +1,74 @@
+"""懂车帝平台 API 端点"""
+
+from typing import Optional, Dict, Any
+
+
+class DongchediAPI:
+    """
+    懂车帝平台 API 集合
+
+    包含懂车帝作品搜索、作品详情、用户作品列表、用户搜索等接口。
+    """
+
+    def __init__(self, client):
+        self._client = client
+
+    def search_works(
+        self,
+        keyword: str,
+        offset: str = "0",
+        source_type: str = "1",
+    ) -> dict:
+        """
+        懂车帝关键词搜索作品
+
+        :param keyword: 搜索关键词（必填）
+        :param offset: 翻页偏移量，从 0 开始，翻页 +20
+        :param source_type: 内容类型：1=综合，2=视频
+        :return: 搜索结果字典
+        """
+        data: Dict[str, Any] = {
+            "keyword": keyword,
+            "offset": str(offset),
+            "sourceType": str(source_type),
+        }
+        return self._client.post("/story/api/dongchedi/searchWork", data=data)
+
+    def get_work(self, work_id: str, work_type: str) -> dict:
+        """
+        懂车帝作品详情
+
+        :param work_id: 懂车帝作品 ID（必填）
+        :param work_type: 作品类型：video=视频，article=图文（必填）
+        :return: 作品详情字典
+        """
+        return self._client.post(
+            "/story/api/dongchedi/workDetail",
+            data={"workId": work_id, "workType": work_type},
+        )
+
+    def get_user_works(self, user_id: str, cursor: int = 0) -> dict:
+        """
+        懂车帝用户作品列表
+
+        :param user_id: 用户 ID（必填）
+        :param cursor: 游标，首次传 0，翻页传上一页返回的 cursor 值
+        :return: 作品列表字典
+        """
+        return self._client.post(
+            "/story/api/dongchedi/workList",
+            data={"userId": user_id, "cursor": cursor},
+        )
+
+    def search_users(self, keyword: str, offset: int = 0) -> dict:
+        """
+        懂车帝用户搜索
+
+        :param keyword: 搜索关键词（必填）
+        :param offset: 偏移量，从 0 开始
+        :return: 搜索结果字典
+        """
+        return self._client.post(
+            "/story/api/dongchedi/searchUser",
+            data={"keyword": keyword, "offset": offset},
+        )