Support both pydantic 1 and 2

Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>

.github/workflows/codecov.yml

@@ -30,7 +30,9 @@ jobs:
       run: |
         export TOWHEE_WORKER=True
         rm -rf ./coverage.xml
+        pip install coverage pytest pytest-cov pytest-xdist
         pip install -r test_requirements.txt
+        apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends -y ffmpeg libsm6 libxext6
         coverage erase
         coverage run -m pytest
         coverage xml

requirements.txt

@@ -5,4 +5,4 @@ numpy
 twine
 tenacity
 contextvars; python_version <= '3.6'
-pydantic<2
+pydantic

towhee/runtime/check_utils.py

@@ -13,7 +13,8 @@
 # limitations under the License.
 
 from typing import Dict, Any, Set, Tuple, Optional
-from pydantic import BaseModel, constr, validator
+import re
+from pydantic import BaseModel, validator
 
 from towhee.runtime.constants import (
     WindowConst,
@@ -34,15 +35,31 @@ def must_larger_than_zero(cls, v):
 
 
 class TupleForm(BaseModel):
+    """Check schema name
+    """
     data: Optional[Tuple[str, ...]]
-    schema_data: Optional[Tuple[constr(regex='^[a-zA-Z_][a-zA-Z_0-9]*$'), ...]]
+    schema_data: Optional[Tuple[str, ...]]
 
     @validator('*', pre=True)
     def must_be_tuple(cls, v):
         if isinstance(v, str):
             return (v,)
         return v
 
+    @validator('schema_data', pre=True)
+    def check_schema(cls, v):
+        def _check_pattern(v: Tuple[str]):
+            pattern = '^[a-zA-Z_][a-zA-Z_0-9]*$'
+            for item in v:
+                if re.match(pattern, item) is None:
+                    raise ValueError(f'Schema can only consist of letters, numbers, and underscores, the error schema: [{item}]')
+
+        if isinstance(v, str):
+            _check_pattern((v, ))
+        elif isinstance(v, tuple):
+            _check_pattern(v)
+        return v
+
 
 class SetForm(BaseModel):
     data: Set[str]