@daocloud-proto/hydra 0.17.0-rc1-dev-13 → 0.17.0-rc1-dev-15

package/admin-api/usage-aggregation/v1alpha1/usage_aggregation.pb.ts

@@ -0,0 +1,42 @@
+/* eslint-disable */
+// @ts-nocheck
+/*
+* This file is a generated Typescript file for GRPC Gateway, DO NOT MODIFY
+*/
+
+import * as fm from "../../../fetch.pb"
+import * as GoogleProtobufTimestamp from "../../../google/protobuf/timestamp.pb"
+
+export enum UsageAggregationPhase {
+  USAGE_AGGREGATION_PHASE_UNSPECIFIED = "USAGE_AGGREGATION_PHASE_UNSPECIFIED",
+  USAGE_AGGREGATION_PHASE_READY = "USAGE_AGGREGATION_PHASE_READY",
+  USAGE_AGGREGATION_PHASE_PENDING = "USAGE_AGGREGATION_PHASE_PENDING",
+  USAGE_AGGREGATION_PHASE_SHRINKING = "USAGE_AGGREGATION_PHASE_SHRINKING",
+  USAGE_AGGREGATION_PHASE_BACKFILLING = "USAGE_AGGREGATION_PHASE_BACKFILLING",
+  USAGE_AGGREGATION_PHASE_INVALID = "USAGE_AGGREGATION_PHASE_INVALID",
+}
+
+export type GetUsageAggregationConfigRequest = {
+}
+
+export type UpdateUsageAggregationConfigRequest = {
+  retentionDays?: number
+}
+
+export type UsageAggregationConfig = {
+  retentionDays?: number
+  desiredCoveredFrom?: GoogleProtobufTimestamp.Timestamp
+  coveredFrom?: GoogleProtobufTimestamp.Timestamp
+  coveredThrough?: GoogleProtobufTimestamp.Timestamp
+  phase?: UsageAggregationPhase
+  message?: string
+}
+
+export class UsageAggregationConfigManagement {
+  static GetUsageAggregationConfig(req: GetUsageAggregationConfigRequest, initReq?: fm.InitReq): Promise<UsageAggregationConfig> {
+    return fm.fetchReq<GetUsageAggregationConfigRequest, UsageAggregationConfig>(`/apis/admin.hydra.io/v1alpha1/usage-aggregation/config?${fm.renderURLSearchParams(req, [])}`, {...initReq, method: "GET"})
+  }
+  static UpdateUsageAggregationConfig(req: UpdateUsageAggregationConfigRequest, initReq?: fm.InitReq): Promise<UsageAggregationConfig> {
+    return fm.fetchReq<UpdateUsageAggregationConfigRequest, UsageAggregationConfig>(`/apis/admin.hydra.io/v1alpha1/usage-aggregation/config`, {...initReq, method: "PUT", body: JSON.stringify(req, fm.replacer)})
+  }
+}

package/management-api/apikey/v1alpha1/apikey.pb.ts

@@ -18,6 +18,12 @@ type OneOf<T> =
         : never)
     : never);
 
+export enum APIKeyUsageQueryMode {
+  API_KEY_USAGE_QUERY_MODE_UNSPECIFIED = "API_KEY_USAGE_QUERY_MODE_UNSPECIFIED",
+  API_KEY_USAGE_QUERY_MODE_FAST = "API_KEY_USAGE_QUERY_MODE_FAST",
+  API_KEY_USAGE_QUERY_MODE_COMPLETE = "API_KEY_USAGE_QUERY_MODE_COMPLETE",
+}
+
 export enum TimePeriod {
   TIME_PERIOD_UNSPECIFIED = "TIME_PERIOD_UNSPECIFIED",
   TIME_PERIOD_HOUR = "TIME_PERIOD_HOUR",
@@ -26,6 +32,12 @@ export enum TimePeriod {
   TIME_PERIOD_MONTH = "TIME_PERIOD_MONTH",
 }
 
+export enum APIKeyUsageQueryStatus {
+  API_KEY_USAGE_QUERY_STATUS_UNSPECIFIED = "API_KEY_USAGE_QUERY_STATUS_UNSPECIFIED",
+  API_KEY_USAGE_QUERY_STATUS_COMPLETED = "API_KEY_USAGE_QUERY_STATUS_COMPLETED",
+  API_KEY_USAGE_QUERY_STATUS_CONFIRMATION_REQUIRED = "API_KEY_USAGE_QUERY_STATUS_CONFIRMATION_REQUIRED",
+}
+
 
 /* hydra modified */ export type BaseCreateWSAPIKeyRequest = {
   name?: string
@@ -64,6 +76,7 @@ export type APIKey = {
   lastUsedTime?: GoogleProtobufTimestamp.Timestamp
   expired?: boolean
   disabled?: boolean
+  quotaLimitedModels?: string[]
 }
 
 export type AllAPIKeyStatus = {
@@ -173,6 +186,7 @@ export type WSAPIKeyUsageStatisticsRequest = {
   models?: string[]
   period?: TimePeriod
   workspace?: number
+  queryMode?: APIKeyUsageQueryMode
 }
 
 export type APIKeyUsageStatisticsRequest = {
@@ -181,11 +195,23 @@ export type APIKeyUsageStatisticsRequest = {
   ids?: string[]
   models?: string[]
   period?: TimePeriod
+  queryMode?: APIKeyUsageQueryMode
 }
 
 export type APIKeyUsageStatisticsResponse = {
   dataPoints?: DataPoint[]
   totalUsage?: Usage
+  queryStatus?: APIKeyUsageQueryStatus
+  warnings?: APIKeyUsageWarning[]
+}
+
+export type APIKeyUsageWarning = {
+  code?: string
+  message?: string
+  requestedStart?: GoogleProtobufTimestamp.Timestamp
+  hourlyCoveredFrom?: GoogleProtobufTimestamp.Timestamp
+  uncoveredStart?: GoogleProtobufTimestamp.Timestamp
+  uncoveredEnd?: GoogleProtobufTimestamp.Timestamp
 }
 
 export type Usage = {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@daocloud-proto/hydra",
-  "version": "v0.17.0-rc1-dev-13",
+  "version": "v0.17.0-rc1-dev-15",
   "description": "",
   "author": "",
   "license": "ISC"