hiddenlayer-sdk 1.2.2__py3-none-any.whl → 2.0.1__py3-none-any.whl

Sign up to get free protection for your applications and to get access to all the features.
Files changed (38) hide show
  1. hiddenlayer/__init__.py +0 -10
  2. hiddenlayer/sdk/exceptions.py +1 -1
  3. hiddenlayer/sdk/models.py +2 -3
  4. hiddenlayer/sdk/rest/__init__.py +16 -11
  5. hiddenlayer/sdk/rest/api/__init__.py +0 -1
  6. hiddenlayer/sdk/rest/api/model_supply_chain_api.py +1706 -571
  7. hiddenlayer/sdk/rest/api/sensor_api.py +214 -1320
  8. hiddenlayer/sdk/rest/models/__init__.py +16 -10
  9. hiddenlayer/sdk/rest/models/{scan_model_request.py → begin_multi_file_upload200_response.py} +9 -9
  10. hiddenlayer/sdk/rest/models/{get_multipart_upload_response.py → begin_multipart_file_upload200_response.py} +9 -9
  11. hiddenlayer/sdk/rest/models/{multipart_upload_part.py → begin_multipart_file_upload200_response_parts_inner.py} +11 -10
  12. hiddenlayer/sdk/rest/models/errors_inner.py +91 -0
  13. hiddenlayer/sdk/rest/models/file_details_v3.py +8 -2
  14. hiddenlayer/sdk/rest/models/{scan_results_v2.py → file_result_v3.py} +21 -32
  15. hiddenlayer/sdk/rest/models/{model_scan_api_v3_scan_query200_response.py → get_condensed_model_scan_reports200_response.py} +4 -4
  16. hiddenlayer/sdk/rest/models/inventory_v3.py +97 -0
  17. hiddenlayer/sdk/rest/models/model_inventory_info.py +1 -1
  18. hiddenlayer/sdk/rest/models/{detections.py → multi_file_upload_request_v3.py} +14 -22
  19. hiddenlayer/sdk/rest/models/{model_scan_api_v3_scan_model_version_id_patch200_response.py → notify_model_scan_completed200_response.py} +4 -4
  20. hiddenlayer/sdk/rest/models/pagination_v3.py +95 -0
  21. hiddenlayer/sdk/rest/models/problem_details.py +103 -0
  22. hiddenlayer/sdk/rest/models/scan_detection_v31.py +155 -0
  23. hiddenlayer/sdk/rest/models/scan_model_details_v3.py +1 -1
  24. hiddenlayer/sdk/rest/models/scan_results_map_v3.py +105 -0
  25. hiddenlayer/sdk/rest/models/scan_results_v3.py +120 -0
  26. hiddenlayer/sdk/rest/models/{model.py → sensor.py} +4 -4
  27. hiddenlayer/sdk/rest/models/{model_query_response.py → sensor_query_response.py} +7 -7
  28. hiddenlayer/sdk/services/aidr_predictive.py +57 -3
  29. hiddenlayer/sdk/services/model_scan.py +94 -132
  30. hiddenlayer/sdk/version.py +1 -1
  31. {hiddenlayer_sdk-1.2.2.dist-info → hiddenlayer_sdk-2.0.1.dist-info}/METADATA +12 -2
  32. {hiddenlayer_sdk-1.2.2.dist-info → hiddenlayer_sdk-2.0.1.dist-info}/RECORD +35 -31
  33. hiddenlayer/sdk/rest/api/model_scan_api.py +0 -591
  34. hiddenlayer/sdk/rest/models/scan_results.py +0 -118
  35. hiddenlayer/sdk/services/model.py +0 -149
  36. {hiddenlayer_sdk-1.2.2.dist-info → hiddenlayer_sdk-2.0.1.dist-info}/LICENSE +0 -0
  37. {hiddenlayer_sdk-1.2.2.dist-info → hiddenlayer_sdk-2.0.1.dist-info}/WHEEL +0 -0
  38. {hiddenlayer_sdk-1.2.2.dist-info → hiddenlayer_sdk-2.0.1.dist-info}/top_level.txt +0 -0
hiddenlayer/__init__.py CHANGED
@@ -7,7 +7,6 @@ from requests.auth import HTTPBasicAuth
7
7
  from hiddenlayer.sdk.rest.api_client import ApiClient
8
8
  from hiddenlayer.sdk.rest.configuration import Configuration
9
9
  from hiddenlayer.sdk.services.aidr_predictive import AIDRPredictive
10
- from hiddenlayer.sdk.services.model import ModelAPI
11
10
  from hiddenlayer.sdk.services.model_scan import ModelScanAPI
12
11
  from hiddenlayer.sdk.utils import is_saas
13
12
  from hiddenlayer.sdk.version import VERSION
@@ -57,7 +56,6 @@ class HiddenlayerServiceClient:
57
56
 
58
57
  self._api_client = ApiClient(configuration=self._config)
59
58
  self._aidr_predictive = AIDRPredictive(self._api_client)
60
- self._model = ModelAPI(self._api_client)
61
59
  self._model_scan = ModelScanAPI(self._api_client)
62
60
 
63
61
  def _get_jwt(self, *, api_id: str, api_key: str) -> str:
@@ -101,11 +99,3 @@ class HiddenlayerServiceClient:
101
99
  "AIDR for Predictive Models is currently only supported with the SaaS version of the platform."
102
100
  )
103
101
  return self._aidr_predictive
104
-
105
- @property
106
- def model(self) -> ModelAPI:
107
- if not self.is_saas:
108
- raise HiddenlayerUnsupportedPlatformError(
109
- "Model management is currently only supported with the SaaS version of the platform."
110
- )
111
- return self._model
@@ -4,7 +4,7 @@ class HiddenlayerConflictError(Exception):
4
4
  pass
5
5
 
6
6
 
7
- class ModelDoesNotExistError(Exception):
7
+ class SensorDoesNotExistError(Exception):
8
8
  pass
9
9
 
10
10
 
hiddenlayer/sdk/models.py CHANGED
@@ -18,16 +18,15 @@ class ScanResults(ScanReportV3):
18
18
 
19
19
  file_name: Optional[str] = None
20
20
  file_path: Optional[str] = None
21
- model_id: Optional[str] = None
22
21
 
23
22
  @classmethod
24
23
  def from_scanreportv3(
25
24
  cls, *, scan_report_v3: ScanReportV3, model_id: Optional[str] = None
26
25
  ) -> Self:
27
26
  scan_results_dict = scan_report_v3.to_dict()
28
- scan_results_dict["model_id"] = model_id
27
+ result = cls(**scan_results_dict)
29
28
 
30
- return cls(**scan_results_dict)
29
+ return result
31
30
 
32
31
 
33
32
  class EmptyScanResults(ScanResults):
@@ -21,7 +21,6 @@ from hiddenlayer.sdk.rest.api.health_api import HealthApi
21
21
  from hiddenlayer.sdk.rest.api.model_supply_chain_api import ModelSupplyChainApi
22
22
  from hiddenlayer.sdk.rest.api.readiness_api import ReadinessApi
23
23
  from hiddenlayer.sdk.rest.api.aidr_predictive_api import AidrPredictiveApi
24
- from hiddenlayer.sdk.rest.api.model_scan_api import ModelScanApi
25
24
  from hiddenlayer.sdk.rest.api.sensor_api import SensorApi
26
25
 
27
26
  # import ApiClient
@@ -42,24 +41,29 @@ from hiddenlayer.sdk.rest.models.artifact_change import ArtifactChange
42
41
  from hiddenlayer.sdk.rest.models.artifact_content import ArtifactContent
43
42
  from hiddenlayer.sdk.rest.models.artifact_location import ArtifactLocation
44
43
  from hiddenlayer.sdk.rest.models.attachment import Attachment
44
+ from hiddenlayer.sdk.rest.models.begin_multi_file_upload200_response import BeginMultiFileUpload200Response
45
+ from hiddenlayer.sdk.rest.models.begin_multipart_file_upload200_response import BeginMultipartFileUpload200Response
46
+ from hiddenlayer.sdk.rest.models.begin_multipart_file_upload200_response_parts_inner import BeginMultipartFileUpload200ResponsePartsInner
45
47
  from hiddenlayer.sdk.rest.models.code_flow import CodeFlow
46
48
  from hiddenlayer.sdk.rest.models.configuration_override import ConfigurationOverride
47
49
  from hiddenlayer.sdk.rest.models.conversion import Conversion
48
50
  from hiddenlayer.sdk.rest.models.create_sensor_request import CreateSensorRequest
49
- from hiddenlayer.sdk.rest.models.detections import Detections
50
51
  from hiddenlayer.sdk.rest.models.edge import Edge
51
52
  from hiddenlayer.sdk.rest.models.edge_traversal import EdgeTraversal
53
+ from hiddenlayer.sdk.rest.models.errors_inner import ErrorsInner
52
54
  from hiddenlayer.sdk.rest.models.exception import Exception
53
55
  from hiddenlayer.sdk.rest.models.external_properties import ExternalProperties
54
56
  from hiddenlayer.sdk.rest.models.external_property_file_reference import ExternalPropertyFileReference
55
57
  from hiddenlayer.sdk.rest.models.external_property_file_references import ExternalPropertyFileReferences
56
58
  from hiddenlayer.sdk.rest.models.file_details_v3 import FileDetailsV3
59
+ from hiddenlayer.sdk.rest.models.file_result_v3 import FileResultV3
57
60
  from hiddenlayer.sdk.rest.models.file_scan_report_v3 import FileScanReportV3
58
61
  from hiddenlayer.sdk.rest.models.file_scan_reports_v3 import FileScanReportsV3
59
62
  from hiddenlayer.sdk.rest.models.fix import Fix
60
- from hiddenlayer.sdk.rest.models.get_multipart_upload_response import GetMultipartUploadResponse
63
+ from hiddenlayer.sdk.rest.models.get_condensed_model_scan_reports200_response import GetCondensedModelScanReports200Response
61
64
  from hiddenlayer.sdk.rest.models.graph import Graph
62
65
  from hiddenlayer.sdk.rest.models.graph_traversal import GraphTraversal
66
+ from hiddenlayer.sdk.rest.models.inventory_v3 import InventoryV3
63
67
  from hiddenlayer.sdk.rest.models.invocation import Invocation
64
68
  from hiddenlayer.sdk.rest.models.location import Location
65
69
  from hiddenlayer.sdk.rest.models.location_inner import LocationInner
@@ -67,17 +71,16 @@ from hiddenlayer.sdk.rest.models.location_relationship import LocationRelationsh
67
71
  from hiddenlayer.sdk.rest.models.logical_location import LogicalLocation
68
72
  from hiddenlayer.sdk.rest.models.mitre_atlas_inner import MITREAtlasInner
69
73
  from hiddenlayer.sdk.rest.models.message import Message
70
- from hiddenlayer.sdk.rest.models.model import Model
71
74
  from hiddenlayer.sdk.rest.models.model_inventory_info import ModelInventoryInfo
72
- from hiddenlayer.sdk.rest.models.model_query_response import ModelQueryResponse
73
- from hiddenlayer.sdk.rest.models.model_scan_api_v3_scan_model_version_id_patch200_response import ModelScanApiV3ScanModelVersionIdPatch200Response
74
- from hiddenlayer.sdk.rest.models.model_scan_api_v3_scan_query200_response import ModelScanApiV3ScanQuery200Response
75
+ from hiddenlayer.sdk.rest.models.multi_file_upload_request_v3 import MultiFileUploadRequestV3
75
76
  from hiddenlayer.sdk.rest.models.multiformat_message_string import MultiformatMessageString
76
- from hiddenlayer.sdk.rest.models.multipart_upload_part import MultipartUploadPart
77
77
  from hiddenlayer.sdk.rest.models.node import Node
78
78
  from hiddenlayer.sdk.rest.models.notification import Notification
79
+ from hiddenlayer.sdk.rest.models.notify_model_scan_completed200_response import NotifyModelScanCompleted200Response
79
80
  from hiddenlayer.sdk.rest.models.paged_response_with_total import PagedResponseWithTotal
81
+ from hiddenlayer.sdk.rest.models.pagination_v3 import PaginationV3
80
82
  from hiddenlayer.sdk.rest.models.physical_location import PhysicalLocation
83
+ from hiddenlayer.sdk.rest.models.problem_details import ProblemDetails
81
84
  from hiddenlayer.sdk.rest.models.property_bag import PropertyBag
82
85
  from hiddenlayer.sdk.rest.models.rectangle import Rectangle
83
86
  from hiddenlayer.sdk.rest.models.region import Region
@@ -94,16 +97,18 @@ from hiddenlayer.sdk.rest.models.run_automation_details import RunAutomationDeta
94
97
  from hiddenlayer.sdk.rest.models.sarif210 import Sarif210
95
98
  from hiddenlayer.sdk.rest.models.scan_create_request import ScanCreateRequest
96
99
  from hiddenlayer.sdk.rest.models.scan_detection_v3 import ScanDetectionV3
100
+ from hiddenlayer.sdk.rest.models.scan_detection_v31 import ScanDetectionV31
97
101
  from hiddenlayer.sdk.rest.models.scan_header_v3 import ScanHeaderV3
98
102
  from hiddenlayer.sdk.rest.models.scan_job import ScanJob
99
103
  from hiddenlayer.sdk.rest.models.scan_job_inventory import ScanJobInventory
100
104
  from hiddenlayer.sdk.rest.models.scan_model_details_v3 import ScanModelDetailsV3
101
105
  from hiddenlayer.sdk.rest.models.scan_model_ids_v3 import ScanModelIdsV3
102
- from hiddenlayer.sdk.rest.models.scan_model_request import ScanModelRequest
103
106
  from hiddenlayer.sdk.rest.models.scan_report_v3 import ScanReportV3
104
- from hiddenlayer.sdk.rest.models.scan_results import ScanResults
105
- from hiddenlayer.sdk.rest.models.scan_results_v2 import ScanResultsV2
107
+ from hiddenlayer.sdk.rest.models.scan_results_map_v3 import ScanResultsMapV3
108
+ from hiddenlayer.sdk.rest.models.scan_results_v3 import ScanResultsV3
106
109
  from hiddenlayer.sdk.rest.models.security_posture import SecurityPosture
110
+ from hiddenlayer.sdk.rest.models.sensor import Sensor
111
+ from hiddenlayer.sdk.rest.models.sensor_query_response import SensorQueryResponse
107
112
  from hiddenlayer.sdk.rest.models.sensor_sor_model_card_query_response import SensorSORModelCardQueryResponse
108
113
  from hiddenlayer.sdk.rest.models.sensor_sor_model_card_response import SensorSORModelCardResponse
109
114
  from hiddenlayer.sdk.rest.models.sensor_sor_query_filter import SensorSORQueryFilter
@@ -5,6 +5,5 @@ from hiddenlayer.sdk.rest.api.health_api import HealthApi
5
5
  from hiddenlayer.sdk.rest.api.model_supply_chain_api import ModelSupplyChainApi
6
6
  from hiddenlayer.sdk.rest.api.readiness_api import ReadinessApi
7
7
  from hiddenlayer.sdk.rest.api.aidr_predictive_api import AidrPredictiveApi
8
- from hiddenlayer.sdk.rest.api.model_scan_api import ModelScanApi
9
8
  from hiddenlayer.sdk.rest.api.sensor_api import SensorApi
10
9