diff --git a/python/core/component.py b/python/core/component.py index da864415..6ebedc73 100644 --- a/python/core/component.py +++ b/python/core/component.py @@ -24,7 +24,6 @@ from appbuilder.core.utils import ttl_lru_cache from appbuilder.core._client import HTTPClient, AsyncHTTPClient from appbuilder.core.message import Message -from urllib.parse import urlparse, unquote class ComponentArguments(BaseModel): """ @@ -580,15 +579,4 @@ def create_output(cls, type, text, role="tool", name="", visible_scope="all", ra "metrics": metrics }] } - return ComponentOutput(**result) - - @staticmethod - def get_filename_from_url(url): - """从给定URL中提取文件名""" - parsed_url = urlparse(url) - # 提取路径部分 - path = parsed_url.path - # 从路径中获取文件名 - filename = path.split('/')[-1] - # 解码URL编码的文件名 - return unquote(filename) + return ComponentOutput(**result) \ No newline at end of file diff --git a/python/core/components/v2/text_to_image/component.py b/python/core/components/v2/text_to_image/component.py index 07972597..f05cbb18 100644 --- a/python/core/components/v2/text_to_image/component.py +++ b/python/core/components/v2/text_to_image/component.py @@ -21,6 +21,7 @@ from appbuilder.core.component import Component from appbuilder.core.message import Message from appbuilder.core._client import HTTPClient +from appbuilder.core.utils import get_filename_from_url from appbuilder.core._exception import AppBuilderServerException, RiskInputException from appbuilder.core.components.text_to_image.model import Text2ImageSubmitRequest, Text2ImageQueryRequest, \ Text2ImageQueryResponse, Text2ImageSubmitResponse, Text2ImageOutMessage @@ -197,7 +198,7 @@ def tool_eval( yield self.create_output( type='image', text={ - 'filename': self.get_filename_from_url(img_urls[url_number]), + 'filename': get_filename_from_url(img_urls[url_number]), 'url': img_urls[url_number], }, raw_data=raw_data, diff --git a/python/core/components/v2/tree_mind/component.py b/python/core/components/v2/tree_mind/component.py index d3f56e4c..5bf8e674 100644 --- a/python/core/components/v2/tree_mind/component.py +++ b/python/core/components/v2/tree_mind/component.py @@ -20,6 +20,7 @@ from appbuilder.core.message import Message from appbuilder.core._client import HTTPClient from appbuilder.core._exception import * +from appbuilder.core.utils import get_filename_from_url from appbuilder.utils.trace.tracer_wrapper import components_run_trace, components_run_stream_trace from appbuilder.core.components.v2.tree_mind.model import TreeMindRequest, TreeMindResponse @@ -84,17 +85,6 @@ def _post(self, query, **kwargs): img_link = treemind_response.info.downloadInfo.fileInfo.pic return img_link, jump_link - @staticmethod - def get_filename_from_url(url): - """从给定URL中提取文件名""" - parsed_url = urlparse(url) - # 提取路径部分 - path = parsed_url.path - # 从路径中获取文件名 - filename = path.split('/')[-1] - # 解码URL编码的文件名 - return unquote(filename) - @components_run_stream_trace def tool_eval( self, @@ -127,7 +117,7 @@ def tool_eval( img_link_result = self.create_output( type="image", text={ - "filename": self.get_filename_from_url(img_link), + "filename": get_filename_from_url(img_link), "url": img_link }, visible_scope='all', diff --git a/python/core/utils.py b/python/core/utils.py index 1b25c7f1..1c2896b5 100644 --- a/python/core/utils.py +++ b/python/core/utils.py @@ -14,7 +14,7 @@ import time import itertools from typing import List -from urllib.parse import urlparse +from urllib.parse import urlparse, unquote from appbuilder.core._client import HTTPClient from appbuilder.core._exception import TypeNotSupportedException, ModelNotSupportedException from appbuilder.utils.model_util import GetModelListRequest, Models, RemoteModelCollector @@ -57,6 +57,15 @@ def get_model_list(secret_key: str = "", api_type_filter: List[str] = [], is_ava models.append(model.name) return models +def get_filename_from_url(url): + """从给定URL中提取文件名""" + parsed_url = urlparse(url) + # 提取路径部分 + path = parsed_url.path + # 从路径中获取文件名 + filename = path.split('/')[-1] + # 解码URL编码的文件名 + return unquote(filename) def convert_cloudhub_url(client: HTTPClient, qianfan_url: str) -> str: """将千帆url转换为AppBuilder url""" diff --git a/setup.py b/setup.py index 742ad192..cb263a93 100755 --- a/setup.py +++ b/setup.py @@ -54,7 +54,7 @@ setup( name="appbuilder-sdk", # NOTE(chengmo): 修改此版本号时,请注意同时修改 __init__.py 中的 __version__ - version="1.0.1", + version="1.0.1.1", author="dongdaxiang", author_email="dongdaxiang@baidu.com", packages=packages,