mirror of
https://github.com/langgenius/dify.git
synced 2026-04-25 05:06:15 +08:00
3165f3adbe
- Introduced CachedPresignStorage to cache presigned download URLs, reducing repeated API calls. - Updated AppAssetService to utilize CachedPresignStorage for improved performance in asset download URL generation. - Refactored asset builders and packagers to support the new storage mechanism. - Removed unused AppAssetsAttrsInitializer to streamline initialization processes. - Added unit tests for CachedPresignStorage to ensure functionality and reliability.
40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
from __future__ import annotations
|
|
|
|
from core.app.entities.app_asset_entities import AppAssetFileTree, AssetNodeType
|
|
from core.app_assets.entities import FileAsset
|
|
from core.app_assets.entities.assets import AssetItem
|
|
from core.app_assets.paths import AssetPaths
|
|
|
|
|
|
def tree_to_asset_items(
|
|
tree: AppAssetFileTree,
|
|
tenant_id: str,
|
|
app_id: str,
|
|
) -> list[AssetItem]:
|
|
"""
|
|
Convert AppAssetFileTree to list of FileAsset for packaging.
|
|
|
|
Args:
|
|
tree: The asset file tree to convert
|
|
tenant_id: Tenant ID for storage key generation
|
|
app_id: App ID for storage key generation
|
|
|
|
Returns:
|
|
List of FileAsset items ready for packaging
|
|
"""
|
|
items: list[AssetItem] = []
|
|
for node in tree.nodes:
|
|
if node.node_type == AssetNodeType.FILE:
|
|
path = tree.get_path(node.id)
|
|
storage_key = AssetPaths.draft_file(tenant_id, app_id, node.id)
|
|
items.append(
|
|
FileAsset(
|
|
asset_id=node.id,
|
|
path=path,
|
|
file_name=node.name,
|
|
extension=node.extension or "",
|
|
storage_key=storage_key,
|
|
)
|
|
)
|
|
return items
|