Merge pull request #11 from nebula-contrib/fix

Fix issue #10 fix vid type issue on INSERT

nebula_carina/models/models.py

@@ -5,7 +5,7 @@
 from typing import Iterable
 
 from nebula3.common.ttypes import Vertex, Tag, Edge
-from pydantic import BaseModel
+from pydantic import BaseModel, StrictStr, StrictInt
 from pydantic.fields import ModelField
 from pydantic.main import ModelMetaclass
 
@@ -186,7 +186,7 @@ class NebulaRecordModel(BaseModel, NebulaConvertableProtocol, metaclass=NebulaRe
 
 class VertexModel(NebulaRecordModel):
 
-    vid: int | str
+    vid: StrictInt | StrictStr
     objects = BaseVertexManager()
 
     @classmethod
@@ -310,8 +310,8 @@ def get_sources(self, edge_type, src_vertex_model, *, distinct=False, limit: Lim
 
 
 class EdgeModel(NebulaRecordModel):
-    src_vid: int | str
-    dst_vid: int | str
+    src_vid: StrictInt | StrictStr
+    dst_vid: StrictInt | StrictStr
     ranking: int = 0
     edge_type_name: str | None  # for view only
     edge_type: EdgeTypeModel  # only one edge type

setup.py

@@ -7,7 +7,7 @@
 
 setup(
     name='nebula-carina',
-    version='0.3.0',
+    version='0.3.1',
     author='Sword Elucidator',
     author_email='[email protected]',
     url='https://github.com/SwordElucidator/nebula-carina',