feat auto update sdk

Author: liyan.90210

Changelog

@@ -1,5 +1,8 @@
 ### Change log
 
+2024-11-28 Bumped to version v1.0.163
+- Updated apis for viking_db/visual/tls
+
 2024-11-21 Bumped to version v1.0.162
 - Updated apis for viking_db
 

volcengine/__init__.py

@@ -1,2 +1,2 @@
 # coding:utf-8
-VERSION='v1.0.162'
+VERSION='v1.0.163'

volcengine/example/tls/example_test.py

@@ -0,0 +1,22 @@
+import os
+import unittest
+
+from volcengine.tls.TLSService import TLSService
+
+class TestTLSService(unittest.TestCase):
+
+    def test_init(self):
+        endpoint = os.environ["VOLCENGINE_ENDPOINT"]
+        region = os.environ["VOLCENGINE_REGION"]
+        access_key_id = os.environ["VOLCENGINE_ACCESS_KEY_ID"]
+        access_key_secret = os.environ["VOLCENGINE_ACCESS_KEY_SECRET"]
+
+        tls_client1 = TLSService(endpoint, access_key_id, access_key_secret, region)
+        tls_client2 = TLSService(endpoint + "test", access_key_id + "test", access_key_secret + "test", region + "test")
+
+        self.assertNotEqual(tls_client1, tls_client2)
+        self.assertEqual(region, tls_client1.get_region())
+        self.assertEqual(region + "test", tls_client2.get_region())
+
+if __name__ == '__main__':
+    unittest.main()

volcengine/example/viking_db/example.py

@@ -13,14 +13,19 @@
 
 if __name__ == '__main__':
     vikingdb_service = VikingDBService()
-    vikingdb_service.set_ak("")
-    vikingdb_service.set_sk("")
+    vikingdb_service.set_ak()
+    vikingdb_service.set_sk()
+
+    # id = vikingdb_service.create_task(TaskType.Filter_Delete, {"collection_name": "example", "filter": {"like": [2]}})
+    # print(id)
+
+    # res = vikingdb_service.update_task("ae6f9a4e-9a68-5374-9687-36017c1ddd3e", task_status=TaskStatus.Confirmed)
 
     # id = vikingdb_service.create_task(TaskType.Data_Import, {"tos_path": "demo-1028/demo_1030", "file_type":"json", "ignore_error":False, "collection_name":"sparse"})
     # print(id)
 
-    # task = vikingdb_service.get_task("01f7f554-46b9-55d6-af6c-b2aa9502d229")
-    # print(task.process_info["err_msg"])
+    # task = vikingdb_service.get_task("bc5e952d-3f95-5e0c-b310-548933890308")
+    # print(task.task_status)
 
     # tasks = vikingdb_service.list_tasks()
     # for item in tasks:
@@ -153,7 +158,7 @@ def gen_random_vector(dim):
     #         default_val=True
     #     ),
     # ]
-    # res = vikingdb_service.create_collection("example1", fields, "This is an example")
+    # res = vikingdb_service.create_collection("example", fields, "This is an example")
     # # 返回一个collection实例
     # print(res)
     #
@@ -196,13 +201,13 @@ def gen_random_vector(dim):
     #         res[i] = random.random() - 0.5
     #     return res
     # collection = vikingdb_service.get_collection("example")
-    # field1 = {"doc_id": "111", "text_vector": gen_random_vector(10), "like": 1, "price": 1.11,
+    # field1 = {"doc_id": "111", "text_vector": gen_random_vector(12), "like": 1, "price": 1.11,
     #           "author": ["gy"], "aim": True}
-    # field2 = {"doc_id": "222", "text_vector": gen_random_vector(10), "like": 2, "price": 2.22,
+    # field2 = {"doc_id": "222", "text_vector": gen_random_vector(12), "like": 2, "price": 2.22,
     #           "author": ["gy", "xjq"], "aim": False}
-    # field3 = {"doc_id": "333", "text_vector": gen_random_vector(10), "like": 3, "price": 3.33,
+    # field3 = {"doc_id": "333", "text_vector": gen_random_vector(12), "like": 3, "price": 3.33,
     #           "author": ["gy", "xjq"], "aim": False}
-    # field4 = {"doc_id": "444", "text_vector": gen_random_vector(10), "like": 4, "price": 4.44,
+    # field4 = {"doc_id": "444", "text_vector": gen_random_vector(12), "like": 4, "price": 4.44,
     #           "author": ["gy", "xjq"], "aim": False}
     # data1 = Data(field1)
     # data2 = Data(field2)
@@ -217,7 +222,7 @@ def gen_random_vector(dim):
 
     #
     # collection = vikingdb_service.get_collection("example")
-    # res = collection.fetch_data("111")
+    # res = collection.fetch_data("333")
     # print(res.fields)
     # res = collection.fetch_data(["111", "222", "333", "444"])
     # # 返回一个列表

volcengine/example/visual/example_cert_pro_liveness_verify_query.py

@@ -0,0 +1,21 @@
+# coding:utf-8
+from __future__ import print_function
+
+from volcengine.visual.VisualService import VisualService
+
+if __name__ == '__main__':
+    visual_service = VisualService()
+
+    # call below method if you don't set ak and sk in $HOME/.volc/config
+    visual_service.set_ak('')
+    visual_service.set_sk('')
+
+    form = {
+        "req_key": "cert_pro_liveness_verify_query",
+        "byted_token": "",
+        "omit_data": False,
+        "omit_image_data": False,
+        "omit_video_data": False
+    }
+    resp = visual_service.cert_pro_liveness_verify_query(form)
+    print(resp)

volcengine/tls/TLSService.py

@@ -133,14 +133,6 @@ def calc_backoff_ms(expected_quit_timestamp_ms):
             sleep_ms = 0
         return sleep_ms
 
-    def __new__(cls, *args, **kwargs):
-        if not hasattr(TLSService, "_instance"):
-            with TLSService._instance_lock:
-                if not hasattr(TLSService, "_instance"):
-                    TLSService._instance = object.__new__(cls)
-
-        return TLSService._instance
-
     def __init__(self, endpoint: str, access_key_id: str, access_key_secret: str, region: str,
                  security_token: str = None, scheme: str = "https", timeout: int = 60,
                  api_version=API_VERSION_V_0_3_0):

volcengine/viking_db/VikingDBService.py

@@ -124,7 +124,8 @@ def get_api_info():
                                    {'Accept': 'application/json', 'Content-Type': 'application/json'}),
             "DropTask": ApiInfo("POST", "/api/task/drop", {}, {},
                                    {'Accept': 'application/json', 'Content-Type': 'application/json'}),
-                                   
+            "UpdateTask": ApiInfo("POST", "/api/task/update", {}, {},
+                                   {'Accept': 'application/json', 'Content-Type': 'application/json'}),                                    
         }
         return api_info
 
@@ -1195,3 +1196,10 @@ def list_tasks(self):
     def drop_task(self, task_id):
         params = {"task_id": task_id}
         res = self.json_exception("DropTask", {}, json.dumps(params))
+
+    def update_task(self, task_id, task_status):
+        params = {
+            "task_id": task_id,
+            "task_status": task_status.value
+        }
+        res = self.json_exception("UpdateTask", {}, json.dumps(params))

volcengine/viking_db/common.py

@@ -67,6 +67,16 @@ class Order(Enum):
 
 class TaskType(Enum):
     Data_Import = "data_import"
+    Filter_Delete = "filter_delete"
+
+class TaskStatus(Enum):
+    Init = "init"
+    Queued = "queued"
+    Running = "running"
+    Done = "done"
+    Fail = "fail"
+    Confirm = "confirm"
+    Confirmed = "confirmed"
 
 class EmbModel(object):
     def __init__(self, model_name, params=None):

volcengine/visual/VisualService.py

@@ -163,6 +163,8 @@ def get_api_info():
             "CVSync2AsyncGetResult": ApiInfo("POST", "/", {"Action": "CVSync2AsyncGetResult", "Version": "2022-08-31"},{}, {}),
             "CVSync2AsyncSubmitTask": ApiInfo("POST", "/",{"Action": "CVSync2AsyncSubmitTask", "Version": "2022-08-31"}, {}, {}),
             "CVProcess": ApiInfo("POST", "/", {"Action": "CVProcess", "Version": "2022-08-31"}, {}, {}),
+            "CertLivenessVerifyQuery": ApiInfo("POST", "/",
+                                               {"Action": "CertLivenessVerifyQuery", "Version": "2022-08-31"}, {}, {}),
         }
         return api_info
 
@@ -257,6 +259,13 @@ def cv_sync2async_get_result(self, form):
         except Exception as e:
             raise Exception(str(e))
 
+    def cert_pro_liveness_verify_query(self, form):
+        try:
+            res_json = self.common_json_handler("CertLivenessVerifyQuery", form)
+            return res_json
+        except Exception as e:
+            raise Exception(str(e))
+
     def img2img_xl_sft(self, form):
         try:
             res_json = self.common_json_handler("Img2ImgXLSft", form)