# Copyright (c) 2023 Baidu, Inc. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from enum import Enum
from typing import Any, Dict
from qianfan.resources import ChatCompletion, Completion, Embedding, Text2Image
[docs]class ActionState(str, Enum):
"""
This class list the key point during an action execution
At default, ActionState should be get through event_handler's
dispatched event.
"""
Preceding = "Preceding"
"""`Preceding` stands for the point before exec"""
Running = "Running"
"""`Running` stands for the point during exec"""
Done = "Done"
"""`Done` stands for the state of doing exec"""
Error = "Error"
"""`Error` stands for the state when errors occur."""
Stopped = "Stopped"
"""`Stopped` stands for the state when stop() is called."""
[docs]class FinetuneStatus(str, Enum):
Unknown = "Unknown"
"""未知状态"""
DatasetLoading = "DatasetLoading"
"""数据集加载中"""
DatasetLoaded = "DatasetLoaded"
"""数据集加载完成"""
DatasetLoadFailed = "DatasetLoadFailed"
"""数据集加载失败"""
DatasetLoadStopped = "DatasetLoadStopped"
"""数据集停止加载"""
TrainCreated = "TrainCreated"
"""任务创建，初始化"""
Training = "Training"
"""训练中 对应训练任务运行时API状态的Running"""
TrainFinished = "TrainFinished"
"""训练完成 对应训练任务运行时API的状态的Done"""
TrainFailed = "TrainFailed"
"""训练任务失败，对应训练任务运行时API的状态的Failed"""
TrainStopped = "TrainStopped"
"""训练任务失败，对应训练任务运行时API的状态的Stop"""
ModelPublishing = "ModelPublishing"
"""模型发布中，对应获取模型运行时的Creating"""
ModelPublishFailed = "ModelPublishFailed"
"""模型发布失败"""
ModelPublished = "ModelPublished"
"""模型发布成功"""
[docs]class ServiceStatus(str, Enum):
Unknown = "Unknown"
"""未知状态"""
Created = "Created"
"""任务创建，初始化"""
Deploying = "Deploying"
"""模型服务发布中"""
Deployed = "Deployed"
"""模型服务发布成功"""
DeployFailed = "DeployFailed"
"""模型服务发布失败"""
DeployStopped = "DeployStopped"
"""服务发布任务停止"""
# model train type -> default train config
ModelTypeMapping = {
"ERNIE-Bot-turbo-0725": "ERNIE-Bot-turbo",
"ERNIE-Bot-turbo-0516": "ERNIE-Bot-turbo",
"ERNIE-Bot-turbo-0704": "ERNIE-Bot-turbo",
"Llama-2-7b": "Llama-2",
"Llama-2-13b": "Llama-2",
"SQLCoder-7B": "SQLCoder",
"ChatGLM2-6B": "ChatGLM2",
"Baichuan2-13B": "Baichuan2-13B",
"BLOOMZ-7B": "BLOOMZ",
}
[docs]class ServiceType(str, Enum):
Chat = "Chat"
"""Corresponding to the `ChatCompletion`"""
Completion = "Completion"
"""Corresponding to the `Completion`"""
Embedding = "Embedding"
"""Corresponding to the `Embedding`"""
Text2Image = "Text2Image"
"""Corresponding to the `Text2Image"""
# service type -> resources class
ServiceTypeResourcesMapping: Dict[ServiceType, Any] = {
ServiceType.Chat: ChatCompletion,
ServiceType.Completion: Completion,
ServiceType.Embedding: Embedding,
ServiceType.Text2Image: Text2Image,
}