some fix

Committed-by: xiaolei.zl from Dev container
alibaba · Oct 8, 2024 · 71372fc · 71372fc
1 parent e8acc7d
commit 71372fc
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 15 deletions.
diff --git a/docs/flex/interactive/development/python/VertexApi.md b/docs/flex/interactive/development/python/VertexApi.md
@@ -32,8 +32,8 @@ vertex_request = [
         label="person",
         primary_key_value=8,
         properties=[
-            ModelProperty(name="name", type="string", value="mike"),
-            ModelProperty(name="age", type="integer", value=12),
+            ModelProperty(name="name",value="mike"),
+            ModelProperty(name="age", value=12),
         ],
     ),
 ]
@@ -172,8 +172,8 @@ vertex_request = VertexRequest(
     label="person",
     primary_key_value=1,
     properties=[
-        ModelProperty(name="name", type="string", value="Cindy"),
-        ModelProperty(name="age", type="integer", value=24),
+        ModelProperty(name="name", value="Cindy"),
+        ModelProperty(name="age", value=24),
     ],
 )
 # update vertex

diff --git a/flex/interactive/sdk/examples/python/basic_example.py b/flex/interactive/sdk/examples/python/basic_example.py
@@ -146,8 +146,8 @@ def addVertex(sess: Session, graph_id: str):
             label="person",
             primary_key_value=8,
             properties=[
-                ModelProperty(name="name", type="string", value="mike"),
-                ModelProperty(name="age", type="integer", value=1),
+                ModelProperty(name="name", value="mike"),
+                ModelProperty(name="age", value=1),
             ],
         ),
     ]
@@ -172,8 +172,8 @@ def addVertex(sess: Session, graph_id: str):
 
 
 def updateVertex(sess: Session, graph_id: str):
-    name_property = ModelProperty(name="name", type="string", value="Cindy")
-    age_property = ModelProperty(name="age", type="integer", value=24)
+    name_property = ModelProperty(name="name", value="Cindy")
+    age_property = ModelProperty(name="age", value=24)
     vertex_request = VertexRequest(
         label="person", primary_key_value=1, properties=[name_property, age_property]
     )

diff --git a/flex/interactive/sdk/python/gs_interactive/client/session.py b/flex/interactive/sdk/python/gs_interactive/client/session.py
@@ -44,7 +44,7 @@
                                    JobStatus, QueryRequest, SchemaMapping,
                                    ServiceStatus, StartServiceRequest,
                                    UpdateProcedureRequest, UploadFileResponse,
-                                   VertexEdgeRequest, VertexRequest)
+                                   VertexEdgeRequest, VertexRequest, VertexData)
 
 
 class EdgeInterface(metaclass=ABCMeta):
@@ -128,7 +128,7 @@ def get_vertex(
         primary_key_value: Annotated[
             Any, Field(description="The primary key value of vertex.")
         ],
-    ) -> Result[VertexRequest]:
+    ) -> Result[VertexData]:
         raise NotImplementedError
 
     @abstractmethod
@@ -369,7 +369,7 @@ def get_vertex(
         primary_key_value: Annotated[
             Any, Field(description="The primary key value of vertex.")
         ],
-    ) -> Result[VertexRequest]:
+    ) -> Result[VertexData]:
         """
         Get a vertex from the specified graph with primary key value.
         """

diff --git a/flex/interactive/sdk/python/gs_interactive/tests/test_driver.py b/flex/interactive/sdk/python/gs_interactive/tests/test_driver.py
@@ -550,8 +550,8 @@ def callVertexEdgeQuery(self):
                 label="person",
                 primary_key_value=8,
                 properties=[
-                    ModelProperty(name="name", type="string", value="mike"),
-                    ModelProperty(name="age", type="integer", value=12),
+                    ModelProperty(name="name", value="mike"),
+                    ModelProperty(name="age", value=12),
                 ],
             ),
         ]
@@ -590,8 +590,8 @@ def callVertexEdgeQuery(self):
             label="person",
             primary_key_value=1,
             properties=[
-                ModelProperty(name="name", type="string", value="Cindy"),
-                ModelProperty(name="age", type="integer", value=24),
+                ModelProperty(name="name", value="Cindy"),
+                ModelProperty(name="age", value=24),
             ],
         )
         # update vertex