Skip to content

Commit 5b98ea9

Browse files
author
fengyikai
committed
星流平台(aicp)：模型API用量接口更新
1 parent 5f594dc commit 5b98ea9

3 files changed

Lines changed: 139 additions & 7 deletions

File tree

‎ksyun/__init__.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,4 +14,4 @@
1414
# limitations under the License.
1515

1616

17-
__version__ = '1.5.8.43'
17+
__version__ = '1.5.8.44'

‎ksyun/client/aicp/v20240612/client.py‎

Lines changed: 71 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -446,7 +446,7 @@ def DescribeApikeys(self, request):
446446

447447

448448
def QueryTokenData(self, request):
449-
"""查询模型API token用量
449+
"""查询模型API token用量（限定半年内的）
450450
:param request: Request instance for QueryTokenData.
451451
:type request: :class:`ksyun.client.aicp.v20240612.models.QueryTokenDataRequest`
452452
"""
@@ -1089,8 +1089,54 @@ def DescribeInferenceEndpoints(self, request):
10891089
raise KsyunSDKException(e.message, e.message)
10901090

10911091

1092+
def EnableEndpointRateLimit(self, request):
1093+
"""开启接入点限流
1094+
:param request: Request instance for EnableEndpointRateLimit.
1095+
:type request: :class:`ksyun.client.aicp.v20240612.models.EnableEndpointRateLimitRequest`
1096+
"""
1097+
try:
1098+
params = request._serialize()
1099+
body = self.call_judge("EnableEndpointRateLimit", params, "application/json")
1100+
response = json.loads(body)
1101+
if "Error" not in response:
1102+
return body
1103+
else:
1104+
code = response["Error"]["Code"]
1105+
message = response["Error"]["Message"]
1106+
req_id = response["RequestId"]
1107+
raise KsyunSDKException(code, message, req_id)
1108+
except Exception as e:
1109+
if isinstance(e, KsyunSDKException):
1110+
raise
1111+
else:
1112+
raise KsyunSDKException(e.message, e.message)
1113+
1114+
1115+
def UpdateInferenceEndpoint(self, request):
1116+
"""创建接入点
1117+
:param request: Request instance for UpdateInferenceEndpoint.
1118+
:type request: :class:`ksyun.client.aicp.v20240612.models.UpdateInferenceEndpointRequest`
1119+
"""
1120+
try:
1121+
params = request._serialize()
1122+
body = self.call_judge("UpdateInferenceEndpoint", params, "application/json")
1123+
response = json.loads(body)
1124+
if "Error" not in response:
1125+
return body
1126+
else:
1127+
code = response["Error"]["Code"]
1128+
message = response["Error"]["Message"]
1129+
req_id = response["RequestId"]
1130+
raise KsyunSDKException(code, message, req_id)
1131+
except Exception as e:
1132+
if isinstance(e, KsyunSDKException):
1133+
raise
1134+
else:
1135+
raise KsyunSDKException(e.message, e.message)
1136+
1137+
10921138
def StartInferenceEndpoint(self, request):
1093-
"""关闭接入点
1139+
"""开启接入点
10941140
:param request: Request instance for StartInferenceEndpoint.
10951141
:type request: :class:`ksyun.client.aicp.v20240612.models.StartInferenceEndpointRequest`
10961142
"""
@@ -1112,6 +1158,29 @@ def StartInferenceEndpoint(self, request):
11121158
raise KsyunSDKException(e.message, e.message)
11131159

11141160

1161+
def StopInferenceEndpoint(self, request):
1162+
"""关闭接入点
1163+
:param request: Request instance for StopInferenceEndpoint.
1164+
:type request: :class:`ksyun.client.aicp.v20240612.models.StopInferenceEndpointRequest`
1165+
"""
1166+
try:
1167+
params = request._serialize()
1168+
body = self.call_judge("StopInferenceEndpoint", params, "application/json")
1169+
response = json.loads(body)
1170+
if "Error" not in response:
1171+
return body
1172+
else:
1173+
code = response["Error"]["Code"]
1174+
message = response["Error"]["Message"]
1175+
req_id = response["RequestId"]
1176+
raise KsyunSDKException(code, message, req_id)
1177+
except Exception as e:
1178+
if isinstance(e, KsyunSDKException):
1179+
raise
1180+
else:
1181+
raise KsyunSDKException(e.message, e.message)
1182+
1183+
11151184
def DeleteInferenceEndpoint(self, request):
11161185
"""删除接入点
11171186
:param request: Request instance for DeleteInferenceEndpoint.

‎ksyun/client/aicp/v20240612/models.py‎

Lines changed: 67 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -721,13 +721,16 @@ def __init__(self):
721721
:type PathPrefix: String
722722
:param DefaultKey: 是否默认只查默认Key
723723
:type PathPrefix: Boolean
724+
:param KeyId: 按apiKeyId 查询
725+
:type PathPrefix: Filter
724726
"""
725727
self.Marker = None
726728
self.MaxResults = None
727729
self.AssociatedModelId = None
728730
self.Status = None
729731
self.Namekeyword = None
730732
self.DefaultKey = None
733+
self.KeyId = None
731734

732735
def _deserialize(self, params):
733736
if params.get("Marker"):
@@ -742,14 +745,16 @@ def _deserialize(self, params):
742745
self.Namekeyword = params.get("Namekeyword")
743746
if params.get("DefaultKey"):
744747
self.DefaultKey = params.get("DefaultKey")
748+
if params.get("KeyId"):
749+
self.KeyId = params.get("KeyId")
745750

746751

747752
class QueryTokenDataRequest(AbstractModel):
748753
"""QueryTokenData请求参数结构体
749754
"""
750755

751756
def __init__(self):
752-
r"""查询模型API token用量
757+
r"""查询模型API token用量（限定半年内的）
753758
:param StartTimestamp: 开始时间，毫秒时间戳，仅支持最近180天内时间。
754759
:type PathPrefix: Int
755760
:param EndTimestamp: 截止时间，毫秒时间戳
@@ -767,6 +772,8 @@ def __init__(self):
767772
:type PathPrefix: Int
768773
:param IsGlobalServer: 是否国际版：false-国内版，true-国际版
769774
:type PathPrefix: Boolean
775+
:param ModelName:
776+
:type PathPrefix: String
770777
"""
771778
self.StartTimestamp = None
772779
self.EndTimestamp = None
@@ -776,6 +783,7 @@ def __init__(self):
776783
self.ReasoningType = None
777784
self.Marker = None
778785
self.IsGlobalServer = None
786+
self.ModelName = None
779787

780788
def _deserialize(self, params):
781789
if params.get("StartTimestamp"):
@@ -794,6 +802,8 @@ def _deserialize(self, params):
794802
self.Marker = params.get("Marker")
795803
if params.get("IsGlobalServer"):
796804
self.IsGlobalServer = params.get("IsGlobalServer")
805+
if params.get("ModelName"):
806+
self.ModelName = params.get("ModelName")
797807

798808

799809
class DisableApikeyStatusRequest(AbstractModel):
@@ -1645,12 +1655,33 @@ def _deserialize(self, params):
16451655
self.Filter = params.get("Filter")
16461656

16471657

1648-
class StartInferenceEndpointRequest(AbstractModel):
1649-
"""StartInferenceEndpoint请求参数结构体
1658+
class EnableEndpointRateLimitRequest(AbstractModel):
1659+
"""EnableEndpointRateLimit请求参数结构体
16501660
"""
16511661

16521662
def __init__(self):
1653-
r"""关闭接入点
1663+
r"""开启接入点限流
1664+
:param EndpointId: 接入点id
1665+
:type PathPrefix: String
1666+
:param RateLimit:
1667+
:type PathPrefix: Object
1668+
"""
1669+
self.EndpointId = None
1670+
self.RateLimit = None
1671+
1672+
def _deserialize(self, params):
1673+
if params.get("EndpointId"):
1674+
self.EndpointId = params.get("EndpointId")
1675+
if params.get("RateLimit"):
1676+
self.RateLimit = params.get("RateLimit")
1677+
1678+
1679+
class UpdateInferenceEndpointRequest(AbstractModel):
1680+
"""UpdateInferenceEndpoint请求参数结构体
1681+
"""
1682+
1683+
def __init__(self):
1684+
r"""创建接入点
16541685
:param EndpointName: 推理接入点名称
16551686
:type PathPrefix: String
16561687
:param ProjectId: 项目制Id
@@ -1681,6 +1712,38 @@ def _deserialize(self, params):
16811712
self.EndpointId = params.get("EndpointId")
16821713

16831714

1715+
class StartInferenceEndpointRequest(AbstractModel):
1716+
"""StartInferenceEndpoint请求参数结构体
1717+
"""
1718+
1719+
def __init__(self):
1720+
r"""开启接入点
1721+
:param EndpointId:
1722+
:type PathPrefix: String
1723+
"""
1724+
self.EndpointId = None
1725+
1726+
def _deserialize(self, params):
1727+
if params.get("EndpointId"):
1728+
self.EndpointId = params.get("EndpointId")
1729+
1730+
1731+
class StopInferenceEndpointRequest(AbstractModel):
1732+
"""StopInferenceEndpoint请求参数结构体
1733+
"""
1734+
1735+
def __init__(self):
1736+
r"""关闭接入点
1737+
:param EndpointId: 推理接入点名称
1738+
:type PathPrefix: String
1739+
"""
1740+
self.EndpointId = None
1741+
1742+
def _deserialize(self, params):
1743+
if params.get("EndpointId"):
1744+
self.EndpointId = params.get("EndpointId")
1745+
1746+
16841747
class DeleteInferenceEndpointRequest(AbstractModel):
16851748
"""DeleteInferenceEndpoint请求参数结构体
16861749
"""

0 commit comments

Comments
 (0)