mirror of
https://github.com/langgenius/dify.git
synced 2024-11-16 19:59:50 +08:00
30 lines
1.1 KiB
Python
30 lines
1.1 KiB
Python
from collections.abc import Sequence
|
|
|
|
import requests
|
|
from yarl import URL
|
|
|
|
from configs import dify_config
|
|
from core.helper.download import download_with_size_limit
|
|
from core.plugin.entities.marketplace import MarketplacePluginDeclaration
|
|
|
|
|
|
def get_plugin_pkg_url(plugin_unique_identifier: str):
|
|
return (URL(str(dify_config.MARKETPLACE_API_URL)) / "api/v1/plugins/download").with_query(
|
|
unique_identifier=plugin_unique_identifier
|
|
)
|
|
|
|
|
|
def download_plugin_pkg(plugin_unique_identifier: str):
|
|
url = str(get_plugin_pkg_url(plugin_unique_identifier))
|
|
return download_with_size_limit(url, dify_config.PLUGIN_MAX_PACKAGE_SIZE)
|
|
|
|
|
|
def batch_fetch_plugin_manifests(plugin_ids: list[str]) -> Sequence[MarketplacePluginDeclaration]:
|
|
if len(plugin_ids) == 0:
|
|
return []
|
|
|
|
url = str(URL(str(dify_config.MARKETPLACE_API_URL)) / "api/v1/plugins/batch")
|
|
response = requests.post(url, json={"plugin_ids": plugin_ids})
|
|
response.raise_for_status()
|
|
return [MarketplacePluginDeclaration(**plugin) for plugin in response.json()["data"]["plugins"]]
|