@@ -83,14 +83,15 @@ async def getVertexAttrs(self, vertexType: str) -> list:
83
83
if logger .level == logging .DEBUG :
84
84
logger .debug ("params: " + self ._locals (locals ()))
85
85
86
- et = await self .getVertexType (vertexType )
86
+ vt = await self .getVertexType (vertexType )
87
87
ret = []
88
88
89
- for at in et ["Attributes" ]:
90
- at ["AttributeType" ]["AttributeType" ] = at ["AttributeType" ].pop ("Name" )
91
- ret .append (
92
- (at ["AttributeName" ], at ["AttributeType" ])
93
- )
89
+ if "Attributes" in vt :
90
+ for at in vt ["Attributes" ]:
91
+ at ["AttributeType" ]["AttributeType" ] = at ["AttributeType" ].pop ("Name" )
92
+ ret .append (
93
+ (at ["AttributeName" ], at ["AttributeType" ])
94
+ )
94
95
95
96
if logger .level == logging .DEBUG :
96
97
logger .debug ("return: " + str (ret ))
@@ -117,21 +118,22 @@ async def getVertexVectors(self, vertexType: str) -> list:
117
118
if logger .level == logging .DEBUG :
118
119
logger .debug ("params: " + self ._locals (locals ()))
119
120
120
- et = await self .getVertexType (vertexType )
121
+ vt = await self .getVertexType (vertexType )
121
122
ret = []
122
123
123
- for vt in et ["EmbeddingAttributes" ]:
124
- ret .append (
125
- (vt ["Name" ], vt )
126
- )
124
+ if "EmbeddingAttributes" in vt :
125
+ for et in vt ["EmbeddingAttributes" ]:
126
+ ret .append (
127
+ (et ["Name" ], et )
128
+ )
127
129
128
130
if logger .level == logging .DEBUG :
129
131
logger .debug ("return: " + str (ret ))
130
132
logger .info ("exit: getVertexVectors" )
131
133
132
134
return ret
133
135
134
- async def getVectorStatus (self , vertexType : str , vectorAttr : str = "" ) -> bool :
136
+ async def getVectorStatus (self , vertexType : str , vectorName : str = "" ) -> bool :
135
137
"""Check the rebuild status of the vertex type or the embedding attribute
136
138
137
139
Args:
0 commit comments