chkp-ai-security-sdk 1.0.0__tar.gz

chkp_ai_security_sdk-1.0.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Check Point Software Technologies Ltd.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

chkp_ai_security_sdk-1.0.0/PKG-INFO

@@ -0,0 +1,233 @@
+Metadata-Version: 2.4
+Name: chkp-ai-security-sdk
+Version: 1.0.0
+Summary: Check Point AI Security Official Python SDK (Workforce AI + Browse Security)
+Home-page: https://github.com/CheckPointSW/ai-security-py-sdk
+Author: Haim Kastner
+Author-email: haimk@checkpoint.com
+Maintainer: Haim Kastner
+Maintainer-email: haimk@checkpoint.com
+License: MIT
+Keywords: python,ai,security,sdk,checkpoint,genai,workforce ai,ai security,browse security
+Requires-Python: >=3.9,<4.0
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: aenum~=3.1
+Requires-Dist: certifi<2026.0,>=2024.0
+Requires-Dist: charset-normalizer~=3.3
+Requires-Dist: frozendict~=2.3
+Requires-Dist: idna~=3.7
+Requires-Dist: MarkupSafe~=2.1
+Requires-Dist: python-dateutil~=2.8
+Requires-Dist: python-dotenv~=1.0
+Requires-Dist: requests~=2.32
+Requires-Dist: typing-extensions>=4.12
+Requires-Dist: pyjwt~=2.8
+Requires-Dist: unitsnet-py>=0.1.82
+Requires-Dist: urllib3~=2.2
+Requires-Dist: aiohttp~=3.9
+Requires-Dist: aiohttp-retry~=2.8
+Dynamic: author
+Dynamic: author-email
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: license
+Dynamic: license-file
+Dynamic: maintainer
+Dynamic: maintainer-email
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# Check Point - AI Security Python SDK (Workforce AI + Browse Security)
+
+[![License](https://img.shields.io/github/license/CheckPointSW/ai-security-py-sdk.svg?style=plastic)](https://github.com/CheckPointSW/ai-security-py-sdk/blob/release/LICENSE) [![Latest Release](https://img.shields.io/github/v/release/CheckPointSW/ai-security-py-sdk?style=plastic)](https://github.com/CheckPointSW/ai-security-py-sdk/releases) [![PyPI version](https://img.shields.io/pypi/v/chkp-ai-security-sdk.svg?style=plastic)](https://pypi.org/project/chkp-ai-security-sdk/)
+
+[![Build SDK Package](https://github.com/CheckPointSW/ai-security-py-sdk/actions/workflows/build.yml/badge.svg)](https://github.com/CheckPointSW/ai-security-py-sdk/actions/workflows/build.yml) [![Publish package to PyPI](https://github.com/CheckPointSW/ai-security-py-sdk/actions/workflows/release.yml/badge.svg)](https://github.com/CheckPointSW/ai-security-py-sdk/actions/workflows/release.yml)
+
+This is the official Check Point AI Security SDK for Python, covering both **Workforce AI** and **Browse Security** products.
+
+The SDK is based on the public [AI Security OpenAPI](https://app.swaggerhub.com/apis/Check-Point/checkpoint-ai-security) and [Browse Security OpenAPI](https://app.swaggerhub.com/apis/Check-Point/checkpoint-browse-security) specifications.
+
+With the SDK, you do not have to manage log in, send keep alive requests, or worry about session expiration.
+
+> The SDK supports simultaneous instances with different tenants, and both products can be used independently or together.
+
+## SDK installation
+
+Via PIP (PyPi registry)
+```bash
+pip install chkp-ai-security-sdk
+```
+
+## Getting started
+
+Import the SDK classes for the product(s) you need:
+
+```python
+from chkp_ai_security_sdk import AISecurity, BrowseSecurity
+```
+
+- **`AISecurity`** — Workforce AI Security (Chats policy, Access policy, Agents policy, etc.)
+- **`BrowseSecurity`** — Browse Security (Secure Browsing policy, DLP policy, web access policy, etc.)
+
+Each class manages its own session independently. Create an instance and provide CloudInfra API credentials to connect.
+
+### Obtaining API credentials
+
+1. Go to the [Infinity Portal API Keys page](https://portal.checkpoint.com/dashboard/settings/api-keys).
+2. Click **New** > **New Account API Key**.
+3. In the **Service** dropdown select:
+   - **Workforce AI Security** for `AISecurity`
+   - **Browser Security** for `BrowseSecurity`
+4. Copy the **Client ID**, **Secret Key**, and **Authentication URL** (gateway).
+
+For more information, see [Infinity Portal Administration Guide](https://sc1.checkpoint.com/documents/Infinity_Portal/WebAdminGuides/EN/Infinity-Portal-Admin-Guide/Content/Topics-Infinity-Portal/API-Keys.htm?tocpath=Global%20Settings%7C_____7#API_Keys).
+
+### Available gateways
+
+| Region | Gateway URL |
+|---|---|
+| Europe | `https://cloudinfra-gw.portal.checkpoint.com` |
+| United States | `https://cloudinfra-gw-us.portal.checkpoint.com` |
+
+Once the Client ID, Secret Key, and Authentication URL are obtained, the SDK can be used.
+
+All API operations can be explored on SwaggerHub:
+- [Workforce AI Security APIs](https://app.swaggerhub.com/apis/Check-Point/checkpoint-ai-security)
+- [Browse Security APIs](https://app.swaggerhub.com/apis/Check-Point/checkpoint-browse-security)
+
+### Workforce AI Security example
+
+```python
+from chkp_ai_security_sdk import AISecurity, InfinityPortalAuth
+
+ai = AISecurity()
+ai.connect(infinity_portal_auth=InfinityPortalAuth(
+    client_id="your-client-id",
+    access_key="your-secret-key",
+    gateway="https://cloudinfra-gw-us.portal.checkpoint.com",
+))
+
+rulebase = ai.chats_policy_api.get_chats_rulebase_external_v1_chats_rulebase_get()
+print(rulebase)
+
+ai.disconnect()
+```
+
+### Browse Security example
+
+```python
+from chkp_ai_security_sdk import BrowseSecurity, InfinityPortalAuth
+
+browse = BrowseSecurity()
+browse.connect(infinity_portal_auth=InfinityPortalAuth(
+    client_id="your-client-id",
+    access_key="your-secret-key",
+    gateway="https://cloudinfra-gw-us.portal.checkpoint.com",
+))
+
+rulebase = browse.dlp_policy_api.get_dlp_rulebase_external_v1_dlp_rulebase_get()
+print(rulebase)
+
+browse.disconnect()
+```
+
+### Using both products together
+
+```python
+from chkp_ai_security_sdk import AISecurity, BrowseSecurity, InfinityPortalAuth
+
+auth = InfinityPortalAuth(
+    client_id="your-client-id",
+    access_key="your-secret-key",
+    gateway="https://cloudinfra-gw-us.portal.checkpoint.com",
+)
+
+ai = AISecurity()
+browse = BrowseSecurity()
+
+ai.connect(infinity_portal_auth=auth)
+browse.connect(infinity_portal_auth=auth)
+
+# Workforce AI
+chats = ai.chats_policy_api.get_chats_rulebase_external_v1_chats_rulebase_get()
+
+# Browse Security
+dlp = browse.dlp_policy_api.get_dlp_rulebase_external_v1_dlp_rulebase_get()
+
+ai.disconnect()
+browse.disconnect()
+```
+
+### Async support
+
+The SDK provides async variants for both products:
+
+```python
+import asyncio
+from chkp_ai_security_sdk import AsyncAISecurity, AsyncBrowseSecurity, InfinityPortalAuth
+
+async def main():
+    auth = InfinityPortalAuth(
+        client_id="your-client-id",
+        access_key="your-secret-key",
+        gateway="https://cloudinfra-gw-us.portal.checkpoint.com",
+    )
+
+    ai = AsyncAISecurity()
+    browse = AsyncBrowseSecurity()
+
+    await asyncio.gather(ai.connect(auth), browse.connect(auth))
+
+    # All API calls are awaitable
+    chats = await ai.chats_policy_api.get_chats_rulebase_external_v1_chats_rulebase_get()
+    dlp = await browse.dlp_policy_api.get_dlp_rulebase_external_v1_dlp_rulebase_get()
+
+    await asyncio.gather(ai.disconnect(), browse.disconnect())
+
+asyncio.run(main())
+```
+
+## Troubleshooting and logging
+
+The full version and build info of the SDK is available via the `info()` static method on each class:
+```python
+from chkp_ai_security_sdk import AISecurity, BrowseSecurity
+
+print(AISecurity.info())       # Workforce AI build info
+print(BrowseSecurity.info())   # Browse Security build info
+```
+
+AI Security SDK uses the official python logger package for logging.
+
+There are 3 loggers, for general info, errors and to inspect network.
+
+As default they will be disabled, in order to enable logging, set to the `AI_SECURITY_SDK_LOGGER` environment variable the following string before loading the SDK:
+```bash
+AI_SECURITY_SDK_LOGGER="*"
+```
+
+And for a specific/s logger set the logger name followed by a comma as following:
+```bash
+AI_SECURITY_SDK_LOGGER="info,error,network"
+```
+
+or activate logger programmatically using SDK methods:
+```python
+from chkp_ai_security_sdk import activate_all_loggers, activate_info_logger, activate_error_logger, activate_network_logger
+...
+activate_all_loggers()    # Will activate all loggers
+activate_info_logger()    # Will activate the info logger only
+activate_error_logger()   # Will activate the error logger only
+activate_network_logger() # Will activate the network logger only
+```
+
+## Report Bug
+
+In case of an issue or a bug found in the SDK, please open an [issue](https://github.com/CheckPointSW/ai-security-py-sdk/issues).
+
+## Contributors
+- Haim Kastner - [haimk@checkpoint.com](mailto:haimk@checkpoint.com)

chkp_ai_security_sdk-1.0.0/README.md

@@ -0,0 +1,190 @@
+# Check Point - AI Security Python SDK (Workforce AI + Browse Security)
+
+[![License](https://img.shields.io/github/license/CheckPointSW/ai-security-py-sdk.svg?style=plastic)](https://github.com/CheckPointSW/ai-security-py-sdk/blob/release/LICENSE) [![Latest Release](https://img.shields.io/github/v/release/CheckPointSW/ai-security-py-sdk?style=plastic)](https://github.com/CheckPointSW/ai-security-py-sdk/releases) [![PyPI version](https://img.shields.io/pypi/v/chkp-ai-security-sdk.svg?style=plastic)](https://pypi.org/project/chkp-ai-security-sdk/)
+
+[![Build SDK Package](https://github.com/CheckPointSW/ai-security-py-sdk/actions/workflows/build.yml/badge.svg)](https://github.com/CheckPointSW/ai-security-py-sdk/actions/workflows/build.yml) [![Publish package to PyPI](https://github.com/CheckPointSW/ai-security-py-sdk/actions/workflows/release.yml/badge.svg)](https://github.com/CheckPointSW/ai-security-py-sdk/actions/workflows/release.yml)
+
+This is the official Check Point AI Security SDK for Python, covering both **Workforce AI** and **Browse Security** products.
+
+The SDK is based on the public [AI Security OpenAPI](https://app.swaggerhub.com/apis/Check-Point/checkpoint-ai-security) and [Browse Security OpenAPI](https://app.swaggerhub.com/apis/Check-Point/checkpoint-browse-security) specifications.
+
+With the SDK, you do not have to manage log in, send keep alive requests, or worry about session expiration.
+
+> The SDK supports simultaneous instances with different tenants, and both products can be used independently or together.
+
+## SDK installation
+
+Via PIP (PyPi registry)
+```bash
+pip install chkp-ai-security-sdk
+```
+
+## Getting started
+
+Import the SDK classes for the product(s) you need:
+
+```python
+from chkp_ai_security_sdk import AISecurity, BrowseSecurity
+```
+
+- **`AISecurity`** — Workforce AI Security (Chats policy, Access policy, Agents policy, etc.)
+- **`BrowseSecurity`** — Browse Security (Secure Browsing policy, DLP policy, web access policy, etc.)
+
+Each class manages its own session independently. Create an instance and provide CloudInfra API credentials to connect.
+
+### Obtaining API credentials
+
+1. Go to the [Infinity Portal API Keys page](https://portal.checkpoint.com/dashboard/settings/api-keys).
+2. Click **New** > **New Account API Key**.
+3. In the **Service** dropdown select:
+   - **Workforce AI Security** for `AISecurity`
+   - **Browser Security** for `BrowseSecurity`
+4. Copy the **Client ID**, **Secret Key**, and **Authentication URL** (gateway).
+
+For more information, see [Infinity Portal Administration Guide](https://sc1.checkpoint.com/documents/Infinity_Portal/WebAdminGuides/EN/Infinity-Portal-Admin-Guide/Content/Topics-Infinity-Portal/API-Keys.htm?tocpath=Global%20Settings%7C_____7#API_Keys).
+
+### Available gateways
+
+| Region | Gateway URL |
+|---|---|
+| Europe | `https://cloudinfra-gw.portal.checkpoint.com` |
+| United States | `https://cloudinfra-gw-us.portal.checkpoint.com` |
+
+Once the Client ID, Secret Key, and Authentication URL are obtained, the SDK can be used.
+
+All API operations can be explored on SwaggerHub:
+- [Workforce AI Security APIs](https://app.swaggerhub.com/apis/Check-Point/checkpoint-ai-security)
+- [Browse Security APIs](https://app.swaggerhub.com/apis/Check-Point/checkpoint-browse-security)
+
+### Workforce AI Security example
+
+```python
+from chkp_ai_security_sdk import AISecurity, InfinityPortalAuth
+
+ai = AISecurity()
+ai.connect(infinity_portal_auth=InfinityPortalAuth(
+    client_id="your-client-id",
+    access_key="your-secret-key",
+    gateway="https://cloudinfra-gw-us.portal.checkpoint.com",
+))
+
+rulebase = ai.chats_policy_api.get_chats_rulebase_external_v1_chats_rulebase_get()
+print(rulebase)
+
+ai.disconnect()
+```
+
+### Browse Security example
+
+```python
+from chkp_ai_security_sdk import BrowseSecurity, InfinityPortalAuth
+
+browse = BrowseSecurity()
+browse.connect(infinity_portal_auth=InfinityPortalAuth(
+    client_id="your-client-id",
+    access_key="your-secret-key",
+    gateway="https://cloudinfra-gw-us.portal.checkpoint.com",
+))
+
+rulebase = browse.dlp_policy_api.get_dlp_rulebase_external_v1_dlp_rulebase_get()
+print(rulebase)
+
+browse.disconnect()
+```
+
+### Using both products together
+
+```python
+from chkp_ai_security_sdk import AISecurity, BrowseSecurity, InfinityPortalAuth
+
+auth = InfinityPortalAuth(
+    client_id="your-client-id",
+    access_key="your-secret-key",
+    gateway="https://cloudinfra-gw-us.portal.checkpoint.com",
+)
+
+ai = AISecurity()
+browse = BrowseSecurity()
+
+ai.connect(infinity_portal_auth=auth)
+browse.connect(infinity_portal_auth=auth)
+
+# Workforce AI
+chats = ai.chats_policy_api.get_chats_rulebase_external_v1_chats_rulebase_get()
+
+# Browse Security
+dlp = browse.dlp_policy_api.get_dlp_rulebase_external_v1_dlp_rulebase_get()
+
+ai.disconnect()
+browse.disconnect()
+```
+
+### Async support
+
+The SDK provides async variants for both products:
+
+```python
+import asyncio
+from chkp_ai_security_sdk import AsyncAISecurity, AsyncBrowseSecurity, InfinityPortalAuth
+
+async def main():
+    auth = InfinityPortalAuth(
+        client_id="your-client-id",
+        access_key="your-secret-key",
+        gateway="https://cloudinfra-gw-us.portal.checkpoint.com",
+    )
+
+    ai = AsyncAISecurity()
+    browse = AsyncBrowseSecurity()
+
+    await asyncio.gather(ai.connect(auth), browse.connect(auth))
+
+    # All API calls are awaitable
+    chats = await ai.chats_policy_api.get_chats_rulebase_external_v1_chats_rulebase_get()
+    dlp = await browse.dlp_policy_api.get_dlp_rulebase_external_v1_dlp_rulebase_get()
+
+    await asyncio.gather(ai.disconnect(), browse.disconnect())
+
+asyncio.run(main())
+```
+
+## Troubleshooting and logging
+
+The full version and build info of the SDK is available via the `info()` static method on each class:
+```python
+from chkp_ai_security_sdk import AISecurity, BrowseSecurity
+
+print(AISecurity.info())       # Workforce AI build info
+print(BrowseSecurity.info())   # Browse Security build info
+```
+
+AI Security SDK uses the official python logger package for logging.
+
+There are 3 loggers, for general info, errors and to inspect network.
+
+As default they will be disabled, in order to enable logging, set to the `AI_SECURITY_SDK_LOGGER` environment variable the following string before loading the SDK:
+```bash
+AI_SECURITY_SDK_LOGGER="*"
+```
+
+And for a specific/s logger set the logger name followed by a comma as following:
+```bash
+AI_SECURITY_SDK_LOGGER="info,error,network"
+```
+
+or activate logger programmatically using SDK methods:
+```python
+from chkp_ai_security_sdk import activate_all_loggers, activate_info_logger, activate_error_logger, activate_network_logger
+...
+activate_all_loggers()    # Will activate all loggers
+activate_info_logger()    # Will activate the info logger only
+activate_error_logger()   # Will activate the error logger only
+activate_network_logger() # Will activate the network logger only
+```
+
+## Report Bug
+
+In case of an issue or a bug found in the SDK, please open an [issue](https://github.com/CheckPointSW/ai-security-py-sdk/issues).
+
+## Contributors
+- Haim Kastner - [haimk@checkpoint.com](mailto:haimk@checkpoint.com)

chkp_ai_security_sdk-1.0.0/chkp_ai_security_sdk/__init__.py

@@ -0,0 +1,25 @@
+from chkp_ai_security_sdk.core.ai_security import AISecurity
+from chkp_ai_security_sdk.core.async_ai_security import AsyncAISecurity
+from chkp_ai_security_sdk.core.browse_security import BrowseSecurity
+from chkp_ai_security_sdk.core.async_browse_security import AsyncBrowseSecurity
+from chkp_ai_security_sdk.classes.infinity_portal_auth import InfinityPortalAuth
+from chkp_ai_security_sdk.classes.workforceai_sdk_info import WorkforceAISDKInfo
+from chkp_ai_security_sdk.classes.sdk_connection_state import SDKConnectionState
+from chkp_ai_security_sdk.classes.workforceai_api_exception import WorkforceAIApiException, WorkforceAIErrorScope
+from chkp_ai_security_sdk.core.logger import activate_all_loggers, activate_info_logger, activate_error_logger, activate_network_logger
+
+__all__ = [
+    'AISecurity',
+    'AsyncAISecurity',
+    'BrowseSecurity',
+    'AsyncBrowseSecurity',
+    'InfinityPortalAuth',
+    'WorkforceAISDKInfo',
+    'SDKConnectionState',
+    'WorkforceAIApiException',
+    'WorkforceAIErrorScope',
+    'activate_all_loggers',
+    'activate_info_logger',
+    'activate_error_logger',
+    'activate_network_logger',
+]

chkp_ai_security_sdk-1.0.0/chkp_ai_security_sdk/__init__template.py

@@ -0,0 +1,25 @@
+from chkp_ai_security_sdk.core.ai_security import AISecurity
+from chkp_ai_security_sdk.core.async_ai_security import AsyncAISecurity
+from chkp_ai_security_sdk.core.browse_security import BrowseSecurity
+from chkp_ai_security_sdk.core.async_browse_security import AsyncBrowseSecurity
+from chkp_ai_security_sdk.classes.infinity_portal_auth import InfinityPortalAuth
+from chkp_ai_security_sdk.classes.workforceai_sdk_info import WorkforceAISDKInfo
+from chkp_ai_security_sdk.classes.sdk_connection_state import SDKConnectionState
+from chkp_ai_security_sdk.classes.workforceai_api_exception import WorkforceAIApiException, WorkforceAIErrorScope
+from chkp_ai_security_sdk.core.logger import activate_all_loggers, activate_info_logger, activate_error_logger, activate_network_logger
+
+__all__ = [
+    'AISecurity',
+    'AsyncAISecurity',
+    'BrowseSecurity',
+    'AsyncBrowseSecurity',
+    'InfinityPortalAuth',
+    'WorkforceAISDKInfo',
+    'SDKConnectionState',
+    'WorkforceAIApiException',
+    'WorkforceAIErrorScope',
+    'activate_all_loggers',
+    'activate_info_logger',
+    'activate_error_logger',
+    'activate_network_logger',
+]

chkp_ai_security_sdk-1.0.0/chkp_ai_security_sdk/classes/infinity_portal_auth.py

@@ -0,0 +1,5 @@
+class InfinityPortalAuth:
+    def __init__(self, client_id: str, access_key: str, gateway: str):
+        self.client_id = client_id
+        self.access_key = access_key
+        self.gateway = gateway

chkp_ai_security_sdk-1.0.0/chkp_ai_security_sdk/classes/sdk_connection_state.py

@@ -0,0 +1,8 @@
+from aenum import Enum
+
+
+class SDKConnectionState(Enum):
+    CONNECTED = 'CONNECTED'
+    CONNECTING = 'CONNECTING'
+    DISCONNECTED = 'DISCONNECTED'
+    CONNECTION_ISSUE = 'CONNECTION_ISSUE'

chkp_ai_security_sdk-1.0.0/chkp_ai_security_sdk/classes/workforceai_api_exception.py

@@ -0,0 +1,35 @@
+from typing import Optional, Any
+from enum import Enum
+
+
+class WorkforceAIErrorScope(Enum):
+    NETWORKING = 'NETWORKING'
+    SERVICE = 'SERVICE'
+    SESSION = 'SESSION'
+    INVALID_PARAMS = 'INVALID_PARAMS'
+
+
+class WorkforceAIApiException(Exception):
+    def __init__(
+        self,
+        error_scope: WorkforceAIErrorScope,
+        request_id: str = None,
+        message: Optional[str] = None,
+        url: Optional[str] = None,
+        status_code: Optional[int] = None,
+        payload_error: Optional[Any] = None,
+        network_error: Optional[Any] = None,
+    ):
+        self.error_scope = error_scope
+        self.request_id = request_id
+        self.message = message
+        self.url = url
+        self.status_code = status_code
+        self.payload_error = payload_error
+        self.network_error = network_error
+
+    def __str__(self):
+        return (
+            f"WorkforceAIApiException: error_scope='{self.error_scope}', message='{self.message}', "
+            f"url='{self.url}', status_code='{self.status_code}'"
+        )

chkp_ai_security_sdk-1.0.0/chkp_ai_security_sdk/classes/workforceai_sdk_info.py

@@ -0,0 +1,13 @@
+class WorkforceAISDKInfo:
+    def __init__(self, sdk_build: str, sdk_version: str, spec: str, spec_version: str, released_on: str):
+        self.sdk_build = sdk_build
+        self.sdk_version = sdk_version
+        self.spec = spec
+        self.spec_version = spec_version
+        self.released_on = released_on
+
+    def __str__(self):
+        return (
+            f'sdk_build:"{self.sdk_build}", sdk_version:"{self.sdk_version}", '
+            f'spec:"{self.spec}", spec_version:"{self.spec_version}", released_on:"{self.released_on}"'
+        )

chkp_ai_security_sdk-1.0.0/chkp_ai_security_sdk/core/ai_security.py

@@ -0,0 +1,40 @@
+from chkp_ai_security_sdk.classes.infinity_portal_auth import InfinityPortalAuth
+from chkp_ai_security_sdk.classes.sdk_connection_state import SDKConnectionState
+from chkp_ai_security_sdk.core.session_manager import SessionManager
+from chkp_ai_security_sdk.generated.api import _ApiMixin
+
+print_message = True
+
+
+class AISecurity(_ApiMixin):
+    """Check Point AI Security SDK - manage AI Security policies and assets."""
+
+    def __init__(self):
+        self._session_manager = SessionManager()
+        if print_message:
+            print('Check Point AI Security SDK - https://github.com/CheckPointSW/ai-security-py-sdk')
+
+    def connect(self, infinity_portal_auth: InfinityPortalAuth):
+        """Connect to AI Security service using CloudInfra API credentials."""
+        self._session_manager.connect(infinity_portal_auth)
+
+    def disconnect(self):
+        """Disconnect and stop all background session management."""
+        self._session_manager.disconnect()
+
+    def connection_state(self) -> SDKConnectionState:
+        """Returns the current connection state."""
+        return self._session_manager.connection_state()
+
+    @staticmethod
+    def info() -> str:
+        """Returns SDK build and spec information."""
+        try:
+            from chkp_ai_security_sdk.generated.sdk_build import sdk_build_info
+            return str(sdk_build_info())
+        except Exception:
+            return 'SDK info not available (run sdk_generator/generate_sdk.py first)'
+
+    def _get_api_client(self):
+        from chkp_ai_security_sdk.generated.api_client import ApiClient
+        return ApiClient(self._session_manager.client_configuration)

chkp_ai_security_sdk-1.0.0/chkp_ai_security_sdk/core/async_ai_security.py

@@ -0,0 +1,51 @@
+import asyncio
+from chkp_ai_security_sdk.classes.infinity_portal_auth import InfinityPortalAuth
+from chkp_ai_security_sdk.classes.sdk_connection_state import SDKConnectionState
+from chkp_ai_security_sdk.core.session_manager import SessionManager
+from chkp_ai_security_sdk.generated_async.api import _ApiMixin
+
+
+class AsyncAISecurity(_ApiMixin):
+    """Check Point AI Security SDK (async) - manage AI Security policies and assets."""
+
+    def __init__(self):
+        self._session_manager = SessionManager()
+        self._api_client = None
+
+    async def connect(self, infinity_portal_auth: InfinityPortalAuth):
+        """Connect to AI Security service using CloudInfra API credentials."""
+        await asyncio.to_thread(self._session_manager.connect, infinity_portal_auth)
+
+    async def disconnect(self):
+        """Disconnect and stop all background session management."""
+        if self._api_client is not None:
+            await self._api_client.close()
+            self._api_client = None
+        await asyncio.to_thread(self._session_manager.disconnect)
+
+    def connection_state(self) -> SDKConnectionState:
+        """Returns the current connection state."""
+        return self._session_manager.connection_state()
+
+    @staticmethod
+    def info() -> str:
+        """Returns SDK build and spec information."""
+        try:
+            from chkp_ai_security_sdk.generated_async.sdk_build import sdk_build_info
+            return str(sdk_build_info())
+        except Exception:
+            return 'SDK info not available (run sdk_generator/generate_sdk.py first)'
+
+    def _get_api_client(self):
+        sync_cfg = self._session_manager.client_configuration
+        if self._api_client is None:
+            from chkp_ai_security_sdk.generated_async.api_client import ApiClient
+            from chkp_ai_security_sdk.generated_async.configuration import Configuration
+            cfg = Configuration()
+            cfg.host = sync_cfg.host
+            cfg.access_token = sync_cfg.access_token
+            cfg.client_id = getattr(sync_cfg, 'client_id', None)
+            self._api_client = ApiClient(cfg)
+        else:
+            self._api_client.configuration.access_token = sync_cfg.access_token
+        return self._api_client