UncountablePythonSDK 0.0.93__py3-none-any.whl → 0.0.96__py3-none-any.whl

{UncountablePythonSDK-0.0.93.dist-info → UncountablePythonSDK-0.0.96.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: UncountablePythonSDK
-Version: 0.0.93
+Version: 0.0.96
 Summary: Uncountable SDK
 Project-URL: Homepage, https://github.com/uncountableinc/uncountable-python-sdk
 Project-URL: Repository, https://github.com/uncountableinc/uncountable-python-sdk.git
@@ -38,6 +38,7 @@ Requires-Dist: simplejson==3.*
 Requires-Dist: grpcio==1.67.1
 Requires-Dist: protobuf>=4.21.1
 Requires-Dist: azure-storage-blob==12.*
+Requires-Dist: boto3-stubs[essential,ses]==1.36.4
 Provides-Extra: test
 Requires-Dist: mypy==1.*; extra == "test"
 Requires-Dist: ruff==0.*; extra == "test"

{UncountablePythonSDK-0.0.93.dist-info → UncountablePythonSDK-0.0.96.dist-info}/RECORD

@@ -28,7 +28,7 @@ examples/integration-server/jobs/materials_auto/example_wh.py,sha256=Hx5nonavOh2
 examples/integration-server/jobs/materials_auto/profile.yaml,sha256=XlOXSRplMJ13T6900pv1wDKxeE9V1hZZTMuvup1MiBM,896
 pkgs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pkgs/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pkgs/argument_parser/__init__.py,sha256=AOL6yZkJg9Z8Q4v4JY4agEbMDQWPptVauSA_itShv64,693
+pkgs/argument_parser/__init__.py,sha256=nQGVudX2XjaFRNXU7V3e-Ap-zmUpRVrwHqlzJ1FDgvk,816
 pkgs/argument_parser/_is_enum.py,sha256=Gw6jJa8nBwYGqXwwCZbSnWL8Rvr5alkg5lSVAqXtOZM,257
 pkgs/argument_parser/_is_namedtuple.py,sha256=Rjc1bKanIPPogl3qG5JPBxglG1TqWYOo1nxxhBASQWY,265
 pkgs/argument_parser/argument_parser.py,sha256=_W0-HLC1F8jXgeFJnhUREorqcXzuhZi__eX_7FFkkVs,19363
@@ -38,7 +38,7 @@ pkgs/filesystem_utils/_blob_session.py,sha256=CtoB7PIocuZo8vvFIS_Rc-YR6KwzFB0rHU
 pkgs/filesystem_utils/_gdrive_session.py,sha256=XTAKppTfIiJtPkzq90yftoMHa-kcGe9MDZQlmx1bviQ,11058
 pkgs/filesystem_utils/_local_session.py,sha256=xFEYhAvNqrOYqwt4jrEYOuYkjJn0zclZhTelW_Q1-rw,2325
 pkgs/filesystem_utils/_s3_session.py,sha256=DdD6M90z1VyTiuRdTmjvBNJOEFHBPyEU8wGKRKwHvOU,4027
-pkgs/filesystem_utils/_sftp_session.py,sha256=6qwBGYu8UTFYw9YAmYAD0qL9qtivRRQpvzmlInmJZ_k,4709
+pkgs/filesystem_utils/_sftp_session.py,sha256=l1hS8KjnINqCX2CZU1t66H-is0mTwTLeMcQMyXW8DjM,4847
 pkgs/filesystem_utils/file_type_utils.py,sha256=oklu7Wtcxg9pBz84cAlXpNlhYEM0AbsO5DoEqvFjoXY,1896
 pkgs/filesystem_utils/filesystem_session.py,sha256=BQ2Go8Mu9-GcnaWh2Pm4x7ugLVsres6XrOQ8RoiEpcE,1045
 pkgs/serialization/__init__.py,sha256=b2sNfYICz0D6Y8shzEUfyUoHZukfuAvfuaZvg8dZHic,1163
@@ -51,10 +51,11 @@ pkgs/serialization/serial_generic.py,sha256=vAC0-o9fkco2lM_YvBavY0JDYncDxbW3jFbo
 pkgs/serialization/serial_union.py,sha256=IdWsSsM8j-qGfl5Wv6haX37ZS1SKob8XT8cilBBOVIo,2670
 pkgs/serialization/yaml.py,sha256=yoJtu7_ixnJV6uTxA_U1PpK5F_ixT08AKVh5ocyYwXM,1466
 pkgs/serialization_util/__init__.py,sha256=YykkhqGNKiLCo-D5vSTq6WiPNfCyPXyr-mQO5e5Klak,513
+pkgs/serialization_util/_env_flags.py,sha256=5kBchfeYS_MRWWSdUE3EljXTtOZX-YYH7inh48d-Qz8,90
 pkgs/serialization_util/_get_type_for_serialization.py,sha256=dW5_W9MFd6wgWfW5qlWork-GBb-QFLtiOZkjk2Zqn2M,1177
 pkgs/serialization_util/convert_to_snakecase.py,sha256=H2BAo5ZdcCDN77RpLb-uP0s7-FQ5Ukwnsd3VYc1vD0M,583
 pkgs/serialization_util/dataclasses.py,sha256=uhNGXQPQLZblDFQuuwkAGmKOPiRyfDzCdg72CVtYJGA,390
-pkgs/serialization_util/serialization_helpers.py,sha256=d11gYBe0WDlPDMqi8LatD4I5RluIcheD8xtMXnPLF1g,6537
+pkgs/serialization_util/serialization_helpers.py,sha256=gP4U5zLYA6Y3y1qtcH2qoRsXPh-UnhR5ITEIRUDj06A,6649
 pkgs/strenum_compat/__init__.py,sha256=wXRFeNvBm8RU6dy1PFJ5sRLgUIEeH_DVR95Sv5qpGbk,59
 pkgs/strenum_compat/strenum_compat.py,sha256=uOUAgpYTjHs1MX8dG81jRlyTkt3KNbkV_25zp7xTX2s,36
 pkgs/type_spec/__init__.py,sha256=h5DmJTca4QVV10sZR1x0-MlkZfuGYDfapR3zHvXfzto,19
@@ -64,7 +65,7 @@ pkgs/type_spec/config.py,sha256=K-DFjRVIW5RCJOuB0QhkWZMOrsIZavFURqmhgAoULgw,5103
 pkgs/type_spec/emit_io_ts.py,sha256=CUvBs0boB_X-Kndh66yYcqFfq3oC_LGs8YffLkJ0ZXA,5707
 pkgs/type_spec/emit_open_api.py,sha256=9AgtHB3ZIObs9cV-RqNZt_q1icmIGwPeovQ63gTl-l8,24541
 pkgs/type_spec/emit_open_api_util.py,sha256=34zslFjTiOI5q8YreBOLD-afmaW4lrlK7pBpNAd3CxA,2379
-pkgs/type_spec/emit_python.py,sha256=BohmwGN89EDHiNlT6-z7xv2C0vIRhQ2SXr4kdoue2a4,52103
+pkgs/type_spec/emit_python.py,sha256=0JKXPapSOVVn6laO5dOpKfON18YkK5KtfhpE-Sqb_5E,52001
 pkgs/type_spec/emit_typescript.py,sha256=_7-SLAOP0jMIrDWDKCPJpyykJh3n0hynm5b-Ltq23hM,10620
 pkgs/type_spec/emit_typescript_util.py,sha256=sugxboeMwXiIdk4RDyUR_CxU31Le3OQ7i5HmuHeSLvc,10447
 pkgs/type_spec/load_types.py,sha256=ero3SXB0M7xKHni5Qxop5Yh5A1TYjjoqwkEgUCQx71Q,3657
@@ -87,7 +88,7 @@ uncountable/__init__.py,sha256=8l8XWNCKsu7TG94c-xa2KHpDegvxDC2FyQISdWC763Y,89
 uncountable/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 uncountable/core/__init__.py,sha256=RFv0kO6rKFf1PtBPu83hCGmxqkJamRtsgQ9_-ztw7tA,341
 uncountable/core/async_batch.py,sha256=9pYGFzVCQXt8059qFHgutweGIFPquJ5Xfq6NT5P-1K0,1206
-uncountable/core/client.py,sha256=cypErG4AXAcgkzetvlSLWA75-LKjqreobZgjSfSUKbY,13004
+uncountable/core/client.py,sha256=ZlGKzpAnPaU0ABOb_bhIDUKZ9x8rlezqVaK0mfDvdw8,13253
 uncountable/core/environment.py,sha256=6cc-nUvUIbJMI0OSEg3dWI-iNgSYhCdUKKn607Z3QpM,1040
 uncountable/core/file_upload.py,sha256=bgvXk9vfF5qlhy2NAUcEEG7Q7i-c1wr2HrpaWD7HldU,4516
 uncountable/core/types.py,sha256=s2CjqYJpsmbC7xMwxxT7kJ_V9bwokrjjWVVjpMcQpKI,333
@@ -106,7 +107,7 @@ uncountable/integration/db/connect.py,sha256=mE3bdV0huclH2iT_dXCQdRL4LkjIuf_myAR
 uncountable/integration/db/session.py,sha256=96cGQXpe6IugBTdSsjdP0S5yhJ6toSmbVB6qhc3FJzE,693
 uncountable/integration/executors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 uncountable/integration/executors/executors.py,sha256=Kzisp1eKufGCWrHIw4mmAj-l1UQ2oJsJR7I-_mksnVs,5441
-uncountable/integration/executors/generic_upload_executor.py,sha256=OnVY4KfmAgaf9LPSAgj7H2wFRXwNXBpg7egVy85z5lk,10327
+uncountable/integration/executors/generic_upload_executor.py,sha256=KRUVeQkywIQUxzuzCIp6Vli1JWFelUCKTeJCxjMmkuI,11179
 uncountable/integration/executors/script_executor.py,sha256=BBQ9f0l7uH2hgKf60jtm-pONzwk-EeOhM2qBAbv_URo,846
 uncountable/integration/queue_runner/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 uncountable/integration/queue_runner/job_scheduler.py,sha256=2acmeWscG87MZsifxwMRMGJ4sQgQVNJNr5CqIPUg75E,6288
@@ -143,13 +144,13 @@ uncountable/types/calculations.py,sha256=fApOFpgBemt_t7IVneVR0VdI3X5EOxiG6Xhzr6R
 uncountable/types/calculations_t.py,sha256=LE_HWYPf7yxNTpH5sVPaC0e98OxJ5hUOhHezeJNKDtk,648
 uncountable/types/chemical_structure.py,sha256=ujyragaD26-QG5jgKnWhO7TN3N1V9b_04T2WhqNYxxo,281
 uncountable/types/chemical_structure_t.py,sha256=iYmGER_vXqoksv2Nr189qU1Zm70s-U49Eiv0DkWaB1I,773
-uncountable/types/client_base.py,sha256=h5nOrnuw6Szq-fEu9mxzMhqMlm3HDpoi5PlL9Or8ypk,68699
+uncountable/types/client_base.py,sha256=5UJI7ByJitlpoW6ET6AIB14pgzB8zHXTQ0yn291cWmc,68932
 uncountable/types/client_config.py,sha256=qLpHt4O_B098CyN6qQajoxZ2zjZ1DILXLUEGyyGP0TQ,280
 uncountable/types/client_config_t.py,sha256=k_UkjemqiVuJBiFUjJYk3h673Nahr0nGZvfEZFzsC0k,699
 uncountable/types/curves.py,sha256=QyEyC20jsG-LGKVx6miiF-w70vKMwNkILFBDIJ5Ok9g,345
 uncountable/types/curves_t.py,sha256=2_9qdrSl1XAvIG57lo45KWNpa0wXgZ97OkSRCPRrudc,1347
 uncountable/types/entity.py,sha256=Zclk1LYcRaYrMDhqyCjMSLEg0fE6_q8LHvV22Qvscgs,566
-uncountable/types/entity_t.py,sha256=dTf525ZlAnhWwuCPElR5xooLt0q0LtdtTMkgMQv15nM,17422
+uncountable/types/entity_t.py,sha256=MWtA4b9ZoayKDKJ2j1MNbt7lefSFJy3mWaswoOOx3ks,17512
 uncountable/types/experiment_groups.py,sha256=qUpFOx1AKgzaT_4khCOv5Xs6jwiQGbvHH-GUh3v1nv4,288
 uncountable/types/experiment_groups_t.py,sha256=_fAYZwqYLR3cFdv2vwLOYs5TvH5CEWDEbh3kFpg26zY,700
 uncountable/types/field_values.py,sha256=hERqHIDmWPQpdW7BLitdA-DNiutmll-ggecwEO7FoVU,1181
@@ -169,7 +170,7 @@ uncountable/types/inputs_t.py,sha256=0b3U77JcZT4hgUvbP_i-E5RoJbJnRxn3OKBqGWu32TM
 uncountable/types/integration_server.py,sha256=VonA8h8TGnVBiss5W8-K82lA01JQa7TLk0ubFo8iiBQ,364
 uncountable/types/integration_server_t.py,sha256=37zyqeet54P9m6pxaZfLOgZCqqZA2dxJ5gl6NCoelQ0,1188
 uncountable/types/job_definition.py,sha256=6BkLZrmTfIYh45XFGZ5HOYveued0YXvl17YTlXblXjw,1646
-uncountable/types/job_definition_t.py,sha256=n8lwjNtt6eHg01yARRIgDiLO9W8ev5upb3qs2RPHZXg,7753
+uncountable/types/job_definition_t.py,sha256=ed_fCyLFQBo6XDfqtfUnk4ghmx0uGtDHVEnNB1N_yv4,7837
 uncountable/types/outputs.py,sha256=I6zP2WHXg_jXgMqmuEJuJOlsjKjQGHjfs1JOwW9YxBM,260
 uncountable/types/outputs_t.py,sha256=ZJhKKkksJ-K7iuuumCly9edU8TRv-eWol4PKG7XPo3E,734
 uncountable/types/overrides.py,sha256=fOvj8P9K9ul8fnTwA--l140EWHuc1BFq8tXgtBkYld4,410
@@ -217,8 +218,8 @@ uncountable/types/api/batch/execute_batch_load_async.py,sha256=rxr2pvJ4w_2Xnqzff
 uncountable/types/api/chemical/__init__.py,sha256=gCgbynxG3jA8FQHzercKtrHKHkiIKr8APdZYUniAor8,55
 uncountable/types/api/chemical/convert_chemical_formats.py,sha256=B1MT10IfFLapomSwzJyAKvyjrwKVDqEuESIhCJPJtgc,1778
 uncountable/types/api/entity/__init__.py,sha256=gCgbynxG3jA8FQHzercKtrHKHkiIKr8APdZYUniAor8,55
-uncountable/types/api/entity/create_entities.py,sha256=2MVtjfpegFwwUU0NGu-cpvgwusV-6B0zhCPWclXTEsc,1562
-uncountable/types/api/entity/create_entity.py,sha256=cVEYcn3ao1vEiF7CiKd85OYBqqqd7ElwRyD_V5C_BHg,1647
+uncountable/types/api/entity/create_entities.py,sha256=R0Xa4gBhOJJ9hNwyZZq2l3BCn-6zy-04JzDo4NxHjVs,1611
+uncountable/types/api/entity/create_entity.py,sha256=YBaFJDZGFnAPadeiqOGZ7LTHET_3RJlXJOUyUpm42PU,1696
 uncountable/types/api/entity/get_entities_data.py,sha256=OUeBo0YkZmVTHT2EHAQt19rm2PU4OpJzOj3aCmZMw0Y,1433
 uncountable/types/api/entity/grant_entity_permissions.py,sha256=Zx3FDHSjiBN_wRtmxbWSWdmT-CkzEoi2uoVLCIdS2Q8,1391
 uncountable/types/api/entity/list_entities.py,sha256=1I6T6fjfolXx1lg8iRiCO5J0ieU0bdudPp31ZmMkXIQ,1895
@@ -295,7 +296,7 @@ uncountable/types/api/triggers/__init__.py,sha256=gCgbynxG3jA8FQHzercKtrHKHkiIKr
 uncountable/types/api/triggers/run_trigger.py,sha256=diX1ix_5hkti1F1uYoZhP5iyc6GHAU5coKgqq5syLhI,1059
 uncountable/types/api/uploader/__init__.py,sha256=gCgbynxG3jA8FQHzercKtrHKHkiIKr8APdZYUniAor8,55
 uncountable/types/api/uploader/invoke_uploader.py,sha256=-loZzBihKqx63eP-f5RuV1mu6tgkRTZmIc545kklZLk,1273
-UncountablePythonSDK-0.0.93.dist-info/METADATA,sha256=Pwy0kXB5Pc8CCUeDNAzQXwTcJwZb9N_7jTpUigM7KFw,2064
-UncountablePythonSDK-0.0.93.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-UncountablePythonSDK-0.0.93.dist-info/top_level.txt,sha256=1UVGjAU-6hJY9qw2iJ7nCBeEwZ793AEN5ZfKX9A1uj4,31
-UncountablePythonSDK-0.0.93.dist-info/RECORD,,
+UncountablePythonSDK-0.0.96.dist-info/METADATA,sha256=noOdXVOZuFMPGoDTMwFBACEpGQ0lj08XV7gBEefA4ko,2114
+UncountablePythonSDK-0.0.96.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+UncountablePythonSDK-0.0.96.dist-info/top_level.txt,sha256=1UVGjAU-6hJY9qw2iJ7nCBeEwZ793AEN5ZfKX9A1uj4,31
+UncountablePythonSDK-0.0.96.dist-info/RECORD,,

pkgs/argument_parser/__init__.py

@@ -1,3 +1,4 @@
+from ._is_enum import is_string_enum_class as is_string_enum_class
 from .argument_parser import CachedParser as CachedParser
 from .argument_parser import ParserError as ParserError
 from .argument_parser import ParserExtraFieldsError as ParserExtraFieldsError
@@ -5,6 +6,7 @@ from .argument_parser import ParserFunction as ParserFunction
 from .argument_parser import ParserOptions as ParserOptions
 from .argument_parser import SourceEncoding as SourceEncoding
 from .argument_parser import build_parser as build_parser
+from .argument_parser import is_optional as is_optional
 from .argument_parser import is_union as is_union
 from .case_convert import camel_to_snake_case as camel_to_snake_case
 from .case_convert import kebab_to_pascal_case as kebab_to_pascal_case

pkgs/filesystem_utils/_sftp_session.py

@@ -54,6 +54,10 @@ def list_sftp_files(
             os.path.join(dir_path, file)
             for file in connection.listdir(dir_path)
             if connection.isfile(os.path.join(dir_path, file))
+            and (
+                valid_extensions is None
+                or os.path.splitext(file)[1] in valid_extensions
+            )
         ])
     return file_paths
 

pkgs/serialization_util/_env_flags.py

@@ -0,0 +1,3 @@
+import os
+
+USE_FASTJSON: bool = os.environ.get("USE_FASTJSON", "false").lower() == "true"

pkgs/serialization_util/serialization_helpers.py

@@ -23,6 +23,7 @@ from pkgs.serialization import (
     get_serial_class_data,
 )
 
+from ._env_flags import USE_FASTJSON
 from ._get_type_for_serialization import SerializationType, get_serialization_type
 from .dataclasses import iterate_fields
 
@@ -109,7 +110,9 @@ def _get_dataclass_conversion_lookups(dataclass_type: Any) -> DataclassConversio
         if scd.has_to_string_value(key):
             value_conversion_functions[key] = _to_string_value
         elif scd.has_unconverted_value(key):
-            value_conversion_functions[key] = identity
+            value_conversion_functions[key] = (
+                serialize_for_storage if USE_FASTJSON else identity
+            )
         else:
             value_conversion_functions[key] = serialize_for_api
 

pkgs/type_spec/emit_python.py

@@ -859,7 +859,7 @@ def _emit_type(ctx: Context, stype: builder.SpecType) -> None:
     if not stype.base.is_base:
         base_class = f"({refer_to(ctx, stype.base)})"
     elif len(generics) > 0:
-        base_class = f"(typing.Generic[{', '.join(generics)}])"
+        base_class = f"[{', '.join(generics)}]"
     class_out.write(f"class {stype.name}{base_class}:\n")
 
     emitted_properties_metadata = _emit_type_properties(
@@ -1288,7 +1288,7 @@ def _emit_api_argument_lookup(
     for endpoint_root in builder.api_endpoints:
         routes_output = config.routes_output[endpoint_root]
 
-        imports = ["import typing", "import dataclasses"]
+        imports = ["import dataclasses"]
         mappings = []
         for namespace in sorted(
             builder.namespaces.values(),
@@ -1337,9 +1337,6 @@ def _emit_api_argument_lookup(
         argument_lookup_out.write(
             f"""{LINE_BREAK.join(imports)}
 
-AT = typing.TypeVar("AT")
-DT = typing.TypeVar("DT")
-
 
 @dataclasses.dataclass(kw_only=True, frozen=True)
 class ApiEndpointKey:
@@ -1348,7 +1345,7 @@ class ApiEndpointKey:
 
 
 @dataclasses.dataclass(kw_only=True)
-class ApiEndpointSpec(typing.Generic[AT, DT]):
+class ApiEndpointSpec[AT, DT]:
     route: str
     arguments_type: type[AT]
     data_type: type[DT]

uncountable/core/client.py

@@ -197,9 +197,10 @@ class Client(ClientMethods):
             logger=self._cfg.logger,
         )
 
-    def _get_response_json(
-        self, response: requests.Response, request_id: str
-    ) -> dict[str, JsonValue]:
+    @classmethod
+    def _validate_response_status(
+        cls, response: requests.Response, request_id: str
+    ) -> None:
         if response.status_code < 200 or response.status_code > 299:
             extra_details: dict[str, JsonValue] | None = None
             try:
@@ -214,6 +215,11 @@ class Client(ClientMethods):
                 extra_details=extra_details,
                 request_id=request_id,
             )
+
+    def _get_response_json(
+        self, response: requests.Response, request_id: str
+    ) -> dict[str, JsonValue]:
+        self._validate_response_status(response, request_id)
         try:
             return typing.cast(dict[str, JsonValue], response.json())
         except JSONDecodeError as e:
@@ -361,6 +367,8 @@ class Client(ClientMethods):
         assert isinstance(http_request, HTTPGetRequest)
         request.params = http_request.query_params
         response = self._send_request(request)
+        self._validate_response_status(response, request_id)
+
         content = response.content
         content_disposition = response.headers.get("Content-Disposition", None)
         return [

uncountable/integration/executors/generic_upload_executor.py

@@ -184,16 +184,33 @@ class GenericUploadJob(Job[None]):
     def _construct_filesystem_session(self, args: JobArguments) -> FileSystemSession:
         match self.data_source:
             case GenericUploadDataSourceSFTP():
-                pem_secret = retrieve_secret(
-                    self.data_source.pem_secret, profile_metadata=args.profile_metadata
-                )
-                pem_key = paramiko.RSAKey.from_private_key(io.StringIO(pem_secret))
-                sftp_config = FileSystemSFTPConfig(
-                    ip=self.data_source.host,
-                    username=self.data_source.username,
-                    pem_path=None,
-                    pem_key=pem_key,
-                )
+                if self.data_source.pem_secret is not None:
+                    pem_secret = retrieve_secret(
+                        self.data_source.pem_secret,
+                        profile_metadata=args.profile_metadata,
+                    )
+                    pem_key = paramiko.RSAKey.from_private_key(io.StringIO(pem_secret))
+                    sftp_config = FileSystemSFTPConfig(
+                        ip=self.data_source.host,
+                        username=self.data_source.username,
+                        pem_path=None,
+                        pem_key=pem_key,
+                    )
+                elif self.data_source.password_secret is not None:
+                    password_secret = retrieve_secret(
+                        self.data_source.password_secret,
+                        profile_metadata=args.profile_metadata,
+                    )
+                    sftp_config = FileSystemSFTPConfig(
+                        ip=self.data_source.host,
+                        username=self.data_source.username,
+                        pem_path=None,
+                        password=password_secret,
+                    )
+                else:
+                    raise ValueError(
+                        "Either pem_secret or password_secret must be specified for sftp data source"
+                    )
                 return SFTPSession(sftp_config=sftp_config)
             case GenericUploadDataSourceS3():
                 if self.data_source.access_key_secret is not None:

uncountable/types/api/entity/create_entities.py

@@ -32,6 +32,7 @@ ENDPOINT_PATH = "api/external/entity/external_create_entities"
 @dataclasses.dataclass(kw_only=True)
 class EntityToCreate:
     field_values: list[field_values_t.FieldRefNameValue] | None = None
+    parent_entity: entity_t.Entity | None = None
 
 
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/api/entity/create_entity.py

@@ -47,6 +47,7 @@ class Arguments:
     definition_id: base_t.ObjectId | None = None
     definition_key: identifier_t.IdentifierKey | None = None
     field_values: list[field_values_t.FieldRefNameValue] | None = None
+    parent_entity: entity_t.Entity | None = None
 
 
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/client_base.py

@@ -290,18 +290,21 @@ class ClientMethods(ABC):
         definition_id: base_t.ObjectId | None = None,
         definition_key: identifier_t.IdentifierKey | None = None,
         field_values: list[field_values_t.FieldRefNameValue] | None = None,
+        parent_entity: entity_t.Entity | None = None,
     ) -> create_entity_t.Data:
         """Creates a new Uncountable entity
 
         :param definition_id: Definition id of the entity to create
         :param definition_key: Key for the definition of the entity to create
         :param entity_type: The type of the entities requested
+        :param parent_entity: The parent entity associated for this new entity. This only applies if there is a pre-defined relationship.
         """
         args = create_entity_t.Arguments(
             definition_id=definition_id,
             definition_key=definition_key,
             entity_type=entity_type,
             field_values=field_values,
+            parent_entity=parent_entity,
         )
         api_request = APIRequest(
             method=create_entity_t.ENDPOINT_METHOD,

uncountable/types/entity_t.py

@@ -346,7 +346,7 @@ class EntityType(StrEnum):
 
 # DO NOT MODIFY -- This file is generated by type_spec
 LimitedEntityType = typing.Annotated[
-    typing.Literal[EntityType.LAB_REQUEST] | typing.Literal[EntityType.APPROVAL] | typing.Literal[EntityType.CUSTOM_ENTITY] | typing.Literal[EntityType.INVENTORY_AMOUNT] | typing.Literal[EntityType.TASK] | typing.Literal[EntityType.PROJECT] | typing.Literal[EntityType.EQUIPMENT] | typing.Literal[EntityType.INV_LOCAL_LOCATIONS] | typing.Literal[EntityType.FIELD_OPTION_SET] | typing.Literal[EntityType.WEBHOOK] | typing.Literal[EntityType.SPECS] | typing.Literal[EntityType.GOAL] | typing.Literal[EntityType.INGREDIENT_TAG_MAP] | typing.Literal[EntityType.INGREDIENT_TAG] | typing.Literal[EntityType.CONDITION_PARAMETER] | typing.Literal[EntityType.OUTPUT] | typing.Literal[EntityType.OUTPUT_CONDITION_PARAMETER] | typing.Literal[EntityType.ASYNC_JOB],
+    typing.Literal[EntityType.LAB_REQUEST] | typing.Literal[EntityType.APPROVAL] | typing.Literal[EntityType.CUSTOM_ENTITY] | typing.Literal[EntityType.INGREDIENT_ATTRIBUTE] | typing.Literal[EntityType.INVENTORY_AMOUNT] | typing.Literal[EntityType.TASK] | typing.Literal[EntityType.PROJECT] | typing.Literal[EntityType.EQUIPMENT] | typing.Literal[EntityType.INV_LOCAL_LOCATIONS] | typing.Literal[EntityType.FIELD_OPTION_SET] | typing.Literal[EntityType.WEBHOOK] | typing.Literal[EntityType.SPECS] | typing.Literal[EntityType.GOAL] | typing.Literal[EntityType.INGREDIENT_TAG_MAP] | typing.Literal[EntityType.INGREDIENT_TAG] | typing.Literal[EntityType.CONDITION_PARAMETER] | typing.Literal[EntityType.OUTPUT] | typing.Literal[EntityType.OUTPUT_CONDITION_PARAMETER] | typing.Literal[EntityType.ASYNC_JOB] | typing.Literal[EntityType.CONSTRAINT],
     serial_alias_annotation(
     named_type_path="sdk.entity.LimitedEntityType",
     ),

uncountable/types/job_definition_t.py

@@ -98,7 +98,8 @@ class GenericUploadDataSourceSFTP(GenericUploadDataSourceBase):
     type: typing.Literal[GenericUploadDataSourceType.SFTP] = GenericUploadDataSourceType.SFTP
     host: str
     username: str
-    pem_secret: secret_retrieval_t.SecretRetrieval
+    pem_secret: secret_retrieval_t.SecretRetrieval | None = None
+    password_secret: secret_retrieval_t.SecretRetrieval | None = None
 
 
 # DO NOT MODIFY -- This file is generated by type_spec