[WORKFLOW] AI update based on proto changes from commit f7cb07e

Author: viambot

src/app/data-client.spec.ts

@@ -72,6 +72,7 @@ import {
   DataCaptureUploadRequest,
   DataCaptureUploadResponse,
   DataType,
+  MimeType,
   SensorData,
   SensorMetadata,
   UploadMetadata,
@@ -93,10 +94,19 @@ import {
   ListDataPipelineRunsResponse,
 } from '../gen/app/datapipelines/v1/data_pipelines_pb';
 import { DataPipelinesService } from '../gen/app/datapipelines/v1/data_pipelines_connect';
+import { DataManagerService } from '../gen/service/datamanager/v1/data_manager_connect';
+import {
+  UploadBinaryDataToDatasetsRequest,
+  UploadBinaryDataToDatasetsResponse,
+} from '../gen/service/datamanager/v1/data_manager_pb';
+import { DataManagerClient } from '../services/data-manager/client';
+
 vi.mock('../gen/app/data/v1/data_pb_service');
 
 let mockTransport: Transport;
 const subject = () => new DataClient(mockTransport);
+const dataManagerSubject = () =>
+  new DataManagerClient({} as any, 'data_manager', {}); // Mock RobotClient
 
 describe('DataClient tests', () => {
   const filter = subject().createFilter({
@@ -1716,3 +1726,46 @@ describe('DataPipelineClient tests', () => {
     });
   });
 });
+
+describe('DataManagerClient tests', () => {
+  const name = 'data_manager';
+  const binaryData = new Uint8Array([1, 2, 3]);
+  const tags = ['tag1', 'tag2'];
+  const datasetIds = ['dataset1', 'dataset2'];
+  const mimeType = MimeType.MIME_TYPE_JPEG;
+  const extra = { key: 'value' };
+
+  describe('uploadBinaryDataToDatasets', () => {
+    let capReq: UploadBinaryDataToDatasetsRequest;
+    beforeEach(() => {
+      mockTransport = createRouterTransport(({ service }) => {
+        service(DataManagerService, {
+          uploadBinaryDataToDatasets: (req) => {
+            capReq = req;
+            return new UploadBinaryDataToDatasetsResponse();
+          },
+        });
+      });
+    });
+
+    it('uploads binary data to datasets', async () => {
+      const expectedRequest = new UploadBinaryDataToDatasetsRequest({
+        name,
+        binaryData,
+        tags,
+        datasetIds,
+        mimeType,
+        extra: Struct.fromJson(extra),
+      });
+
+      await dataManagerSubject().uploadBinaryDataToDatasets(
+        binaryData,
+        tags,
+        datasetIds,
+        mimeType,
+        extra
+      );
+      expect(capReq).toStrictEqual(expectedRequest);
+    });
+  });
+});

src/app/data-client.ts

@@ -604,6 +604,7 @@ export class DataClient {
    * @param tags The tags to add to the data
    * @param filter Optional `pb.Filter` specifying binary data to add tags to.
    *   No `filter` implies all binary data.
+   * @deprecated
    */
   async addTagsToBinaryDataByFilter(tags: string[], filter?: Filter) {
     await this.dataClient.addTagsToBinaryDataByFilter({
@@ -679,6 +680,7 @@ export class DataClient {
    * @param filter Optional `pb.Filter` specifying binary data to add tags to.
    *   No `filter` implies all binary data.
    * @returns The number of items deleted
+   * @deprecated
    */
   async removeTagsFromBinaryDataByFilter(tags: string[], filter?: Filter) {
     const resp = await this.dataClient.removeTagsFromBinaryDataByFilter({
@@ -705,6 +707,7 @@ export class DataClient {
    * @param filter Optional `pb.Filter` specifying what data to get tags from.
    *   No `filter` implies all data.
    * @returns The list of tags
+   * @deprecated
    */
   async tagsByFilter(filter?: Filter) {
     const resp = await this.dataClient.tagsByFilter({ filter });
@@ -826,6 +829,7 @@ export class DataClient {
    * @param filter Optional `pb.Filter` specifying what data to get tags from.
    *   No `filter` implies all labels.
    * @returns The list of bounding box labels
+   * @deprecated
    */
   async boundingBoxLabelsByFilter(filter?: Filter) {
     const resp = await this.dataClient.boundingBoxLabelsByFilter({

src/services/data-manager/client.ts

@@ -1,11 +1,15 @@
 import { Struct, type JsonValue } from '@bufbuild/protobuf';
 import type { CallOptions, PromiseClient } from '@connectrpc/connect';
 import { DataManagerService } from '../../gen/service/datamanager/v1/data_manager_connect.js';
-import { SyncRequest } from '../../gen/service/datamanager/v1/data_manager_pb.js';
+import {
+  SyncRequest,
+  UploadBinaryDataToDatasetsRequest,
+} from '../../gen/service/datamanager/v1/data_manager_pb.js';
 import type { RobotClient } from '../../robot';
 import type { Options } from '../../types';
 import { doCommandFromClient } from '../../utils';
 import type { DataManager } from './data-manager';
+import { MimeType } from '../../app/datasync/v1/data_sync_pb.js';
 
 export class DataManagerClient implements DataManager {
   private client: PromiseClient<typeof DataManagerService>;
@@ -80,4 +84,55 @@ export class DataManagerClient implements DataManager {
       callOptions
     );
   }
-}
+
+  /**
+   * Uploads binary data to specified datasets.
+   *
+   * @example
+   *
+   * ```ts
+   * const dataManager = new VIAM.DataManagerClient(
+   *   machine,
+   *   'my_data_manager'
+   * );
+   * const binaryData = new Uint8Array([1, 2, 3]);
+   * const tags = ['tag1', 'tag2'];
+   * const datasetIds = ['dataset1', 'dataset2'];
+   * const mimeType = MimeType.MIME_TYPE_JPEG;
+   * await dataManager.uploadBinaryDataToDatasets(
+   *   binaryData,
+   *   tags,
+   *   datasetIds,
+   *   mimeType
+   * );
+   * ```
+   *
+   * @param binaryData - The binary data to upload.
+   * @param tags - Tags to associate with the binary data.
+   * @param datasetIds - IDs of the datasets to upload the data to.
+   * @param mimeType - The MIME type of the binary data.
+   * @param extra - Additional arguments to pass to the method.
+   * @param callOptions - Call options for the request.
+   */
+  async uploadBinaryDataToDatasets(
+    binaryData: Uint8Array,
+    tags: string[],
+    datasetIds: string[],
+    mimeType: MimeType,
+    extra = {},
+    callOptions = this.callOptions
+  ) {
+    const request = new UploadBinaryDataToDatasetsRequest({
+      name: this.name,
+      binaryData,
+      tags,
+      datasetIds,
+      mimeType,
+      extra: Struct.fromJson(extra),
+    });
+
+    this.options.requestLogger?.(request);
+
+    await this.client.uploadBinaryDataToDatasets(request, callOptions);
+  }
+}

src/services/data-manager/data-manager.ts

@@ -1,6 +1,23 @@
 import type { Struct } from '@bufbuild/protobuf';
 import type { Resource } from '../../types';
+import type { MimeType } from '../../app/datasync/v1/data_sync_pb.js';
 
 export interface DataManager extends Resource {
   sync: (extra?: Struct) => Promise<void>;
+  /**
+   * Uploads binary data to specified datasets.
+   *
+   * @param binaryData - The binary data to upload.
+   * @param tags - Tags to associate with the binary data.
+   * @param datasetIds - IDs of the datasets to upload the data to.
+   * @param mimeType - The MIME type of the binary data.
+   * @param extra - Additional arguments to pass to the method.
+   */
+  uploadBinaryDataToDatasets: (
+    binaryData: Uint8Array,
+    tags: string[],
+    datasetIds: string[],
+    mimeType: MimeType,
+    extra?: Struct
+  ) => Promise<void>;
 }