numind 0.0.2__tar.gz → 0.1.0__tar.gz

{numind-0.0.2 → numind-0.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: numind
-Version: 0.0.2
+Version: 0.1.0
 Summary: SDK to interact with the NuMind models API.
 Project-URL: Homepage, https://github.com/numindai/nuextract-platform-sdk
 Project-URL: Repository, https://github.com/numindai/nuextract-platform-sdk.git
@@ -43,6 +43,8 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Requires-Python: >=3.9
+Requires-Dist: aiohttp-retry>=2.8.3
+Requires-Dist: aiohttp>=3.8.4
 Requires-Dist: lazy-imports<2,>=1
 Requires-Dist: pydantic>=2
 Requires-Dist: python-dateutil>=2.8.2
@@ -54,6 +56,7 @@ Requires-Dist: pyyaml; extra == 'build'
 Requires-Dist: toml; extra == 'build'
 Requires-Dist: uv; extra == 'build'
 Provides-Extra: tests
+Requires-Dist: pytest-asyncio; extra == 'tests'
 Requires-Dist: pytest-cov; extra == 'tests'
 Requires-Dist: pytest-dependency; extra == 'tests'
 Requires-Dist: pytest-rerunfailures; extra == 'tests'
@@ -86,6 +89,28 @@ from numind import NuMind
 client = NuMind(api_key=os.environ["NUMIND_API_KEY"])
 ```
 
+### Create an async client
+
+You can create an **async** client by using the `NuMindAsync` class:
+
+```python
+import asyncio
+from numind import NuMindAsync
+
+client = NuMindAsync(api_key="API_KEY")
+requests = [{}]
+
+async def main():
+    return [
+        await client.extract(project_id, **request_kwargs)
+        for request_kwargs in requests
+    ]
+
+responses = asyncio.run(main())
+```
+
+The methods and their usages are the same as for the sync `NuMind` client except that API methods are coroutines that must be awaited.
+
 ### Extract structured information "on the fly"
 
 If you want to extract structured information from data without projects but just by providing the input template, you can use the `extract` method which provides a more user-friendly way to interact with the API:
@@ -330,6 +355,7 @@ Class | Method | HTTP request | Description
 *FilesApi* | [**post_api_files**](docs/FilesApi.md#post_api_files) | **POST** /api/files | 
 *FilesApi* | [**post_api_files_fileid_convert_to_document**](docs/FilesApi.md#post_api_files_fileid_convert_to_document) | **POST** /api/files/{fileId}/convert-to-document | 
 *InferenceApi* | [**post_api_infer_template**](docs/InferenceApi.md#post_api_infer_template) | **POST** /api/infer-template | 
+*InferenceApi* | [**post_api_infer_template_async**](docs/InferenceApi.md#post_api_infer_template_async) | **POST** /api/infer-template-async | 
 *InferenceApi* | [**post_api_infer_template_document_documentid**](docs/InferenceApi.md#post_api_infer_template_document_documentid) | **POST** /api/infer-template/document/{documentId} | 
 *InferenceApi* | [**post_api_infer_template_file**](docs/InferenceApi.md#post_api_infer_template_file) | **POST** /api/infer-template/file | 
 *InferenceApi* | [**post_api_projects_projectid_infer_document_async_documentid**](docs/InferenceApi.md#post_api_projects_projectid_infer_document_async_documentid) | **POST** /api/projects/{projectId}/infer-document-async/{documentId} | 
@@ -397,6 +423,7 @@ Class | Method | HTTP request | Description
  - [JobIdResponse](docs/JobIdResponse.md)
  - [JobResponse](docs/JobResponse.md)
  - [MemberResponse](docs/MemberResponse.md)
+ - [ModelUsageResponse](docs/ModelUsageResponse.md)
  - [OrganizationResponse](docs/OrganizationResponse.md)
  - [PlaygroundItemResponse](docs/PlaygroundItemResponse.md)
  - [ProjectResponse](docs/ProjectResponse.md)

{numind-0.0.2 → numind-0.1.0}/README.md

@@ -25,6 +25,28 @@ from numind import NuMind
 client = NuMind(api_key=os.environ["NUMIND_API_KEY"])
 ```
 
+### Create an async client
+
+You can create an **async** client by using the `NuMindAsync` class:
+
+```python
+import asyncio
+from numind import NuMindAsync
+
+client = NuMindAsync(api_key="API_KEY")
+requests = [{}]
+
+async def main():
+    return [
+        await client.extract(project_id, **request_kwargs)
+        for request_kwargs in requests
+    ]
+
+responses = asyncio.run(main())
+```
+
+The methods and their usages are the same as for the sync `NuMind` client except that API methods are coroutines that must be awaited.
+
 ### Extract structured information "on the fly"
 
 If you want to extract structured information from data without projects but just by providing the input template, you can use the `extract` method which provides a more user-friendly way to interact with the API:
@@ -269,6 +291,7 @@ Class | Method | HTTP request | Description
 *FilesApi* | [**post_api_files**](docs/FilesApi.md#post_api_files) | **POST** /api/files | 
 *FilesApi* | [**post_api_files_fileid_convert_to_document**](docs/FilesApi.md#post_api_files_fileid_convert_to_document) | **POST** /api/files/{fileId}/convert-to-document | 
 *InferenceApi* | [**post_api_infer_template**](docs/InferenceApi.md#post_api_infer_template) | **POST** /api/infer-template | 
+*InferenceApi* | [**post_api_infer_template_async**](docs/InferenceApi.md#post_api_infer_template_async) | **POST** /api/infer-template-async | 
 *InferenceApi* | [**post_api_infer_template_document_documentid**](docs/InferenceApi.md#post_api_infer_template_document_documentid) | **POST** /api/infer-template/document/{documentId} | 
 *InferenceApi* | [**post_api_infer_template_file**](docs/InferenceApi.md#post_api_infer_template_file) | **POST** /api/infer-template/file | 
 *InferenceApi* | [**post_api_projects_projectid_infer_document_async_documentid**](docs/InferenceApi.md#post_api_projects_projectid_infer_document_async_documentid) | **POST** /api/projects/{projectId}/infer-document-async/{documentId} | 
@@ -336,6 +359,7 @@ Class | Method | HTTP request | Description
  - [JobIdResponse](docs/JobIdResponse.md)
  - [JobResponse](docs/JobResponse.md)
  - [MemberResponse](docs/MemberResponse.md)
+ - [ModelUsageResponse](docs/ModelUsageResponse.md)
  - [OrganizationResponse](docs/OrganizationResponse.md)
  - [PlaygroundItemResponse](docs/PlaygroundItemResponse.md)
  - [ProjectResponse](docs/ProjectResponse.md)

{numind-0.0.2 → numind-0.1.0}/docs/ExtractionApi.md

@@ -9,12 +9,13 @@ Method | HTTP request | Description
 
 
 # **post_api_projects_projectid_extract**
-> ExtractionResponse post_api_projects_projectid_extract(project_id, body, temperature=temperature, rasterization_dpi=rasterization_dpi, max_output_tokens=max_output_tokens)
+> ExtractionResponse post_api_projects_projectid_extract(project_id, body, temperature=temperature, rasterization_dpi=rasterization_dpi, max_output_tokens=max_output_tokens, degraded_mode=degraded_mode, max_tokens_smart_example=max_tokens_smart_example)
 
 
  Extract information from the provided text or file. Some files are converted to images - 
-the **rasterizationDPI** parameter controls their resolution. When **temperature**, **rasterizationDPI** 
- or **maxOutputTokens** parameters are not specified, they are set to their project-setting values.
+the **rasterizationDPI** parameter controls their resolution. When **temperature**, **rasterizationDPI**, 
+**maxOutputTokens**, **degradedMode** and **maxTokensSmartExample** parameters are not specified, 
+they are set to their project-setting values.
 
 #### Response:
 Returns a JSON representing the extracted information.
@@ -61,9 +62,11 @@ with numind.openapi_client.ApiClient(configuration) as api_client:
     temperature = 3.4 # float | Model temperature (optional). Controls output diversity.  When not specified, the project value is used.   Ranges between 0 and 1. (optional)
     rasterization_dpi = 56 # int | Resolution used to convert formatted documents (PDFs, etc.) to images, in dot per inch (optional).   Ranges between 1 and 300. (optional)
     max_output_tokens = 56 # int | Maximum number of output tokens (optional). Must be positive. Set to 0 for no limit. (optional)
+    degraded_mode = 'degraded_mode_example' # str | Controls whether a response is returned when smart example is not functionning. Rejects by default. (optional)
+    max_tokens_smart_example = 56 # int | Controls the maximum number of tokens that can be allocated to the examples.  Must be positive. Ranges in the context window of the model. (optional)
 
     try:
-        api_response = api_instance.post_api_projects_projectid_extract(project_id, body, temperature=temperature, rasterization_dpi=rasterization_dpi, max_output_tokens=max_output_tokens)
+        api_response = api_instance.post_api_projects_projectid_extract(project_id, body, temperature=temperature, rasterization_dpi=rasterization_dpi, max_output_tokens=max_output_tokens, degraded_mode=degraded_mode, max_tokens_smart_example=max_tokens_smart_example)
         print("The response of ExtractionApi->post_api_projects_projectid_extract:\n")
         pprint(api_response)
     except Exception as e:
@@ -82,6 +85,8 @@ Name | Type | Description  | Notes
  **temperature** | **float**| Model temperature (optional). Controls output diversity.  When not specified, the project value is used.   Ranges between 0 and 1. | [optional] 
  **rasterization_dpi** | **int**| Resolution used to convert formatted documents (PDFs, etc.) to images, in dot per inch (optional).   Ranges between 1 and 300. | [optional] 
  **max_output_tokens** | **int**| Maximum number of output tokens (optional). Must be positive. Set to 0 for no limit. | [optional] 
+ **degraded_mode** | **str**| Controls whether a response is returned when smart example is not functionning. Rejects by default. | [optional] 
+ **max_tokens_smart_example** | **int**| Controls the maximum number of tokens that can be allocated to the examples.  Must be positive. Ranges in the context window of the model. | [optional] 
 
 ### Return type
 
@@ -101,13 +106,13 @@ Name | Type | Description  | Notes
 | Status code | Description | Response headers |
 |-------------|-------------|------------------|
 **200** |  |  -  |
-**400** | Invalid value for: query parameter temperature, Invalid value for: query parameter rasterizationDPI, Invalid value for: query parameter maxOutputTokens, Invalid value for: body |  -  |
+**400** | Invalid value for: query parameter temperature, Invalid value for: query parameter rasterizationDPI, Invalid value for: query parameter maxOutputTokens, Invalid value for: query parameter maxTokensSmartExample, Invalid value for: body |  -  |
 **0** |  |  -  |
 
 [[Back to top]](#) [[Back to API list]](../README.md#documentation-for-api-endpoints) [[Back to Model list]](../README.md#documentation-for-models) [[Back to README]](../README.md)
 
 # **post_api_projects_projectid_extract_async**
-> JobIdResponse post_api_projects_projectid_extract_async(project_id, body, temperature=temperature, dpi=dpi, max_output_tokens=max_output_tokens)
+> JobIdResponse post_api_projects_projectid_extract_async(project_id, body, temperature=temperature, dpi=dpi, max_output_tokens=max_output_tokens, degraded_mode=degraded_mode, max_tokens_smart_example=max_tokens_smart_example)
 
 Start file extraction as async job
 
@@ -143,9 +148,11 @@ with numind.openapi_client.ApiClient(configuration) as api_client:
     temperature = 3.4 # float | Model temperature (optional). Controls output diversity.  When not specified, the project value is used.   Ranges between 0 and 1. (optional)
     dpi = 56 # int | DPI for file rasterization (optional).  When not specified, the project value is used.   If the file is already an image or a text, this parameter is ignored. (optional)
     max_output_tokens = 56 # int | Maximum number of output tokens (optional).  When not specified, the project value is used.   Use 0 to indicate no limit. (optional)
+    degraded_mode = 'degraded_mode_example' # str | Controls whether a response is returned when smart example is not functionning. Rejects by default. (optional)
+    max_tokens_smart_example = 56 # int | Controls the maximum number of tokens that can be allocated to the examples.  Must be positive. Ranges in the context window of the model. (optional)
 
     try:
-        api_response = api_instance.post_api_projects_projectid_extract_async(project_id, body, temperature=temperature, dpi=dpi, max_output_tokens=max_output_tokens)
+        api_response = api_instance.post_api_projects_projectid_extract_async(project_id, body, temperature=temperature, dpi=dpi, max_output_tokens=max_output_tokens, degraded_mode=degraded_mode, max_tokens_smart_example=max_tokens_smart_example)
         print("The response of ExtractionApi->post_api_projects_projectid_extract_async:\n")
         pprint(api_response)
     except Exception as e:
@@ -164,6 +171,8 @@ Name | Type | Description  | Notes
  **temperature** | **float**| Model temperature (optional). Controls output diversity.  When not specified, the project value is used.   Ranges between 0 and 1. | [optional] 
  **dpi** | **int**| DPI for file rasterization (optional).  When not specified, the project value is used.   If the file is already an image or a text, this parameter is ignored. | [optional] 
  **max_output_tokens** | **int**| Maximum number of output tokens (optional).  When not specified, the project value is used.   Use 0 to indicate no limit. | [optional] 
+ **degraded_mode** | **str**| Controls whether a response is returned when smart example is not functionning. Rejects by default. | [optional] 
+ **max_tokens_smart_example** | **int**| Controls the maximum number of tokens that can be allocated to the examples.  Must be positive. Ranges in the context window of the model. | [optional] 
 
 ### Return type
 
@@ -183,7 +192,7 @@ Name | Type | Description  | Notes
 | Status code | Description | Response headers |
 |-------------|-------------|------------------|
 **200** |  |  -  |
-**400** | Invalid value for: query parameter temperature, Invalid value for: query parameter dpi, Invalid value for: query parameter maxOutputTokens, Invalid value for: body |  -  |
+**400** | Invalid value for: query parameter temperature, Invalid value for: query parameter dpi, Invalid value for: query parameter maxOutputTokens, Invalid value for: query parameter maxTokensSmartExample, Invalid value for: body |  -  |
 **0** |  |  -  |
 
 [[Back to top]](#) [[Back to API list]](../README.md#documentation-for-api-endpoints) [[Back to Model list]](../README.md#documentation-for-models) [[Back to README]](../README.md)

{numind-0.0.2 → numind-0.1.0}/docs/InferenceApi.md

@@ -5,6 +5,7 @@ All URIs are relative to *https://nuextract.ai*
 Method | HTTP request | Description
 ------------- | ------------- | -------------
 [**post_api_infer_template**](InferenceApi.md#post_api_infer_template) | **POST** /api/infer-template | 
+[**post_api_infer_template_async**](InferenceApi.md#post_api_infer_template_async) | **POST** /api/infer-template-async | 
 [**post_api_infer_template_document_documentid**](InferenceApi.md#post_api_infer_template_document_documentid) | **POST** /api/infer-template/document/{documentId} | 
 [**post_api_infer_template_file**](InferenceApi.md#post_api_infer_template_file) | **POST** /api/infer-template/file | 
 [**post_api_projects_projectid_infer_document_async_documentid**](InferenceApi.md#post_api_projects_projectid_infer_document_async_documentid) | **POST** /api/projects/{projectId}/infer-document-async/{documentId} | 
@@ -97,6 +98,91 @@ Name | Type | Description  | Notes
 
 [[Back to top]](#) [[Back to API list]](../README.md#documentation-for-api-endpoints) [[Back to Model list]](../README.md#documentation-for-models) [[Back to README]](../README.md)
 
+# **post_api_infer_template_async**
+> JobIdResponse post_api_infer_template_async(template_request)
+
+
+ Derive a template from the provided natural language description.
+ Potentially, this endpoint can equally be used to correct the template to conform to the NuExtract standard.
+ The resulting template is a JSON object that can be used as a project template.
+
+
+#### Response:
+ Returns a JSON representing the derived template.
+ The response is an empty template if the derivation fails.
+
+   
+
+### Example
+
+* OAuth Authentication (oauth2Auth):
+
+```python
+import numind.openapi_client
+from numind.openapi_client.models.job_id_response import JobIdResponse
+from numind.openapi_client.models.template_request import TemplateRequest
+from numind.openapi_client.rest import ApiException
+from pprint import pprint
+
+# Defining the host is optional and defaults to https://nuextract.ai
+# See configuration.py for a list of all supported configuration parameters.
+configuration = numind.openapi_client.Configuration(
+    host = "https://nuextract.ai"
+)
+
+# The client must configure the authentication and authorization parameters
+# in accordance with the API server security policy.
+# Examples for each auth method are provided below, use the example that
+# satisfies your auth use case.
+
+configuration.access_token = os.environ["ACCESS_TOKEN"]
+
+# Enter a context with an instance of the API client
+with numind.openapi_client.ApiClient(configuration) as api_client:
+    # Create an instance of the API class
+    api_instance = numind.openapi_client.InferenceApi(api_client)
+    template_request = {description=[EXAMPLE ONLY] Create a template that extracts key information from an order confirmation email. The template should be able to pull details like the order ID, customer ID, date and time of the order, status, total amount, currency, item details (product ID, quantity, and unit price), shipping address, any customer requests or delivery preferences, and the estimated delivery date.} # TemplateRequest | 
+
+    try:
+        api_response = api_instance.post_api_infer_template_async(template_request)
+        print("The response of InferenceApi->post_api_infer_template_async:\n")
+        pprint(api_response)
+    except Exception as e:
+        print("Exception when calling InferenceApi->post_api_infer_template_async: %s\n" % e)
+```
+
+
+
+### Parameters
+
+
+Name | Type | Description  | Notes
+------------- | ------------- | ------------- | -------------
+ **template_request** | [**TemplateRequest**](TemplateRequest.md)|  | 
+
+### Return type
+
+[**JobIdResponse**](JobIdResponse.md)
+
+### Authorization
+
+[oauth2Auth](../README.md#oauth2Auth)
+
+### HTTP request headers
+
+ - **Content-Type**: application/json
+ - **Accept**: application/json, text/plain
+
+### HTTP response details
+
+| Status code | Description | Response headers |
+|-------------|-------------|------------------|
+**200** |  |  -  |
+**400** | Invalid value for: body |  -  |
+**0** |  |  -  |
+
+[[Back to top]](#) [[Back to API list]](../README.md#documentation-for-api-endpoints) [[Back to Model list]](../README.md#documentation-for-models) [[Back to README]](../README.md)
+
 # **post_api_infer_template_document_documentid**
 > object post_api_infer_template_document_documentid(document_id)
 

numind-0.1.0/docs/ModelUsageResponse.md

@@ -0,0 +1,31 @@
+# ModelUsageResponse
+
+
+## Properties
+
+Name | Type | Description | Notes
+------------ | ------------- | ------------- | -------------
+**usage** | **int** |  | 
+**balance** | **int** |  | 
+**overage** | **int** |  | 
+
+## Example
+
+```python
+from numind.openapi_client.models.model_usage_response import ModelUsageResponse
+
+# TODO update the JSON string below
+json = "{}"
+# create an instance of ModelUsageResponse from a JSON string
+model_usage_response_instance = ModelUsageResponse.from_json(json)
+# print the JSON string representation of the object
+print(ModelUsageResponse.to_json())
+
+# convert the object into a dict
+model_usage_response_dict = model_usage_response_instance.to_dict()
+# create an instance of ModelUsageResponse from a dict
+model_usage_response_from_dict = ModelUsageResponse.from_dict(model_usage_response_dict)
+```
+[[Back to Model list]](../README.md#documentation-for-models) [[Back to API list]](../README.md#documentation-for-api-endpoints) [[Back to README]](../README.md)
+
+

{numind-0.0.2 → numind-0.1.0}/docs/ProjectManagementApi.md

@@ -398,7 +398,7 @@ with numind.openapi_client.ApiClient(configuration) as api_client:
     # Create an instance of the API class
     api_instance = numind.openapi_client.ProjectManagementApi(api_client)
     project_id = 'project_id_example' # str | Unique project identifier.
-    update_project_settings_request = {"temperature":0,"rasterizationDPI":115,"maxOutputTokens":0} # UpdateProjectSettingsRequest | 
+    update_project_settings_request = {"temperature":0,"rasterizationDPI":115,"maxOutputTokens":0,"degradedMode":"Reject","maxTokensSmartExample":16000} # UpdateProjectSettingsRequest | 
 
     try:
         api_response = api_instance.patch_api_projects_projectid_settings(project_id, update_project_settings_request)
@@ -464,6 +464,8 @@ A **Project** is created with default settings:
  `temperature` | 0.0 |
  `rasterizationDPI` | 115 |
  `maxOutputTokens` | 0 (no limit) |
+ `degradedMode` | Reject |
+ `maxTokensSmartExample` | 16000 |
 
 
 If *ownerOrganization* is not provided, the **Project** will be owned by the authenticated user.
@@ -733,6 +735,8 @@ Default values are:
  `temperature` | 0.0 |
  `rasterizationDPI` | 115 |
  `maxOutputTokens` | 0 (no limit) |
+ `degradedMode` | Reject |
+ `maxTokensSmartExample` | 16000 |
 
 #### Error Responses:
 `404 Not Found` - If a **Project** with the specified `projectId` does not exist.

{numind-0.0.2 → numind-0.1.0}/docs/ProjectSettingsResponse.md

@@ -9,6 +9,8 @@ Name | Type | Description | Notes
 **temperature** | **float** | Model temperature. | 
 **rasterization_dpi** | **int** | Resolution used to convert formatted documents to images. | 
 **max_output_tokens** | **int** | Maximum number of output tokens (optional). Must be positive. Set to 0 for no limit. | 
+**degraded_mode** | **str** | Controls whether a response is returned when smart example is not functionning. Rejects by default. | 
+**max_tokens_smart_example** | **int** | Maximum number of output tokens for smart examples (optional). Must be positive. | 
 
 ## Example
 

{numind-0.0.2 → numind-0.1.0}/docs/UpdateProjectSettingsRequest.md

@@ -8,6 +8,8 @@ Name | Type | Description | Notes
 **temperature** | **float** | Model temperature (optional). | [optional] 
 **rasterization_dpi** | **int** | Resolution used to convert formatted documents to images (optional). | [optional] 
 **max_output_tokens** | **int** | Maximum number of output tokens (optional). Must be positive. Set to 0 for no limit. | [optional] 
+**degraded_mode** | **str** | Controls whether a response is returned when smart example is not functionning. Rejects by default. | [optional] 
+**max_tokens_smart_example** | **int** | Maximum number of output tokens for smart examples (optional). Must be positive. | [optional] 
 
 ## Example
 

{numind-0.0.2 → numind-0.1.0}/pyproject.toml

@@ -4,12 +4,12 @@ build-backend = "hatchling.build"
 
 [project]
 name = "numind"
-version = "0.0.2"
+version = "0.1.0"
 description = "SDK to interact with the NuMind models API."
 requires-python = ">=3.9"
 keywords = [ "artificial intelligence", "deep learning", "structured",]
 classifiers = [ "Intended Audience :: Developers", "Intended Audience :: Science/Research", "Topic :: Scientific/Engineering :: Artificial Intelligence", "License :: OSI Approved :: MIT License", "Programming Language :: Python", "Programming Language :: Python :: 3 :: Only", "Programming Language :: Python :: 3.9", "Programming Language :: Python :: 3.10", "Programming Language :: Python :: 3.11", "Programming Language :: Python :: 3.12", "Programming Language :: Python :: 3.13", "Operating System :: OS Independent",]
-dependencies = [ "urllib3 >= 2.1.0, < 3.0.0", "python_dateutil >= 2.8.2", "pydantic >= 2", "typing-extensions >= 4.7.1", "lazy-imports >= 1, < 2",]
+dependencies = [ "urllib3 >= 2.1.0, < 3.0.0", "python_dateutil >= 2.8.2", "aiohttp >= 3.8.4", "aiohttp-retry >= 2.8.3", "pydantic >= 2", "typing-extensions >= 4.7.1", "lazy-imports >= 1, < 2",]
 [[project.authors]]
 name = "Nathan Fradet"
 email = "nathan@numind.ai"
@@ -23,7 +23,7 @@ file = "LICENSE"
 
 [project.optional-dependencies]
 build = [ "openapi-generator-cli", "uv", "PyYAML", "toml",]
-tests = [ "pytest-cov", "pytest-rerunfailures", "pytest-dependency",]
+tests = [ "pytest-asyncio", "pytest-cov", "pytest-rerunfailures", "pytest-dependency",]
 
 [project.urls]
 Homepage = "https://github.com/numindai/nuextract-platform-sdk"
@@ -35,7 +35,7 @@ Issues = "https://github.com/numindai/nuextract-platform-sdk/issues"
 target-version = "py39"
 
 [tool.mypy]
-files = [ "numind.openapi_client", "tests",]
+files = [ "numind.openapi_client_async", "tests",]
 warn_unused_configs = true
 warn_redundant_casts = true
 warn_unused_ignores = true
@@ -46,7 +46,7 @@ disallow_subclassing_any = true
 disallow_untyped_decorators = true
 disallow_any_generics = true
 [[tool.mypy.overrides]]
-module = [ "numind.openapi_client.configuration",]
+module = [ "numind.openapi_client_async.configuration",]
 warn_unused_ignores = true
 strict_equality = true
 extra_checks = true

numind-0.1.0/sdk_generation/adapt_openapi_client_init.py

@@ -0,0 +1,150 @@
+"""Adds async imports to the openapi_client package."""  # noqa:INP001
+
+import re
+from pathlib import Path
+
+
+def modify_init_file(file_path: Path) -> None:
+    """
+    Modify the __init__.py file to add async versions of imports.
+
+    Args:
+        file_path (str): Path to the __init__.py file
+
+    """
+    with file_path.open() as f:
+        content = f.read()
+
+    # First, we need to update the __all__ list to include async versions
+    all_pattern = r"__all__\s*=\s*\[(.*?)\]"
+    all_match = re.search(all_pattern, content, re.DOTALL)
+
+    if all_match:
+        all_content = all_match.group(1)
+        # Extract existing items and add async versions
+        items = re.findall(r'"([^"]*)"', all_content)
+        async_items = []
+
+        for item in items:
+            if item.endswith("Api"):
+                async_items.append(f'"{item}Async"')
+            elif item == "ApiClient":
+                async_items.append('"ApiClientAsync"')
+
+        # Create new __all__ content
+        all_items = [f'"{item}"' for item in items] + async_items
+        new_all_content = "__all__ = [\n    " + ",\n    ".join(all_items) + ",\n]"
+
+        content = re.sub(all_pattern, new_all_content, content, flags=re.DOTALL)
+
+    # Now handle the TYPE_CHECKING block imports
+    type_checking_pattern = r'(if __import__\("typing"\)\.TYPE_CHECKING:.*?)(else:)'
+    type_checking_match = re.search(type_checking_pattern, content, re.DOTALL)
+
+    if type_checking_match:
+        type_checking_block = type_checking_match.group(1)
+
+        # Find all API imports and create async versions
+        api_imports = re.findall(
+            r"from numind\.openapi_client\.api\.(\w+) import (\w+) as (\w+)",
+            type_checking_block,
+        )
+
+        # Find ApiClient import
+        client_import = re.search(
+            (
+                r"from numind\.openapi_client\.api_client import (ApiClient) as "
+                r"(ApiClient)"
+            ),
+            type_checking_block,
+        )
+
+        async_imports = []
+
+        # Add async API imports
+        for module, class_name, alias in api_imports:
+            async_import = (
+                f"    from numind.openapi_client.api_async.{module} import "
+                f"{class_name} as {alias}Async"
+            )
+            async_imports.append(async_import)
+
+        # Add async ApiClient import
+        if client_import:
+            async_import = (
+                "    from numind.openapi_client.api_client_async import ApiClient as "
+                "ApiClientAsync"
+            )
+            async_imports.append(async_import)
+
+        # Insert async imports after the existing imports
+        if async_imports:
+            async_imports_str = "\n" + "\n".join(async_imports)
+            # Find the position just before 'else:'
+            else_pos = content.find("else:", type_checking_match.start())
+            content = content[:else_pos] + async_imports_str + "\n" + content[else_pos:]
+
+    # Now handle the lazy loading block
+    lazy_pattern = (
+        r'(load\(\s*LazyModule\(\s*\*as_package\(__file__\),.*?""")(.*?)(""",)'
+    )
+    lazy_match = re.search(lazy_pattern, content, re.DOTALL)
+
+    if lazy_match:
+        lazy_imports = lazy_match.group(2)
+
+        # Find all API imports and create async versions
+        api_imports = re.findall(
+            r"from numind\.openapi_client\.api\.(\w+) import (\w+) as (\w+)",
+            lazy_imports,
+        )
+
+        # Find ApiClient import
+        client_import = re.search(
+            (
+                r"from numind\.openapi_client\.api_client import (ApiClient) as "
+                r"(ApiClient)"
+            ),
+            lazy_imports,
+        )
+
+        async_imports = []
+
+        # Add async API imports
+        for module, class_name, alias in api_imports:
+            async_import = (
+                f"from numind.openapi_client.api_async.{module} import {class_name} "
+                f"as {alias}Async"
+            )
+            async_imports.append(async_import)
+
+        # Add async ApiClient import
+        if client_import:
+            async_import = (
+                "from numind.openapi_client.api_client_async import ApiClient as "
+                "ApiClientAsync"
+            )
+            async_imports.append(async_import)
+
+        # Insert async imports
+        if async_imports:
+            async_imports_str = "\n" + "\n".join(async_imports)
+            new_lazy_imports = lazy_imports + async_imports_str
+            content = content.replace(lazy_imports, new_lazy_imports)
+
+    with file_path.open("w") as f:
+        f.write(content)
+
+
+if __name__ == "__main__":
+    from argparse import ArgumentParser
+
+    parser = ArgumentParser(description="Dataset creation script")
+    parser.add_argument(
+        "--init-path", type=str, default="src/numind/openapi_client/__init__.py"
+    )
+    args = vars(parser.parse_args())
+
+    modify_init_file(
+        Path(args["init_path"]),
+    )

{numind-0.0.2/src → numind-0.1.0/sdk_generation}/adapt_pyproject.py

@@ -1,4 +1,4 @@
-"""Collate the project and generated client pyproject.toml for the complete package."""
+"""Collate the project and generated client pyproject.toml for the complete package."""  # noqa:INP001
 
 from pathlib import Path
 

{numind-0.0.2/src → numind-0.1.0/sdk_generation}/collate_documentation_readme.py

@@ -1,4 +1,4 @@
-"""Script removing problematic models from an OpenAPI specification file."""
+"""Script removing problematic models from an OpenAPI specification file."""  # noqa:INP001
 
 from pathlib import Path
 
@@ -9,7 +9,7 @@ OPENAPI_GEN_MENTION = (
 
 
 def read_readme(readme_path: Path) -> str:
-    # TODO find the
+    """Read the README file."""
     with readme_path.open() as file:
         readme = file.read().splitlines()
     for idx, line in enumerate(readme):
@@ -20,6 +20,7 @@ def read_readme(readme_path: Path) -> str:
 
 
 def read_and_preprocess_generated_readme(readme_gen_path: Path) -> str:
+    """Read and preprocess the generated README file."""
     with readme_gen_path.open() as file:
         doc_readme = file.read().splitlines()
 

{numind-0.0.2 → numind-0.1.0/sdk_generation}/config.json

@@ -1,7 +1,5 @@
 {
-  "packageName": "numind.openapi_client",
   "projectName": "numind",
-  "packageVersion": "1.0.0",
   "packageUrl": "https://github.com/numindai/numind",
   "setEnsureAsciiToFalse": true
 }