task: add download artifacts method

Author: na66im

packages/sdk/src/entities/run-item/process-run-item.spec.ts

@@ -14,6 +14,7 @@ function buildItem(
     termination_reason: overrides.termination_reason ?? undefined,
     error_code: null,
     error_message: null,
+    input_artifacts: [],
     output_artifacts: [],
   } as ItemResultReadResponse;
 }

packages/sdk/src/platform-sdk.test.ts

@@ -535,4 +535,29 @@ describe('PlatformSDK', () => {
     await expect(sdk.listRunResults('test-run-id')).rejects.toThrow(AuthenticationError);
     await expect(sdk.listRunResults('test-run-id')).rejects.toThrow(errorMessage);
   });
+
+  it('should download artifact successfully', async () => {
+    mockTokenProvider.mockResolvedValue('mocked-token');
+    setMockScenario('success');
+
+    const result = await sdk.downloadArtifact('test-run-id', 'test-artifact-id');
+    expect(result).toBeDefined();
+  });
+
+  it('should handle download artifact failure', async () => {
+    mockTokenProvider.mockResolvedValue('mocked-token');
+    setMockScenario('notFoundError');
+
+    await expect(sdk.downloadArtifact('test-run-id', 'test-artifact-id')).rejects.toThrow(
+      'Resource not found: '
+    );
+  });
+
+  it('should handle no token for download artifact', async () => {
+    mockTokenProvider.mockResolvedValue(null);
+
+    await expect(sdk.downloadArtifact('test-run-id', 'test-artifact-id')).rejects.toThrow(
+      AuthenticationError
+    );
+  });
 });

packages/sdk/src/platform-sdk.ts

@@ -121,6 +121,7 @@ export interface PlatformSDK {
     applicationId: string,
     version: string
   ): Promise<VersionReadResponse>;
+  downloadArtifact(runId: string, artifactId: string): Promise<unknown>;
 }
 /**
  * Main SDK class for interacting with the Aignostics Platform
@@ -615,4 +616,17 @@ export class PlatformSDKHttp implements PlatformSDK {
   getConfig(): PlatformSDKConfig {
     return { ...this.#config };
   }
+
+  async downloadArtifact(runId: string, artifactId: string): Promise<unknown> {
+    const client = await this.#getClient();
+    try {
+      const response = await client.getArtifactUrlV1RunsRunIdArtifactsArtifactIdFileGet({
+        runId,
+        artifactId,
+      });
+      return response;
+    } catch (error) {
+      handleRequestError(error);
+    }
+  }
 }

packages/sdk/src/test-utils/http-mocks.ts

@@ -118,6 +118,7 @@ const itemResultFactory = Factory.define<ItemResultReadResponse>(() => ({
   terminated_at: faker.date.recent().toISOString(),
   output_artifacts: outputArtifactFactory.buildList(faker.number.int({ min: 0, max: 3 })),
   error_code: null,
+  input_artifacts: [],
 }));
 
 const runFactory = Factory.define<RunReadResponse>(() => ({
@@ -246,6 +247,12 @@ export const handlers = {
     http.get('*/v1/runs/:runId/items', () => {
       return HttpResponse.json(mockResponses.runResultsSuccess, { status: 200 });
     }),
+    http.get('*/v1/runs/:runId/artifacts/:artifactId/file', () => {
+      return HttpResponse.json(
+        { url: 'https://storage.example.com/presigned-url' },
+        { status: 200 }
+      );
+    }),
   ],
 
   // Empty responses
@@ -265,6 +272,12 @@ export const handlers = {
     http.get('*/v1/runs/:runId/items', () => {
       return HttpResponse.json([], { status: 200 });
     }),
+    http.get('*/v1/runs/:runId/artifacts/:artifactId/file', () => {
+      return HttpResponse.json(
+        { url: 'https://storage.example.com/presigned-url' },
+        { status: 200 }
+      );
+    }),
   ],
 
   // Not found responses
@@ -293,6 +306,9 @@ export const handlers = {
     http.get('*/v1/runs/:runId/items', () => {
       return HttpResponse.json(mockResponses.error, { status: 404 });
     }),
+    http.get('*/v1/runs/:runId/artifacts/:artifactId/file', () => {
+      return HttpResponse.json(mockResponses.error, { status: 404 });
+    }),
   ],
 
   validationError: [
@@ -320,6 +336,9 @@ export const handlers = {
     http.get('*/v1/runs/:runId/items', () => {
       return HttpResponse.json(mockResponses.validationError, { status: 422 });
     }),
+    http.get('*/v1/runs/:runId/artifacts/:artifactId/file', () => {
+      return HttpResponse.json(mockResponses.validationError, { status: 422 });
+    }),
   ],
 
   internalServerError: [
@@ -347,6 +366,9 @@ export const handlers = {
     http.get('*/v1/runs/:runId/items', () => {
       return HttpResponse.json(mockResponses.error, { status: 500 });
     }),
+    http.get('*/v1/runs/:runId/artifacts/:artifactId/file', () => {
+      return HttpResponse.json(mockResponses.error, { status: 500 });
+    }),
   ],
 
   // Network error (connection failure)
@@ -375,6 +397,9 @@ export const handlers = {
     http.get('*/v1/runs/:runId/items', () => {
       return HttpResponse.error();
     }),
+    http.get('*/v1/runs/:runId/artifacts/:artifactId/file', () => {
+      return HttpResponse.error();
+    }),
   ],
 };