superb-ai-onprem 0.5.10__py3-none-any.whl → 0.5.11__py3-none-any.whl

{superb_ai_onprem-0.5.10.dist-info → superb_ai_onprem-0.5.11.dist-info}/RECORD

@@ -1,14 +1,14 @@
-spb_onprem/__init__.py,sha256=lGZRdgpRdBFM97SawPJIC-a5uRzrd5jdb9i5yHDeveQ,2649
-spb_onprem/_version.py,sha256=ssU-HhfV0VSycUpIh7CbiGRJhaZLd00QnDLLKR6dd90,706
+spb_onprem/__init__.py,sha256=ugmA-tqXYbVB8OoRYW8SjxRitLTIwHvWQZ-U2KyKjdo,2885
+spb_onprem/_version.py,sha256=Sc1qEPGqm9lq7B7gOJTSc6KB6ru5lGdWZqZtirZLNXQ,706
 spb_onprem/base_model.py,sha256=XLtyoxRBs68LrvbFH8V4EvQGPc2W17koC310MnS37jc,442
 spb_onprem/base_service.py,sha256=dPfr3mGXYlqadOXycu6RBFX1HcZ1qzEsskLoOxERLOU,5737
 spb_onprem/base_types.py,sha256=5HO6uy6qf08b4KSElwIaGy7UkoQG2KqVO6gcHbsqqSo,269
-spb_onprem/entities.py,sha256=8C3DiGq04fryNjktk3zthum0Sa_sQb292jJEkZhd5zg,967
+spb_onprem/entities.py,sha256=LoywYC22ErWIWcJZQYg_foO_5c25pd77_11tFIKhsZk,1084
 spb_onprem/exceptions.py,sha256=jx5rTGsVZ5shOdbgQzk8GcSyMWFtb_3xhPq6Sylwc5o,478
 spb_onprem/searches.py,sha256=-ARAdCumAa1RViwhcinh-E7VHLs4xv5gk9dPyNze6-4,1448
 spb_onprem/activities/__init__.py,sha256=iTHUza8mtiBd_QEPH-0tzrOsd_VSwSQNMbzVWfebQ6c,175
 spb_onprem/activities/queries.py,sha256=L01PRDUQdbbIGoKP_M-CmZ1ORQzuq9NOr-rdWLW0ejE,6083
-spb_onprem/activities/service.py,sha256=HX_0IgdqFp0ProgQxfsf6rJk00uLDGnlqMOfRc3cKRM,11469
+spb_onprem/activities/service.py,sha256=jWfgAZGrULR8IETdPtUoNjwJDm-Zs8ZiC23eQEYBveA,11431
 spb_onprem/activities/entities/__init__.py,sha256=lJXPl4UqurNscZOYBcCMfRZBp27JKbPEbab5iXtL3ME,237
 spb_onprem/activities/entities/activity.py,sha256=xVJCrtU47ZG-ALRxsOauT0fIxc9EzqkTyw1xt5qVqJQ,1358
 spb_onprem/activities/entities/activity_history.py,sha256=6Iyt8mUjpWvC7KpCJj56Fn73MXuL_uIQKjz90o86cso,1037
@@ -22,17 +22,18 @@ spb_onprem/activities/params/start_activity.py,sha256=uBf78FYkjGkM5g8shIoMgz8qoC
 spb_onprem/activities/params/update_activity.py,sha256=QU3h2YkPFkqHEHXv6078V2PWEa272h60ST6FYPPZLWU,2615
 spb_onprem/activities/params/update_activity_history.py,sha256=ScWxOG7pZ-7WcTzvJleZIm-TsZVTRkwNrTCyNDyyVSc,1620
 spb_onprem/contents/__init__.py,sha256=9EfIMQxbJuUZVUqsTa3Ji-yDidFPQQB5gnJI4R01YWI,74
-spb_onprem/contents/queries.py,sha256=hk7JHfUBinIDp4xswYxmTMQwxsRbKWTxFeZQtUVl9K4,925
-spb_onprem/contents/service.py,sha256=a_Y_l18sULozh14obwAsEJG9x2BLSAgerqCNCf5TH2Q,5024
+spb_onprem/contents/queries.py,sha256=DuBlNTvgbo1gYfYIodScu2G5FngerZL6DzteBejV0-g,1188
+spb_onprem/contents/service.py,sha256=Waf3Kx3r4JXL7K8JUeg-LsJ5jEI3OqyoxgM3JWM78QI,5521
 spb_onprem/contents/entities/__init__.py,sha256=HsQ9J8UDxCx4xYTdMKQto7HCVUQilNozQCkIsceWezk,117
 spb_onprem/contents/entities/base_content.py,sha256=nM7NALpeRjtUKPv7eU0-hlqT1rPtD2mwB6Bvv_2Zc1E,346
 spb_onprem/contents/entities/content.py,sha256=YhTGHE9ykiOgFjvxbnLzSLg2665jPYrDATteB9PbGPE,534
-spb_onprem/contents/params/__init__.py,sha256=TYVs27fn47eTjnXGL6SQvkbWQqWaGwY6hJjnDLiLl-U,161
+spb_onprem/contents/params/__init__.py,sha256=YCkHhN01Dkx9ytCG35oSh2FmDDZcW-Dv2FhT1bFmzio,240
 spb_onprem/contents/params/create.py,sha256=JLMbNiIecOsVvVSYY6DfUpWjZxZKuO9-NSpuRlSIxS0,408
+spb_onprem/contents/params/delete_content.py,sha256=_M9d6FJpA47_pcM5YyHbMmTZv1F2wUwPryQTL9RxZwM,298
 spb_onprem/contents/params/get_download_url.py,sha256=3oaiqmjJRxw0U59SoTxoih1pcU-cuQvcU-g89lCcqtQ,98
 spb_onprem/data/__init__.py,sha256=5XKxNm2BlKufrX5uRQreUEzJ-nerTrozKpG1RJL5wt8,68
-spb_onprem/data/queries.py,sha256=FgmTnTL1LeIyOW7WZNT6djpW5hIE7gx-f-a1QJZ43T8,15433
-spb_onprem/data/service.py,sha256=EWKRCVW4XCrC1F-DUSj3IK-8xoFE3LzMUD5oZVbWDjA,22402
+spb_onprem/data/queries.py,sha256=mGnBBDuFNNVMpATid-uNpqVEzeu3R3u0wZnQDe8wR-o,17154
+spb_onprem/data/service.py,sha256=nUGw9Scn5FLFLURAXOid5sGrlfQLGH7r59uOCzTHVjk,25799
 spb_onprem/data/entities/__init__.py,sha256=ZQ2YWAHIan2ZLa1z5ozPSJwW3pqdlsJCxmAT3NuG89c,376
 spb_onprem/data/entities/annotation.py,sha256=pS97Nx3TWMBrb-iiBfRct4RXl7hPMy2IOzTIpXkmBJU,751
 spb_onprem/data/entities/data.py,sha256=TOdnete4bfwROQ7C7oew7vV8bZyewQvFdjccyx4mec8,1217
@@ -46,7 +47,7 @@ spb_onprem/data/enums/data_meta_type.py,sha256=9rd12-7C1udbbIGvnuGURKmd5-lndtW7o
 spb_onprem/data/enums/data_status.py,sha256=uV_yLtigSmht-tKFaLwvO4MpV5j4GzwuBRDILGrxMBA,292
 spb_onprem/data/enums/data_type.py,sha256=9ErNmOK-RIOrhyt25SE2WVPSx_nTxt3ncTG5UtW92iM,225
 spb_onprem/data/enums/scene_type.py,sha256=pTibjST4nJc3ylUrjQifYBdtLIcS2D0q71bdhyDjbIk,205
-spb_onprem/data/params/__init__.py,sha256=G83X-qgnWnuOMuMpsDKO0NZ9xWqQtmk6bdaT47XUvic,2386
+spb_onprem/data/params/__init__.py,sha256=R8x9zC7VZK4R5jm41wvBvCQ8qDCTMR1GdKyV13lnYaU,2598
 spb_onprem/data/params/change_data_labeler.py,sha256=jUG6bm0uheR1jVCy-RwFY97C9K93fdDdf3KDHJIelM4,570
 spb_onprem/data/params/change_data_reviewer.py,sha256=dFix3NfS2_8lpaoWxz6anLEJWMLdmRKM34ib8XwQiVQ,577
 spb_onprem/data/params/change_data_status.py,sha256=34gGtmTJT8wiDDpHOgQYvgP0-ziuYEBDFlpfY8SAZtI,593
@@ -58,6 +59,8 @@ spb_onprem/data/params/delete_data.py,sha256=HN4xXW9o4ZFOSjlzRZJ-F0wwtd_RzgI30gV
 spb_onprem/data/params/delete_prediction.py,sha256=X3sV2mrSUqg8FOlZYRtb0o1hAwFos48ydFK2Il8UCp4,479
 spb_onprem/data/params/delete_scene.py,sha256=pD8TtS5Jm2uQC3F2EQ6-nAbQuQbJLG2pAwFqmsgucyM,478
 spb_onprem/data/params/delete_slice_annotation_version.py,sha256=5M12bHKg9HJlh6hQ8XgLwFrfhTU7CiLM__gLZVBYrMg,595
+spb_onprem/data/params/get_data_detail.py,sha256=7_aWRcHHsX0U2chl3CWrVLRUfzeoPzK8JtjwUGQdv0Y,388
+spb_onprem/data/params/get_evaluation_value_list.py,sha256=zsnk5y2FrZXHbb8jIsxQ8BTTX9-HIAVXZe4B37mxvmk,1115
 spb_onprem/data/params/insert_annotation_version.py,sha256=bxw9LndaDJ0DLBo8U8kkGctHz1ToEgD_UxWWVpmJpks,987
 spb_onprem/data/params/insert_data_to_slice.py,sha256=_hLNC80wJbxGSouOpjwTY19mAf-itfRR4sxHnlLiiqk,478
 spb_onprem/data/params/insert_prediction.py,sha256=68q5ZSSq4arAsozy1i52phkUoBfUSzNVVxCuygRYDfI,621
@@ -74,14 +77,15 @@ spb_onprem/data/params/update_slice_annotation.py,sha256=7eu2DFvznOv0HuvKiZTHDOT
 spb_onprem/data/params/update_slice_annotation_version.py,sha256=M2STViLDSQDZ0WNYLcenum-NHpKQXUOfGh1XE6An_7c,1318
 spb_onprem/data/params/upsert_data_meta.py,sha256=faBI9UuH9onOxmEHc4C63nFJXqZ8YMnD2q-HKj_ZMxM,1033
 spb_onprem/datasets/__init__.py,sha256=Sjrb1tewB3CoODtHjRYOe-w2HpZi9UgbCyE2p8MzHfw,74
-spb_onprem/datasets/queries.py,sha256=PSj-3tOJfWUw8YXsv682LbcZoGbGHGgSQbWQGXjLePY,1912
-spb_onprem/datasets/service.py,sha256=ehc3IMYCN72pWZXvSs-j5jeYWQ6CxkU-0AA1-OwmtIk,3872
+spb_onprem/datasets/queries.py,sha256=uJKFD8awmmO6wiltX7fN8ovR0_yewarKkpHAwR7LRXY,2211
+spb_onprem/datasets/service.py,sha256=r5dim9mQcyb_gwBQ9oG4je0K8iHs1elvcWFLW9wvzYA,4439
 spb_onprem/datasets/entities/__init__.py,sha256=yx5tsyX4GLYrjqx6-ZEoceJ7jxQzIh15OgyFFCs7vow,59
 spb_onprem/datasets/entities/dataset.py,sha256=HO7EU5vwSBnILCI9xXhgZb36akIi6AC0ApamQlQczhU,538
-spb_onprem/datasets/params/__init__.py,sha256=7wmc4xQi3GOSeREelqfAveI0buYJWn7huX1Q9_jl_50,291
+spb_onprem/datasets/params/__init__.py,sha256=uqy7QijSgvOzrRY0xIg9GxRUo5Bquscs3dTiboLeTCM,370
 spb_onprem/datasets/params/create_dataset.py,sha256=YGhLvY4arthjZwKQ28HLv7ch0Gd2lJ-vyGHBZnMuy4E,719
 spb_onprem/datasets/params/dataset.py,sha256=WTOUl5M5cc6rtTwhLw_z31Cs209LkBq8Ja4LJGzrmGE,668
 spb_onprem/datasets/params/datasets.py,sha256=Hx4YlLxfb-Qwi4Y5AFl5pyyjupvuoVcCtxLPGjIV7UY,1580
+spb_onprem/datasets/params/delete_dataset.py,sha256=hzHV66_2TCJj1UK1czTNDzYm8fbeAthlvRvlVonRySI,323
 spb_onprem/datasets/params/update_dataset.py,sha256=1oaj2qB9hvnypl4-WtcTNCa4iSuEkJjEalq2JsTm5Ro,924
 spb_onprem/exports/__init__.py,sha256=l_eUjnrFJXs-vdOeMNiWYK-UdJkxArv7hxmKDG4Dfio,181
 spb_onprem/exports/queries.py,sha256=sFT6VX2UwAYyVNkiBteQ_JtKYnhMrt64ww_NuXhUhLM,4084
@@ -94,6 +98,26 @@ spb_onprem/exports/params/delete_export.py,sha256=EusUB86HNLtFYu4gIDJqZsODRETtTY
 spb_onprem/exports/params/export.py,sha256=0EP6nkQc6vFI-f8218Yq4NxfFcw8MQtHMNkYlGOXqo4,799
 spb_onprem/exports/params/exports.py,sha256=oOQo-2Cqsm3Th3s-0gNcVGjbOyZqB6ujPRLXipulB4Y,2417
 spb_onprem/exports/params/update_export.py,sha256=iOlZoHEN2iiY83hCZWdgTn-O9J8hcSUqd5K7_gEzvP4,3057
+spb_onprem/inferences/__init__.py,sha256=5SBXFgLWbTQV-lHoP7it4AbSAAAw_7IRm4FGRPPAeYw,68
+spb_onprem/inferences/service.py,sha256=ZkORci7Yt8Ruo0aY2KPsvynZG_uuitMKJO3FR22Nplc,1719
+spb_onprem/models/__init__.py,sha256=7Z6SiJqR0CVoK44Lx_J1WIPBO72Pe0mR4ZrWyb9tqAk,109
+spb_onprem/models/entities.py,sha256=TgsNdN5zJtgZc3FXmB2yv8rFjFwRzLaqhubdK40Zupw,205
+spb_onprem/models/queries.py,sha256=Pwymn30vaxXI89Yl_glkKSJKD5xfVj3PCLes2pUdXrQ,899
+spb_onprem/models/service.py,sha256=pvAHIJFw_0S8sBlUjDb2dUVT0whqR-ZiWfbWdw1cLuw,2499
+spb_onprem/models/params/__init__.py,sha256=iwZ8TWZEt5A4cMmidslkAdsBFspTxhs-WKTNzjAWPys,154
+spb_onprem/models/params/delete_model.py,sha256=onUSod8jKNeOH3zIziK82GCv2BzLF-fiKUjSu7bX87U,403
+spb_onprem/models/params/get_models.py,sha256=IZtCDQZd4lUumDj0NEr0q_PFefJvdHSj2hqmdymyu1E,734
+spb_onprem/predictions/__init__.py,sha256=8BDi5TI_OuCT5aHffosRQVO41v3K909mY48qTWsP6JU,135
+spb_onprem/predictions/entities.py,sha256=KABfAB4L4uP65lEkuZ5TzVS3oejCS-metmhj1YIbB2M,441
+spb_onprem/predictions/queries.py,sha256=oL1d8zI5be6kUTeyYV5GDefPW-r_npPP8DwFDwS-vkY,3477
+spb_onprem/predictions/service.py,sha256=uVECUKGUMgqOw4AxlyV5cnYPhBSF3KRbKlx08tP0DPo,8064
+spb_onprem/predictions/params/__init__.py,sha256=aU5-aQosfVekHqEBBlPImUraeFsmxSThNXd_fhhCsMc,648
+spb_onprem/predictions/params/create_prediction_set.py,sha256=DNp7oHt8X1revLBH-5f__xE5suJS3eCZrTjUlCTkD8A,1374
+spb_onprem/predictions/params/delete_prediction_from_data.py,sha256=wwkivSuFybhbDVmSOlH_08U6ze78yP_BPdiyKdBtqHE,575
+spb_onprem/predictions/params/delete_prediction_set.py,sha256=KMiuc8QIWCMLce_Xn5LZjy0PwJWxuLT9qRjKrYUg0LM,457
+spb_onprem/predictions/params/get_prediction_set.py,sha256=byoUu5CkBRfvv2ucIzSBL-8E_mKjCLMo20Lb7IIP4k4,435
+spb_onprem/predictions/params/get_prediction_sets.py,sha256=w3yLmrVWiTPGtXhERe6hwWd_rIv2M-EbwgeiAWCHPBc,761
+spb_onprem/predictions/params/update_prediction_set_data_info.py,sha256=oGl2pZPjsMCeCr3gjYlg8ms0ukl811ogwpSbDhUyxDk,767
 spb_onprem/slices/__init__.py,sha256=xgpNGYzqgwQ8C-Bgw9AZWMAgBW38UU-U4Wube8hkodI,69
 spb_onprem/slices/queries.py,sha256=13JcjACvCg7AbdGfUODDJeq77gGZlNjngFB-9UmgLZk,3080
 spb_onprem/slices/service.py,sha256=TnWGMVflesdaUXFKOBZTpdHSIKkQwj93Q3q9aGts5xI,5615
@@ -108,20 +132,28 @@ spb_onprem/slices/params/update_slice.py,sha256=kryOmCnRTQ_OU0qDKgugppLrpeUpuLwm
 spb_onprem/users/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 spb_onprem/users/entities/__init__.py,sha256=X8HZsCTlQnuPszok3AwI-i7bsQi0Ehul5L_2jZaol5E,57
 spb_onprem/users/entities/auth.py,sha256=_KP-7yUErBxhJMm-dE3ObprPEG6e0JI2qNg6g8aK1qM,3371
-superb_ai_onprem-0.5.10.dist-info/licenses/LICENSE,sha256=CdinbFiHKGkGl6cPde6WgXhMuzyUXEG6tzy2-7udZ8o,1066
+superb_ai_onprem-0.5.11.dist-info/licenses/LICENSE,sha256=CdinbFiHKGkGl6cPde6WgXhMuzyUXEG6tzy2-7udZ8o,1066
 tests/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
 tests/activities/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
 tests/activities/real_test.py,sha256=0gQHg7rIEuZndGZyNHMWSD5nUZPMsUGigfCjWFxMthQ,1786
 tests/activities/test_get_activity_history.py,sha256=gNLc4OQLgw9wqToRxV0y5REAHPudZK58h-plj5oczoE,5292
-tests/activities/test_params.py,sha256=L3mcnrN2c8dT0AVTjnGedu6LHF3eZ4sSDzz8eaMHJmg,2369
-tests/activities/test_service.py,sha256=AbWFwjaUjQRs5oDZ2ba0C23XJxCqsFNrxH1LamMtWmU,4698
+tests/activities/test_params.py,sha256=P_k03Ll-We3gyFTeACalErP1UzwS2gB6ayIerKMMLEU,2335
+tests/activities/test_service.py,sha256=8nijwzJlAFL9S0TqyuAvOJZzBvPPjxJbrj3EyziT_Ng,4407
+tests/data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+tests/data/test_data_service.py,sha256=JgcNozcwECKN9-LR80FXkH4RslXVWAGJh-FyODcmRCk,13702
+tests/datasets/__init__.py,sha256=nMsjhPXe9rU87gI_wasqCtz7SsTae2LzpUUnXXzS5zI,23
+tests/datasets/test_dataset_service.py,sha256=Sgw0HWZZJaPWVsbEq6w8tG4WlRBA98b1EoXLT8hRz_E,4881
 tests/exports/__init__.py,sha256=iBqr781TtnT1Tm5wLTCsdftpuRgQifIibxWrB_I7tiw,23
 tests/exports/real_test.py,sha256=KeHWvQP-vHCvFWvD4_7qFmreAX9o_Dd_YowaGFogZu8,3765
 tests/exports/test_entities.py,sha256=hG7G4kVkyHKT3mv4lvrzUqOW8ILeHiYj87QZjQcmg9E,8836
 tests/exports/test_integration.py,sha256=cCcEgwIIHyQRlc04EAXSKz7RcblQvhI2GBR3uVaOOq8,6201
 tests/exports/test_params.py,sha256=oRRa6nEru_FImlB3TrmFiBidz6ZstCx4rVaCK-EMGfQ,11070
-tests/exports/test_service.py,sha256=IDcKxrmByh00jk9142P2ThuGureMoijTHNdU0rERGG8,14628
-superb_ai_onprem-0.5.10.dist-info/METADATA,sha256=JJIfex7sDBjoqRwth4WEEkTyMzsu1qALX9arBRqbd0g,5818
-superb_ai_onprem-0.5.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-superb_ai_onprem-0.5.10.dist-info/top_level.txt,sha256=LbqU6FjWKaxO7FPS5-71e3OIS8VgBi5VrtQMWFOW25Q,17
-superb_ai_onprem-0.5.10.dist-info/RECORD,,
+tests/exports/test_service.py,sha256=sAoFbLWGgRzE61QjnaZ1yymIAuDz7nrbsw42G3klTCQ,14660
+tests/models/__init__.py,sha256=Z2naiUgIjI-nHGsldwze1BqpV5JfDKTaBk-vg6_h6t4,21
+tests/models/test_model_service.py,sha256=6Wm9nmY7nbqImSu62d0q4VtNiCjKpMji6eSIgSi7lfA,7587
+tests/predictions/__init__.py,sha256=mFAXy0xOEXVYM7quYfIhwSih3kcEDdD4rLPq5G91Igg,26
+tests/predictions/test_prediction_service.py,sha256=9maX0JGxfUJ1QO_YEdS9OI5-o1tXlMWyeEbvZB8LB48,12679
+superb_ai_onprem-0.5.11.dist-info/METADATA,sha256=FK0vXshMvz4-5X8q-xxbJSLXKuvxki54YnRLwrKwEFk,5818
+superb_ai_onprem-0.5.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+superb_ai_onprem-0.5.11.dist-info/top_level.txt,sha256=LbqU6FjWKaxO7FPS5-71e3OIS8VgBi5VrtQMWFOW25Q,17
+superb_ai_onprem-0.5.11.dist-info/RECORD,,

tests/activities/test_params.py

@@ -20,7 +20,7 @@ class TestUpdateActivityHistoryParams:
         )
         
         # Then
-        assert params["activity_history_id"] == activity_history_id
+        assert params["id"] == activity_history_id
         assert params["status"] == status
         assert params["meta"] == meta
         assert "progress" not in params
@@ -37,7 +37,7 @@ class TestUpdateActivityHistoryParams:
         )
         
         # Then
-        assert params["activity_history_id"] == activity_history_id
+        assert params["id"] == activity_history_id
         assert params["progress"] == progress
         assert "status" not in params
         assert "meta" not in params

tests/activities/test_service.py

@@ -15,12 +15,10 @@ class TestActivityService:
     def test_create_activity(self, activity_service):
         # Given
         mock_response = {
-            "createJob": {
-                "id": "test_id",
-                "type": "test_type",
-                "name": "test_name",
-                "description": "test_description"
-            }
+            "id": "test_id",
+            "activity_type": "test_type",
+            "name": "test_name",
+            "description": "test_description"
         }
         activity_service.request_gql = MagicMock(return_value=mock_response)
         
@@ -34,29 +32,27 @@ class TestActivityService:
         # Then
         assert isinstance(activity, Activity)
         assert activity.id == "test_id"
-        assert activity.type == "test_type"
+        assert activity.activity_type == "test_type"
         assert activity.name == "test_name"
         assert activity.description == "test_description"
     
     def test_get_activities(self, activity_service):
         # Given
         mock_response = {
-            "jobs": {
-                "activities": [
-                    {
-                        "id": "test_id_1",
-                        "type": "test_type",
-                        "name": "test_name_1"
-                    },
-                    {
-                        "id": "test_id_2",
-                        "type": "test_type",
-                        "name": "test_name_2"
-                    }
-                ],
-                "next": "next_cursor",
-                "totalCount": 2
-            }
+            "jobs": [
+                {
+                    "id": "test_id_1",
+                    "activity_type": "test_type",
+                    "name": "test_name_1"
+                },
+                {
+                    "id": "test_id_2",
+                    "activity_type": "test_type",
+                    "name": "test_name_2"
+                }
+            ],
+            "next": "next_cursor",
+            "totalCount": 2
         }
         activity_service.request_gql = MagicMock(return_value=mock_response)
         
@@ -74,11 +70,9 @@ class TestActivityService:
     def test_start_activity(self, activity_service):
         # Given
         mock_response = {
-            "startJob": {
-                "id": "test_history_id",
-                "jobId": "test_activity_id",
-                "status": ActivityStatus.RUNNING.value
-            }
+            "id": "test_history_id",
+            "jobId": "test_activity_id",
+            "status": ActivityStatus.RUNNING.value
         }
         activity_service.request_gql = MagicMock(return_value=mock_response)
         
@@ -97,11 +91,9 @@ class TestActivityService:
     def test_update_activity_history_status(self, activity_service):
         # Given
         mock_response = {
-            "updateJobHistory": {
-                "id": "test_history_id",
-                "jobId": "test_activity_id",
-                "status": ActivityStatus.SUCCESS.value
-            }
+            "id": "test_history_id",
+            "jobId": "test_activity_id",
+            "status": ActivityStatus.SUCCESS.value
         }
         activity_service.request_gql = MagicMock(return_value=mock_response)
         
@@ -119,11 +111,9 @@ class TestActivityService:
     def test_update_activity_history_progress(self, activity_service):
         # Given
         mock_response = {
-            "updateJobHistory": {
-                "id": "test_history_id",
-                "jobId": "test_activity_id",
-                "progress": {"current": 50, "total": 100}
-            }
+            "id": "test_history_id",
+            "jobId": "test_activity_id",
+            "progress": {"current": 50, "total": 100}
         }
         activity_service.request_gql = MagicMock(return_value=mock_response)