UncountablePythonSDK 0.0.95__tar.gz → 0.0.96__tar.gz

{uncountablepythonsdk-0.0.95/UncountablePythonSDK.egg-info → uncountablepythonsdk-0.0.96}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: UncountablePythonSDK
-Version: 0.0.95
+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

{uncountablepythonsdk-0.0.95 → uncountablepythonsdk-0.0.96/UncountablePythonSDK.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: UncountablePythonSDK
-Version: 0.0.95
+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

{uncountablepythonsdk-0.0.95 → uncountablepythonsdk-0.0.96}/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

{uncountablepythonsdk-0.0.95 → uncountablepythonsdk-0.0.96}/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
 

{uncountablepythonsdk-0.0.95 → uncountablepythonsdk-0.0.96}/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]

{uncountablepythonsdk-0.0.95 → uncountablepythonsdk-0.0.96}/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:

{uncountablepythonsdk-0.0.95 → uncountablepythonsdk-0.0.96}/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

{uncountablepythonsdk-0.0.95 → uncountablepythonsdk-0.0.96}/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

{uncountablepythonsdk-0.0.95 → uncountablepythonsdk-0.0.96}/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,

{uncountablepythonsdk-0.0.95 → uncountablepythonsdk-0.0.96}/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",
     ),

{uncountablepythonsdk-0.0.95 → uncountablepythonsdk-0.0.96}/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