将get_filename_from_url实现迁移至utils

baidubce · Jan 17, 2025 · 1cef2cb · 1cef2cb
1 parent 3fd3095
commit 1cef2cb
Show file tree

Hide file tree

Showing 5 changed files with 16 additions and 28 deletions.
diff --git 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)
diff --git 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
@@ -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
@@ -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
@@ -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="[email protected]",
     packages=packages,