@meetkai/mka1 0.49.16 → 0.49.18

package/src/models/components/servingvalidationerror.ts

@@ -7,18 +7,44 @@ import { safeParse } from "../../lib/schemas.js";
 import { Result as SafeParseResult } from "../../types/fp.js";
 import { SDKValidationError } from "../errors/sdkvalidationerror.js";
 
-export type ServingValidationErrorLoc = string | number;
-
 export type Context = {};
 
+export type ServingValidationErrorLoc = string | number;
+
 export type ServingValidationError = {
+  ctx?: Context | undefined;
+  input?: any | undefined;
   loc: Array<string | number>;
   msg: string;
   type: string;
-  input?: any | undefined;
-  ctx?: Context | undefined;
 };
 
+/** @internal */
+export const Context$inboundSchema: z.ZodType<Context, z.ZodTypeDef, unknown> =
+  z.object({});
+/** @internal */
+export type Context$Outbound = {};
+
+/** @internal */
+export const Context$outboundSchema: z.ZodType<
+  Context$Outbound,
+  z.ZodTypeDef,
+  Context
+> = z.object({});
+
+export function contextToJSON(context: Context): string {
+  return JSON.stringify(Context$outboundSchema.parse(context));
+}
+export function contextFromJSON(
+  jsonString: string,
+): SafeParseResult<Context, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => Context$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'Context' from JSON`,
+  );
+}
+
 /** @internal */
 export const ServingValidationErrorLoc$inboundSchema: z.ZodType<
   ServingValidationErrorLoc,
@@ -52,51 +78,25 @@ export function servingValidationErrorLocFromJSON(
   );
 }
 
-/** @internal */
-export const Context$inboundSchema: z.ZodType<Context, z.ZodTypeDef, unknown> =
-  z.object({});
-/** @internal */
-export type Context$Outbound = {};
-
-/** @internal */
-export const Context$outboundSchema: z.ZodType<
-  Context$Outbound,
-  z.ZodTypeDef,
-  Context
-> = z.object({});
-
-export function contextToJSON(context: Context): string {
-  return JSON.stringify(Context$outboundSchema.parse(context));
-}
-export function contextFromJSON(
-  jsonString: string,
-): SafeParseResult<Context, SDKValidationError> {
-  return safeParse(
-    jsonString,
-    (x) => Context$inboundSchema.parse(JSON.parse(x)),
-    `Failed to parse 'Context' from JSON`,
-  );
-}
-
 /** @internal */
 export const ServingValidationError$inboundSchema: z.ZodType<
   ServingValidationError,
   z.ZodTypeDef,
   unknown
 > = z.object({
+  ctx: z.lazy(() => Context$inboundSchema).optional(),
+  input: z.any().optional(),
   loc: z.array(z.union([z.string(), z.number().int()])),
   msg: z.string(),
   type: z.string(),
-  input: z.any().optional(),
-  ctx: z.lazy(() => Context$inboundSchema).optional(),
 });
 /** @internal */
 export type ServingValidationError$Outbound = {
+  ctx?: Context$Outbound | undefined;
+  input?: any | undefined;
   loc: Array<string | number>;
   msg: string;
   type: string;
-  input?: any | undefined;
-  ctx?: Context$Outbound | undefined;
 };
 
 /** @internal */
@@ -105,11 +105,11 @@ export const ServingValidationError$outboundSchema: z.ZodType<
   z.ZodTypeDef,
   ServingValidationError
 > = z.object({
+  ctx: z.lazy(() => Context$outboundSchema).optional(),
+  input: z.any().optional(),
   loc: z.array(z.union([z.string(), z.number().int()])),
   msg: z.string(),
   type: z.string(),
-  input: z.any().optional(),
-  ctx: z.lazy(() => Context$outboundSchema).optional(),
 });
 
 export function servingValidationErrorToJSON(

package/src/models/components/volume.ts

@@ -12,33 +12,33 @@ import { SDKValidationError } from "../errors/sdkvalidationerror.js";
  * A persistent volume.
  */
 export type Volume = {
+  createdAt: Date;
   id: string;
   name: string;
   sizeGb?: number | null | undefined;
-  createdAt: Date;
 };
 
 /** @internal */
 export const Volume$inboundSchema: z.ZodType<Volume, z.ZodTypeDef, unknown> = z
   .object({
-    id: z.string(),
-    name: z.string(),
-    size_gb: z.nullable(z.number().int()).optional(),
     created_at: z.string().datetime({ offset: true }).transform(v =>
       new Date(v)
     ),
+    id: z.string(),
+    name: z.string(),
+    size_gb: z.nullable(z.number().int()).optional(),
   }).transform((v) => {
     return remap$(v, {
-      "size_gb": "sizeGb",
       "created_at": "createdAt",
+      "size_gb": "sizeGb",
     });
   });
 /** @internal */
 export type Volume$Outbound = {
+  created_at: string;
   id: string;
   name: string;
   size_gb?: number | null | undefined;
-  created_at: string;
 };
 
 /** @internal */
@@ -47,14 +47,14 @@ export const Volume$outboundSchema: z.ZodType<
   z.ZodTypeDef,
   Volume
 > = z.object({
+  createdAt: z.date().transform(v => v.toISOString()),
   id: z.string(),
   name: z.string(),
   sizeGb: z.nullable(z.number().int()).optional(),
-  createdAt: z.date().transform(v => v.toISOString()),
 }).transform((v) => {
   return remap$(v, {
-    sizeGb: "size_gb",
     createdAt: "created_at",
+    sizeGb: "size_gb",
   });
 });
 

package/src/models/operations/deregisterdeployment.ts

@@ -0,0 +1,71 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod/v3";
+import { remap as remap$ } from "../../lib/primitives.js";
+import { safeParse } from "../../lib/schemas.js";
+import { Result as SafeParseResult } from "../../types/fp.js";
+import { SDKValidationError } from "../errors/sdkvalidationerror.js";
+
+export type DeregisterDeploymentRequest = {
+  deploymentId: string;
+  /**
+   * Optional external end-user identifier forwarded by the API gateway.
+   */
+  xOnBehalfOf?: string | undefined;
+};
+
+/** @internal */
+export const DeregisterDeploymentRequest$inboundSchema: z.ZodType<
+  DeregisterDeploymentRequest,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  deployment_id: z.string(),
+  "X-On-Behalf-Of": z.string().optional(),
+}).transform((v) => {
+  return remap$(v, {
+    "deployment_id": "deploymentId",
+    "X-On-Behalf-Of": "xOnBehalfOf",
+  });
+});
+/** @internal */
+export type DeregisterDeploymentRequest$Outbound = {
+  deployment_id: string;
+  "X-On-Behalf-Of"?: string | undefined;
+};
+
+/** @internal */
+export const DeregisterDeploymentRequest$outboundSchema: z.ZodType<
+  DeregisterDeploymentRequest$Outbound,
+  z.ZodTypeDef,
+  DeregisterDeploymentRequest
+> = z.object({
+  deploymentId: z.string(),
+  xOnBehalfOf: z.string().optional(),
+}).transform((v) => {
+  return remap$(v, {
+    deploymentId: "deployment_id",
+    xOnBehalfOf: "X-On-Behalf-Of",
+  });
+});
+
+export function deregisterDeploymentRequestToJSON(
+  deregisterDeploymentRequest: DeregisterDeploymentRequest,
+): string {
+  return JSON.stringify(
+    DeregisterDeploymentRequest$outboundSchema.parse(
+      deregisterDeploymentRequest,
+    ),
+  );
+}
+export function deregisterDeploymentRequestFromJSON(
+  jsonString: string,
+): SafeParseResult<DeregisterDeploymentRequest, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => DeregisterDeploymentRequest$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'DeregisterDeploymentRequest' from JSON`,
+  );
+}

package/src/models/operations/getfinetuneevents.ts

@@ -10,6 +10,7 @@ import { SDKValidationError } from "../errors/sdkvalidationerror.js";
 
 export type GetFinetuneEventsRequest = {
   jobId: string;
+  includeLineage?: boolean | undefined;
   /**
    * Optional external end-user identifier forwarded by the API gateway.
    */
@@ -23,16 +24,19 @@ export const GetFinetuneEventsRequest$inboundSchema: z.ZodType<
   unknown
 > = z.object({
   job_id: z.string(),
+  include_lineage: z.boolean().default(false),
   "X-On-Behalf-Of": z.string().optional(),
 }).transform((v) => {
   return remap$(v, {
     "job_id": "jobId",
+    "include_lineage": "includeLineage",
     "X-On-Behalf-Of": "xOnBehalfOf",
   });
 });
 /** @internal */
 export type GetFinetuneEventsRequest$Outbound = {
   job_id: string;
+  include_lineage: boolean;
   "X-On-Behalf-Of"?: string | undefined;
 };
 
@@ -43,10 +47,12 @@ export const GetFinetuneEventsRequest$outboundSchema: z.ZodType<
   GetFinetuneEventsRequest
 > = z.object({
   jobId: z.string(),
+  includeLineage: z.boolean().default(false),
   xOnBehalfOf: z.string().optional(),
 }).transform((v) => {
   return remap$(v, {
     jobId: "job_id",
+    includeLineage: "include_lineage",
     xOnBehalfOf: "X-On-Behalf-Of",
   });
 });

package/src/models/operations/index.ts

@@ -102,6 +102,7 @@ export * from "./deletetexttospeechhistory.js";
 export * from "./deletevectorstore.js";
 export * from "./deletevectorstorefile.js";
 export * from "./deletevolume.js";
+export * from "./deregisterdeployment.js";
 export * from "./downloadworkspacearchive.js";
 export * from "./downloadworkspacefile.js";
 export * from "./embed.js";
@@ -254,6 +255,7 @@ export * from "./proxybrowserportrequest.js";
 export * from "./putorgautomodel.js";
 export * from "./regeneratescopedapikey.js";
 export * from "./regenerateserviceaccountapikey.js";
+export * from "./registerdeployment.js";
 export * from "./registermodel.js";
 export * from "./reissueownerkey.js";
 export * from "./removeteammember.js";

package/src/models/operations/registerdeployment.ts

@@ -0,0 +1,69 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod/v3";
+import { remap as remap$ } from "../../lib/primitives.js";
+import { safeParse } from "../../lib/schemas.js";
+import { Result as SafeParseResult } from "../../types/fp.js";
+import { SDKValidationError } from "../errors/sdkvalidationerror.js";
+
+export type RegisterDeploymentRequest = {
+  deploymentId: string;
+  /**
+   * Optional external end-user identifier forwarded by the API gateway.
+   */
+  xOnBehalfOf?: string | undefined;
+};
+
+/** @internal */
+export const RegisterDeploymentRequest$inboundSchema: z.ZodType<
+  RegisterDeploymentRequest,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  deployment_id: z.string(),
+  "X-On-Behalf-Of": z.string().optional(),
+}).transform((v) => {
+  return remap$(v, {
+    "deployment_id": "deploymentId",
+    "X-On-Behalf-Of": "xOnBehalfOf",
+  });
+});
+/** @internal */
+export type RegisterDeploymentRequest$Outbound = {
+  deployment_id: string;
+  "X-On-Behalf-Of"?: string | undefined;
+};
+
+/** @internal */
+export const RegisterDeploymentRequest$outboundSchema: z.ZodType<
+  RegisterDeploymentRequest$Outbound,
+  z.ZodTypeDef,
+  RegisterDeploymentRequest
+> = z.object({
+  deploymentId: z.string(),
+  xOnBehalfOf: z.string().optional(),
+}).transform((v) => {
+  return remap$(v, {
+    deploymentId: "deployment_id",
+    xOnBehalfOf: "X-On-Behalf-Of",
+  });
+});
+
+export function registerDeploymentRequestToJSON(
+  registerDeploymentRequest: RegisterDeploymentRequest,
+): string {
+  return JSON.stringify(
+    RegisterDeploymentRequest$outboundSchema.parse(registerDeploymentRequest),
+  );
+}
+export function registerDeploymentRequestFromJSON(
+  jsonString: string,
+): SafeParseResult<RegisterDeploymentRequest, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => RegisterDeploymentRequest$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'RegisterDeploymentRequest' from JSON`,
+  );
+}

package/src/models/operations/updateorgratelimit.ts

@@ -11,7 +11,7 @@ import { SDKValidationError } from "../errors/sdkvalidationerror.js";
 export type UpdateOrgRateLimitRequestBody = {
   rateLimitEnabled?: boolean | undefined;
   rateLimitTimeWindow?: number | undefined;
-  rateLimitMax?: number | undefined;
+  rateLimitMax?: number | null | undefined;
 };
 
 export type UpdateOrgRateLimitRequest = {
@@ -41,13 +41,13 @@ export const UpdateOrgRateLimitRequestBody$inboundSchema: z.ZodType<
 > = z.object({
   rateLimitEnabled: z.boolean().optional(),
   rateLimitTimeWindow: z.number().optional(),
-  rateLimitMax: z.number().int().optional(),
+  rateLimitMax: z.nullable(z.number().int()).optional(),
 });
 /** @internal */
 export type UpdateOrgRateLimitRequestBody$Outbound = {
   rateLimitEnabled?: boolean | undefined;
   rateLimitTimeWindow?: number | undefined;
-  rateLimitMax?: number | undefined;
+  rateLimitMax?: number | null | undefined;
 };
 
 /** @internal */
@@ -58,7 +58,7 @@ export const UpdateOrgRateLimitRequestBody$outboundSchema: z.ZodType<
 > = z.object({
   rateLimitEnabled: z.boolean().optional(),
   rateLimitTimeWindow: z.number().optional(),
-  rateLimitMax: z.number().int().optional(),
+  rateLimitMax: z.nullable(z.number().int()).optional(),
 });
 
 export function updateOrgRateLimitRequestBodyToJSON(

package/src/react-query/index.ts

@@ -294,12 +294,14 @@ export * from "./sandboxUploadFile.js";
 export * from "./servingAcceleratorsList.js";
 export * from "./servingDeploymentsCreate.js";
 export * from "./servingDeploymentsDelete.js";
+export * from "./servingDeploymentsDeregister.js";
 export * from "./servingDeploymentsGet.js";
 export * from "./servingDeploymentsGetLogs.js";
 export * from "./servingDeploymentsGetMetrics.js";
 export * from "./servingDeploymentsList.js";
 export * from "./servingDeploymentsListReplicas.js";
 export * from "./servingDeploymentsListRevisions.js";
+export * from "./servingDeploymentsRegister.js";
 export * from "./servingDeploymentsRollback.js";
 export * from "./servingDeploymentsUpdate.js";
 export * from "./servingDeploymentsUpdateScaling.js";

package/src/react-query/servingDeploymentsDeregister.ts

@@ -0,0 +1,111 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  MutationKey,
+  useMutation,
+  UseMutationResult,
+} from "@tanstack/react-query";
+import { SDKCore } from "../core.js";
+import { servingDeploymentsDeregister } from "../funcs/servingDeploymentsDeregister.js";
+import { combineSignals } from "../lib/primitives.js";
+import { RequestOptions } from "../lib/sdks.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 { unwrapAsync } from "../types/fp.js";
+import { useSDKContext } from "./_context.js";
+import { MutationHookOptions } from "./_types.js";
+
+export type ServingDeploymentsDeregisterMutationVariables = {
+  request: operations.DeregisterDeploymentRequest;
+  options?: RequestOptions;
+};
+
+export type ServingDeploymentsDeregisterMutationData = void;
+
+export type ServingDeploymentsDeregisterMutationError =
+  | errors.ServingError
+  | errors.ServingHTTPValidationError
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * Deregister the model from the LLM gateway
+ *
+ * @remarks
+ * Withdraw the deployment's model from the LLM gateway registry: inference stops resolving, compute keeps running (re-register any time).
+ */
+export function useServingDeploymentsDeregisterMutation(
+  options?: MutationHookOptions<
+    ServingDeploymentsDeregisterMutationData,
+    ServingDeploymentsDeregisterMutationError,
+    ServingDeploymentsDeregisterMutationVariables
+  >,
+): UseMutationResult<
+  ServingDeploymentsDeregisterMutationData,
+  ServingDeploymentsDeregisterMutationError,
+  ServingDeploymentsDeregisterMutationVariables
+> {
+  const client = useSDKContext();
+  return useMutation({
+    ...buildServingDeploymentsDeregisterMutation(client, options),
+    ...options,
+  });
+}
+
+export function mutationKeyServingDeploymentsDeregister(): MutationKey {
+  return ["@meetkai/mka1", "deployments", "deregister"];
+}
+
+export function buildServingDeploymentsDeregisterMutation(
+  client$: SDKCore,
+  hookOptions?: RequestOptions,
+): {
+  mutationKey: MutationKey;
+  mutationFn: (
+    variables: ServingDeploymentsDeregisterMutationVariables,
+  ) => Promise<ServingDeploymentsDeregisterMutationData>;
+} {
+  return {
+    mutationKey: mutationKeyServingDeploymentsDeregister(),
+    mutationFn: function servingDeploymentsDeregisterMutationFn({
+      request,
+      options,
+    }): Promise<ServingDeploymentsDeregisterMutationData> {
+      const mergedOptions = {
+        ...hookOptions,
+        ...options,
+        fetchOptions: {
+          ...hookOptions?.fetchOptions,
+          ...options?.fetchOptions,
+          signal: combineSignals(
+            hookOptions?.fetchOptions?.signal,
+            options?.fetchOptions?.signal,
+          ),
+        },
+      };
+      return unwrapAsync(servingDeploymentsDeregister(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}

package/src/react-query/servingDeploymentsRegister.ts

@@ -0,0 +1,112 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  MutationKey,
+  useMutation,
+  UseMutationResult,
+} from "@tanstack/react-query";
+import { SDKCore } from "../core.js";
+import { servingDeploymentsRegister } from "../funcs/servingDeploymentsRegister.js";
+import { combineSignals } from "../lib/primitives.js";
+import { RequestOptions } from "../lib/sdks.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 { unwrapAsync } from "../types/fp.js";
+import { useSDKContext } from "./_context.js";
+import { MutationHookOptions } from "./_types.js";
+
+export type ServingDeploymentsRegisterMutationVariables = {
+  request: operations.RegisterDeploymentRequest;
+  options?: RequestOptions;
+};
+
+export type ServingDeploymentsRegisterMutationData = components.Deployment;
+
+export type ServingDeploymentsRegisterMutationError =
+  | errors.ServingError
+  | errors.ServingHTTPValidationError
+  | SDKError
+  | ResponseValidationError
+  | ConnectionError
+  | RequestAbortedError
+  | RequestTimeoutError
+  | InvalidRequestError
+  | UnexpectedClientError
+  | SDKValidationError;
+
+/**
+ * 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 useServingDeploymentsRegisterMutation(
+  options?: MutationHookOptions<
+    ServingDeploymentsRegisterMutationData,
+    ServingDeploymentsRegisterMutationError,
+    ServingDeploymentsRegisterMutationVariables
+  >,
+): UseMutationResult<
+  ServingDeploymentsRegisterMutationData,
+  ServingDeploymentsRegisterMutationError,
+  ServingDeploymentsRegisterMutationVariables
+> {
+  const client = useSDKContext();
+  return useMutation({
+    ...buildServingDeploymentsRegisterMutation(client, options),
+    ...options,
+  });
+}
+
+export function mutationKeyServingDeploymentsRegister(): MutationKey {
+  return ["@meetkai/mka1", "deployments", "register"];
+}
+
+export function buildServingDeploymentsRegisterMutation(
+  client$: SDKCore,
+  hookOptions?: RequestOptions,
+): {
+  mutationKey: MutationKey;
+  mutationFn: (
+    variables: ServingDeploymentsRegisterMutationVariables,
+  ) => Promise<ServingDeploymentsRegisterMutationData>;
+} {
+  return {
+    mutationKey: mutationKeyServingDeploymentsRegister(),
+    mutationFn: function servingDeploymentsRegisterMutationFn({
+      request,
+      options,
+    }): Promise<ServingDeploymentsRegisterMutationData> {
+      const mergedOptions = {
+        ...hookOptions,
+        ...options,
+        fetchOptions: {
+          ...hookOptions?.fetchOptions,
+          ...options?.fetchOptions,
+          signal: combineSignals(
+            hookOptions?.fetchOptions?.signal,
+            options?.fetchOptions?.signal,
+          ),
+        },
+      };
+      return unwrapAsync(servingDeploymentsRegister(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}

package/src/react-query/servingFinetuneJobsGetEvents.core.ts

@@ -43,6 +43,7 @@ export function buildServingFinetuneJobsGetEventsQuery(
 } {
   return {
     queryKey: queryKeyServingFinetuneJobsGetEvents(request.jobId, {
+      includeLineage: request.includeLineage,
       xOnBehalfOf: request.xOnBehalfOf,
     }),
     queryFn: async function servingFinetuneJobsGetEventsQueryFn(
@@ -70,7 +71,10 @@ export function buildServingFinetuneJobsGetEventsQuery(
 
 export function queryKeyServingFinetuneJobsGetEvents(
   jobId: string,
-  parameters: { xOnBehalfOf?: string | undefined },
+  parameters: {
+    includeLineage?: boolean | undefined;
+    xOnBehalfOf?: string | undefined;
+  },
 ): QueryKey {
   return ["@meetkai/mka1", "finetuneJobs", "getEvents", jobId, parameters];
 }

package/src/react-query/servingFinetuneJobsGetEvents.ts

@@ -111,7 +111,10 @@ export function setServingFinetuneJobsGetEventsData(
   client: QueryClient,
   queryKeyBase: [
     jobId: string,
-    parameters: { xOnBehalfOf?: string | undefined },
+    parameters: {
+      includeLineage?: boolean | undefined;
+      xOnBehalfOf?: string | undefined;
+    },
   ],
   data: ServingFinetuneJobsGetEventsQueryData,
 ): ServingFinetuneJobsGetEventsQueryData | undefined {
@@ -123,7 +126,13 @@ export function setServingFinetuneJobsGetEventsData(
 export function invalidateServingFinetuneJobsGetEvents(
   client: QueryClient,
   queryKeyBase: TupleToPrefixes<
-    [jobId: string, parameters: { xOnBehalfOf?: string | undefined }]
+    [
+      jobId: string,
+      parameters: {
+        includeLineage?: boolean | undefined;
+        xOnBehalfOf?: string | undefined;
+      },
+    ]
   >,
   filters?: Omit<InvalidateQueryFilters, "queryKey" | "predicate" | "exact">,
 ): Promise<void> {