UncountablePythonSDK 0.0.15__py3-none-any.whl → 0.0.17__py3-none-any.whl

pkgs/type_spec/actions_registry/emit_typescript.py

@@ -0,0 +1,106 @@
+import io
+from collections import defaultdict
+
+from main.base.types import actions_registry_t
+
+from ...type_spec import builder
+from ..emit_typescript_util import INDENT, MODIFY_NOTICE, ts_name
+from ..util import encode_common_string
+
+
+def _action_symbol_name(action_definition: actions_registry_t.ActionDefinition) -> str:
+    return f"{ts_name(action_definition.ref_name, name_case=builder.NameCase.convert)}"
+
+
+def _action_module_name(module: str) -> str:
+    return f"ActionsRegistryT.ActionsRegistryModule.{ts_name(module, name_case=builder.NameCase.convert)}"
+
+
+def emit_action_definitions(
+    action_definitions: defaultdict[str, list[actions_registry_t.ActionDefinition]],
+) -> str:
+    out = io.StringIO()
+    out.write(MODIFY_NOTICE)
+    out.write("\n")
+    out.write('import { ActionsRegistryT } from "unc_mat/types"\n\n')
+    out.write(MODIFY_NOTICE)
+    out.write("export const actionDefinitions = {\n")
+    modules = []
+    for key, values in action_definitions.items():
+        out.write(MODIFY_NOTICE)
+        out.write(f"{INDENT}[{_action_module_name(key)}]: {{\n")
+        modules.append(key)
+        for action_definition in values:
+            out.write(_emit_action_definition(action_definition, INDENT * 2))
+        out.write(f"{INDENT}}},\n")
+
+    out.write("}\n")
+    out.write(MODIFY_NOTICE)
+    out.write("\n")
+    out.write(_emit_action_definition_types(modules, indent=""))
+    out.write(MODIFY_NOTICE)
+    out.write("\n")
+
+    return out.getvalue()
+
+
+def _emit_action_definition(
+    action_definition: actions_registry_t.ActionDefinition, indent: str
+) -> str:
+    out = io.StringIO()
+
+    sub_indent = indent + INDENT
+    out.write(f"{indent}{_action_symbol_name(action_definition)}: {{\n")
+    out.write(f"{sub_indent}name: {encode_common_string(action_definition.name)},\n")
+    if action_definition.icon is not None:
+        out.write(f"{sub_indent}icon: {encode_common_string(action_definition.icon)},\n")
+    out.write(
+        f"{sub_indent}shortDescription: {encode_common_string(action_definition.short_description)},\n"
+    )
+    out.write(
+        f"{sub_indent}description: {encode_common_string(action_definition.description)},\n"
+    )
+    out.write(
+        f"{sub_indent}refName: {encode_common_string(action_definition.ref_name)},\n"
+    )
+    out.write(f"{sub_indent}module: {_action_module_name(action_definition.module)},\n")
+    if (
+        action_definition.visibility_scope is not None
+        and len(action_definition.visibility_scope) > 0
+    ):
+        out.write(
+            f"{sub_indent}visibilityScope: {_emit_visibility_scope(action_definition.visibility_scope)},\n"
+        )
+    out.write(f"{indent}}},\n")
+    return out.getvalue()
+
+
+def _emit_action_definition_types(modules: list[str], indent: str) -> str:
+    out = io.StringIO()
+
+    sub_indent = indent + INDENT
+    out.write(
+        f"{indent}type RefNameKeys<M extends ActionsRegistryT.ActionsRegistryModule> = keyof (typeof actionDefinitions)[M]\n"
+    )
+    out.write(
+        f"{indent}type ActionDefinitionIdentifierGetter<M extends ActionsRegistryT.ActionsRegistryModule> = {{ module: M; refName: RefNameKeys<M> }}\n"
+    )
+    out.write(f"{indent}export type ActionDefinitionIdentifier =\n")
+    for module in modules:
+        out.write(
+            f"{sub_indent}| ActionDefinitionIdentifierGetter<{_action_module_name(module)}>\n"
+        )
+    out.write("\n")
+    return out.getvalue()
+
+
+def _emit_visibility_scope(
+    visibility_scope: list[actions_registry_t.ActionDefinitionVisibilityScope],
+) -> str:
+    visibility_scope_types = ",".join([
+        f"ActionsRegistryT.ActionDefinitionVisibilityScope.{ts_name(visibility_item, name_case=builder.NameCase.convert)}"
+        for visibility_item in visibility_scope
+        if visibility_item is not None
+    ])
+
+    return f"[ {visibility_scope_types} ]"

pkgs/type_spec/builder.py

@@ -747,6 +747,10 @@ class SpecEndpoint:
         self.is_external = self.path_root == "api/external"
         self.has_attachment = data.get("has_attachment", False)
 
+        assert (
+            not is_sdk or self.desc is not None
+        ), f"Endpoint description required for SDK endpoints, missing: {path}"
+
 
 def _parse_const(
     builder: SpecBuilder,

pkgs/type_spec/emit_python.py

@@ -347,6 +347,34 @@ def _validate_supports_handler_generation(
     return stype
 
 
+def _emit_endpoint_invocation_docstring(
+    ctx: Context,
+    endpoint: builder.SpecEndpoint,
+    arguments_type: builder.SpecTypeDefnObject,
+) -> None:
+    has_argument_desc = arguments_type.properties is not None and any(
+        prop.desc is not None for prop in arguments_type.properties.values()
+    )
+    has_endpoint_desc = endpoint.desc
+    if not has_argument_desc and not has_endpoint_desc:
+        return
+
+    FULL_INDENT = INDENT * 2
+    ctx.out.write(FULL_INDENT)
+    ctx.out.write('"""')
+
+    if endpoint.desc is not None and has_endpoint_desc:
+        ctx.out.write(f"{endpoint.desc}\n")
+        ctx.out.write("\n")
+
+    if arguments_type.properties is not None and has_argument_desc:
+        for prop in arguments_type.properties.values():
+            if prop.desc:
+                ctx.out.write(f"{FULL_INDENT}:param {prop.name}: {prop.desc}\n")
+
+    ctx.out.write(f'{FULL_INDENT}"""\n')
+
+
 def _emit_endpoint_invocation_function(
     ctx: Context, namespace: builder.SpecNamespace
 ) -> None:
@@ -393,8 +421,14 @@ def _emit_endpoint_invocation_function(
             class_out=ctx.out,
         )
     ctx.out.write(f"{INDENT}) -> {refer_to(ctx=ctx, stype=data_type)}:")
-
     ctx.out.write("\n")
+
+    _emit_endpoint_invocation_docstring(
+        ctx=ctx,
+        endpoint=endpoint,
+        arguments_type=arguments_type,
+    )
+
     ctx.out.write(f"{INDENT}{INDENT}args = {refer_to(ctx=ctx, stype=arguments_type)}(")
     if has_arguments:
         assert arguments_type.properties is not None

pkgs/type_spec/emit_typescript.py

@@ -226,20 +226,20 @@ def _emit_endpoint(
 
     if is_binary:
         tsx_response_part = f"""import {{ {wrap_name} }} from "unc_base/api"
-import {{ Arguments }} from "{type_path}"
+import type {{ Arguments }} from "{type_path}"
 
 export type {{ Arguments }}
 """
     elif has_data and endpoint.has_attachment:
-        tsx_response_part = f"""import {{ {wrap_name}, AttachmentResponse }} from "unc_base/api"
-import {{ Arguments, Data }} from "{type_path}"
+        tsx_response_part = f"""import {{ {wrap_name}, type AttachmentResponse }} from "unc_base/api"
+import type {{ Arguments, Data }} from "{type_path}"
 
 export type {{ Arguments, Data }}
 export type Response = AttachmentResponse<Data>
 """
     elif has_data:
-        tsx_response_part = f"""import {{ {wrap_name}, JsonResponse }} from "unc_base/api"
-import {{ Arguments, Data }} from "{type_path}"
+        tsx_response_part = f"""import {{ {wrap_name}, type JsonResponse }} from "unc_base/api"
+import type {{ Arguments, Data }} from "{type_path}"
 
 export type {{ Arguments, Data }}
 export type Response = JsonResponse<Data>
@@ -248,7 +248,7 @@ export type Response = JsonResponse<Data>
     else:
         assert has_deprecated_result
         tsx_response_part = f"""import {{ {wrap_name} }} from "unc_base/api"
-import {{ Arguments, DeprecatedResult }} from "{type_path}"
+import type {{ Arguments, DeprecatedResult }} from "{type_path}"
 
 export type {{ Arguments }}
 export type Response = DeprecatedResult
@@ -278,7 +278,6 @@ export const apiCall = {wrap_call}(
     if need_index:
         with open(index_path, "a") as index:
             print(f"Updated API Index {index_path}")
-            index.write("\n// eslint-disable-next-line import/first\n")
             index.write(f'import * as {api_name} from "./{namespace.path[-1]}"\n\n')
             index.write(f"export {{ {api_name} }}\n")
 

pkgs/type_spec/value_spec/emit_python.py

@@ -22,6 +22,7 @@ def emit_functions(functions: list[value_spec_t.Function]) -> str:
     out.write(
         f"""{MODIFY_NOTICE}
 {LINT_HEADER}
+import datetime
 from typing import cast, Union
 
 from decimal import Decimal

type_spec/external/api/chemical/convert_chemical_formats.yaml

@@ -0,0 +1,33 @@
+$endpoint:
+    is_sdk: true
+    method: post
+    path: ${external}/chemical/convert_chemical_formats
+    function: main.site.app.external.chemical.convert_chemical_formats.convert_chemical_formats
+    desc: Converts chemical formats, into the format used by Uncountable and usable in other APIs for eg. set_input_attribute_values
+
+
+ChemicalStructureFile:
+    type: Object
+    properties:
+        struct_file:
+            type: String
+            desc: Stringified chemical structure. MOL v2000/3000, SMILES, Part of SDF.
+
+Arguments:
+    type: Object
+    properties:
+        source_chemical_structures:
+            type: List<ChemicalStructureFile>
+
+UncountableChemicalStructure:
+    type: Alias
+    alias: Dict<String, JsonValue>
+    desc: Uncountable format for chemical structures compatible with set ingredient attributes api etc.
+
+Data:
+    type: Object
+    properties:
+        chemical_structures:
+            type: List<UncountableChemicalStructure>
+            desc: The parsed chemical structures in the Uncountable format
+            convert_value: no_convert

type_spec/external/api/entity/create_entities.yaml

@@ -19,7 +19,7 @@ Arguments:
             type: ObjectId
             desc: "Definition id for the entities to create"
         entity_type:
-            type: Union<Literal<entity.EntityType.lab_request>, Literal<entity.EntityType.approval>, Literal<entity.EntityType.custom_entity>, Literal<entity.EntityType.task>, Literal<entity.EntityType.project>>
+            type: Union<Literal<entity.EntityType.lab_request>, Literal<entity.EntityType.approval>, Literal<entity.EntityType.custom_entity>, Literal<entity.EntityType.task>, Literal<entity.EntityType.project>, Literal<entity.EntityType.equipment>, Literal<entity.EntityType.inv_local_locations>>
             desc: "The type of the entities to create"
         entities_to_create:
             type: List<EntityToCreate>

type_spec/external/api/entity/create_entity.yaml

@@ -26,7 +26,7 @@ Arguments:
             type: ObjectId
             desc: "Definition id of the entity to create"
         entity_type:
-            type: Union<Literal<entity.EntityType.lab_request>, Literal<entity.EntityType.approval>, Literal<entity.EntityType.custom_entity>, Literal<entity.EntityType.task>, Literal<entity.EntityType.project>>
+            type: Union<Literal<entity.EntityType.lab_request>, Literal<entity.EntityType.approval>, Literal<entity.EntityType.custom_entity>, Literal<entity.EntityType.task>, Literal<entity.EntityType.project>, Literal<entity.EntityType.equipment>, Literal<entity.EntityType.inv_local_locations>>
             desc: "The type of the entities requested"
         field_values?:
             type: Optional<List<field_values.FieldRefNameValue>>

type_spec/external/api/recipe_links/create_recipe_link.yaml

@@ -0,0 +1,25 @@
+$endpoint:
+    is_sdk: true
+    method: post
+    path: ${external}/recipe_links/create_recipe_link
+    function: main.site.app.external.recipe_links.create_recipe_link.create_recipe_link
+    desc: Create a link between two recipes.  Skip if the link already exists
+
+Arguments:
+    type: Object
+    properties:
+        recipe_from_key:
+            type: identifier.IdentifierKey
+            desc: "Identifier for the recipe the link comes from"
+        recipe_to_key:
+            type: identifier.IdentifierKey
+            desc: "Identifier for the recipe the link goes to"
+        link_type:
+            type: Union<Literal<recipe_links.RecipeLinkType.child>, Literal<recipe_links.RecipeLinkType.control>, Literal<recipe_links.RecipeLinkType.user_link>>
+            desc: "The type of link being created"
+        name:
+            type: String
+            desc: "The name used for the link"
+
+Data:
+  type: async_batch.AsyncBatchActionReturn

type_spec/external/api/recipes/associate_recipe_as_input.yaml

@@ -0,0 +1,19 @@
+$endpoint:
+    is_sdk: true
+    method: post
+    path: ${external}/recipes/associate_recipe_as_input
+    function: main.site.app.external.recipes.associate_recipe_as_input.associate_recipe_as_input
+    desc: Create or return the input association for a recipe
+
+Arguments:
+    type: Object
+    properties:
+        recipe_key:
+            type: identifier.IdentifierKey
+            desc: "Identifier for the recipe"
+
+Data:
+  type: Object
+  properties:
+    result_id:
+      type: ObjectId

type_spec/external/api/recipes/associate_recipe_as_lot.yaml

@@ -0,0 +1,19 @@
+$endpoint:
+    is_sdk: true
+    method: post
+    path: ${external}/recipes/associate_recipe_as_lot
+    function: main.site.app.external.recipes.associate_recipe_as_lot.associate_recipe_as_lot
+    desc: Create a new lot association for the provided recipe with the provided ingredient
+
+Arguments:
+    type: Object
+    properties:
+        recipe_key:
+            type: identifier.IdentifierKey
+            desc: "Identifier for the recipe"
+        ingredient_key:
+            type: identifier.IdentifierKey
+            desc: "Identifier for the ingredient"
+
+Data:
+  type: async_batch.AsyncBatchActionReturn

type_spec/external/api/recipes/create_recipe.yaml

@@ -0,0 +1,35 @@
+$endpoint:
+    is_sdk: true
+    method: post
+    path: ${external}/recipes/create_recipe
+    function: main.site.app.external.recipes.create_recipe.create_recipe
+    desc: Returns the id of the recipe being created.
+
+
+Arguments:
+    type: Object
+    properties:
+        name?:
+            type: String
+            desc: The name for the recipe
+        material_family_id:
+            type: ObjectId
+            desc: The material family for the recipe
+        workflow_id:
+            type: ObjectId
+            desc: The identifier of the workflow to create the recipe with
+        workflow_variant_id?:
+            type: Optional<ObjectId>
+            desc: The identifier of the workflow variant to create the recipe with
+        recipe_metadata?:
+            type: List<recipe_metadata.MetadataValue>
+            desc: Metadata values to populate the recipe with
+        identifiers:
+            type: recipe_identifiers.RecipeIdentifiers
+            desc: A recipe won't be created if it matches the identifier.  An identifier must be unique in the schema
+
+Data:
+  type: Object
+  properties:
+    result_id:
+      type: ObjectId

type_spec/external/api/recipes/set_recipe_inputs.yaml

@@ -25,6 +25,9 @@ RecipeInputValue:
         set_actual_value?:
             type: Boolean
             desc: "If True, modify the actual value for the input. If not provided or False, modify the set value for the input."
+        lot_recipe_id?:
+            type: ObjectId
+            desc: The recipe id for a lot to be associated to this recipe input. If the recipe is not a lot, it will be created as a lot.
 
 Arguments:
     type: Object

type_spec/external/api/recipes/set_recipe_metadata.yaml

@@ -0,0 +1,19 @@
+$endpoint:
+    is_sdk: true
+    method: post
+    path: ${external}/recipes/set_recipe_metadata
+    function: main.site.app.external.recipes.set_recipe_metadata.set_recipe_metadata
+    desc: "Set metadata values on a recipe"
+
+Arguments:
+    type: Object
+    properties:
+        recipe_key:
+            type: identifier.IdentifierKey
+            desc: "Identifier for the recipe"
+        recipe_metadata:
+            type: List<recipe_metadata.MetadataValue>
+            desc: "Metadata values to populate the recipe with"
+
+Data:
+  type: Object

uncountable/core/client.py

@@ -1,15 +1,15 @@
+import base64
+import json
+import typing
 from dataclasses import dataclass
 from enum import StrEnum
-import json
 from urllib.parse import urljoin
-import base64
-import typing
+
 import requests
-from uncountable.types.base import JsonValue
-from uncountable.types.client_base import APIRequest, ClientMethods
+
 from pkgs.argument_parser import CachedParser
 from pkgs.serialization_util import serialize_for_api
-
+from uncountable.types.client_base import APIRequest, ClientMethods
 
 DT = typing.TypeVar("DT")
 
@@ -24,8 +24,8 @@ class HTTPRequestBase:
     method: EndpointMethod
     url: str
     headers: dict[str, str]
-    body: typing.Optional[str] = None
-    query_params: typing.Optional[dict[str, str]]
+    body: typing.Optional[typing.Union[str, dict[str, str]]] = None
+    query_params: typing.Optional[dict[str, str]] = None
 
 
 @dataclass(kw_only=True)
@@ -37,6 +37,7 @@ class HTTPGetRequest(HTTPRequestBase):
 @dataclass(kw_only=True)
 class HTTPPostRequest(HTTPRequestBase):
     method: typing.Literal[EndpointMethod.POST]
+    body: typing.Union[str, dict[str, str]]
 
 
 HTTPRequest = HTTPPostRequest | HTTPGetRequest
@@ -98,9 +99,7 @@ class Client(ClientMethods):
         match self._auth_details:
             case AuthDetailsApiKey():
                 encoded = base64.standard_b64encode(
-                    f"{self._auth_details.api_id}:{self._auth_details.api_secret_key}".encode(
-                        "utf-8"
-                    )
+                    f"{self._auth_details.api_id}:{self._auth_details.api_secret_key}".encode()
                 ).decode("utf-8")
                 return {"Authorization": f"Basic {encoded}"}
         typing.assert_never(self._auth_details)
@@ -108,21 +107,21 @@ class Client(ClientMethods):
     def _build_http_request(self, *, api_request: APIRequest) -> HTTPRequest:
         headers = self._build_auth_headers()
         method = api_request.method.lower()
-        query_params = {"data": json.dumps(serialize_for_api(api_request.args))}
+        data = {"data": json.dumps(serialize_for_api(api_request.args))}
         match method:
             case "get":
                 return HTTPGetRequest(
                     method=EndpointMethod.GET,
                     url=urljoin(self._base_url, api_request.endpoint),
-                    query_params=query_params,
+                    query_params=data,
                     headers=headers,
                 )
             case "post":
                 return HTTPPostRequest(
                     method=EndpointMethod.POST,
                     url=urljoin(self._base_url, api_request.endpoint),
+                    body=data,
                     headers=headers,
-                    query_params=query_params
                 )
             case _:
                 raise ValueError(f"unsupported request method: {method}")

uncountable/integration/executors/script_executor.py

@@ -1,18 +1,19 @@
 
-
+import os
 import importlib
 import inspect
 from uncountable.integration.job import Job
-from uncountable.integration.types import JobExecutorScript
+from uncountable.integration.types import JobExecutorScript, ProfileMetadata
 
 
-def resolve_script_executor(executor: JobExecutorScript) -> type[Job]:
-    job_module = importlib.import_module(executor.import_path)
+def resolve_script_executor(executor: JobExecutorScript, profile_metadata: ProfileMetadata) -> type[Job]:
+    job_module_path = ".".join([os.environ["UNC_PROFILES_MODULE"], profile_metadata.name, executor.import_path])
+    job_module = importlib.import_module(job_module_path)
     found_jobs: list[type[Job]] = []
     for _, job_class in inspect.getmembers(job_module, inspect.isclass):
-        if Job in job_class.__bases__:
+        if getattr(job_class, "_unc_job_registered", False):
             found_jobs.append(job_class())
     assert (
         len(found_jobs) == 1
-    ), f"expected exactly one job class in {executor.import_path}"
+    ), f"expected exactly one job class in {executor.import_path}, found {len(found_jobs)}"
     return found_jobs[0]

uncountable/types/__init__.py

@@ -3,11 +3,18 @@
 # ruff: noqa: E402
 # fmt: off
 # isort: skip_file
+from .api.recipes import associate_recipe_as_input as associate_recipe_as_input_t
+from .api.recipes import associate_recipe_as_lot as associate_recipe_as_lot_t
+from . import async_batch as async_batch_t
 from . import base as base_t
 from . import calculations as calculations_t
+from . import chemical_structure as chemical_structure_t
+from .api.chemical import convert_chemical_formats as convert_chemical_formats_t
 from .api.entity import create_entities as create_entities_t
 from .api.entity import create_entity as create_entity_t
 from .api.inputs import create_inputs as create_inputs_t
+from .api.recipes import create_recipe as create_recipe_t
+from .api.recipe_links import create_recipe_link as create_recipe_link_t
 from .api.recipes import create_recipes as create_recipes_t
 from . import curves as curves_t
 from . import entity as entity_t
@@ -31,11 +38,13 @@ from .api.recipe_metadata import get_recipe_metadata_data as get_recipe_metadata
 from .api.recipes import get_recipe_names as get_recipe_names_t
 from .api.recipes import get_recipe_output_metadata as get_recipe_output_metadata_t
 from .api.recipes import get_recipes_data as get_recipes_data_t
+from . import identifier as identifier_t
 from . import input_attributes as input_attributes_t
 from . import inputs as inputs_t
 from .api.entity import list_entities as list_entities_t
 from . import outputs as outputs_t
 from . import phases as phases_t
+from . import recipe_identifiers as recipe_identifiers_t
 from . import recipe_links as recipe_links_t
 from . import recipe_metadata as recipe_metadata_t
 from . import recipe_output_metadata as recipe_output_metadata_t
@@ -45,6 +54,7 @@ from .api.outputs import resolve_output_conditions as resolve_output_conditions_
 from . import response as response_t
 from .api.inputs import set_input_attribute_values as set_input_attribute_values_t
 from .api.recipes import set_recipe_inputs as set_recipe_inputs_t
+from .api.recipes import set_recipe_metadata as set_recipe_metadata_t
 from .api.recipes import set_recipe_outputs as set_recipe_outputs_t
 from .api.entity import set_values as set_values_t
 from . import units as units_t
@@ -53,11 +63,18 @@ from . import workflows as workflows_t
 
 
 __all__: list[str] = [
+    "associate_recipe_as_input_t",
+    "associate_recipe_as_lot_t",
+    "async_batch_t",
     "base_t",
     "calculations_t",
+    "chemical_structure_t",
+    "convert_chemical_formats_t",
     "create_entities_t",
     "create_entity_t",
     "create_inputs_t",
+    "create_recipe_t",
+    "create_recipe_link_t",
     "create_recipes_t",
     "curves_t",
     "entity_t",
@@ -81,11 +98,13 @@ __all__: list[str] = [
     "get_recipe_names_t",
     "get_recipe_output_metadata_t",
     "get_recipes_data_t",
+    "identifier_t",
     "input_attributes_t",
     "inputs_t",
     "list_entities_t",
     "outputs_t",
     "phases_t",
+    "recipe_identifiers_t",
     "recipe_links_t",
     "recipe_metadata_t",
     "recipe_output_metadata_t",
@@ -95,6 +114,7 @@ __all__: list[str] = [
     "response_t",
     "set_input_attribute_values_t",
     "set_recipe_inputs_t",
+    "set_recipe_metadata_t",
     "set_recipe_outputs_t",
     "set_values_t",
     "units_t",

uncountable/types/api/chemical/__init__.py

@@ -0,0 +1 @@
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/chemical/convert_chemical_formats.py

@@ -0,0 +1,50 @@
+# DO NOT MODIFY -- This file is generated by type_spec
+# flake8: noqa: F821
+# ruff: noqa: E402
+# fmt: off
+# isort: skip_file
+from __future__ import annotations
+import typing  # noqa: F401
+import datetime  # noqa: F401
+from decimal import Decimal  # noqa: F401
+from dataclasses import dataclass
+from pkgs.serialization import serial_class
+from ... import base as base_t
+
+__all__: list[str] = [
+    "Arguments",
+    "ChemicalStructureFile",
+    "Data",
+    "ENDPOINT_METHOD",
+    "ENDPOINT_PATH",
+    "UncountableChemicalStructure",
+]
+
+ENDPOINT_METHOD = "POST"
+ENDPOINT_PATH = "api/external/chemical/convert_chemical_formats"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@dataclass(kw_only=True)
+class ChemicalStructureFile:
+    struct_file: str
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@dataclass(kw_only=True)
+class Arguments:
+    source_chemical_structures: list[ChemicalStructureFile]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+UncountableChemicalStructure = dict[str, base_t.JsonValue]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    unconverted_values={"chemical_structures"},
+)
+@dataclass(kw_only=True)
+class Data:
+    chemical_structures: list[UncountableChemicalStructure]
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/entity/create_entities.py

@@ -34,7 +34,7 @@ class EntityToCreate:
 @dataclass(kw_only=True)
 class Arguments:
     definition_id: base_t.ObjectId
-    entity_type: typing.Union[typing.Literal[entity_t.EntityType.LAB_REQUEST], typing.Literal[entity_t.EntityType.APPROVAL], typing.Literal[entity_t.EntityType.CUSTOM_ENTITY], typing.Literal[entity_t.EntityType.TASK], typing.Literal[entity_t.EntityType.PROJECT]]
+    entity_type: typing.Union[typing.Literal[entity_t.EntityType.LAB_REQUEST], typing.Literal[entity_t.EntityType.APPROVAL], typing.Literal[entity_t.EntityType.CUSTOM_ENTITY], typing.Literal[entity_t.EntityType.TASK], typing.Literal[entity_t.EntityType.PROJECT], typing.Literal[entity_t.EntityType.EQUIPMENT], typing.Literal[entity_t.EntityType.INV_LOCAL_LOCATIONS]]
     entities_to_create: list[EntityToCreate]
 
 

uncountable/types/api/entity/create_entity.py

@@ -40,7 +40,7 @@ class EntityFieldInitialValue:
 @dataclass(kw_only=True)
 class Arguments:
     definition_id: base_t.ObjectId
-    entity_type: typing.Union[typing.Literal[entity_t.EntityType.LAB_REQUEST], typing.Literal[entity_t.EntityType.APPROVAL], typing.Literal[entity_t.EntityType.CUSTOM_ENTITY], typing.Literal[entity_t.EntityType.TASK], typing.Literal[entity_t.EntityType.PROJECT]]
+    entity_type: typing.Union[typing.Literal[entity_t.EntityType.LAB_REQUEST], typing.Literal[entity_t.EntityType.APPROVAL], typing.Literal[entity_t.EntityType.CUSTOM_ENTITY], typing.Literal[entity_t.EntityType.TASK], typing.Literal[entity_t.EntityType.PROJECT], typing.Literal[entity_t.EntityType.EQUIPMENT], typing.Literal[entity_t.EntityType.INV_LOCAL_LOCATIONS]]
     field_values: typing.Optional[typing.Optional[list[field_values_t.FieldRefNameValue]]] = None