@avaprotocol/sdk-js 1.3.2 → 1.3.3

package/dist/index.js

@@ -53,6 +53,8 @@ var require_avs_pb = __commonJS({
     }.call(null);
     var google_protobuf_wrappers_pb = require("google-protobuf/google/protobuf/wrappers_pb.js");
     goog.object.extend(proto, google_protobuf_wrappers_pb);
+    var google_protobuf_timestamp_pb = require("google-protobuf/google/protobuf/timestamp_pb.js");
+    goog.object.extend(proto, google_protobuf_timestamp_pb);
     goog.exportSymbol("proto.aggregator.BlockCondition", null, global);
     goog.exportSymbol("proto.aggregator.BranchNode", null, global);
     goog.exportSymbol("proto.aggregator.Condition", null, global);
@@ -66,6 +68,7 @@ var require_avs_pb = __commonJS({
     goog.exportSymbol("proto.aggregator.ETHTransferNode", null, global);
     goog.exportSymbol("proto.aggregator.Error", null, global);
     goog.exportSymbol("proto.aggregator.EventCondition", null, global);
+    goog.exportSymbol("proto.aggregator.EventCondition.Matcher", null, global);
     goog.exportSymbol("proto.aggregator.Execution", null, global);
     goog.exportSymbol("proto.aggregator.Execution.Step", null, global);
     goog.exportSymbol("proto.aggregator.ExecutionReq", null, global);
@@ -132,12 +135,19 @@ var require_avs_pb = __commonJS({
       proto.aggregator.BlockCondition.displayName = "proto.aggregator.BlockCondition";
     }
     proto.aggregator.EventCondition = function(opt_data) {
-      jspb.Message.initialize(this, opt_data, 0, -1, null, null);
+      jspb.Message.initialize(this, opt_data, 0, -1, proto.aggregator.EventCondition.repeatedFields_, null);
     };
     goog.inherits(proto.aggregator.EventCondition, jspb.Message);
     if (goog.DEBUG && !COMPILED) {
       proto.aggregator.EventCondition.displayName = "proto.aggregator.EventCondition";
     }
+    proto.aggregator.EventCondition.Matcher = function(opt_data) {
+      jspb.Message.initialize(this, opt_data, 0, -1, proto.aggregator.EventCondition.Matcher.repeatedFields_, null);
+    };
+    goog.inherits(proto.aggregator.EventCondition.Matcher, jspb.Message);
+    if (goog.DEBUG && !COMPILED) {
+      proto.aggregator.EventCondition.Matcher.displayName = "proto.aggregator.EventCondition.Matcher";
+    }
     proto.aggregator.TaskTrigger = function(opt_data) {
       jspb.Message.initialize(this, opt_data, 0, -1, null, proto.aggregator.TaskTrigger.oneofGroups_);
     };
@@ -669,13 +679,19 @@ var require_avs_pb = __commonJS({
     proto.aggregator.BlockCondition.prototype.setInterval = function(value) {
       return jspb.Message.setProto3IntField(this, 1, value);
     };
+    proto.aggregator.EventCondition.repeatedFields_ = [1];
     if (jspb.Message.GENERATE_TO_OBJECT) {
       proto.aggregator.EventCondition.prototype.toObject = function(opt_includeInstance) {
         return proto.aggregator.EventCondition.toObject(opt_includeInstance, this);
       };
       proto.aggregator.EventCondition.toObject = function(includeInstance, msg) {
         var f, obj = {
-          expression: jspb.Message.getFieldWithDefault(msg, 1, "")
+          matcherList: jspb.Message.toObjectList(
+            msg.getMatcherList(),
+            proto.aggregator.EventCondition.Matcher.toObject,
+            includeInstance
+          ),
+          expression: jspb.Message.getFieldWithDefault(msg, 2, "")
         };
         if (includeInstance) {
           obj.$jspbMessageInstance = msg;
@@ -696,6 +712,11 @@ var require_avs_pb = __commonJS({
         var field = reader.getFieldNumber();
         switch (field) {
           case 1:
+            var value = new proto.aggregator.EventCondition.Matcher();
+            reader.readMessage(value, proto.aggregator.EventCondition.Matcher.deserializeBinaryFromReader);
+            msg.addMatcher(value);
+            break;
+          case 2:
             var value = (
               /** @type {string} */
               reader.readString()
@@ -716,23 +737,141 @@ var require_avs_pb = __commonJS({
     };
     proto.aggregator.EventCondition.serializeBinaryToWriter = function(message, writer) {
       var f = void 0;
+      f = message.getMatcherList();
+      if (f.length > 0) {
+        writer.writeRepeatedMessage(
+          1,
+          f,
+          proto.aggregator.EventCondition.Matcher.serializeBinaryToWriter
+        );
+      }
       f = message.getExpression();
+      if (f.length > 0) {
+        writer.writeString(
+          2,
+          f
+        );
+      }
+    };
+    proto.aggregator.EventCondition.Matcher.repeatedFields_ = [2];
+    if (jspb.Message.GENERATE_TO_OBJECT) {
+      proto.aggregator.EventCondition.Matcher.prototype.toObject = function(opt_includeInstance) {
+        return proto.aggregator.EventCondition.Matcher.toObject(opt_includeInstance, this);
+      };
+      proto.aggregator.EventCondition.Matcher.toObject = function(includeInstance, msg) {
+        var f, obj = {
+          type: jspb.Message.getFieldWithDefault(msg, 1, ""),
+          valueList: (f = jspb.Message.getRepeatedField(msg, 2)) == null ? void 0 : f
+        };
+        if (includeInstance) {
+          obj.$jspbMessageInstance = msg;
+        }
+        return obj;
+      };
+    }
+    proto.aggregator.EventCondition.Matcher.deserializeBinary = function(bytes) {
+      var reader = new jspb.BinaryReader(bytes);
+      var msg = new proto.aggregator.EventCondition.Matcher();
+      return proto.aggregator.EventCondition.Matcher.deserializeBinaryFromReader(msg, reader);
+    };
+    proto.aggregator.EventCondition.Matcher.deserializeBinaryFromReader = function(msg, reader) {
+      while (reader.nextField()) {
+        if (reader.isEndGroup()) {
+          break;
+        }
+        var field = reader.getFieldNumber();
+        switch (field) {
+          case 1:
+            var value = (
+              /** @type {string} */
+              reader.readString()
+            );
+            msg.setType(value);
+            break;
+          case 2:
+            var value = (
+              /** @type {string} */
+              reader.readString()
+            );
+            msg.addValue(value);
+            break;
+          default:
+            reader.skipField();
+            break;
+        }
+      }
+      return msg;
+    };
+    proto.aggregator.EventCondition.Matcher.prototype.serializeBinary = function() {
+      var writer = new jspb.BinaryWriter();
+      proto.aggregator.EventCondition.Matcher.serializeBinaryToWriter(this, writer);
+      return writer.getResultBuffer();
+    };
+    proto.aggregator.EventCondition.Matcher.serializeBinaryToWriter = function(message, writer) {
+      var f = void 0;
+      f = message.getType();
       if (f.length > 0) {
         writer.writeString(
           1,
           f
         );
       }
+      f = message.getValueList();
+      if (f.length > 0) {
+        writer.writeRepeatedString(
+          2,
+          f
+        );
+      }
     };
-    proto.aggregator.EventCondition.prototype.getExpression = function() {
+    proto.aggregator.EventCondition.Matcher.prototype.getType = function() {
       return (
         /** @type {string} */
         jspb.Message.getFieldWithDefault(this, 1, "")
       );
     };
-    proto.aggregator.EventCondition.prototype.setExpression = function(value) {
+    proto.aggregator.EventCondition.Matcher.prototype.setType = function(value) {
       return jspb.Message.setProto3StringField(this, 1, value);
     };
+    proto.aggregator.EventCondition.Matcher.prototype.getValueList = function() {
+      return (
+        /** @type {!Array<string>} */
+        jspb.Message.getRepeatedField(this, 2)
+      );
+    };
+    proto.aggregator.EventCondition.Matcher.prototype.setValueList = function(value) {
+      return jspb.Message.setField(this, 2, value || []);
+    };
+    proto.aggregator.EventCondition.Matcher.prototype.addValue = function(value, opt_index) {
+      return jspb.Message.addToRepeatedField(this, 2, value, opt_index);
+    };
+    proto.aggregator.EventCondition.Matcher.prototype.clearValueList = function() {
+      return this.setValueList([]);
+    };
+    proto.aggregator.EventCondition.prototype.getMatcherList = function() {
+      return (
+        /** @type{!Array<!proto.aggregator.EventCondition.Matcher>} */
+        jspb.Message.getRepeatedWrapperField(this, proto.aggregator.EventCondition.Matcher, 1)
+      );
+    };
+    proto.aggregator.EventCondition.prototype.setMatcherList = function(value) {
+      return jspb.Message.setRepeatedWrapperField(this, 1, value);
+    };
+    proto.aggregator.EventCondition.prototype.addMatcher = function(opt_value, opt_index) {
+      return jspb.Message.addToRepeatedWrapperField(this, 1, opt_value, proto.aggregator.EventCondition.Matcher, opt_index);
+    };
+    proto.aggregator.EventCondition.prototype.clearMatcherList = function() {
+      return this.setMatcherList([]);
+    };
+    proto.aggregator.EventCondition.prototype.getExpression = function() {
+      return (
+        /** @type {string} */
+        jspb.Message.getFieldWithDefault(this, 2, "")
+      );
+    };
+    proto.aggregator.EventCondition.prototype.setExpression = function(value) {
+      return jspb.Message.setProto3StringField(this, 2, value);
+    };
     proto.aggregator.TaskTrigger.oneofGroups_ = [[2, 3, 4, 5, 6]];
     proto.aggregator.TaskTrigger.TriggerTypeCase = {
       TRIGGER_TYPE_NOT_SET: 0,
@@ -759,7 +898,8 @@ var require_avs_pb = __commonJS({
           fixedTime: (f = msg.getFixedTime()) && proto.aggregator.FixedTimeCondition.toObject(includeInstance, f),
           cron: (f = msg.getCron()) && proto.aggregator.CronCondition.toObject(includeInstance, f),
           block: (f = msg.getBlock()) && proto.aggregator.BlockCondition.toObject(includeInstance, f),
-          event: (f = msg.getEvent()) && proto.aggregator.EventCondition.toObject(includeInstance, f)
+          event: (f = msg.getEvent()) && proto.aggregator.EventCondition.toObject(includeInstance, f),
+          id: jspb.Message.getFieldWithDefault(msg, 7, "")
         };
         if (includeInstance) {
           obj.$jspbMessageInstance = msg;
@@ -813,6 +953,13 @@ var require_avs_pb = __commonJS({
             reader.readMessage(value, proto.aggregator.EventCondition.deserializeBinaryFromReader);
             msg.setEvent(value);
             break;
+          case 7:
+            var value = (
+              /** @type {string} */
+              reader.readString()
+            );
+            msg.setId(value);
+            break;
           default:
             reader.skipField();
             break;
@@ -874,6 +1021,13 @@ var require_avs_pb = __commonJS({
           proto.aggregator.EventCondition.serializeBinaryToWriter
         );
       }
+      f = message.getId();
+      if (f.length > 0) {
+        writer.writeString(
+          7,
+          f
+        );
+      }
     };
     proto.aggregator.TaskTrigger.prototype.getName = function() {
       return (
@@ -959,6 +1113,15 @@ var require_avs_pb = __commonJS({
     proto.aggregator.TaskTrigger.prototype.hasEvent = function() {
       return jspb.Message.getField(this, 6) != null;
     };
+    proto.aggregator.TaskTrigger.prototype.getId = function() {
+      return (
+        /** @type {string} */
+        jspb.Message.getFieldWithDefault(this, 7, "")
+      );
+    };
+    proto.aggregator.TaskTrigger.prototype.setId = function(value) {
+      return jspb.Message.setProto3StringField(this, 7, value);
+    };
     if (jspb.Message.GENERATE_TO_OBJECT) {
       proto.aggregator.ETHTransferNode.prototype.toObject = function(opt_includeInstance) {
         return proto.aggregator.ETHTransferNode.toObject(opt_includeInstance, this);
@@ -1827,7 +1990,8 @@ var require_avs_pb = __commonJS({
       };
       proto.aggregator.FilterNode.toObject = function(includeInstance, msg) {
         var f, obj = {
-          expression: jspb.Message.getFieldWithDefault(msg, 1, "")
+          expression: jspb.Message.getFieldWithDefault(msg, 1, ""),
+          input: jspb.Message.getFieldWithDefault(msg, 2, "")
         };
         if (includeInstance) {
           obj.$jspbMessageInstance = msg;
@@ -1854,6 +2018,13 @@ var require_avs_pb = __commonJS({
             );
             msg.setExpression(value);
             break;
+          case 2:
+            var value = (
+              /** @type {string} */
+              reader.readString()
+            );
+            msg.setInput(value);
+            break;
           default:
             reader.skipField();
             break;
@@ -1875,6 +2046,13 @@ var require_avs_pb = __commonJS({
           f
         );
       }
+      f = message.getInput();
+      if (f.length > 0) {
+        writer.writeString(
+          2,
+          f
+        );
+      }
     };
     proto.aggregator.FilterNode.prototype.getExpression = function() {
       return (
@@ -1885,6 +2063,15 @@ var require_avs_pb = __commonJS({
     proto.aggregator.FilterNode.prototype.setExpression = function(value) {
       return jspb.Message.setProto3StringField(this, 1, value);
     };
+    proto.aggregator.FilterNode.prototype.getInput = function() {
+      return (
+        /** @type {string} */
+        jspb.Message.getFieldWithDefault(this, 2, "")
+      );
+    };
+    proto.aggregator.FilterNode.prototype.setInput = function(value) {
+      return jspb.Message.setProto3StringField(this, 2, value);
+    };
     proto.aggregator.LoopNode.oneofGroups_ = [[10, 11, 12, 13, 14, 15]];
     proto.aggregator.LoopNode.RunnerCase = {
       RUNNER_NOT_SET: 0,
@@ -5218,8 +5405,10 @@ var require_avs_pb = __commonJS({
       proto.aggregator.GetKeyReq.toObject = function(includeInstance, msg) {
         var f, obj = {
           owner: jspb.Message.getFieldWithDefault(msg, 1, ""),
-          expiredAt: jspb.Message.getFieldWithDefault(msg, 2, 0),
-          signature: jspb.Message.getFieldWithDefault(msg, 3, "")
+          chainId: jspb.Message.getFieldWithDefault(msg, 2, 0),
+          issuedAt: (f = msg.getIssuedAt()) && google_protobuf_timestamp_pb.Timestamp.toObject(includeInstance, f),
+          expiredAt: (f = msg.getExpiredAt()) && google_protobuf_timestamp_pb.Timestamp.toObject(includeInstance, f),
+          signature: jspb.Message.getFieldWithDefault(msg, 5, "")
         };
         if (includeInstance) {
           obj.$jspbMessageInstance = msg;
@@ -5251,9 +5440,19 @@ var require_avs_pb = __commonJS({
               /** @type {number} */
               reader.readInt64()
             );
-            msg.setExpiredAt(value);
+            msg.setChainId(value);
             break;
           case 3:
+            var value = new google_protobuf_timestamp_pb.Timestamp();
+            reader.readMessage(value, google_protobuf_timestamp_pb.Timestamp.deserializeBinaryFromReader);
+            msg.setIssuedAt(value);
+            break;
+          case 4:
+            var value = new google_protobuf_timestamp_pb.Timestamp();
+            reader.readMessage(value, google_protobuf_timestamp_pb.Timestamp.deserializeBinaryFromReader);
+            msg.setExpiredAt(value);
+            break;
+          case 5:
             var value = (
               /** @type {string} */
               reader.readString()
@@ -5281,17 +5480,33 @@ var require_avs_pb = __commonJS({
           f
         );
       }
-      f = message.getExpiredAt();
+      f = message.getChainId();
       if (f !== 0) {
         writer.writeInt64(
           2,
           f
         );
       }
+      f = message.getIssuedAt();
+      if (f != null) {
+        writer.writeMessage(
+          3,
+          f,
+          google_protobuf_timestamp_pb.Timestamp.serializeBinaryToWriter
+        );
+      }
+      f = message.getExpiredAt();
+      if (f != null) {
+        writer.writeMessage(
+          4,
+          f,
+          google_protobuf_timestamp_pb.Timestamp.serializeBinaryToWriter
+        );
+      }
       f = message.getSignature();
       if (f.length > 0) {
         writer.writeString(
-          3,
+          5,
           f
         );
       }
@@ -5305,23 +5520,53 @@ var require_avs_pb = __commonJS({
     proto.aggregator.GetKeyReq.prototype.setOwner = function(value) {
       return jspb.Message.setProto3StringField(this, 1, value);
     };
-    proto.aggregator.GetKeyReq.prototype.getExpiredAt = function() {
+    proto.aggregator.GetKeyReq.prototype.getChainId = function() {
       return (
         /** @type {number} */
         jspb.Message.getFieldWithDefault(this, 2, 0)
       );
     };
-    proto.aggregator.GetKeyReq.prototype.setExpiredAt = function(value) {
+    proto.aggregator.GetKeyReq.prototype.setChainId = function(value) {
       return jspb.Message.setProto3IntField(this, 2, value);
     };
+    proto.aggregator.GetKeyReq.prototype.getIssuedAt = function() {
+      return (
+        /** @type{?proto.google.protobuf.Timestamp} */
+        jspb.Message.getWrapperField(this, google_protobuf_timestamp_pb.Timestamp, 3)
+      );
+    };
+    proto.aggregator.GetKeyReq.prototype.setIssuedAt = function(value) {
+      return jspb.Message.setWrapperField(this, 3, value);
+    };
+    proto.aggregator.GetKeyReq.prototype.clearIssuedAt = function() {
+      return this.setIssuedAt(void 0);
+    };
+    proto.aggregator.GetKeyReq.prototype.hasIssuedAt = function() {
+      return jspb.Message.getField(this, 3) != null;
+    };
+    proto.aggregator.GetKeyReq.prototype.getExpiredAt = function() {
+      return (
+        /** @type{?proto.google.protobuf.Timestamp} */
+        jspb.Message.getWrapperField(this, google_protobuf_timestamp_pb.Timestamp, 4)
+      );
+    };
+    proto.aggregator.GetKeyReq.prototype.setExpiredAt = function(value) {
+      return jspb.Message.setWrapperField(this, 4, value);
+    };
+    proto.aggregator.GetKeyReq.prototype.clearExpiredAt = function() {
+      return this.setExpiredAt(void 0);
+    };
+    proto.aggregator.GetKeyReq.prototype.hasExpiredAt = function() {
+      return jspb.Message.getField(this, 4) != null;
+    };
     proto.aggregator.GetKeyReq.prototype.getSignature = function() {
       return (
         /** @type {string} */
-        jspb.Message.getFieldWithDefault(this, 3, "")
+        jspb.Message.getFieldWithDefault(this, 5, "")
       );
     };
     proto.aggregator.GetKeyReq.prototype.setSignature = function(value) {
-      return jspb.Message.setProto3StringField(this, 3, value);
+      return jspb.Message.setProto3StringField(this, 5, value);
     };
     if (jspb.Message.GENERATE_TO_OBJECT) {
       proto.aggregator.KeyResp.prototype.toObject = function(opt_includeInstance) {
@@ -5996,6 +6241,7 @@ var require_avs_grpc_pb = __commonJS({
     var grpc2 = require("@grpc/grpc-js");
     var avs_pb20 = require_avs_pb();
     var google_protobuf_wrappers_pb = require("google-protobuf/google/protobuf/wrappers_pb.js");
+    var google_protobuf_timestamp_pb = require("google-protobuf/google/protobuf/timestamp_pb.js");
     function serialize_aggregator_CreateTaskReq(arg) {
       if (!(arg instanceof avs_pb20.CreateTaskReq)) {
         throw new Error("Expected argument of type aggregator.CreateTaskReq");
@@ -6382,13 +6628,21 @@ var grpc = __toESM(require("@grpc/grpc-js"));
 var import_grpc_js = require("@grpc/grpc-js");
 
 // src/auth.ts
-var getKeyRequestMessage = (address, expiredAt) => {
-  return `key request for ${address} expired at ${expiredAt}`;
+var getKeyRequestMessage = (chainId, address, issuedAt, expiredAt) => {
+  return `Please sign the below text for ownership verification.
+
+URI: https://app.avaprotocol.org
+Chain ID: ${chainId}
+Version: 1
+Issued At: ${issuedAt.toISOString()}
+Expire At: ${expiredAt.toISOString()}
+Wallet: ${address}`;
 };
 
 // src/index.ts
 var import_avs_grpc_pb = __toESM(require_avs_grpc_pb());
 var avs_pb19 = __toESM(require_avs_pb());
+var import_timestamp_pb = require("google-protobuf/google/protobuf/timestamp_pb");
 
 // src/models/workflow.ts
 var import_lodash3 = __toESM(require("lodash"));
@@ -6425,6 +6679,7 @@ var Trigger = class {
    * @param props
    */
   constructor(props) {
+    this.id = props.id;
     this.name = props.name;
     this.type = props.type;
     this.data = props.data;
@@ -6466,6 +6721,7 @@ var BlockTrigger = class _BlockTrigger extends interface_default {
   toRequest() {
     const request = new avs_pb3.TaskTrigger();
     request.setName(this.name);
+    request.setId(this.id);
     if (!this.data) {
       throw new Error(`Trigger data is missing for ${this.type}`);
     }
@@ -6499,6 +6755,7 @@ var CronTrigger = class _CronTrigger extends interface_default {
   toRequest() {
     const request = new avs_pb4.TaskTrigger();
     request.setName(this.name);
+    request.setId(this.id);
     if (!this.data) {
       throw new Error(`Trigger data is missing for ${this.type}`);
     }
@@ -6534,11 +6791,25 @@ var EventTrigger = class _EventTrigger extends interface_default {
   toRequest() {
     const request = new avs_pb5.TaskTrigger();
     request.setName(this.name);
+    request.setId(this.id);
     if (!this.data) {
       throw new Error(`Trigger data is missing for ${this.type}`);
     }
     const condition = new avs_pb5.EventCondition();
-    condition.setExpression(this.data.expression);
+    const expression = this.data.expression;
+    const matcher = this.data.matcher;
+    if (expression && expression != "") {
+      condition.setExpression(expression);
+    }
+    if (matcher && matcher.length >= 1) {
+      console.log("hit here", matcher);
+      condition.setMatcherList(matcher.map((element) => {
+        const m = new avs_pb5.EventCondition.Matcher();
+        m.setType(element["type"]);
+        m.setValueList(element["value"]);
+        return m;
+      }));
+    }
     request.setEvent(condition);
     console.log("EventTrigger.toRequest.request:", request.toObject());
     return request;
@@ -6546,10 +6817,22 @@ var EventTrigger = class _EventTrigger extends interface_default {
   static fromResponse(raw) {
     const obj = raw.toObject();
     console.log("EventTrigger.fromResponse.obj:", obj);
+    let data = {};
+    if (raw.getEvent().getExpression()) {
+      data.expression = raw.getEvent().getExpression();
+    }
+    if (raw.getEvent().getMatcherList()) {
+      data.matcher = raw.getEvent().getMatcherList().map((item) => {
+        return {
+          type: item.getType(),
+          value: item.getValueList()
+        };
+      });
+    }
     return new _EventTrigger({
       ...obj,
       type: "event" /* Event */,
-      data: raw.getEvent().toObject()
+      data
     });
   }
 };
@@ -6569,6 +6852,7 @@ var FixedTimeTrigger = class _FixedTimeTrigger extends interface_default {
   toRequest() {
     const request = new avs_pb6.TaskTrigger();
     request.setName(this.name);
+    request.setId(this.id);
     if (!this.data) {
       throw new Error(`Trigger data is missing for ${this.type}`);
     }
@@ -7253,10 +7537,20 @@ var BaseClient = class {
    * @param expiredAtEpoch - The expiration epoch
    * @returns {Promise<GetKeyResponse>} - The response from the auth call
    */
-  async authWithAPIKey(address, apiKey, expiredAtEpoch) {
+  async authWithAPIKey({
+    chainId,
+    address,
+    issuedAt,
+    expiredAt,
+    apiKey
+  }) {
     const request = new avs_pb19.GetKeyReq();
+    request.setChainId(chainId);
     request.setOwner(address);
-    request.setExpiredAt(expiredAtEpoch);
+    const issueTs = import_timestamp_pb.Timestamp.fromDate(issuedAt);
+    const expiredTs = import_timestamp_pb.Timestamp.fromDate(expiredAt);
+    request.setIssuedAt(issueTs);
+    request.setExpiredAt(expiredTs);
     request.setSignature(apiKey);
     const result = await this.sendGrpcRequest("getKey", request);
     return { authKey: result.getKey() };
@@ -7268,10 +7562,20 @@ var BaseClient = class {
    * @param expiredAtEpoch - The expiration epoch
    * @returns {Promise<GetKeyResponse>} - The response from the auth call
    */
-  async authWithSignature(address, signature, expiredAtEpoch) {
+  async authWithSignature({
+    chainId,
+    address,
+    issuedAt,
+    expiredAt,
+    signature
+  }) {
     const request = new avs_pb19.GetKeyReq();
+    request.setChainId(chainId);
     request.setOwner(address);
-    request.setExpiredAt(expiredAtEpoch);
+    const issueTs = import_timestamp_pb.Timestamp.fromDate(issuedAt);
+    const expiredTs = import_timestamp_pb.Timestamp.fromDate(expiredAt);
+    request.setIssuedAt(issueTs);
+    request.setExpiredAt(expiredTs);
     request.setSignature(signature);
     const result = await this.sendGrpcRequest(
       "getKey",