@daocloud-proto/hydra 0.17.0-rc1-dev-10 → 0.17.0-rc1-dev-14

package/admin-api/guardrails/v1alpha1/guardrails.pb.ts

@@ -162,6 +162,7 @@ type BaseExternalDetectPlugin = {
   provider?: ExternalDetectProvider
   checkInput?: boolean
   checkOutput?: boolean
+  maxInputTokens?: number
 }
 
 export type ExternalDetectPlugin = BaseExternalDetectPlugin

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
@@ -66,6 +78,23 @@ export type APIKey = {
   disabled?: boolean
 }
 
+export type AllAPIKeyStatus = {
+  id?: string
+  expired?: boolean
+  disabled?: boolean
+  deleted?: boolean
+  name?: string
+}
+
+export type ListAllWSAPIKeysStatusRequest = {
+  workspace?: number
+  userId?: string
+}
+
+export type ListAllWSAPIKeysStatusResponse = {
+  items?: AllAPIKeyStatus[]
+}
+
 export type ListWSAPIKeyRequest = {
   workspace?: number
   page?: HydraCommonCommon.Pagination
@@ -156,6 +185,7 @@ export type WSAPIKeyUsageStatisticsRequest = {
   models?: string[]
   period?: TimePeriod
   workspace?: number
+  queryMode?: APIKeyUsageQueryMode
 }
 
 export type APIKeyUsageStatisticsRequest = {
@@ -164,11 +194,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 = {
@@ -231,6 +273,9 @@ export class WSAPIKeyManagement {
   static ListMyWSAPIKeys(req: ListWSAPIKeyRequest, initReq?: fm.InitReq): Promise<ListAPIKeyResponse> {
     return fm.fetchReq<ListWSAPIKeyRequest, ListAPIKeyResponse>(`/apis/hydra.io/v1alpha1/workspaces/${req["workspace"]}/my-apikeys?${fm.renderURLSearchParams(req, ["workspace"])}`, {...initReq, method: "GET"})
   }
+  static ListAllWSAPIKeysStatus(req: ListAllWSAPIKeysStatusRequest, initReq?: fm.InitReq): Promise<ListAllWSAPIKeysStatusResponse> {
+    return fm.fetchReq<ListAllWSAPIKeysStatusRequest, ListAllWSAPIKeysStatusResponse>(`/apis/hydra.io/v1alpha1/workspaces/${req["workspace"]}/all-apikey-statuses?${fm.renderURLSearchParams(req, ["workspace"])}`, {...initReq, method: "GET"})
+  }
   static ListWSAPIKeyMasked(req: ListWSAPIKeyRequest, initReq?: fm.InitReq): Promise<ListAPIKeyResponse> {
     return fm.fetchReq<ListWSAPIKeyRequest, ListAPIKeyResponse>(`/apis/hydra.io/v1alpha1/workspaces/${req["workspace"]}/masked-apikeys?${fm.renderURLSearchParams(req, ["workspace"])}`, {...initReq, method: "GET"})
   }

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

@@ -54,6 +54,7 @@ export enum ContextLengthRange {
   CTX_16K_64K = "CTX_16K_64K",
   CTX_64K_128K = "CTX_64K_128K",
   CTX_GE_128K = "CTX_GE_128K",
+  CTX_OTHER = "CTX_OTHER",
 }
 
 export enum ModelOperationTag {

package/package.json

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