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

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-12",
   "description": "",
   "author": "",
   "license": "ISC"