From f7ad293330ec36526fc70ea4041ed23e647a40a0 Mon Sep 17 00:00:00 2001 From: baowei04 Date: Sun, 26 Oct 2025 22:46:46 +0800 Subject: [PATCH] support v2 url --- python/core/utils.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/python/core/utils.py b/python/core/utils.py index 661156e9..dc4dddd3 100644 --- a/python/core/utils.py +++ b/python/core/utils.py @@ -87,6 +87,17 @@ def convert_cloudhub_url(client: HTTPClient, qianfan_url: str) -> str: return "{}/{}{}".format(client.gateway, cloudhub_url_prefix, url_suffix) +def convert_cloudhub_url_v2(client: HTTPClient, qianfan_url: str) -> str: + """将千帆url转换为AppBuilder url""" + qianfan_url_prefix = "rpc/2.0/ai_custom/v1/wenxinworkshop" + cloudhub_url_prefix = "rpc/2.0/cloud_hub/v1/bce/wenxinworkshop/ai_custom/v1" + index = str.find(qianfan_url, qianfan_url_prefix) + if index == -1: + raise ValueError(f"url format error, {qianfan_url} is not a valid qianfan url") + url_suffix = qianfan_url[index + len(qianfan_url_prefix):] + return "{}/{}{}".format(client.gateway, cloudhub_url_prefix, url_suffix) + + def is_url(string): """ 判断字符串是否是URL @@ -129,7 +140,8 @@ def get_model_url(self, model_name: str) -> str: origin_name = short_name for model in self.model_list: if model.name == origin_name: - return convert_cloudhub_url(self.client, model.url) + return model.url + # return convert_cloudhub_url(self.client, model.url) raise ModelNotSupportedException(f"Model[{model_name}] not available! " f"You can query available models through: appbuilder.get_model_list()")