@daocloud-proto/hydra 0.16.0-dev-11 → 0.16.0-dev-13

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

@@ -85,6 +85,16 @@ export type DeleteWSAPIKeyRequest = {
   workspace?: number
 }
 
+export type GetWSAPIKeySecretRequest = {
+  id?: string
+  workspace?: number
+}
+
+export type APIKeySecret = {
+  id?: string
+  key?: string
+}
+
 export type UpdateWSAPIKeyQuotaRequest = {
   id?: string
   workspace?: number
@@ -177,6 +187,12 @@ 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 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"})
+  }
+  static GetWSAPIKeySecret(req: GetWSAPIKeySecretRequest, initReq?: fm.InitReq): Promise<APIKeySecret> {
+    return fm.fetchReq<GetWSAPIKeySecretRequest, APIKeySecret>(`/apis/hydra.io/v1alpha1/workspaces/${req["workspace"]}/apikeys/${req["id"]}/secret?${fm.renderURLSearchParams(req, ["workspace", "id"])}`, {...initReq, method: "GET"})
+  }
   static DeleteWSAPIKey(req: DeleteWSAPIKeyRequest, initReq?: fm.InitReq): Promise<APIKey> {
     return fm.fetchReq<DeleteWSAPIKeyRequest, APIKey>(`/apis/hydra.io/v1alpha1/workspaces/${req["workspace"]}/apikeys/${req["id"]}`, {...initReq, method: "DELETE", body: JSON.stringify(req, fm.replacer)})
   }

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

@@ -9,6 +9,15 @@ import * as fm from "../../../fetch.pb"
 import * as GoogleProtobufEmpty from "../../../google/protobuf/empty.pb"
 import * as GoogleProtobufTimestamp from "../../../google/protobuf/timestamp.pb"
 
+type Absent<T, K extends keyof T> = { [k in Exclude<keyof T, K>]?: undefined };
+type OneOf<T> =
+  | { [k in keyof T]?: undefined }
+  | (
+    keyof T extends infer K ?
+      (K extends string & keyof T ? { [k in K]: T[K] } & Absent<T, K>
+        : never)
+    : never);
+
 export enum ModelSupportFeature {
   MODEL_SUPPORT_TYPE_UNSPECIFIED = "MODEL_SUPPORT_TYPE_UNSPECIFIED",
   TEXT_GENERATION = "TEXT_GENERATION",
@@ -148,6 +157,9 @@ export type DeployTemplate = {
   runtimeCommand?: string
   templateName?: string
   templateDescription?: string
+  workerCommand?: string
+  workerEnv?: EnvVar[]
+  distributed?: boolean
 }
 
 export type EnvVar = {
@@ -251,13 +263,17 @@ export type DeleteModelTemplateRequest = {
   modelId?: string
 }
 
-export type ListModelTemplatesRequest = {
+
+/* hydra modified */ export type BaseListModelTemplatesRequest = {
   workspace?: number
   modelId?: string
   inferenceRuntime?: string
   page?: HydraCommonCommon.Pagination
 }
 
+export type ListModelTemplatesRequest = BaseListModelTemplatesRequest
+  & OneOf<{ distributed: boolean }>
+
 export type ListModelTemplatesResponse = {
   items?: ModelTemplate[]
   page?: HydraCommonCommon.Pagination

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

@@ -65,6 +65,13 @@ export enum DeployDetectConditionType {
   MODEL_IMAGE_DOWNLOADED = "MODEL_IMAGE_DOWNLOADED",
 }
 
+export enum BuiltinEnvVarScope {
+  SCOPE_ALL = "SCOPE_ALL",
+  SCOPE_DISTRIBUTED = "SCOPE_DISTRIBUTED",
+  SCOPE_LEADER = "SCOPE_LEADER",
+  SCOPE_WORKER = "SCOPE_WORKER",
+}
+
 export enum ModelServingActionRequestAction {
   MODEL_SERVING_ACTION_UNSPECIFIED = "MODEL_SERVING_ACTION_UNSPECIFIED",
   START = "START",
@@ -157,6 +164,8 @@ export type RuntimeConfig = {
   imagePullSecretName?: string[]
   runtimeCommand?: string
   env?: EnvVar[]
+  workerCommand?: string
+  workerEnv?: EnvVar[]
 }
 
 export type CreateModelServingRequest = {
@@ -349,7 +358,9 @@ export type BuiltinEnvVar = {
 }
 
 export type ListBuiltinEnvVarsResponse = {
-  items?: BuiltinEnvVar[]
+  single?: BuiltinEnvVar[]
+  leader?: BuiltinEnvVar[]
+  worker?: BuiltinEnvVar[]
 }
 
 export class ModelServingManagement {

package/package.json

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