rapidata 1.4.6__py3-none-any.whl → 1.6.0__py3-none-any.whl

rapidata/service/openapi_service.py

@@ -1,5 +1,7 @@
 import json
+import time
 import requests
+import threading
 from rapidata.api_client.api.campaign_api import CampaignApi
 from rapidata.api_client.api.dataset_api import DatasetApi
 from rapidata.api_client.api.order_api import OrderApi
@@ -13,6 +15,8 @@ from rapidata.api_client.configuration import Configuration
 
 class OpenAPIService:
 
+    _TOKEN_EXPIRATION_MINUTES = 30
+
     def __init__(
         self,
         client_id: str,
@@ -20,18 +24,41 @@ class OpenAPIService:
         endpoint: str,
         token_url: str,
         oauth_scope: str,
-        cert_path: str | None = None,
+        cert_path: str | None = None
     ):
         client_configuration = Configuration(host=endpoint, ssl_ca_cert=cert_path)
         self.api_client = ApiClient(configuration=client_configuration)
 
-        token = self.__fetch_token(client_id, client_secret, oauth_scope, token_url, cert_path)
+        self._client_id = client_id
+        self._client_secret = client_secret
+        self._oauth_scope = oauth_scope
+        self._token_url = token_url
+        self._cert_path = cert_path
 
-        client_configuration.api_key["bearer"] = f"Bearer {token}"
         self._api_client = ApiClient()
         self._order_api = OrderApi(self.api_client)
         self._dataset_api = DatasetApi(self.api_client)
 
+        api_token = self.__fetch_token(
+            self._client_id, self._client_secret, self._oauth_scope, self._token_url, self._cert_path
+        )
+        self.api_client.configuration.api_key["bearer"] = f"Bearer {api_token}"
+
+        refresh_thread = threading.Thread(
+            target=lambda: self.__refresh_token_periodically(self._TOKEN_EXPIRATION_MINUTES - 1)
+        )
+        refresh_thread.daemon = True
+        refresh_thread.start()
+
+    def __refresh_token_periodically(self, refresh_interval):
+        while True:
+            new_token = self.__fetch_token(
+                self._client_id, self._client_secret, self._oauth_scope, self._token_url, self._cert_path
+            )
+            self.api_client.configuration.api_key["bearer"] = f"Bearer {new_token}"
+
+            time.sleep(refresh_interval)
+
     @property
     def order_api(self) -> OrderApi:
         return self._order_api

{rapidata-1.4.6.dist-info → rapidata-1.6.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rapidata
-Version: 1.4.6
+Version: 1.6.0
 Summary: Rapidata package containing the Rapidata Python Client to interact with the Rapidata Web API in an easy way.
 License: Apache-2.0
 Author: Rapidata AG

{rapidata-1.4.6.dist-info → rapidata-1.6.0.dist-info}/RECORD

@@ -1,5 +1,5 @@
-rapidata/__init__.py,sha256=xkV2cgV_RBZCW8CNxWFMlf4EFJUWXcJnqfzQtJsG6NA,579
-rapidata/api_client/__init__.py,sha256=JmHmClUXDsPZxpAoI7L3s4fvVSNrfxNHXXmFo8Q9k14,23276
+rapidata/__init__.py,sha256=PwuTW_WL75klRHmgTPKVqIVQJg32Uq48TBxxYI6qUvI,621
+rapidata/api_client/__init__.py,sha256=qjZpyMWRZgTrH3dtHsT1niqRKVVU92e_HCsoH2H8ypc,23396
 rapidata/api_client/api/__init__.py,sha256=S0oVoAVMys10M-Z1SqirMdnHMYSHH3Lz6iph1CfILc0,1004
 rapidata/api_client/api/campaign_api.py,sha256=DxPFqt9F6c9OpXu_Uxhsrib2NVwnbcZFa3Vkrj7cIuA,40474
 rapidata/api_client/api/coco_api.py,sha256=4QYkW7c0SZvs-HOYmj585yL0KNr6Xc16ajS7b72yI6w,24972
@@ -14,13 +14,13 @@ rapidata/api_client/api/rapid_api.py,sha256=q7LJiweMnln2fs_KOzwVlO4bEk564kNkvPsc
 rapidata/api_client/api/rapidata_identity_api_api.py,sha256=-kgoDuLdh-R4MQ7JPi3kQ0WDFKbmI0MkCjxwHXBmksA,9824
 rapidata/api_client/api/simple_workflow_api.py,sha256=BIc4Cx05cbd_5BV0Rsb44M3uMlQZ5IB_6iHyNbbAZiU,15440
 rapidata/api_client/api/user_info_api.py,sha256=FuuA95Beeky-rnqIoSUe2-WQ7oVTfq0RElX0jfKXT0w,10042
-rapidata/api_client/api/validation_api.py,sha256=ArZMBUDyryzfoQCVXJGvzw7xUlUpKEE6XCEAvxW36ok,79742
+rapidata/api_client/api/validation_api.py,sha256=U6OQUhZ-5C895GWi2iz4pAfmSTHfyRhdtoYcbFoNcnA,90019
 rapidata/api_client/api/workflow_api.py,sha256=prRrSAVTiPqw2w2PS0-TVFRS7ycS9sMCNE-Sz-ESnfk,52882
 rapidata/api_client/api_client.py,sha256=EDhxAOUc5JFWvFsF1zc726Q7GoEFkuB8uor5SlGx9K4,27503
 rapidata/api_client/api_response.py,sha256=eMxw1mpmJcoGZ3gs9z6jM4oYoZ10Gjk333s9sKxGv7s,652
 rapidata/api_client/configuration.py,sha256=g472vHVPLBotq8EkfSXP4sbp7xnn-3sb8O8BBlRWK1I,15931
 rapidata/api_client/exceptions.py,sha256=eLLd1fxM0Ygf3IIG6aNx9hdy79drst5Cem0UjI_NamM,5978
-rapidata/api_client/models/__init__.py,sha256=DnXGkxLZZNnBDnjOB2YPyLCwbR3glSipc4Hg6aZQuPc,21731
+rapidata/api_client/models/__init__.py,sha256=ORUPTcjrZCCLWcP4flfRBESc2lIjTmzFOe42K6tJRDY,21851
 rapidata/api_client/models/add_campaign_artifact_result.py,sha256=4IvFVS-tLlL6eHsWp-IZ_ul5T30-h3YEwd2B5ioBbgY,2582
 rapidata/api_client/models/add_campaign_model.py,sha256=OJzkfvQlrp6j6ffwVShouUCW-MQZw60BGUJpqjbSGs8,6853
 rapidata/api_client/models/add_validation_rapid_model.py,sha256=-HRHMK-o6dgGjUqfsP_woZcFxfN7nuJ-L1CUaK9nihY,4918
@@ -153,6 +153,7 @@ rapidata/api_client/models/get_pipeline_by_id_result.py,sha256=JB4kJasul1r298p5P
 rapidata/api_client/models/get_pipeline_by_id_result_artifacts_value.py,sha256=tvh4GD4YeU9VnXKWLiqTfqGsC-90hUa_zAd-qkcwEPs,8476
 rapidata/api_client/models/get_public_orders_result.py,sha256=1Eq-R7wX2fKREuu-o1u5pVqHnc89-_--pkz78y80oEM,3001
 rapidata/api_client/models/get_simple_workflow_result_overview_result.py,sha256=ThHj63FCET3FcfC8jBjTG8opL2ZUMwg4BZFIzXFaC8I,6554
+rapidata/api_client/models/get_validation_set_by_id_result.py,sha256=dsonFbNl_Zj4mwUmo4Hc6QSDfpc1MDaiFzbEC3Ta_Uo,2845
 rapidata/api_client/models/get_workflow_by_id_result.py,sha256=l8DvKolvRGJQtMuTXLVwJuCN0eaSET-gi0p6TV6phuQ,2886
 rapidata/api_client/models/get_workflow_by_id_result_workflow.py,sha256=Pc-kq2AnoxqVQmJj52Fptyb-RoTihwq73GpODiw6trk,5735
 rapidata/api_client/models/get_workflow_config_result.py,sha256=2JiFcYzAsiUOheWrg1bL6RYuQqAvrn2WaL9pZpQZh1w,3023
@@ -237,7 +238,7 @@ rapidata/api_client/models/query_validation_rapids_result_paged_result.py,sha256
 rapidata/api_client/models/query_validation_set_model.py,sha256=FIPvd0-wXvK-GTdtj_ZvQEcsLt2B8wTXb6-33WujkBY,4239
 rapidata/api_client/models/query_workflows_model.py,sha256=Y3HuKLbtLlejEQsJyVZ-vLN6AJExWoLo1SQmLFWX4vw,4135
 rapidata/api_client/models/ranked_datapoint_model.py,sha256=c9Q5iqRPvNd-2x_1t9SRE7wI0optxB67Z-tE8V4D2lM,3087
-rapidata/api_client/models/rapid_answer.py,sha256=2vDzsQm5_hFyaexuvT5--V3ZBXpKh9rwJVlXFc3ojUw,3118
+rapidata/api_client/models/rapid_answer.py,sha256=IvDuJNJlkBUjI3Q5STq6SL48FzEsfXQ9N1TsJ_TO4ig,3282
 rapidata/api_client/models/rapid_answer_result.py,sha256=oXYHKpjkXa_cxgJb8xUR_v8jo6RV06nLxLxXEGVrhAY,11634
 rapidata/api_client/models/rapid_result_model.py,sha256=JVUSCNErUbZimSIelKb7pgXjxdaHFX6ZjKahuRLbe7w,3048
 rapidata/api_client/models/rapid_result_model_result.py,sha256=RjA75vGoZW5fEHLxS3-z1TUesyax_qxPXW4aYhGFoPw,11681
@@ -299,6 +300,8 @@ rapidata/api_client/models/validation_import_post_request_blueprint.py,sha256=KI
 rapidata/api_client/models/validation_rapid_selection_config.py,sha256=T6RoGq_YSytk_TAcWsfOsU_7nfQlLmnO4Qn0CkQok7k,3219
 rapidata/api_client/models/validation_selection.py,sha256=sgq3YtlKdAx2Ec41jFlZG1QNkN1NVwOJL-ALGxq3iL4,3131
 rapidata/api_client/models/validation_set.py,sha256=8ogEXVaKNB9pR2xuouf_Ckz_6-ryV0usGxx2qPqKjQk,3386
+rapidata/api_client/models/validation_set_model.py,sha256=79-5JwNDk51WtlXO0CHfjnO70AcyK_ErjZqBqmSYBFs,2916
+rapidata/api_client/models/validation_set_model_paged_result.py,sha256=KFdXjw6qkzGXLIHYlVSxfSYPpBPHTxEi4e6qGBzCIlk,3518
 rapidata/api_client/models/validation_set_overview_model.py,sha256=wlrjXg1dmlFmP8rIKDR88pGzpUEg15t8esr63BfYaaE,2693
 rapidata/api_client/models/validation_set_paged_result.py,sha256=3ch0KcbcH7qn8iJPY90WIiTF_MAIPJxWa3DUNfKwSI0,3477
 rapidata/api_client/models/workflow_aggregation_step_model.py,sha256=6LWLE1aY9P5u3YbbMZFcUDrREP8YiYypOw6Wrmxc0gs,4467
@@ -310,8 +313,8 @@ rapidata/api_client/models/workflow_split_model.py,sha256=zthOSaUl8dbLhLymLK_lrP
 rapidata/api_client/models/workflow_split_model_filter_configs_inner.py,sha256=1Fx9uZtztiiAdMXkj7YeCqt7o6VkG9lKf7D7UP_h088,7447
 rapidata/api_client/models/workflow_state.py,sha256=5LAK1se76RCoozeVB6oxMPb8p_5bhLZJqn7q5fFQWis,850
 rapidata/api_client/rest.py,sha256=zmCIFQC2l1t-KZcq-TgEm3vco3y_LK6vRm3Q07K-xRI,9423
-rapidata/api_client_README.md,sha256=ytnRUXykfU8I78TFE6DyO8xuFtyrK9hLztx5cejwfCM,34597
-rapidata/rapidata_client/__init__.py,sha256=2SsfOqLpkTh6b0wTKwUaBboxwuU9_SvP4TK-eUoocGo,795
+rapidata/api_client_README.md,sha256=wSrVJxrpMg3zHtVmiiPgmFB8_8PQLQkkk4a0595zcHc,34887
+rapidata/rapidata_client/__init__.py,sha256=mmfjHy_GMA7xFU_haNwxjNRaCF505iPXhE4KC2dXObY,841
 rapidata/rapidata_client/assets/__init__.py,sha256=T-XKvMSkmyI8iYLUYDdZ3LrrSInHsGMUY_Tz77hhnlE,240
 rapidata/rapidata_client/assets/base_asset.py,sha256=B2YWH1NgaeYUYHDW3OPpHM_bqawHbH4EjnRCE2BYwiM,298
 rapidata/rapidata_client/assets/media_asset.py,sha256=4xU1k2abdHGwbkJAYNOZYyOPB__5VBRDvRjklegFufQ,887
@@ -322,11 +325,9 @@ rapidata/rapidata_client/country_codes/__init__.py,sha256=FB9Dcks44J6C6YBSYmTmNZ
 rapidata/rapidata_client/country_codes/country_codes.py,sha256=Q0HMX7uHJQDeLCFPP5bq4iYi6pgcDWEcl2ONGhjgoeU,286
 rapidata/rapidata_client/dataset/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rapidata/rapidata_client/dataset/rapidata_dataset.py,sha256=12mvcpHa-ZpDOFhSuqnSb6pFhR1q7rF98fgiDBuONdw,4935
-rapidata/rapidata_client/dataset/rapidata_validation_set.py,sha256=rSRZQ67-8XjLa1_pWutieBrQfmvPA0fOMK7QKQpxsX0,10970
+rapidata/rapidata_client/dataset/rapidata_validation_set.py,sha256=KhvHMML591h61I9lqXSmp_s2_VZAWmuCoixmA2rAI8c,10973
 rapidata/rapidata_client/dataset/validation_rapid_parts.py,sha256=uzpOZFqQu8bG6vmjcWWUNJPZsRe28OmnyalRE6ry8tk,2317
-rapidata/rapidata_client/dataset/validation_set_builder.py,sha256=xPZBzNRPausIbYNRVinIhsVGnKSIIuxP5tjKkV4hOyo,7990
-rapidata/rapidata_client/feature_flags/__init__.py,sha256=IYkcK_bZCl5RfyQFiWjjUdz4y0jipiW9qfeopq4EjQQ,40
-rapidata/rapidata_client/feature_flags/feature_flags.py,sha256=ESwPY0mHDCzY_rjYastl1rWZVbmt-3gc9lHQdwgGfGo,4479
+rapidata/rapidata_client/dataset/validation_set_builder.py,sha256=ApYyEIwf9CQfnKJ2nZwK9FzgnY3Ff3BWetv75OlR-KM,8018
 rapidata/rapidata_client/filter/__init__.py,sha256=F3JsMCbAZWH7SNdgaj98ydTqgCCKXKBhLyt9gQ4x6tQ,301
 rapidata/rapidata_client/filter/age_filter.py,sha256=hncr1zNM_HrO0fCta8h0LgTCB8Ufv_vI5_cGhMb-xr8,478
 rapidata/rapidata_client/filter/base_filter.py,sha256=nXraj72cumyQjjYoo4MMpnlE0aWjAIOmGakKf0MNqps,135
@@ -342,9 +343,9 @@ rapidata/rapidata_client/metadata/prompt_metadata.py,sha256=_FypjKWrC3iKUO_G2CVw
 rapidata/rapidata_client/metadata/public_text_metadata.py,sha256=LTiBQHs6izxQ6-C84d6Pf7lL4ENTDgg__HZnDKvzvMc,511
 rapidata/rapidata_client/metadata/transcription_metadata.py,sha256=THtDEVCON4UlcXHmXrjilaOLHys4TrktUOPGWnXaCcc,631
 rapidata/rapidata_client/order/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-rapidata/rapidata_client/order/rapidata_order.py,sha256=IMozRlrYK_yqGLk0GQbLNYL_RRb7gUcSgDe6NUjOEEE,4812
-rapidata/rapidata_client/order/rapidata_order_builder.py,sha256=QvICzduLAuAkf8qFKxHV3zAag838WnV9lEzWdD4dxI0,15926
-rapidata/rapidata_client/rapidata_client.py,sha256=oSgBXBnszS7f2ITc0ljlOAaRlIpX8DSAE-RFOAKIDIM,6917
+rapidata/rapidata_client/order/rapidata_order.py,sha256=t5ddz_6Dk2DOpsDQ2EiZHJJB1RzU-etKwBL0RErEsSY,4567
+rapidata/rapidata_client/order/rapidata_order_builder.py,sha256=b--9byhsiAW1fL0mVSPzGJT0X4MQ-tCC0BNjIm2vu-Q,16406
+rapidata/rapidata_client/rapidata_client.py,sha256=DwWi7gtACV_OxQDEBHCILv8921tWQy3UtXzJ0ZcjjTo,8018
 rapidata/rapidata_client/referee/__init__.py,sha256=E1VODxTjoQRnxzdgMh3aRlDLouxe1nWuvozEHXD2gq4,150
 rapidata/rapidata_client/referee/base_referee.py,sha256=bMy7cw0a-pGNbFu6u_1_Jplu0A483Ubj4oDQzh8vu8k,493
 rapidata/rapidata_client/referee/early_stopping_referee.py,sha256=Dg2Kk7OiLBtS3kknsLxyJIlS27xmPvsikFR6g4xlbTE,1862
@@ -356,13 +357,14 @@ rapidata/rapidata_client/selection/conditional_validation_selection.py,sha256=wP
 rapidata/rapidata_client/selection/demographic_selection.py,sha256=DnMbLhzRItZ0t4Engo2fav4imG01JqAmn0SHOoPOhnQ,516
 rapidata/rapidata_client/selection/labeling_selection.py,sha256=cqDMQEXfQGMmgIvPgGOYgIGaXflV_J7LZsGOsakLXqo,425
 rapidata/rapidata_client/selection/validation_selection.py,sha256=HswzD2SvZZWisNLoGj--0sT_TIK8crYp3xGGndo6aLY,523
+rapidata/rapidata_client/settings/__init__.py,sha256=pqAmWPURaZ71rz6ogUH2rnfKL3rf7k4IhLyeM8ZHVlI,71
+rapidata/rapidata_client/settings/feature_flags.py,sha256=aKnxl1kv4cC3TaHV-xfqYlvL3ATIB2-H0VKl1rUt8jQ,4658
+rapidata/rapidata_client/settings/settings.py,sha256=oglRs-xgMGW8t2DhHfWIjWGkPW9Bm_fIaTTcaQKy-b8,4410
 rapidata/rapidata_client/simple_builders/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rapidata/rapidata_client/simple_builders/simple_classification_builders.py,sha256=gTb1OzJZjNW48ltrp0v9tcDjLee9_ZiLw78zYziVu_g,5520
 rapidata/rapidata_client/simple_builders/simple_compare_builders.py,sha256=GmNIUoAkoAHp0OnWLOgeI5-wnDiycuye0o1CUa9c3Q0,4808
-rapidata/rapidata_client/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-rapidata/rapidata_client/utils/utils.py,sha256=Fl99gCnh_HnieIp099xEvEv4g2kEIKiFcUp0G2iz6x8,815
 rapidata/rapidata_client/workflow/__init__.py,sha256=xWuzAhBnbcUFfWcgYrzj8ZYLSOXyFtgfepgMrf0hNhU,290
-rapidata/rapidata_client/workflow/base_workflow.py,sha256=3Ow3x3XDkan8ahGq9SkjOGda16ERQ0dtIYcnaZ6QMPc,1416
+rapidata/rapidata_client/workflow/base_workflow.py,sha256=iDBeklFwOIbMuJHA7t-x_cwxfygJCG0SoXLBB7R53fQ,1395
 rapidata/rapidata_client/workflow/classify_workflow.py,sha256=NkyyBrlCDqYVQaCARR9EHYuREEtXond69kD6jbzcN3M,1713
 rapidata/rapidata_client/workflow/compare_workflow.py,sha256=nqrgtohlXBfgHh_cRPOFAlwdI8wVc7PUsH0FDM7wIjg,1431
 rapidata/rapidata_client/workflow/evaluation_workflow.py,sha256=IBQoVFxOaeCDIBfaaboytWk8hNX9QH_xikkXt3lu9GY,619
@@ -370,10 +372,8 @@ rapidata/rapidata_client/workflow/free_text_workflow.py,sha256=VaypoG3yKgsbtVyqx
 rapidata/rapidata_client/workflow/transcription_workflow.py,sha256=_KDtGCdRhauJm3jQHpwhY-Hq79CLg5I8q2RgOz5lo1g,1404
 rapidata/service/__init__.py,sha256=s9bS1AJZaWIhLtJX_ZA40_CK39rAAkwdAmymTMbeWl4,68
 rapidata/service/local_file_service.py,sha256=pgorvlWcx52Uh3cEG6VrdMK_t__7dacQ_5AnfY14BW8,877
-rapidata/service/openapi_service.py,sha256=pGcOCttKZW0PVCSM7Kfehe5loh7CxmmDDbu4UJbamnI,2770
-rapidata/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-rapidata/utils/image_utils.py,sha256=TldO3eJWG8IhfJjm5MfNGO0mEDm1mQTsRoA0HLU1Uxs,404
-rapidata-1.4.6.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-rapidata-1.4.6.dist-info/METADATA,sha256=SFcRi0ilN8gwUDdl1tSA8gt_OVADGVPyVG4_EeJWvfg,1012
-rapidata-1.4.6.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-rapidata-1.4.6.dist-info/RECORD,,
+rapidata/service/openapi_service.py,sha256=l-ga9TLKWR6Gwx9nbDSQ2J1HZRPhgk_mGgnC3kud3uw,3688
+rapidata-1.6.0.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+rapidata-1.6.0.dist-info/METADATA,sha256=lQFu7Zzja9wuFSOzxsCrZ0dezR2YodP6snRFkjXscEo,1012
+rapidata-1.6.0.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+rapidata-1.6.0.dist-info/RECORD,,

rapidata/rapidata_client/utils/utils.py

@@ -1,22 +0,0 @@
-from rapidata.api_client.models.classify_payload import ClassifyPayload
-from rapidata.api_client.models.create_demographic_rapid_model import (
-    CreateDemographicRapidModel,
-)
-from rapidata.service.openapi_service import OpenAPIService
-
-
-class Utils:
-    def __init__(self, openapi_service: OpenAPIService):
-        self.openapi_service = openapi_service
-
-    def new_demographic_rapid(
-        self, question: str, options: list[str], identifier: str, media_path: str
-    ):
-        payload = ClassifyPayload(
-            _t="ClassifyPayload", title=question, possibleCategories=options
-        )
-        model = CreateDemographicRapidModel(identifier=identifier, payload=payload)
-
-        self.openapi_service.rapid_api.rapid_create_demographic_rapid_post(
-            model=model, file=[media_path]
-        )

rapidata/utils/image_utils.py

@@ -1,13 +0,0 @@
-from io import BytesIO
-import PIL.Image as Image
-
-class ImageUtils:
-
-    @staticmethod
-    def convert_PIL_image_to_bytes(image: Image.Image):
-        """
-        Convert a PIL image to bytes with meta data encoded. We can't just use image.tobytes() because this only returns the pixel data.
-        """
-        buffer = BytesIO()
-        image.save(buffer, image.format)
-        return buffer.getvalue()