Add missing type annotation

Author: richard-epsilla

pyepsilla/cloud/client.py

@@ -104,7 +104,7 @@ def insert(self, table_name: str, records: list[dict]):
 
 
     ## query data from table
-    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):
+    def query(self, table_name: str, query_field: str = None, query_vector: Union[list,dict] = None, response_fields: Optional[list] = None, limit: int = 2, filter: Optional[str] = None, with_distance: Optional[bool] = False):
         req_url = "{}/data/query".format(self._baseurl)
         req_data = { "table": table_name }
         if query_field != None:

pyepsilla/enterprise/client.py

@@ -221,7 +221,7 @@ def query(
         self,
         table_name: str,
         query_field: str = None,
-        query_vector: list = None,
+        query_vector: Union[list,dict] = None,
         response_fields: Optional[list] = None,
         limit: int = 2,
         filter: Optional[str] = None,

pyepsilla/vectordb/version.py

@@ -1 +1 @@
-__version__ = "0.2.0"
+__version__ = "0.2.1"