refactor: use enum auto

promptulate/agents/assistant_agent/agent.py

@@ -1,4 +1,4 @@
-from enum import Enum
+from enum import Enum, auto
 from typing import Callable, Dict, List, Optional, TypedDict
 
 from typing_extensions import NotRequired
@@ -17,9 +17,9 @@
 
 
 class StepTypes(str, Enum):
-    PLAN = "plan"
-    EXECUTE = "execute"
-    REVISE = "revise"
+    PLAN = auto()
+    EXECUTE = auto()
+    REVISE = auto()
 
 
 class AdditionalProperties(TypedDict):

promptulate/agents/assistant_agent/schema.py

@@ -1,14 +1,14 @@
-from enum import Enum
+from enum import Enum, auto
 from typing import List, Optional
 
 from promptulate.pydantic_v1 import BaseModel, Field
 
 
 class TaskStatus(str, Enum):
-    TODO = "todo"
-    DONE = "done"
-    ERROR = "error"
-    DISCARDED = "discarded"
+    TODO = auto()
+    DONE = auto()
+    ERROR = auto()
+    DISCARDED = auto()
 
 
 class Task(BaseModel):

promptulate/beta/agents/assistant_agent/agent.py

@@ -1,4 +1,4 @@
-from enum import Enum
+from enum import Enum, auto
 from typing import Callable, Dict, List, Optional, TypedDict
 
 from typing_extensions import NotRequired
@@ -17,9 +17,9 @@
 
 
 class StepTypes(str, Enum):
-    PLAN = "plan"
-    EXECUTE = "execute"
-    REVISE = "revise"
+    PLAN = auto()
+    EXECUTE = auto()
+    REVISE = auto()
 
 
 class AdditionalProperties(TypedDict):

promptulate/beta/agents/assistant_agent/schema.py

@@ -1,14 +1,14 @@
-from enum import Enum
+from enum import Enum, auto
 from typing import List, Optional
 
 from promptulate.pydantic_v1 import BaseModel, Field
 
 
 class TaskStatus(str, Enum):
-    TODO = "todo"
-    DONE = "done"
-    ERROR = "error"
-    DISCARDED = "discarded"
+    TODO = auto()
+    DONE = auto()
+    ERROR = auto()
+    DISCARDED = auto()
 
 
 class Task(BaseModel):

promptulate/schema.py

@@ -1,7 +1,7 @@
 import warnings
 from abc import abstractmethod
 from datetime import datetime
-from enum import Enum
+from enum import Enum, auto
 from typing import Any, Callable, Dict, Iterator, List, Optional, Union
 
 from promptulate.pydantic_v1 import BaseModel, Field
@@ -180,11 +180,11 @@ def type(self) -> str:
 class LLMType(str, Enum):
     """All LLM type here"""
 
-    OpenAI = "OpenAI"
-    ChatOpenAI = "ChatOpenAI"
-    ErnieBot = "ErnieBot"
-    QianFan = "QianFan"
-    ZhiPu = "ZhiPu"
+    OpenAI = auto()
+    ChatOpenAI = auto()
+    ErnieBot = auto()
+    QianFan = auto()
+    ZhiPu = auto()
 
 
 class MessageSet:

promptulate/uacp/schema.py

@@ -1,13 +1,13 @@
-from enum import Enum
+from enum import Enum, auto
 from typing import Any, Dict, List, Optional
 
 from promptulate.pydantic_v1 import BaseModel, Field
 
 
 class Status(Enum):
-    created = "created"
-    running = "running"
-    completed = "completed"
+    created = auto()
+    running = auto()
+    completed = auto()
 
 
 class StepOutput(BaseModel):

pyproject.toml

@@ -9,7 +9,7 @@ name = "promptulate"
 readme = "README.md"
 homepage = "https://github.com/Undertone0809/promptulate"
 repository = "https://github.com/Undertone0809/promptulate"
-version = "1.18.3"
+version = "1.18.4"
 keywords = [
     "promptulate",
     "pne",