@meetkai/mka1 0.49.17 → 0.49.18

package/src/funcs/servingDeploymentsRegister.ts

@@ -0,0 +1,198 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { SDKCore } from "../core.js";
+import { encodeSimple } from "../lib/encodings.js";
+import { matchStatusCode } from "../lib/http.js";
+import * as M from "../lib/matchers.js";
+import { compactMap } from "../lib/primitives.js";
+import { safeParse } from "../lib/schemas.js";
+import { RequestOptions } from "../lib/sdks.js";
+import { extractSecurity, resolveGlobalSecurity } from "../lib/security.js";
+import { pathToFunc } from "../lib/url.js";
+import * as components from "../models/components/index.js";
+import {
+  ConnectionError,
+  InvalidRequestError,
+  RequestAbortedError,
+  RequestTimeoutError,
+  UnexpectedClientError,
+} from "../models/errors/httpclienterrors.js";
+import * as errors from "../models/errors/index.js";
+import { ResponseValidationError } from "../models/errors/responsevalidationerror.js";
+import { SDKError } from "../models/errors/sdkerror.js";
+import { SDKValidationError } from "../models/errors/sdkvalidationerror.js";
+import * as operations from "../models/operations/index.js";
+import { APICall, APIPromise } from "../types/async.js";
+import { Result } from "../types/fp.js";
+
+/**
+ * Register the model with the LLM gateway
+ *
+ * @remarks
+ * Make the deployment's model available for inference: registers it in the LLM gateway's model registry under the deployment's name. Deployment must be ready. Registration is deliberately decoupled from deployment so pricing, budgets, and rate limits can be configured before the model becomes callable.
+ */
+export function servingDeploymentsRegister(
+  client: SDKCore,
+  request: operations.RegisterDeploymentRequest,
+  options?: RequestOptions,
+): APIPromise<
+  Result<
+    components.Deployment,
+    | errors.ServingError
+    | errors.ServingHTTPValidationError
+    | SDKError
+    | ResponseValidationError
+    | ConnectionError
+    | RequestAbortedError
+    | RequestTimeoutError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
+  >
+> {
+  return new APIPromise($do(
+    client,
+    request,
+    options,
+  ));
+}
+
+async function $do(
+  client: SDKCore,
+  request: operations.RegisterDeploymentRequest,
+  options?: RequestOptions,
+): Promise<
+  [
+    Result<
+      components.Deployment,
+      | errors.ServingError
+      | errors.ServingHTTPValidationError
+      | SDKError
+      | ResponseValidationError
+      | ConnectionError
+      | RequestAbortedError
+      | RequestTimeoutError
+      | InvalidRequestError
+      | UnexpectedClientError
+      | SDKValidationError
+    >,
+    APICall,
+  ]
+> {
+  const parsed = safeParse(
+    request,
+    (value) => operations.RegisterDeploymentRequest$outboundSchema.parse(value),
+    "Input validation failed",
+  );
+  if (!parsed.ok) {
+    return [parsed, { status: "invalid" }];
+  }
+  const payload = parsed.value;
+  const body = null;
+
+  const pathParams = {
+    deployment_id: encodeSimple("deployment_id", payload.deployment_id, {
+      explode: false,
+      charEncoding: "percent",
+    }),
+  };
+  const path = pathToFunc(
+    "/api/v1/serving/deployments/{deployment_id}/registration",
+  )(pathParams);
+
+  const headers = new Headers(compactMap({
+    Accept: "application/json",
+    "X-On-Behalf-Of": encodeSimple(
+      "X-On-Behalf-Of",
+      payload["X-On-Behalf-Of"],
+      { explode: false, charEncoding: "none" },
+    ),
+  }));
+
+  const secConfig = await extractSecurity(client._options.bearerAuth);
+  const securityInput = secConfig == null ? {} : { bearerAuth: secConfig };
+  const requestSecurity = resolveGlobalSecurity(securityInput);
+
+  const context = {
+    options: client._options,
+    baseURL: options?.serverURL ?? client._baseURL ?? "",
+    operationID: "registerDeployment",
+    oAuth2Scopes: null,
+
+    resolvedSecurity: requestSecurity,
+
+    securitySource: client._options.bearerAuth,
+    retryConfig: options?.retries
+      || client._options.retryConfig
+      || {
+        strategy: "backoff",
+        backoff: {
+          initialInterval: 500,
+          maxInterval: 5000,
+          exponent: 1.5,
+          maxElapsedTime: 15000,
+        },
+        retryConnectionErrors: true,
+      }
+      || { strategy: "none" },
+    retryCodes: options?.retryCodes || ["429", "500", "502", "503", "504"],
+  };
+
+  const requestRes = client._createRequest(context, {
+    security: requestSecurity,
+    method: "POST",
+    baseURL: options?.serverURL,
+    path: path,
+    headers: headers,
+    body: body,
+    userAgent: client._options.userAgent,
+    timeoutMs: options?.timeoutMs || client._options.timeoutMs || -1,
+  }, options);
+  if (!requestRes.ok) {
+    return [requestRes, { status: "invalid" }];
+  }
+  const req = requestRes.value;
+
+  const doResult = await client._do(req, {
+    context,
+    isErrorStatusCode: (statusCode: number) =>
+      matchStatusCode({ status: statusCode } as Response, ["4XX", "5XX"]),
+    retryConfig: context.retryConfig,
+    retryCodes: context.retryCodes,
+  });
+  if (!doResult.ok) {
+    return [doResult, { status: "request-error", request: req }];
+  }
+  const response = doResult.value;
+
+  const responseFields = {
+    HttpMeta: { Response: response, Request: req },
+  };
+
+  const [result] = await M.match<
+    components.Deployment,
+    | errors.ServingError
+    | errors.ServingHTTPValidationError
+    | SDKError
+    | ResponseValidationError
+    | ConnectionError
+    | RequestAbortedError
+    | RequestTimeoutError
+    | InvalidRequestError
+    | UnexpectedClientError
+    | SDKValidationError
+  >(
+    M.json(200, components.Deployment$inboundSchema),
+    M.jsonErr([403, 404, 409], errors.ServingError$inboundSchema),
+    M.jsonErr(422, errors.ServingHTTPValidationError$inboundSchema),
+    M.fail([401, "4XX"]),
+    M.fail("5XX"),
+  )(response, req, { extraFields: responseFields });
+  if (!result.ok) {
+    return [result, { status: "complete", request: req, response }];
+  }
+
+  return [result, { status: "complete", request: req, response }];
+}

package/src/funcs/servingFinetuneJobsGetEvents.ts

@@ -4,7 +4,7 @@
 
 import * as z from "zod/v3";
 import { SDKCore } from "../core.js";
-import { encodeSimple } from "../lib/encodings.js";
+import { encodeFormQuery, encodeSimple } from "../lib/encodings.js";
 import { matchStatusCode } from "../lib/http.js";
 import * as M from "../lib/matchers.js";
 import { compactMap } from "../lib/primitives.js";
@@ -103,6 +103,10 @@ async function $do(
     pathParams,
   );
 
+  const query = encodeFormQuery({
+    "include_lineage": payload.include_lineage,
+  });
+
   const headers = new Headers(compactMap({
     Accept: "application/json",
     "X-On-Behalf-Of": encodeSimple(
@@ -147,6 +151,7 @@ async function $do(
     baseURL: options?.serverURL,
     path: path,
     headers: headers,
+    query: query,
     body: body,
     userAgent: client._options.userAgent,
     timeoutMs: options?.timeoutMs || client._options.timeoutMs || -1,

package/src/lib/config.ts

@@ -65,7 +65,7 @@ export function serverURLFromOptions(options: SDKOptions): URL | null {
 export const SDK_METADATA = {
   language: "typescript",
   openapiDocVersion: "1.1.0",
-  sdkVersion: "0.49.17",
+  sdkVersion: "0.49.18",
   genVersion: "2.918.4",
-  userAgent: "speakeasy-sdk/typescript 0.49.17 2.918.4 1.1.0 @meetkai/mka1",
+  userAgent: "speakeasy-sdk/typescript 0.49.18 2.918.4 1.1.0 @meetkai/mka1",
 } as const;

package/src/mcp-server/mcp-server.ts

@@ -19,7 +19,7 @@ const routes = buildRouteMap({
 export const app = buildApplication(routes, {
   name: "mcp",
   versionInfo: {
-    currentVersion: "0.49.17",
+    currentVersion: "0.49.18",
   },
 });
 

package/src/mcp-server/server.ts

@@ -301,12 +301,14 @@ import { tool$sandboxUploadFile } from "./tools/sandboxUploadFile.js";
 import { tool$servingAcceleratorsList } from "./tools/servingAcceleratorsList.js";
 import { tool$servingDeploymentsCreate } from "./tools/servingDeploymentsCreate.js";
 import { tool$servingDeploymentsDelete } from "./tools/servingDeploymentsDelete.js";
+import { tool$servingDeploymentsDeregister } from "./tools/servingDeploymentsDeregister.js";
 import { tool$servingDeploymentsGet } from "./tools/servingDeploymentsGet.js";
 import { tool$servingDeploymentsGetLogs } from "./tools/servingDeploymentsGetLogs.js";
 import { tool$servingDeploymentsGetMetrics } from "./tools/servingDeploymentsGetMetrics.js";
 import { tool$servingDeploymentsList } from "./tools/servingDeploymentsList.js";
 import { tool$servingDeploymentsListReplicas } from "./tools/servingDeploymentsListReplicas.js";
 import { tool$servingDeploymentsListRevisions } from "./tools/servingDeploymentsListRevisions.js";
+import { tool$servingDeploymentsRegister } from "./tools/servingDeploymentsRegister.js";
 import { tool$servingDeploymentsRollback } from "./tools/servingDeploymentsRollback.js";
 import { tool$servingDeploymentsUpdate } from "./tools/servingDeploymentsUpdate.js";
 import { tool$servingDeploymentsUpdateScaling } from "./tools/servingDeploymentsUpdateScaling.js";
@@ -345,7 +347,7 @@ export function createMCPServer(deps: {
 }) {
   const server = new McpServer({
     name: "SDK",
-    version: "0.49.17",
+    version: "0.49.18",
   });
 
   const client = new SDKCore({
@@ -662,40 +664,42 @@ export function createMCPServer(deps: {
   tool(tool$authClusterInviteInfo);
   tool(tool$authClusterAcceptInvite);
   tool(tool$authApiKeyGetJwtFromKey);
-  tool(tool$servingDeploymentsCreate);
+  tool(tool$servingAcceleratorsList);
   tool(tool$servingDeploymentsList);
+  tool(tool$servingDeploymentsCreate);
+  tool(tool$servingDeploymentsDelete);
   tool(tool$servingDeploymentsGet);
   tool(tool$servingDeploymentsUpdate);
-  tool(tool$servingDeploymentsDelete);
-  tool(tool$servingDeploymentsUpdateScaling);
-  tool(tool$servingDeploymentsListRevisions);
-  tool(tool$servingDeploymentsRollback);
   tool(tool$servingDeploymentsGetLogs);
   tool(tool$servingDeploymentsGetMetrics);
+  tool(tool$servingDeploymentsDeregister);
+  tool(tool$servingDeploymentsRegister);
   tool(tool$servingDeploymentsListReplicas);
-  tool(tool$servingFinetuneJobsCreate);
+  tool(tool$servingDeploymentsListRevisions);
+  tool(tool$servingDeploymentsRollback);
+  tool(tool$servingDeploymentsUpdateScaling);
   tool(tool$servingFinetuneJobsList);
-  tool(tool$servingFinetuneJobsGet);
+  tool(tool$servingFinetuneJobsCreate);
   tool(tool$servingFinetuneJobsCancel);
-  tool(tool$servingFinetuneJobsGetLogs);
-  tool(tool$servingFinetuneJobsGetEvents);
+  tool(tool$servingFinetuneJobsGet);
   tool(tool$servingFinetuneJobsListCheckpoints);
-  tool(tool$servingImagesBuild);
+  tool(tool$servingFinetuneJobsGetEvents);
+  tool(tool$servingFinetuneJobsGetLogs);
   tool(tool$servingImagesList);
-  tool(tool$servingImagesGet);
+  tool(tool$servingImagesBuild);
   tool(tool$servingImagesDelete);
-  tool(tool$servingModelsRegister);
+  tool(tool$servingImagesGet);
   tool(tool$servingModelsList);
-  tool(tool$servingModelsGet);
+  tool(tool$servingModelsRegister);
   tool(tool$servingModelsDelete);
-  tool(tool$servingVolumesCreate);
-  tool(tool$servingVolumesList);
-  tool(tool$servingVolumesGet);
-  tool(tool$servingVolumesDelete);
-  tool(tool$servingSecretsCreate);
+  tool(tool$servingModelsGet);
   tool(tool$servingSecretsList);
+  tool(tool$servingSecretsCreate);
   tool(tool$servingSecretsDelete);
-  tool(tool$servingAcceleratorsList);
+  tool(tool$servingVolumesList);
+  tool(tool$servingVolumesCreate);
+  tool(tool$servingVolumesDelete);
+  tool(tool$servingVolumesGet);
 
   return server;
 }

package/src/mcp-server/tools/servingDeploymentsDeregister.ts

@@ -0,0 +1,35 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { servingDeploymentsDeregister } from "../../funcs/servingDeploymentsDeregister.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.DeregisterDeploymentRequest$inboundSchema,
+};
+
+export const tool$servingDeploymentsDeregister: ToolDefinition<typeof args> = {
+  name: "serving-deployments-deregister",
+  description: `Deregister the model from the LLM gateway
+
+Withdraw the deployment's model from the LLM gateway registry: inference stops resolving, compute keeps running (re-register any time).`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await servingDeploymentsDeregister(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.signal } },
+    ).$inspect();
+
+    if (!result.ok) {
+      return {
+        content: [{ type: "text", text: result.error.message }],
+        isError: true,
+      };
+    }
+
+    return formatResult(void 0, apiCall);
+  },
+};

package/src/mcp-server/tools/servingDeploymentsRegister.ts

@@ -0,0 +1,37 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { servingDeploymentsRegister } from "../../funcs/servingDeploymentsRegister.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.RegisterDeploymentRequest$inboundSchema,
+};
+
+export const tool$servingDeploymentsRegister: ToolDefinition<typeof args> = {
+  name: "serving-deployments-register",
+  description: `Register the model with the LLM gateway
+
+Make the deployment's model available for inference: registers it in the LLM gateway's model registry under the deployment's name. Deployment must be ready. Registration is deliberately decoupled from deployment so pricing, budgets, and rate limits can be configured before the model becomes callable.`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await servingDeploymentsRegister(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.signal } },
+    ).$inspect();
+
+    if (!result.ok) {
+      return {
+        content: [{ type: "text", text: result.error.message }],
+        isError: true,
+      };
+    }
+
+    const value = result.value;
+
+    return formatResult(value, apiCall);
+  },
+};

package/src/models/components/acceleratorspec.ts

@@ -15,7 +15,6 @@ import { SDKValidationError } from "../errors/sdkvalidationerror.js";
  * Request an accelerator by type and count, with optional fallback types.
  */
 export type AcceleratorSpec = {
-  type: string;
   /**
    * Accelerators per replica
    */
@@ -24,6 +23,7 @@ export type AcceleratorSpec = {
    * Ordered fallback types if the primary is unavailable
    */
   fallback?: Array<string> | undefined;
+  type: string;
 };
 
 /** @internal */
@@ -32,15 +32,15 @@ export const AcceleratorSpec$inboundSchema: z.ZodType<
   z.ZodTypeDef,
   unknown
 > = z.object({
-  type: z.string(),
   count: z.number().int().default(1),
   fallback: z.array(z.string()).optional(),
+  type: z.string(),
 });
 /** @internal */
 export type AcceleratorSpec$Outbound = {
-  type: string;
   count: number;
   fallback?: Array<string> | undefined;
+  type: string;
 };
 
 /** @internal */
@@ -49,9 +49,9 @@ export const AcceleratorSpec$outboundSchema: z.ZodType<
   z.ZodTypeDef,
   AcceleratorSpec
 > = z.object({
-  type: z.string(),
   count: z.number().int().default(1),
   fallback: z.array(z.string()).optional(),
+  type: z.string(),
 });
 
 export function acceleratorSpecToJSON(

package/src/models/components/acceleratortype.ts

@@ -21,14 +21,14 @@ export type Kind = ClosedEnum<typeof Kind>;
  * An offered accelerator type in the catalog.
  */
 export type AcceleratorType = {
-  name: string;
-  vendor: string;
   kind?: Kind | undefined;
-  memoryGb: number;
   /**
    * Max accelerators of this type per replica
    */
   maxCount: number;
+  memoryGb: number;
+  name: string;
+  vendor: string;
 };
 
 /** @internal */
@@ -45,24 +45,24 @@ export const AcceleratorType$inboundSchema: z.ZodType<
   z.ZodTypeDef,
   unknown
 > = z.object({
-  name: z.string(),
-  vendor: z.string(),
   kind: Kind$inboundSchema.default("gpu"),
-  memory_gb: z.number().int(),
   max_count: z.number().int(),
+  memory_gb: z.number().int(),
+  name: z.string(),
+  vendor: z.string(),
 }).transform((v) => {
   return remap$(v, {
-    "memory_gb": "memoryGb",
     "max_count": "maxCount",
+    "memory_gb": "memoryGb",
   });
 });
 /** @internal */
 export type AcceleratorType$Outbound = {
-  name: string;
-  vendor: string;
   kind: string;
-  memory_gb: number;
   max_count: number;
+  memory_gb: number;
+  name: string;
+  vendor: string;
 };
 
 /** @internal */
@@ -71,15 +71,15 @@ export const AcceleratorType$outboundSchema: z.ZodType<
   z.ZodTypeDef,
   AcceleratorType
 > = z.object({
-  name: z.string(),
-  vendor: z.string(),
   kind: Kind$outboundSchema.default("gpu"),
-  memoryGb: z.number().int(),
   maxCount: z.number().int(),
+  memoryGb: z.number().int(),
+  name: z.string(),
+  vendor: z.string(),
 }).transform((v) => {
   return remap$(v, {
-    memoryGb: "memory_gb",
     maxCount: "max_count",
+    memoryGb: "memory_gb",
   });
 });
 

package/src/models/components/checkpoint.ts

@@ -12,10 +12,11 @@ import { SDKValidationError } from "../errors/sdkvalidationerror.js";
  * A saved checkpoint/artifact produced by a job.
  */
 export type Checkpoint = {
+  createdAt: Date;
+  id: string;
   path: string;
-  step?: number | null | undefined;
   sizeBytes?: number | null | undefined;
-  createdAt: Date;
+  step?: number | null | undefined;
 };
 
 /** @internal */
@@ -24,22 +25,24 @@ export const Checkpoint$inboundSchema: z.ZodType<
   z.ZodTypeDef,
   unknown
 > = z.object({
+  created_at: z.string().datetime({ offset: true }).transform(v => new Date(v)),
+  id: z.string(),
   path: z.string(),
-  step: z.nullable(z.number().int()).optional(),
   size_bytes: z.nullable(z.number().int()).optional(),
-  created_at: z.string().datetime({ offset: true }).transform(v => new Date(v)),
+  step: z.nullable(z.number().int()).optional(),
 }).transform((v) => {
   return remap$(v, {
-    "size_bytes": "sizeBytes",
     "created_at": "createdAt",
+    "size_bytes": "sizeBytes",
   });
 });
 /** @internal */
 export type Checkpoint$Outbound = {
+  created_at: string;
+  id: string;
   path: string;
-  step?: number | null | undefined;
   size_bytes?: number | null | undefined;
-  created_at: string;
+  step?: number | null | undefined;
 };
 
 /** @internal */
@@ -48,14 +51,15 @@ export const Checkpoint$outboundSchema: z.ZodType<
   z.ZodTypeDef,
   Checkpoint
 > = z.object({
+  createdAt: z.date().transform(v => v.toISOString()),
+  id: z.string(),
   path: z.string(),
-  step: z.nullable(z.number().int()).optional(),
   sizeBytes: z.nullable(z.number().int()).optional(),
-  createdAt: z.date().transform(v => v.toISOString()),
+  step: z.nullable(z.number().int()).optional(),
 }).transform((v) => {
   return remap$(v, {
-    sizeBytes: "size_bytes",
     createdAt: "created_at",
+    sizeBytes: "size_bytes",
   });
 });