lattica-query 0.6.0__tar.gz

lattica_query-0.6.0/LICENSE.txt

@@ -0,0 +1,44 @@
+TEMPORARY END USER LICENSE AGREEMENT FOR LATTICA-QUERY PACKAGE
+Version 1.0 - Effective until April 7, 2025
+
+IMPORTANT: READ CAREFULLY BEFORE DOWNLOADING, INSTALLING, COPYING, OR USING THE SOFTWARE
+
+This Temporary End User License Agreement ("Agreement") is a legal agreement between you (either an individual or a single entity) and LatticaAI Inc. ("Licensor") for the software package identified above, which includes computer software and may include associated media, printed materials, and electronic documentation (collectively, the "Software").
+
+BY DOWNLOADING, INSTALLING, COPYING, OR OTHERWISE USING THE SOFTWARE, YOU AGREE TO BE BOUND BY THE TERMS OF THIS AGREEMENT. IF YOU DO NOT AGREE TO THE TERMS OF THIS AGREEMENT, DO NOT DOWNLOAD, INSTALL, COPY, OR USE THE SOFTWARE.
+
+1. GRANT OF LICENSE
+   This is a TEMPORARY, NON-EXCLUSIVE, NON-TRANSFERABLE, REVOCABLE license to download and use the Software solely for evaluation purposes. This license explicitly DOES NOT permit any other use of the Software.
+
+2. RESTRICTIONS
+   a. You may NOT use the Software for any commercial purpose.
+   b. You may NOT redistribute, sell, rent, lease, sublicense, or otherwise transfer rights to the Software.
+   c. You may NOT modify, translate, reverse engineer, decompile, disassemble, or create derivative works based on the Software.
+   d. You may NOT copy the Software except as expressly permitted by this Agreement.
+   e. You may NOT remove any proprietary notices or labels on the Software.
+   f. You may NOT use the Software in any manner that could damage, disable, overburden, or impair Licensor's services.
+   g. You may NOT use the Software to violate any applicable laws or regulations.
+   h. You may NOT publish any benchmark or performance tests run on the Software.
+
+3. NO WARRANTY
+   THE SOFTWARE IS PROVIDED "AS IS" WITHOUT WARRANTY OF ANY KIND, EITHER EXPRESS OR IMPLIED, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, AND NONINFRINGEMENT. THE ENTIRE RISK ARISING OUT OF USE OR PERFORMANCE OF THE SOFTWARE REMAINS WITH YOU.
+
+4. LIMITATION OF LIABILITY
+   IN NO EVENT SHALL LICENSOR BE LIABLE FOR ANY DAMAGES WHATSOEVER (INCLUDING, WITHOUT LIMITATION, DAMAGES FOR LOSS OF BUSINESS PROFITS, BUSINESS INTERRUPTION, LOSS OF BUSINESS INFORMATION, OR ANY OTHER PECUNIARY LOSS) ARISING OUT OF THE USE OF OR INABILITY TO USE THE SOFTWARE, EVEN IF LICENSOR HAS BEEN ADVISED OF THE POSSIBILITY OF SUCH DAMAGES.
+
+5. TERMINATION
+   a. This Agreement is effective until April 7, 2025 or until terminated by Licensor.
+   b. This Agreement will terminate automatically without notice if you fail to comply with any provision of this Agreement.
+   c. Upon termination, you must destroy all copies of the Software.
+   d. Licensor reserves the right to replace this temporary Agreement with a permanent license agreement at any time.
+
+6. GOVERNING LAW
+   This Agreement shall be governed by the laws of the jurisdiction in which Licensor is located, without regard to its conflict of law principles.
+
+7. ENTIRE AGREEMENT
+   This Agreement constitutes the entire agreement between you and Licensor and supersedes any prior statements, representations, or agreements, whether oral or written.
+
+8. CONTACT INFORMATION
+   For questions regarding this Agreement, please contact: support@lattica.ai
+
+© 2025 LatticaAI Inc. All rights reserved.

lattica_query-0.6.0/PKG-INFO

@@ -0,0 +1,29 @@
+Metadata-Version: 2.3
+Name: lattica-query
+Version: 0.6.0
+Summary: Lattica client
+License: Proprietary
+Author: LatticaAI Inc.
+Author-email: hello@lattica.ai
+Requires-Python: >=3.12,<4.0
+Classifier: License :: Other/Proprietary License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Dist: lattica-common (>=0.5.0,<0.6.0)
+Requires-Dist: timeit-decorator
+Project-URL: Homepage, https://www.lattica.ai/
+Description-Content-Type: text/markdown
+
+# Lattica Query
+
+This is the official Python client for Lattica's homomorphic encryption service.
+
+## Important Notice
+
+This package is distributed under a restrictive proprietary license. Please see LICENSE.txt for the complete terms and conditions.
+
+**Contact**: support@lattica.ai
+
+© 2025 LatticaAI Inc. All rights reserved.
+

lattica_query-0.6.0/README.md

@@ -0,0 +1,11 @@
+# Lattica Query
+
+This is the official Python client for Lattica's homomorphic encryption service.
+
+## Important Notice
+
+This package is distributed under a restrictive proprietary license. Please see LICENSE.txt for the complete terms and conditions.
+
+**Contact**: support@lattica.ai
+
+© 2025 LatticaAI Inc. All rights reserved.

lattica_query-0.6.0/lattica_query/auth.py

@@ -0,0 +1,18 @@
+from lattica_common.app_api import HttpClient, generate_random_token_name
+
+
+def get_demo_token(model_id: str) -> str:
+    http_client = HttpClient(None)
+    token_name = generate_random_token_name(10)
+    response = http_client.send_http_request(
+        "api/token/generate_token_demo",
+        req_params={
+            'modelId': model_id,
+            'tokenName': token_name
+        })
+    token = response.get('token')
+
+    if token is None:
+        raise ValueError("The response does not contain a 'token' field.")
+
+    return token

lattica_query-0.6.0/lattica_query/external_demos/user_client_init.py

@@ -0,0 +1,14 @@
+from lattica_query import hom_common
+from lattica_common.app_api import LatticaAppAPI
+from lattica_query.worker_api import LatticaWorkerAPI
+
+# TODO (pavel): this script is not up to date and won't work!
+worker_api_client = LatticaWorkerAPI("<YOUR-TOKEN-HERE>")
+app_client = LatticaAppAPI("<YOUR-TOKEN-HERE>")
+
+hom_common.user_client_init(
+    worker_api_client,
+    app_client,
+    secret_key_file_path='my_secret_key.lsk',
+)
+

lattica_query-0.6.0/lattica_query/external_demos/user_client_query.py

@@ -0,0 +1,28 @@
+from lattica_query.internal_demos.lattica_query_client_local import LocalQueryClient
+from lattica_common.internal_demos_common.common_demos_utils import print_query_result
+from lattica_query.worker_api import LatticaWorkerAPI
+import torch
+
+import lattica_query.hom_common as hom_common
+
+
+# TODO (pavel): this script is not up to date and won't work!
+worker_api_client = LatticaWorkerAPI("<YOUR-TOKEN-HERE>")
+query_client = LocalQueryClient("<YOUR-TOKEN-HERE>")
+
+# or you can pass the token as an arg to `get_context':
+serialized_context, serialized_homseq = worker_api_client.get_user_init_data()
+
+print('Loading local secret key...')
+serialized_sk = hom_common.read_byte_arrays_from_file('my_secret_key.lsk')
+
+for i in range(5):
+    print(f'Iteration {i=}')
+
+    # create a plaintext tensor with the correct shape
+    data_pt = torch.rand(10, dtype=torch.float64)
+
+    pt_expected = query_client.apply_clear(data_pt)
+    print(f'Image {i=}')
+    pt_dec = query_client.run_query(serialized_context, serialized_sk, data_pt, serialized_homseq)
+    print_query_result(i, data_pt, pt_expected, pt_dec)

lattica_query-0.6.0/lattica_query/hom_common.py

@@ -0,0 +1,61 @@
+from typing import Optional, Tuple, Union, TYPE_CHECKING
+import os
+
+from lattica_common.app_api import LatticaAppAPI
+
+from lattica import interface_query as toolkit_interface
+from lattica_query.worker_api import LatticaWorkerAPI
+
+
+def _write_byte_arrays_to_file(file_path, byte_array1, byte_array2):
+    with open(file_path, 'wb') as f:
+        # Write the length of the first byte array
+        f.write(len(byte_array1).to_bytes(4, byteorder='little'))
+        # Write the first byte array
+        f.write(byte_array1)
+        # Write the length of the second byte array
+        f.write(len(byte_array2).to_bytes(4, byteorder='little'))
+        # Write the second byte array
+        f.write(byte_array2)
+
+
+def read_byte_arrays_from_file(file_path):
+    with open(file_path, 'rb') as f:
+        # Read the length of the first byte array
+        len1 = int.from_bytes(f.read(4), byteorder='little')
+        # Read the first byte array
+        byte_array1 = f.read(len1)
+        # Read the length of the second byte array
+        len2 = int.from_bytes(f.read(4), byteorder='little')
+        # Read the second byte array
+        byte_array2 = f.read(len2)
+
+    return byte_array1, byte_array2
+
+
+def user_client_init(
+        worker_api_client: LatticaWorkerAPI,
+        app_client:  LatticaAppAPI,
+        secret_key_file_path: Union[str, os.PathLike] = 'my_secret_key.lsk',
+     ) -> None:
+
+    serialized_context, serialized_homseq = worker_api_client.get_user_init_data()
+
+    print(f'Creating client FHE keys...')
+    serialized_secret_key, serialized_public_key = toolkit_interface.generate_key(
+        serialized_homseq, serialized_context)
+
+    # Store locally for later use file
+    _write_byte_arrays_to_file(secret_key_file_path, *serialized_secret_key)
+
+    print(f'Registering FHE public key...')
+    temp_filename = 'my_pk.lpk'
+    # Store as temp file
+    with open(temp_filename, 'wb') as handle:
+        handle.write(serialized_public_key)
+    # Upload to server
+    app_client.upload_user_file(temp_filename)
+    # Delete temp file
+    os.remove(temp_filename)
+    print(f'Preprocessing public key...')
+    worker_api_client.preprocess_pk()