@@ -98,16 +98,22 @@ def insert(self, table_name: str, records: list[dict]):
9898
9999
100100 ## query data from table
101- def query (self , table_name : str , query_field : str = "" , query_vector : list = None , response_fields : list = None , limit : int = 1 , with_distance : bool = False ):
101+ def query (self , table_name : str , query_field : str = None , query_vector : list = None , response_fields : Optional [ list ] = None , limit : int = 2 , filter : Optional [ str ] = None , with_distance : Optional [ bool ] = False ):
102102 req_url = "{}/data/query" .format (self ._baseurl )
103- req_data = {
104- "table" : table_name ,
105- "queryField" : query_field ,
106- "queryVector" : query_vector ,
107- "response" : response_fields ,
108- "limit" : limit ,
109- "withDistance" : with_distance
110- }
103+ req_data = { "table" : table_name }
104+ if query_field != None :
105+ req_data ["queryField" ] = query_field
106+ if query_vector != None :
107+ req_data ["queryVector" ] = query_vector
108+ if response_fields != None :
109+ req_data ["response" ] = response_fields
110+ if limit != None :
111+ req_data ["limit" ] = limit
112+ if filter != None :
113+ req_data ["filter" ] = filter
114+ if with_distance != None :
115+ req_data ["withDistance" ] = with_distance
116+
111117 res = requests .post (url = req_url , data = json .dumps (req_data ), headers = self ._header )
112118 status_code = res .status_code
113119 body = res .json ()
@@ -155,19 +161,14 @@ def get(self, table_name: str, response_fields: Optional[list] = None, primary_k
155161 primary_keys = ids
156162
157163 req_data = {"table" : table_name }
158-
159164 if response_fields != None :
160165 req_data ["response" ] = response_fields
161-
162166 if primary_keys != None :
163167 req_data ["primaryKeys" ] = primary_keys
164-
165168 if filter != None :
166169 req_data ["filter" ] = filter
167-
168170 if skip != None :
169171 req_data ["skip" ] = skip
170-
171172 if limit != None :
172173 req_data ["limit" ] = limit
173174
0 commit comments