|
22 | 22 | from ...exceptions.custom_errors import InsufficientCreditsError, MissingWalletError |
23 | 23 | from ...exceptions.service_errors_utils import DEFAULT_BACKEND_SERVICE_STATUS_CODES |
24 | 24 | from ...models.basic_types import LogStreamingResponse, VersionStr |
25 | | -from ...models.domain.files import File |
| 25 | +from ...models.domain.files import File as DomainFile |
26 | 26 | from ...models.pagination import Page, PaginationParams |
27 | 27 | from ...models.schemas.errors import ErrorGet |
| 28 | +from ...models.schemas.files import File as SchemaFile |
28 | 29 | from ...models.schemas.jobs import ( |
29 | 30 | ArgumentTypes, |
30 | 31 | Job, |
@@ -282,19 +283,21 @@ async def get_job_outputs( |
282 | 283 | results: dict[str, ArgumentTypes] = {} |
283 | 284 | for name, value in outputs.items(): |
284 | 285 | if isinstance(value, BaseFileLink): |
285 | | - file_id: UUID = File.create_id(*value.path.split("/")) |
| 286 | + file_id: UUID = DomainFile.create_id(*value.path.split("/")) |
286 | 287 |
|
287 | 288 | found = await storage_client.search_owned_files( |
288 | 289 | user_id=user_id, file_id=file_id, limit=1 |
289 | 290 | ) |
290 | 291 | if found: |
291 | 292 | assert len(found) == 1 # nosec |
292 | | - results[name] = to_file_api_model(found[0]) |
| 293 | + results[name] = SchemaFile.from_domain_model( |
| 294 | + to_file_api_model(found[0]) |
| 295 | + ) |
293 | 296 | else: |
294 | | - api_file: File = await storage_client.create_soft_link( |
| 297 | + api_file = await storage_client.create_soft_link( |
295 | 298 | user_id=user_id, target_s3_path=value.path, as_file_id=file_id |
296 | 299 | ) |
297 | | - results[name] = api_file |
| 300 | + results[name] = SchemaFile.from_domain_model(api_file) |
298 | 301 | else: |
299 | 302 | results[name] = value |
300 | 303 |
|
|
0 commit comments