tamar-file-hub-client 0.1.2__py3-none-any.whl → 0.1.4__py3-none-any.whl

file_hub_client/services/file/sync_blob_service.py

@@ -9,7 +9,7 @@ from .base_file_service import BaseFileService
 from ...enums import UploadMode
 from ...errors import ValidationError
 from ...rpc import SyncGrpcClient
-from ...schemas import FileUploadResponse, UploadUrlResponse, BatchDownloadUrlResponse, DownloadUrlInfo, GcsUrlInfo, GetGcsUrlResponse, BatchGcsUrlResponse
+from ...schemas import FileUploadResponse, UploadUrlResponse, BatchDownloadUrlResponse, DownloadUrlInfo, GcsUrlInfo, GetGcsUrlResponse, BatchGcsUrlResponse, CompressionStatusResponse, GetVariantsResponse, RecompressionResponse, VariantDownloadUrlResponse, CompressedVariant
 from ...utils import HttpUploader, HttpDownloader, retry_with_backoff, get_file_mime_type
 
 
@@ -37,6 +37,7 @@ class SyncBlobService(BaseFileService):
             mime_type: Optional[str] = None,
             is_temporary: Optional[bool] = False,
             expire_seconds: Optional[int] = None,
+            keep_original_filename: Optional[bool] = False,
             request_id: Optional[str] = None,
             **metadata
     ) -> FileUploadResponse:
@@ -51,6 +52,7 @@ class SyncBlobService(BaseFileService):
             mime_type: MIME类型
             is_temporary: 是否为临时文件
             expire_seconds: 过期秒数
+            keep_original_filename: 是否保留原始文件名（默认False）
             request_id: 请求ID（可选，如果不提供则自动生成）
             **metadata: 额外的元数据（如 x-org-id, x-user-id 等）
 
@@ -84,6 +86,7 @@ class SyncBlobService(BaseFileService):
             mime_type=mime_type or "application/octet-stream",
             is_temporary=is_temporary,
             expire_seconds=expire_seconds,
+            keep_original_filename=keep_original_filename,
         )
 
         if folder_id:
@@ -112,10 +115,12 @@ class SyncBlobService(BaseFileService):
             file_hash: str,
             is_temporary: Optional[bool] = False,
             expire_seconds: Optional[int] = None,
+            keep_original_filename: Optional[bool] = False,
             request_id: Optional[str] = None,
             **metadata
     ) -> FileUploadResponse:
-        """客户端直传实现"""
+        """客户端直传实现（GCS 直传）"""
+        
         # 获取上传URL，以及对应的文件和上传文件信息
         upload_url_resp = self.generate_upload_url(
             file_name=file_name,
@@ -126,6 +131,7 @@ class SyncBlobService(BaseFileService):
             file_hash=file_hash,
             is_temporary=is_temporary,
             expire_seconds=expire_seconds if is_temporary and expire_seconds else None,
+            keep_original_filename=keep_original_filename,
             request_id=request_id,
             **metadata
         )
@@ -137,11 +143,17 @@ class SyncBlobService(BaseFileService):
                 upload_file=upload_url_resp.upload_file
             )
 
+        # 构建HTTP头，包含Content-Type和固定的Cache-Control
+        headers = {
+            "Content-Type": mime_type,
+            "Cache-Control": "public, max-age=86400"  # 24小时公共缓存
+        }
+
         # 上传文件到对象存储
         self.http_uploader.upload(
             url=upload_url_resp.upload_url,
             content=content,
-            headers={"Content-Type": mime_type},
+            headers=headers,
             total_size=file_size,
         )
 
@@ -169,10 +181,12 @@ class SyncBlobService(BaseFileService):
             file_hash: str,
             is_temporary: Optional[bool] = False,
             expire_seconds: Optional[int] = None,
+            keep_original_filename: Optional[bool] = False,
             request_id: Optional[str] = None,
             **metadata
     ) -> FileUploadResponse:
-        """断点续传实现"""
+        """断点续传实现（GCS 直传）"""
+        
         # 获取断点续传URL，以及对应的文件和上传文件信息
         upload_url_resp = self.generate_resumable_upload_url(
             file_name=file_name,
@@ -183,6 +197,7 @@ class SyncBlobService(BaseFileService):
             file_hash=file_hash,
             is_temporary=is_temporary,
             expire_seconds=expire_seconds if is_temporary and expire_seconds else None,
+            keep_original_filename=keep_original_filename,
             request_id=request_id,
             **metadata
         )
@@ -194,6 +209,12 @@ class SyncBlobService(BaseFileService):
                 upload_file=upload_url_resp.upload_file
             )
 
+        # 构建HTTP头，包含Content-Type和固定的Cache-Control
+        headers = {
+            "Content-Type": mime_type,
+            "Cache-Control": "public, max-age=86400"  # 24小时公共缓存
+        }
+
         # 开启断点续传
         upload_url = self.http_uploader.start_resumable_session(
             url=upload_url_resp.upload_url,
@@ -205,7 +226,7 @@ class SyncBlobService(BaseFileService):
         self.http_uploader.upload(
             url=upload_url,
             content=content,
-            headers={"Content-Type": mime_type},
+            headers=headers,
             total_size=file_size,
             is_resume=True
         )
@@ -254,6 +275,7 @@ class SyncBlobService(BaseFileService):
             file_hash: str = None,
             is_temporary: Optional[bool] = False,
             expire_seconds: Optional[int] = None,
+            keep_original_filename: Optional[bool] = False,
             request_id: Optional[str] = None,
             **metadata
     ) -> UploadUrlResponse:
@@ -269,6 +291,8 @@ class SyncBlobService(BaseFileService):
             file_hash: 文件哈希
             is_temporary: 是否为临时文件
             expire_seconds: 过期秒数
+            keep_original_filename: 是否保留原始文件名（默认False）
+            request_id: 请求ID（可选，如果不提供则自动生成）
             **metadata: 额外的元数据（如 x-org-id, x-user-id 等）
 
         Returns:
@@ -286,6 +310,7 @@ class SyncBlobService(BaseFileService):
             file_hash=file_hash,
             is_temporary=is_temporary,
             expire_seconds=expire_seconds if is_temporary and expire_seconds else None,
+            keep_original_filename=keep_original_filename,
         )
 
         if folder_id:
@@ -312,6 +337,7 @@ class SyncBlobService(BaseFileService):
             file_hash: str = None,
             is_temporary: Optional[bool] = False,
             expire_seconds: Optional[int] = None,
+            keep_original_filename: Optional[bool] = False,
             request_id: Optional[str] = None,
             **metadata
     ) -> UploadUrlResponse:
@@ -327,6 +353,7 @@ class SyncBlobService(BaseFileService):
             file_hash: 文件哈希
             is_temporary: 是否为临时文件
             expire_seconds: 过期秒数
+            keep_original_filename: 是否保留原始文件名（默认False）
             request_id: 请求ID（可选，如果不提供则自动生成）
             **metadata: 额外的元数据（如 x-org-id, x-user-id 等）
 
@@ -347,6 +374,7 @@ class SyncBlobService(BaseFileService):
             file_hash=file_hash,
             is_temporary=is_temporary,
             expire_seconds=expire_seconds if is_temporary and expire_seconds else None,
+            keep_original_filename=keep_original_filename,
         )
 
         if folder_id:
@@ -371,6 +399,7 @@ class SyncBlobService(BaseFileService):
             mode: Optional[UploadMode] = UploadMode.NORMAL,
             is_temporary: Optional[bool] = False,
             expire_seconds: Optional[int] = None,
+            keep_original_filename: Optional[bool] = False,
             url: Optional[str] = None,
             file_name: Optional[str] = None,
             request_id: Optional[str] = None,
@@ -385,6 +414,7 @@ class SyncBlobService(BaseFileService):
             mode: 上传模式（NORMAL/DIRECT/RESUMABLE/STREAM）
             is_temporary: 是否为临时文件
             expire_seconds: 过期秒数
+            keep_original_filename: 是否保留原始文件名（默认False）
             url: 要下载并上传的URL（可选）
             file_name: 当使用url参数时指定的文件名（可选）
             request_id: 请求ID（可选，如果不提供则自动生成）
@@ -395,6 +425,8 @@ class SyncBlobService(BaseFileService):
 
         Note:
             必须提供 file 或 url 参数之一
+            
+            Cache-Control 头在 GCS 直传模式（STREAM/RESUMABLE）下自动设置为 "public, max-age=86400"
         """
         # 参数验证：必须提供 file 或 url 之一
         if file is None and not url:
@@ -450,7 +482,7 @@ class SyncBlobService(BaseFileService):
 
         # 根据上传模式执行不同的上传逻辑
         if mode == UploadMode.NORMAL:
-            # 普通上传（通过gRPC）
+            # 普通上传（通过gRPC）- 不需要 Cache-Control
             return self._upload_file(
                 file_name=extracted_file_name,
                 content=content,
@@ -459,12 +491,13 @@ class SyncBlobService(BaseFileService):
                 mime_type=mime_type,
                 is_temporary=is_temporary,
                 expire_seconds=expire_seconds if is_temporary and expire_seconds else None,
+                keep_original_filename=keep_original_filename,
                 request_id=request_id,
                 **metadata
             )
 
         elif mode == UploadMode.STREAM:
-            # 流式上传
+            # 流式上传 - 需要 Cache-Control
             return self._upload_stream(
                 file_name=extracted_file_name,
                 content=content,
@@ -475,12 +508,13 @@ class SyncBlobService(BaseFileService):
                 file_hash=file_hash,
                 is_temporary=is_temporary,
                 expire_seconds=expire_seconds if is_temporary and expire_seconds else None,
+                keep_original_filename=keep_original_filename,
                 request_id=request_id,
                 **metadata
             )
 
         elif mode == UploadMode.RESUMABLE:
-            # 断点续传
+            # 断点续传 - 需要 Cache-Control
             return self._upload_resumable(
                 file_name=extracted_file_name,
                 content=content,
@@ -491,6 +525,7 @@ class SyncBlobService(BaseFileService):
                 file_hash=file_hash,
                 is_temporary=is_temporary,
                 expire_seconds=expire_seconds if is_temporary and expire_seconds else None,
+                keep_original_filename=keep_original_filename,
                 request_id=request_id,
                 **metadata
             )
@@ -566,6 +601,30 @@ class SyncBlobService(BaseFileService):
             chunk_size=chunk_size,
         )
 
+    def download_to_bytes(
+            self,
+            file_id: str,
+            *,
+            request_id: Optional[str] = None,
+            **metadata
+    ) -> bytes:
+        """
+        下载文件并返回字节数据
+
+        Args:
+            file_id: 文件ID
+            request_id: 请求ID（可选，如果不提供则自动生成）
+            **metadata: 额外的元数据
+
+        Returns:
+            文件的字节数据
+        """
+        # 获取下载URL
+        download_url = self.generate_download_url(file_id, request_id=request_id, **metadata)
+
+        # 下载到内存并返回字节数据
+        return self.http_downloader.download(url=download_url, save_path=None)
+
     def batch_generate_download_url(
             self,
             file_ids: List[str],
@@ -688,3 +747,215 @@ class SyncBlobService(BaseFileService):
             ))
 
         return BatchGcsUrlResponse(gcs_urls=gcs_urls)
+
+    def get_compression_status(
+            self,
+            file_id: str,
+            *,
+            request_id: Optional[str] = None,
+            **metadata
+    ) -> CompressionStatusResponse:
+        """
+        获取文件压缩状态
+        
+        Args:
+            file_id: 文件ID
+            request_id: 请求ID，用于追踪
+            **metadata: 额外的gRPC元数据
+            
+        Returns:
+            CompressionStatusResponse: 压缩状态响应
+        """
+        from ...rpc.gen import file_service_pb2, file_service_pb2_grpc
+        
+        stub = self.client.get_stub(file_service_pb2_grpc.FileServiceStub)
+        
+        request = file_service_pb2.CompressionStatusRequest(file_id=file_id)
+        
+        # 构建元数据
+        grpc_metadata = self.client.build_metadata(request_id=request_id, **metadata)
+        
+        response = stub.GetCompressionStatus(request, metadata=grpc_metadata)
+        
+        # 转换压缩变体
+        variants = []
+        for variant in response.variants:
+            variants.append(CompressedVariant(
+                variant_name=variant.variant_name,
+                variant_type=variant.variant_type,
+                media_type=variant.media_type,
+                width=variant.width,
+                height=variant.height,
+                file_size=variant.file_size,
+                format=variant.format,
+                quality=variant.quality if variant.quality else None,
+                duration=variant.duration if variant.duration else None,
+                bitrate=variant.bitrate if variant.bitrate else None,
+                fps=variant.fps if variant.fps else None,
+                compression_ratio=variant.compression_ratio,
+                stored_path=variant.stored_path
+            ))
+        
+        return CompressionStatusResponse(
+            status=response.status,
+            error_message=response.error_message if response.error_message else None,
+            variants=variants
+        )
+
+    def get_compressed_variants(
+            self,
+            file_id: str,
+            *,
+            variant_type: Optional[str] = None,
+            request_id: Optional[str] = None,
+            **metadata
+    ) -> GetVariantsResponse:
+        """
+        获取文件的压缩变体
+        
+        Args:
+            file_id: 文件ID
+            variant_type: 变体类型(image, video, thumbnail)
+            request_id: 请求ID，用于追踪
+            **metadata: 额外的gRPC元数据
+            
+        Returns:
+            GetVariantsResponse: 压缩变体响应
+        """
+        from ...rpc.gen import file_service_pb2, file_service_pb2_grpc
+        
+        stub = self.client.get_stub(file_service_pb2_grpc.FileServiceStub)
+        
+        request = file_service_pb2.GetVariantsRequest(file_id=file_id)
+        if variant_type:
+            request.variant_type = variant_type
+        
+        # 构建元数据
+        grpc_metadata = self.client.build_metadata(request_id=request_id, **metadata)
+        
+        response = stub.GetCompressedVariants(request, metadata=grpc_metadata)
+        
+        # 转换压缩变体
+        variants = []
+        for variant in response.variants:
+            variants.append(CompressedVariant(
+                variant_name=variant.variant_name,
+                variant_type=variant.variant_type,
+                media_type=variant.media_type,
+                width=variant.width,
+                height=variant.height,
+                file_size=variant.file_size,
+                format=variant.format,
+                quality=variant.quality if variant.quality else None,
+                duration=variant.duration if variant.duration else None,
+                bitrate=variant.bitrate if variant.bitrate else None,
+                fps=variant.fps if variant.fps else None,
+                compression_ratio=variant.compression_ratio,
+                stored_path=variant.stored_path
+            ))
+        
+        return GetVariantsResponse(variants=variants)
+
+    def trigger_recompression(
+            self,
+            file_id: str,
+            *,
+            force_reprocess: bool = False,
+            request_id: Optional[str] = None,
+            **metadata
+    ) -> RecompressionResponse:
+        """
+        触发文件重新压缩
+        
+        Args:
+            file_id: 文件ID
+            force_reprocess: 是否强制重新处理
+            request_id: 请求ID，用于追踪
+            **metadata: 额外的gRPC元数据
+            
+        Returns:
+            RecompressionResponse: 重新压缩响应
+        """
+        from ...rpc.gen import file_service_pb2, file_service_pb2_grpc
+        
+        stub = self.client.get_stub(file_service_pb2_grpc.FileServiceStub)
+        
+        request = file_service_pb2.RecompressionRequest(
+            file_id=file_id,
+            force_reprocess=force_reprocess
+        )
+        
+        # 构建元数据
+        grpc_metadata = self.client.build_metadata(request_id=request_id, **metadata)
+        
+        response = stub.TriggerRecompression(request, metadata=grpc_metadata)
+        
+        return RecompressionResponse(
+            task_id=response.task_id,
+            status=response.status
+        )
+
+    def generate_variant_download_url(
+            self,
+            file_id: str,
+            variant_name: str,
+            *,
+            expire_seconds: int = 3600,
+            is_cdn: bool = False,
+            request_id: Optional[str] = None,
+            **metadata
+    ) -> VariantDownloadUrlResponse:
+        """
+        生成变体下载URL
+        
+        Args:
+            file_id: 文件ID
+            variant_name: 变体名称(large/medium/small/thumbnail)
+            expire_seconds: 过期时间（秒）
+            is_cdn: 是否使用CDN
+            request_id: 请求ID，用于追踪
+            **metadata: 额外的gRPC元数据
+            
+        Returns:
+            VariantDownloadUrlResponse: 变体下载URL响应
+        """
+        from ...rpc.gen import file_service_pb2, file_service_pb2_grpc
+        
+        stub = self.client.get_stub(file_service_pb2_grpc.FileServiceStub)
+        
+        request = file_service_pb2.VariantDownloadUrlRequest(
+            file_id=file_id,
+            variant_name=variant_name,
+            expire_seconds=expire_seconds,
+            is_cdn=is_cdn
+        )
+        
+        # 构建元数据
+        grpc_metadata = self.client.build_metadata(request_id=request_id, **metadata)
+        
+        response = stub.GenerateVariantDownloadUrl(request, metadata=grpc_metadata)
+        
+        # 转换变体信息
+        variant_info = None
+        if response.variant_info:
+            variant_info = CompressedVariant(
+                variant_name=response.variant_info.variant_name,
+                variant_type=response.variant_info.variant_type,
+                media_type=response.variant_info.media_type,
+                width=response.variant_info.width,
+                height=response.variant_info.height,
+                file_size=response.variant_info.file_size,
+                format=response.variant_info.format,
+                quality=response.variant_info.quality if response.variant_info.quality else None,
+                duration=response.variant_info.duration if response.variant_info.duration else None,
+                bitrate=response.variant_info.bitrate if response.variant_info.bitrate else None,
+                fps=response.variant_info.fps if response.variant_info.fps else None,
+                compression_ratio=response.variant_info.compression_ratio,
+                stored_path=response.variant_info.stored_path
+            )
+        
+        return VariantDownloadUrlResponse(
+            url=response.url,
+            error=response.error if response.error else None,
+            variant_info=variant_info
+        )

file_hub_client/services/file/sync_file_service.py

@@ -10,6 +10,11 @@ from ...schemas import (
     File,
     FileListResponse,
     GetFileResponse,
+    CompressionStatusResponse,
+    GetVariantsResponse,
+    RecompressionResponse,
+    VariantDownloadUrlResponse,
+    CompressedVariant,
 )
 from ...errors import FileNotFoundError
 
@@ -273,3 +278,215 @@ class SyncFileService(BaseFileService):
         files = [self._convert_file_info(f) for f in response.files]
 
         return FileListResponse(files=files)
+
+    def get_compression_status(
+            self,
+            file_id: str,
+            *,
+            request_id: Optional[str] = None,
+            **metadata
+    ) -> CompressionStatusResponse:
+        """
+        获取文件压缩状态
+        
+        Args:
+            file_id: 文件ID
+            request_id: 请求ID，用于追踪
+            **metadata: 额外的gRPC元数据
+            
+        Returns:
+            CompressionStatusResponse: 压缩状态响应
+        """
+        from ...rpc.gen import file_service_pb2, file_service_pb2_grpc
+        
+        stub = self.client.get_stub(file_service_pb2_grpc.FileServiceStub)
+        
+        request = file_service_pb2.CompressionStatusRequest(file_id=file_id)
+        
+        # 构建元数据
+        grpc_metadata = self.client.build_metadata(request_id=request_id, **metadata)
+        
+        response = stub.GetCompressionStatus(request, metadata=grpc_metadata)
+        
+        # 转换压缩变体
+        variants = []
+        for variant in response.variants:
+            variants.append(CompressedVariant(
+                variant_name=variant.variant_name,
+                variant_type=variant.variant_type,
+                media_type=variant.media_type,
+                width=variant.width,
+                height=variant.height,
+                file_size=variant.file_size,
+                format=variant.format,
+                quality=variant.quality if variant.quality else None,
+                duration=variant.duration if variant.duration else None,
+                bitrate=variant.bitrate if variant.bitrate else None,
+                fps=variant.fps if variant.fps else None,
+                compression_ratio=variant.compression_ratio,
+                stored_path=variant.stored_path
+            ))
+        
+        return CompressionStatusResponse(
+            status=response.status,
+            error_message=response.error_message if response.error_message else None,
+            variants=variants
+        )
+
+    def get_compressed_variants(
+            self,
+            file_id: str,
+            *,
+            variant_type: Optional[str] = None,
+            request_id: Optional[str] = None,
+            **metadata
+    ) -> GetVariantsResponse:
+        """
+        获取文件的压缩变体
+        
+        Args:
+            file_id: 文件ID
+            variant_type: 变体类型(image, video, thumbnail)
+            request_id: 请求ID，用于追踪
+            **metadata: 额外的gRPC元数据
+            
+        Returns:
+            GetVariantsResponse: 压缩变体响应
+        """
+        from ...rpc.gen import file_service_pb2, file_service_pb2_grpc
+        
+        stub = self.client.get_stub(file_service_pb2_grpc.FileServiceStub)
+        
+        request = file_service_pb2.GetVariantsRequest(file_id=file_id)
+        if variant_type:
+            request.variant_type = variant_type
+        
+        # 构建元数据
+        grpc_metadata = self.client.build_metadata(request_id=request_id, **metadata)
+        
+        response = stub.GetCompressedVariants(request, metadata=grpc_metadata)
+        
+        # 转换压缩变体
+        variants = []
+        for variant in response.variants:
+            variants.append(CompressedVariant(
+                variant_name=variant.variant_name,
+                variant_type=variant.variant_type,
+                media_type=variant.media_type,
+                width=variant.width,
+                height=variant.height,
+                file_size=variant.file_size,
+                format=variant.format,
+                quality=variant.quality if variant.quality else None,
+                duration=variant.duration if variant.duration else None,
+                bitrate=variant.bitrate if variant.bitrate else None,
+                fps=variant.fps if variant.fps else None,
+                compression_ratio=variant.compression_ratio,
+                stored_path=variant.stored_path
+            ))
+        
+        return GetVariantsResponse(variants=variants)
+
+    def trigger_recompression(
+            self,
+            file_id: str,
+            *,
+            force_reprocess: bool = False,
+            request_id: Optional[str] = None,
+            **metadata
+    ) -> RecompressionResponse:
+        """
+        触发文件重新压缩
+        
+        Args:
+            file_id: 文件ID
+            force_reprocess: 是否强制重新处理
+            request_id: 请求ID，用于追踪
+            **metadata: 额外的gRPC元数据
+            
+        Returns:
+            RecompressionResponse: 重新压缩响应
+        """
+        from ...rpc.gen import file_service_pb2, file_service_pb2_grpc
+        
+        stub = self.client.get_stub(file_service_pb2_grpc.FileServiceStub)
+        
+        request = file_service_pb2.RecompressionRequest(
+            file_id=file_id,
+            force_reprocess=force_reprocess
+        )
+        
+        # 构建元数据
+        grpc_metadata = self.client.build_metadata(request_id=request_id, **metadata)
+        
+        response = stub.TriggerRecompression(request, metadata=grpc_metadata)
+        
+        return RecompressionResponse(
+            task_id=response.task_id,
+            status=response.status
+        )
+
+    def generate_variant_download_url(
+            self,
+            file_id: str,
+            variant_name: str,
+            *,
+            expire_seconds: int = 3600,
+            is_cdn: bool = False,
+            request_id: Optional[str] = None,
+            **metadata
+    ) -> VariantDownloadUrlResponse:
+        """
+        生成变体下载URL
+        
+        Args:
+            file_id: 文件ID
+            variant_name: 变体名称(large/medium/small/thumbnail)
+            expire_seconds: 过期时间（秒）
+            is_cdn: 是否使用CDN
+            request_id: 请求ID，用于追踪
+            **metadata: 额外的gRPC元数据
+            
+        Returns:
+            VariantDownloadUrlResponse: 变体下载URL响应
+        """
+        from ...rpc.gen import file_service_pb2, file_service_pb2_grpc
+        
+        stub = self.client.get_stub(file_service_pb2_grpc.FileServiceStub)
+        
+        request = file_service_pb2.VariantDownloadUrlRequest(
+            file_id=file_id,
+            variant_name=variant_name,
+            expire_seconds=expire_seconds,
+            is_cdn=is_cdn
+        )
+        
+        # 构建元数据
+        grpc_metadata = self.client.build_metadata(request_id=request_id, **metadata)
+        
+        response = stub.GenerateVariantDownloadUrl(request, metadata=grpc_metadata)
+        
+        # 转换变体信息
+        variant_info = None
+        if response.variant_info:
+            variant_info = CompressedVariant(
+                variant_name=response.variant_info.variant_name,
+                variant_type=response.variant_info.variant_type,
+                media_type=response.variant_info.media_type,
+                width=response.variant_info.width,
+                height=response.variant_info.height,
+                file_size=response.variant_info.file_size,
+                format=response.variant_info.format,
+                quality=response.variant_info.quality if response.variant_info.quality else None,
+                duration=response.variant_info.duration if response.variant_info.duration else None,
+                bitrate=response.variant_info.bitrate if response.variant_info.bitrate else None,
+                fps=response.variant_info.fps if response.variant_info.fps else None,
+                compression_ratio=response.variant_info.compression_ratio,
+                stored_path=response.variant_info.stored_path
+            )
+        
+        return VariantDownloadUrlResponse(
+            url=response.url,
+            error=response.error if response.error else None,
+            variant_info=variant_info
+        )