From 59e421669209e13d69108c1fd41895d4457e70b8 Mon Sep 17 00:00:00 2001 From: chyroc Date: Tue, 24 Sep 2024 13:08:23 +0800 Subject: [PATCH] fix StrEnum --- cozepy/conversation.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/cozepy/conversation.py b/cozepy/conversation.py index e9b5d7e..a682d75 100644 --- a/cozepy/conversation.py +++ b/cozepy/conversation.py @@ -1,4 +1,4 @@ -from enum import StrEnum, Enum +from enum import Enum from typing import Dict, List, Optional from .auth import Auth @@ -13,7 +13,7 @@ class MessageRole(str, Enum): assistant = "assistant" -class MessageType(StrEnum): +class MessageType(str, Enum): # User input content. # 用户输入内容。 question = "question" @@ -37,7 +37,7 @@ class MessageType(StrEnum): verbose = "verbose" -class MessageContentType(StrEnum): +class MessageContentType(str, Enum): # Text. # 文本。 text = "text" @@ -49,7 +49,7 @@ class MessageContentType(StrEnum): card = "card" -class MessageObjectStringType(StrEnum): +class MessageObjectStringType(str, Enum): """ The content type of the multimodal message. """