mirror of
https://github.com/OpenBMB/ChatDev.git
synced 2024-12-26 05:15:03 +03:00
86 lines
2.9 KiB
Python
86 lines
2.9 KiB
Python
# =========== Copyright 2023 @ CAMEL-AI.org. 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.
|
|
# =========== Copyright 2023 @ CAMEL-AI.org. All Rights Reserved. ===========
|
|
from enum import Enum
|
|
|
|
|
|
class TaskType(Enum):
|
|
AI_SOCIETY = "ai_society"
|
|
CODE = "code"
|
|
MISALIGNMENT = "misalignment"
|
|
TRANSLATION = "translation"
|
|
EVALUATION = "evaluation"
|
|
SOLUTION_EXTRACTION = "solution_extraction"
|
|
CHATDEV = "chat_dev"
|
|
DEFAULT = "default"
|
|
|
|
|
|
class RoleType(Enum):
|
|
ASSISTANT = "assistant"
|
|
USER = "user"
|
|
CRITIC = "critic"
|
|
EMBODIMENT = "embodiment"
|
|
DEFAULT = "default"
|
|
CHATDEV = "AgentTech"
|
|
CHATDEV_COUNSELOR = "counselor"
|
|
CHATDEV_CEO = "chief executive officer (CEO)"
|
|
CHATDEV_CHRO = "chief human resource officer (CHRO)"
|
|
CHATDEV_CPO = "chief product officer (CPO)"
|
|
CHATDEV_CTO = "chief technology officer (CTO)"
|
|
CHATDEV_PROGRAMMER = "programmer"
|
|
CHATDEV_REVIEWER = "code reviewer"
|
|
CHATDEV_TESTER = "software test engineer"
|
|
CHATDEV_CCO = "chief creative officer (CCO)"
|
|
|
|
|
|
class ModelType(Enum):
|
|
GPT_3_5_TURBO = "gpt-3.5-turbo-16k-0613"
|
|
GPT_3_5_TURBO_NEW = "gpt-3.5-turbo-16k"
|
|
GPT_4 = "gpt-4"
|
|
GPT_4_32k = "gpt-4-32k"
|
|
GPT_4_TURBO = "gpt-4-1106-preview"
|
|
GPT_4_TURBO_V = "gpt-4-1106-vision-preview"
|
|
|
|
STUB = "stub"
|
|
|
|
@property
|
|
def value_for_tiktoken(self):
|
|
return self.value if self.name != "STUB" else "gpt-3.5-turbo-16k-0613"
|
|
|
|
|
|
class PhaseType(Enum):
|
|
REFLECTION = "reflection"
|
|
RECRUITING_CHRO = "recruiting CHRO"
|
|
RECRUITING_CPO = "recruiting CPO"
|
|
RECRUITING_CTO = "recruiting CTO"
|
|
DEMAND_ANALYSIS = "demand analysis"
|
|
CHOOSING_LANGUAGE = "choosing language"
|
|
RECRUITING_PROGRAMMER = "recruiting programmer"
|
|
RECRUITING_REVIEWER = "recruiting reviewer"
|
|
RECRUITING_TESTER = "recruiting software test engineer"
|
|
RECRUITING_CCO = "recruiting chief creative officer"
|
|
CODING = "coding"
|
|
CODING_COMPLETION = "coding completion"
|
|
CODING_AUTOMODE = "coding auto mode"
|
|
REVIEWING_COMMENT = "review comment"
|
|
REVIEWING_MODIFICATION = "code modification after reviewing"
|
|
ERROR_SUMMARY = "error summary"
|
|
MODIFICATION = "code modification"
|
|
ART_ELEMENT_ABSTRACTION = "art element abstraction"
|
|
ART_ELEMENT_INTEGRATION = "art element integration"
|
|
CREATING_ENVIRONMENT_DOCUMENT = "environment document"
|
|
CREATING_USER_MANUAL = "user manual"
|
|
|
|
|
|
__all__ = ["TaskType", "RoleType", "ModelType", "PhaseType"]
|