Adding estimate cost operation

Author: johnsabath

gooseai/dashboard/dashboard.pb.go

@@ -13,13 +13,7 @@
 
 var jspb = require('google-protobuf');
 var goog = jspb;
-var global = (function() {
-  if (this) { return this; }
-  if (typeof window !== 'undefined') { return window; }
-  if (typeof global !== 'undefined') { return global; }
-  if (typeof self !== 'undefined') { return self; }
-  return Function('return this')();
-}.call(null));
+var global = (function() { return this || window || global || self || Function('return this')(); }).call(null);
 
 goog.exportSymbol('proto.gooseai.EngineInfo', null, global);
 goog.exportSymbol('proto.gooseai.EngineTokenizer', null, global);

gooseai/engines/engines_pb.js

@@ -9,13 +9,15 @@ import * as grpc from "grpc";
 interface IGenerationServiceService extends grpc.ServiceDefinition<grpc.UntypedServiceImplementation> {
   generate: grpc.MethodDefinition<generation_pb.Request, generation_pb.Answer>;
   chainGenerate: grpc.MethodDefinition<generation_pb.ChainRequest, generation_pb.Answer>;
+  estimateCost: grpc.MethodDefinition<generation_pb.Request, generation_pb.EstimateCostResponse>;
 }
 
 export const GenerationServiceService: IGenerationServiceService;
 
 export interface IGenerationServiceServer extends grpc.UntypedServiceImplementation {
   generate: grpc.handleServerStreamingCall<generation_pb.Request, generation_pb.Answer>;
   chainGenerate: grpc.handleServerStreamingCall<generation_pb.ChainRequest, generation_pb.Answer>;
+  estimateCost: grpc.handleUnaryCall<generation_pb.Request, generation_pb.EstimateCostResponse>;
 }
 
 export class GenerationServiceClient extends grpc.Client {
@@ -24,4 +26,7 @@ export class GenerationServiceClient extends grpc.Client {
   generate(argument: generation_pb.Request, metadata?: grpc.Metadata | null, options?: grpc.CallOptions | null): grpc.ClientReadableStream<generation_pb.Answer>;
   chainGenerate(argument: generation_pb.ChainRequest, metadataOrOptions?: grpc.Metadata | grpc.CallOptions | null): grpc.ClientReadableStream<generation_pb.Answer>;
   chainGenerate(argument: generation_pb.ChainRequest, metadata?: grpc.Metadata | null, options?: grpc.CallOptions | null): grpc.ClientReadableStream<generation_pb.Answer>;
+  estimateCost(argument: generation_pb.Request, callback: grpc.requestCallback<generation_pb.EstimateCostResponse>): grpc.ClientUnaryCall;
+  estimateCost(argument: generation_pb.Request, metadataOrOptions: grpc.Metadata | grpc.CallOptions | null, callback: grpc.requestCallback<generation_pb.EstimateCostResponse>): grpc.ClientUnaryCall;
+  estimateCost(argument: generation_pb.Request, metadata: grpc.Metadata | null, options: grpc.CallOptions | null, callback: grpc.requestCallback<generation_pb.EstimateCostResponse>): grpc.ClientUnaryCall;
 }

gooseai/generation/generation.pb.go

@@ -26,6 +26,17 @@ function deserialize_gooseai_ChainRequest(buffer_arg) {
   return generation_pb.ChainRequest.deserializeBinary(new Uint8Array(buffer_arg));
 }
 
+function serialize_gooseai_EstimateCostResponse(arg) {
+  if (!(arg instanceof generation_pb.EstimateCostResponse)) {
+    throw new Error('Expected argument of type gooseai.EstimateCostResponse');
+  }
+  return Buffer.from(arg.serializeBinary());
+}
+
+function deserialize_gooseai_EstimateCostResponse(buffer_arg) {
+  return generation_pb.EstimateCostResponse.deserializeBinary(new Uint8Array(buffer_arg));
+}
+
 function serialize_gooseai_Request(arg) {
   if (!(arg instanceof generation_pb.Request)) {
     throw new Error('Expected argument of type gooseai.Request');
@@ -64,6 +75,17 @@ var GenerationServiceService = exports.GenerationServiceService = {
     responseSerialize: serialize_gooseai_Answer,
     responseDeserialize: deserialize_gooseai_Answer,
   },
+  estimateCost: {
+    path: '/gooseai.GenerationService/EstimateCost',
+    requestStream: false,
+    responseStream: false,
+    requestType: generation_pb.Request,
+    responseType: generation_pb.EstimateCostResponse,
+    requestSerialize: serialize_gooseai_Request,
+    requestDeserialize: deserialize_gooseai_Request,
+    responseSerialize: serialize_gooseai_EstimateCostResponse,
+    responseDeserialize: deserialize_gooseai_EstimateCostResponse,
+  },
 };
 
 exports.GenerationServiceClient = grpc.makeGenericClientConstructor(GenerationServiceService);

gooseai/generation/generation_grpc.pb.go

@@ -643,6 +643,26 @@ export namespace ImageParameters {
   }
 }
 
+export class EstimateCostResponse extends jspb.Message {
+  getTotalCreditCost(): number;
+  setTotalCreditCost(value: number): void;
+
+  serializeBinary(): Uint8Array;
+  toObject(includeInstance?: boolean): EstimateCostResponse.AsObject;
+  static toObject(includeInstance: boolean, msg: EstimateCostResponse): EstimateCostResponse.AsObject;
+  static extensions: {[key: number]: jspb.ExtensionFieldInfo<jspb.Message>};
+  static extensionsBinary: {[key: number]: jspb.ExtensionFieldBinaryInfo<jspb.Message>};
+  static serializeBinaryToWriter(message: EstimateCostResponse, writer: jspb.BinaryWriter): void;
+  static deserializeBinary(bytes: Uint8Array): EstimateCostResponse;
+  static deserializeBinaryFromReader(message: EstimateCostResponse, reader: jspb.BinaryReader): EstimateCostResponse;
+}
+
+export namespace EstimateCostResponse {
+  export type AsObject = {
+    totalCreditCost: number,
+  }
+}
+
 export class ClassifierConcept extends jspb.Message {
   getConcept(): string;
   setConcept(value: string): void;

gooseai/generation/generation_grpc_pb.d.ts

@@ -32,6 +32,7 @@ goog.exportSymbol('proto.gooseai.ClassifierParameters', null, global);
 goog.exportSymbol('proto.gooseai.ConditionerParameters', null, global);
 goog.exportSymbol('proto.gooseai.CutoutParameters', null, global);
 goog.exportSymbol('proto.gooseai.DiffusionSampler', null, global);
+goog.exportSymbol('proto.gooseai.EstimateCostResponse', null, global);
 goog.exportSymbol('proto.gooseai.FinishReason', null, global);
 goog.exportSymbol('proto.gooseai.GuidanceInstanceParameters', null, global);
 goog.exportSymbol('proto.gooseai.GuidanceParameters', null, global);
@@ -392,6 +393,27 @@ if (goog.DEBUG && !COMPILED) {
    */
   proto.gooseai.ImageParameters.displayName = 'proto.gooseai.ImageParameters';
 }
+/**
+ * Generated by JsPbCodeGenerator.
+ * @param {Array=} opt_data Optional initial data array, typically from a
+ * server response, or constructed directly in Javascript. The array is used
+ * in place and becomes part of the constructed object. It is not cloned.
+ * If no data is provided, the constructed object will be empty, but still
+ * valid.
+ * @extends {jspb.Message}
+ * @constructor
+ */
+proto.gooseai.EstimateCostResponse = function(opt_data) {
+  jspb.Message.initialize(this, opt_data, 0, -1, null, null);
+};
+goog.inherits(proto.gooseai.EstimateCostResponse, jspb.Message);
+if (goog.DEBUG && !COMPILED) {
+  /**
+   * @public
+   * @override
+   */
+  proto.gooseai.EstimateCostResponse.displayName = 'proto.gooseai.EstimateCostResponse';
+}
 /**
  * Generated by JsPbCodeGenerator.
  * @param {Array=} opt_data Optional initial data array, typically from a
@@ -5240,6 +5262,136 @@ proto.gooseai.ImageParameters.prototype.clearParametersList = function() {
 
 
 
+if (jspb.Message.GENERATE_TO_OBJECT) {
+/**
+ * Creates an object representation of this proto.
+ * Field names that are reserved in JavaScript and will be renamed to pb_name.
+ * Optional fields that are not set will be set to undefined.
+ * To access a reserved field use, foo.pb_<name>, eg, foo.pb_default.
+ * For the list of reserved names please see:
+ *     net/proto2/compiler/js/internal/generator.cc#kKeyword.
+ * @param {boolean=} opt_includeInstance Deprecated. whether to include the
+ *     JSPB instance for transitional soy proto support:
+ *     http://goto/soy-param-migration
+ * @return {!Object}
+ */
+proto.gooseai.EstimateCostResponse.prototype.toObject = function(opt_includeInstance) {
+  return proto.gooseai.EstimateCostResponse.toObject(opt_includeInstance, this);
+};
+
+
+/**
+ * Static version of the {@see toObject} method.
+ * @param {boolean|undefined} includeInstance Deprecated. Whether to include
+ *     the JSPB instance for transitional soy proto support:
+ *     http://goto/soy-param-migration
+ * @param {!proto.gooseai.EstimateCostResponse} msg The msg instance to transform.
+ * @return {!Object}
+ * @suppress {unusedLocalVariables} f is only used for nested messages
+ */
+proto.gooseai.EstimateCostResponse.toObject = function(includeInstance, msg) {
+  var f, obj = {
+    totalCreditCost: jspb.Message.getFloatingPointFieldWithDefault(msg, 1, 0.0)
+  };
+
+  if (includeInstance) {
+    obj.$jspbMessageInstance = msg;
+  }
+  return obj;
+};
+}
+
+
+/**
+ * Deserializes binary data (in protobuf wire format).
+ * @param {jspb.ByteSource} bytes The bytes to deserialize.
+ * @return {!proto.gooseai.EstimateCostResponse}
+ */
+proto.gooseai.EstimateCostResponse.deserializeBinary = function(bytes) {
+  var reader = new jspb.BinaryReader(bytes);
+  var msg = new proto.gooseai.EstimateCostResponse;
+  return proto.gooseai.EstimateCostResponse.deserializeBinaryFromReader(msg, reader);
+};
+
+
+/**
+ * Deserializes binary data (in protobuf wire format) from the
+ * given reader into the given message object.
+ * @param {!proto.gooseai.EstimateCostResponse} msg The message object to deserialize into.
+ * @param {!jspb.BinaryReader} reader The BinaryReader to use.
+ * @return {!proto.gooseai.EstimateCostResponse}
+ */
+proto.gooseai.EstimateCostResponse.deserializeBinaryFromReader = function(msg, reader) {
+  while (reader.nextField()) {
+    if (reader.isEndGroup()) {
+      break;
+    }
+    var field = reader.getFieldNumber();
+    switch (field) {
+    case 1:
+      var value = /** @type {number} */ (reader.readDouble());
+      msg.setTotalCreditCost(value);
+      break;
+    default:
+      reader.skipField();
+      break;
+    }
+  }
+  return msg;
+};
+
+
+/**
+ * Serializes the message to binary data (in protobuf wire format).
+ * @return {!Uint8Array}
+ */
+proto.gooseai.EstimateCostResponse.prototype.serializeBinary = function() {
+  var writer = new jspb.BinaryWriter();
+  proto.gooseai.EstimateCostResponse.serializeBinaryToWriter(this, writer);
+  return writer.getResultBuffer();
+};
+
+
+/**
+ * Serializes the given message to binary data (in protobuf wire
+ * format), writing to the given BinaryWriter.
+ * @param {!proto.gooseai.EstimateCostResponse} message
+ * @param {!jspb.BinaryWriter} writer
+ * @suppress {unusedLocalVariables} f is only used for nested messages
+ */
+proto.gooseai.EstimateCostResponse.serializeBinaryToWriter = function(message, writer) {
+  var f = undefined;
+  f = message.getTotalCreditCost();
+  if (f !== 0.0) {
+    writer.writeDouble(
+      1,
+      f
+    );
+  }
+};
+
+
+/**
+ * optional double total_credit_cost = 1;
+ * @return {number}
+ */
+proto.gooseai.EstimateCostResponse.prototype.getTotalCreditCost = function() {
+  return /** @type {number} */ (jspb.Message.getFloatingPointFieldWithDefault(this, 1, 0.0));
+};
+
+
+/**
+ * @param {number} value
+ * @return {!proto.gooseai.EstimateCostResponse} returns this
+ */
+proto.gooseai.EstimateCostResponse.prototype.setTotalCreditCost = function(value) {
+  return jspb.Message.setProto3FloatField(this, 1, value);
+};
+
+
+
+
+
 if (jspb.Message.GENERATE_TO_OBJECT) {
 /**
  * Creates an object representation of this proto.