diff --git a/osschat_insert.py b/osschat_insert.py index cb38028..92f7ccc 100644 --- a/osschat_insert.py +++ b/osschat_insert.py @@ -13,13 +13,13 @@ # limitations under the License. from typing import Dict, Optional, Any -from pydantic import BaseModel +from pydantic import BaseModel, Extra from towhee import ops, pipe, AutoPipes, AutoConfig @AutoConfig.register -class OSSChatInsertConfig(BaseModel): +class OSSChatInsertConfig(BaseModel, extra=Extra.allow): """ Config of pipeline """