UncountablePythonSDK 0.0.43__tar.gz → 0.0.44__tar.gz

{uncountablepythonsdk-0.0.43/UncountablePythonSDK.egg-info → uncountablepythonsdk-0.0.44}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: UncountablePythonSDK
-Version: 0.0.43
+Version: 0.0.44
 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.43 → uncountablepythonsdk-0.0.44/UncountablePythonSDK.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: UncountablePythonSDK
-Version: 0.0.43
+Version: 0.0.44
 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.43 → uncountablepythonsdk-0.0.44}/pkgs/filesystem_utils/_sftp_session.py

@@ -63,7 +63,7 @@ class SFTPSession(FileSystemSession):
         super().__init__()
         self.host: str = sftp_config.ip
         self.username: str = sftp_config.username
-        self.key_file: str | paramiko.AgentKey | None = (
+        self.key_file: str | paramiko.RSAKey | None = (
             sftp_config.pem_path
             if sftp_config.pem_path is not None
             else sftp_config.pem_key

{uncountablepythonsdk-0.0.43 → uncountablepythonsdk-0.0.44}/pkgs/filesystem_utils/file_type_utils.py

@@ -55,7 +55,7 @@ class FileSystemSFTPConfig:
     ip: str
     username: str
     pem_path: str | None
-    pem_key: paramiko.AgentKey | None = None
+    pem_key: paramiko.RSAKey | None = None
     password: str | None = None
     valid_extensions: Optional[tuple[str]] = None
     recursive: bool = True

{uncountablepythonsdk-0.0.43 → uncountablepythonsdk-0.0.44}/uncountable/integration/executors/generic_upload_executor.py

@@ -4,6 +4,8 @@ import re
 from datetime import datetime, timezone
 from tempfile import NamedTemporaryFile
 
+import paramiko
+
 from pkgs.filesystem_utils import (
     FileObjectData,
     FileSystemFileReference,
@@ -176,14 +178,14 @@ class GenericUploadJob(Job):
                 pem_secret = retrieve_secret(
                     self.data_source.pem_secret, profile_metadata=args.profile_metadata
                 )
-                with NamedTemporaryFile() as pem_file:
-                    pem_file.write(pem_secret.encode())
-                    sftp_config = FileSystemSFTPConfig(
-                        ip=self.data_source.host,
-                        username=self.data_source.username,
-                        pem_path=pem_file.name,
-                    )
-                    return SFTPSession(sftp_config=sftp_config)
+                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,
+                )
+                return SFTPSession(sftp_config=sftp_config)
 
     def run(self, args: JobArguments) -> JobResult:
         client = args.client