@meetkai/mka1 0.51.1 → 0.51.2

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

@@ -0,0 +1,37 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { servingTenantsList } from "../../funcs/servingTenantsList.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.ListServingTenantsRequest$inboundSchema,
+};
+
+export const tool$servingTenantsList: ToolDefinition<typeof args> = {
+  name: "serving-tenants-list",
+  description: `List serving tenants
+
+List organization-wide serving enablement records. Cluster administrators only; a missing organization is not provisioned.`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await servingTenantsList(
+      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/mcp-server/tools/servingTenantsSet.ts

@@ -0,0 +1,37 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { servingTenantsSet } from "../../funcs/servingTenantsSet.js";
+import * as operations from "../../models/operations/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: operations.SetServingTenantRequest$inboundSchema,
+};
+
+export const tool$servingTenantsSet: ToolDefinition<typeof args> = {
+  name: "serving-tenants-set",
+  description: `Set serving tenant enablement
+
+Enable or disable serving for an organization and all of its teams. Disabling blocks product API access but does not stop existing workloads or callbacks. Cluster administrators only.`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await servingTenantsSet(
+      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/index.ts

@@ -342,6 +342,7 @@ export * from "./pagemodel.js";
 export * from "./pagereplica.js";
 export * from "./pagerevision.js";
 export * from "./pagesecret.js";
+export * from "./pageservingtenant.js";
 export * from "./pagevolume.js";
 export * from "./pendingsafetycheck.js";
 export * from "./preconfiguredskill.js";
@@ -469,6 +470,8 @@ export * from "./searchvectorstorerequest.js";
 export * from "./secret.js";
 export * from "./secretcreate.js";
 export * from "./security.js";
+export * from "./servingtenant.js";
+export * from "./servingtenantset.js";
 export * from "./servingvalidationerror.js";
 export * from "./sessionkind.js";
 export * from "./sessionrecord.js";

package/src/models/components/pageservingtenant.ts

@@ -0,0 +1,74 @@
+/*
+ * 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";
+import {
+  ServingTenant,
+  ServingTenant$inboundSchema,
+  ServingTenant$Outbound,
+  ServingTenant$outboundSchema,
+} from "./servingtenant.js";
+
+export type PageServingTenant = {
+  data: Array<ServingTenant>;
+  nextCursor?: string | null | undefined;
+  total: number;
+};
+
+/** @internal */
+export const PageServingTenant$inboundSchema: z.ZodType<
+  PageServingTenant,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  data: z.array(ServingTenant$inboundSchema),
+  next_cursor: z.nullable(z.string()).optional(),
+  total: z.number().int(),
+}).transform((v) => {
+  return remap$(v, {
+    "next_cursor": "nextCursor",
+  });
+});
+/** @internal */
+export type PageServingTenant$Outbound = {
+  data: Array<ServingTenant$Outbound>;
+  next_cursor?: string | null | undefined;
+  total: number;
+};
+
+/** @internal */
+export const PageServingTenant$outboundSchema: z.ZodType<
+  PageServingTenant$Outbound,
+  z.ZodTypeDef,
+  PageServingTenant
+> = z.object({
+  data: z.array(ServingTenant$outboundSchema),
+  nextCursor: z.nullable(z.string()).optional(),
+  total: z.number().int(),
+}).transform((v) => {
+  return remap$(v, {
+    nextCursor: "next_cursor",
+  });
+});
+
+export function pageServingTenantToJSON(
+  pageServingTenant: PageServingTenant,
+): string {
+  return JSON.stringify(
+    PageServingTenant$outboundSchema.parse(pageServingTenant),
+  );
+}
+export function pageServingTenantFromJSON(
+  jsonString: string,
+): SafeParseResult<PageServingTenant, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => PageServingTenant$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'PageServingTenant' from JSON`,
+  );
+}

package/src/models/components/servingtenant.ts

@@ -0,0 +1,85 @@
+/*
+ * 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";
+
+/**
+ * Organization-wide serving enablement shared by every team in the organization.
+ */
+export type ServingTenant = {
+  disabledAt: Date | null;
+  enabled: boolean;
+  enabledAt: Date | null;
+  enabledBy: string | null;
+  orgId: string;
+};
+
+/** @internal */
+export const ServingTenant$inboundSchema: z.ZodType<
+  ServingTenant,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  disabled_at: z.nullable(
+    z.string().datetime({ offset: true }).transform(v => new Date(v)),
+  ),
+  enabled: z.boolean(),
+  enabled_at: z.nullable(
+    z.string().datetime({ offset: true }).transform(v => new Date(v)),
+  ),
+  enabled_by: z.nullable(z.string()),
+  org_id: z.string(),
+}).transform((v) => {
+  return remap$(v, {
+    "disabled_at": "disabledAt",
+    "enabled_at": "enabledAt",
+    "enabled_by": "enabledBy",
+    "org_id": "orgId",
+  });
+});
+/** @internal */
+export type ServingTenant$Outbound = {
+  disabled_at: string | null;
+  enabled: boolean;
+  enabled_at: string | null;
+  enabled_by: string | null;
+  org_id: string;
+};
+
+/** @internal */
+export const ServingTenant$outboundSchema: z.ZodType<
+  ServingTenant$Outbound,
+  z.ZodTypeDef,
+  ServingTenant
+> = z.object({
+  disabledAt: z.nullable(z.date().transform(v => v.toISOString())),
+  enabled: z.boolean(),
+  enabledAt: z.nullable(z.date().transform(v => v.toISOString())),
+  enabledBy: z.nullable(z.string()),
+  orgId: z.string(),
+}).transform((v) => {
+  return remap$(v, {
+    disabledAt: "disabled_at",
+    enabledAt: "enabled_at",
+    enabledBy: "enabled_by",
+    orgId: "org_id",
+  });
+});
+
+export function servingTenantToJSON(servingTenant: ServingTenant): string {
+  return JSON.stringify(ServingTenant$outboundSchema.parse(servingTenant));
+}
+export function servingTenantFromJSON(
+  jsonString: string,
+): SafeParseResult<ServingTenant, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => ServingTenant$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'ServingTenant' from JSON`,
+  );
+}

package/src/models/components/servingtenantset.ts

@@ -0,0 +1,54 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod/v3";
+import { safeParse } from "../../lib/schemas.js";
+import { Result as SafeParseResult } from "../../types/fp.js";
+import { SDKValidationError } from "../errors/sdkvalidationerror.js";
+
+/**
+ * Desired organization-wide serving state.
+ */
+export type ServingTenantSet = {
+  enabled: boolean;
+};
+
+/** @internal */
+export const ServingTenantSet$inboundSchema: z.ZodType<
+  ServingTenantSet,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  enabled: z.boolean(),
+});
+/** @internal */
+export type ServingTenantSet$Outbound = {
+  enabled: boolean;
+};
+
+/** @internal */
+export const ServingTenantSet$outboundSchema: z.ZodType<
+  ServingTenantSet$Outbound,
+  z.ZodTypeDef,
+  ServingTenantSet
+> = z.object({
+  enabled: z.boolean(),
+});
+
+export function servingTenantSetToJSON(
+  servingTenantSet: ServingTenantSet,
+): string {
+  return JSON.stringify(
+    ServingTenantSet$outboundSchema.parse(servingTenantSet),
+  );
+}
+export function servingTenantSetFromJSON(
+  jsonString: string,
+): SafeParseResult<ServingTenantSet, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => ServingTenantSet$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'ServingTenantSet' from JSON`,
+  );
+}

package/src/models/operations/getservingtenant.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 GetServingTenantRequest = {
+  orgId: string;
+  /**
+   * Optional external end-user identifier forwarded by the API gateway.
+   */
+  xOnBehalfOf?: string | undefined;
+};
+
+/** @internal */
+export const GetServingTenantRequest$inboundSchema: z.ZodType<
+  GetServingTenantRequest,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  org_id: z.string(),
+  "X-On-Behalf-Of": z.string().optional(),
+}).transform((v) => {
+  return remap$(v, {
+    "org_id": "orgId",
+    "X-On-Behalf-Of": "xOnBehalfOf",
+  });
+});
+/** @internal */
+export type GetServingTenantRequest$Outbound = {
+  org_id: string;
+  "X-On-Behalf-Of"?: string | undefined;
+};
+
+/** @internal */
+export const GetServingTenantRequest$outboundSchema: z.ZodType<
+  GetServingTenantRequest$Outbound,
+  z.ZodTypeDef,
+  GetServingTenantRequest
+> = z.object({
+  orgId: z.string(),
+  xOnBehalfOf: z.string().optional(),
+}).transform((v) => {
+  return remap$(v, {
+    orgId: "org_id",
+    xOnBehalfOf: "X-On-Behalf-Of",
+  });
+});
+
+export function getServingTenantRequestToJSON(
+  getServingTenantRequest: GetServingTenantRequest,
+): string {
+  return JSON.stringify(
+    GetServingTenantRequest$outboundSchema.parse(getServingTenantRequest),
+  );
+}
+export function getServingTenantRequestFromJSON(
+  jsonString: string,
+): SafeParseResult<GetServingTenantRequest, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => GetServingTenantRequest$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'GetServingTenantRequest' from JSON`,
+  );
+}

package/src/models/operations/index.ts

@@ -162,6 +162,7 @@ export * from "./getresponsesusage.js";
 export * from "./getsandboxusage.js";
 export * from "./getserviceaccount.js";
 export * from "./getservingmodel.js";
+export * from "./getservingtenant.js";
 export * from "./getsession.js";
 export * from "./getsessionurl.js";
 export * from "./getskill.js";
@@ -233,6 +234,7 @@ export * from "./listsecrets.js";
 export * from "./listserviceaccountapikeys.js";
 export * from "./listserviceaccounts.js";
 export * from "./listservingmodels.js";
+export * from "./listservingtenants.js";
 export * from "./listsessions.js";
 export * from "./listskills.js";
 export * from "./listskillversions.js";
@@ -276,6 +278,7 @@ export * from "./setapikeybudget.js";
 export * from "./setcurrency.js";
 export * from "./setorgbudget.js";
 export * from "./setorgprice.js";
+export * from "./setservingtenant.js";
 export * from "./setupwhatsappappwebhook.js";
 export * from "./startexportfeedback.js";
 export * from "./streamagentrunevents.js";

package/src/models/operations/listservingtenants.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 ListServingTenantsRequest = {
+  cursor?: string | null | undefined;
+  limit?: number | undefined;
+  /**
+   * Optional external end-user identifier forwarded by the API gateway.
+   */
+  xOnBehalfOf?: string | undefined;
+};
+
+/** @internal */
+export const ListServingTenantsRequest$inboundSchema: z.ZodType<
+  ListServingTenantsRequest,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  cursor: z.nullable(z.string()).optional(),
+  limit: z.number().int().default(50),
+  "X-On-Behalf-Of": z.string().optional(),
+}).transform((v) => {
+  return remap$(v, {
+    "X-On-Behalf-Of": "xOnBehalfOf",
+  });
+});
+/** @internal */
+export type ListServingTenantsRequest$Outbound = {
+  cursor?: string | null | undefined;
+  limit: number;
+  "X-On-Behalf-Of"?: string | undefined;
+};
+
+/** @internal */
+export const ListServingTenantsRequest$outboundSchema: z.ZodType<
+  ListServingTenantsRequest$Outbound,
+  z.ZodTypeDef,
+  ListServingTenantsRequest
+> = z.object({
+  cursor: z.nullable(z.string()).optional(),
+  limit: z.number().int().default(50),
+  xOnBehalfOf: z.string().optional(),
+}).transform((v) => {
+  return remap$(v, {
+    xOnBehalfOf: "X-On-Behalf-Of",
+  });
+});
+
+export function listServingTenantsRequestToJSON(
+  listServingTenantsRequest: ListServingTenantsRequest,
+): string {
+  return JSON.stringify(
+    ListServingTenantsRequest$outboundSchema.parse(listServingTenantsRequest),
+  );
+}
+export function listServingTenantsRequestFromJSON(
+  jsonString: string,
+): SafeParseResult<ListServingTenantsRequest, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => ListServingTenantsRequest$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'ListServingTenantsRequest' from JSON`,
+  );
+}

package/src/models/operations/setservingtenant.ts

@@ -0,0 +1,76 @@
+/*
+ * 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 * as components from "../components/index.js";
+import { SDKValidationError } from "../errors/sdkvalidationerror.js";
+
+export type SetServingTenantRequest = {
+  orgId: string;
+  /**
+   * Optional external end-user identifier forwarded by the API gateway.
+   */
+  xOnBehalfOf?: string | undefined;
+  servingTenantSet: components.ServingTenantSet;
+};
+
+/** @internal */
+export const SetServingTenantRequest$inboundSchema: z.ZodType<
+  SetServingTenantRequest,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  org_id: z.string(),
+  "X-On-Behalf-Of": z.string().optional(),
+  ServingTenantSet: components.ServingTenantSet$inboundSchema,
+}).transform((v) => {
+  return remap$(v, {
+    "org_id": "orgId",
+    "X-On-Behalf-Of": "xOnBehalfOf",
+    "ServingTenantSet": "servingTenantSet",
+  });
+});
+/** @internal */
+export type SetServingTenantRequest$Outbound = {
+  org_id: string;
+  "X-On-Behalf-Of"?: string | undefined;
+  ServingTenantSet: components.ServingTenantSet$Outbound;
+};
+
+/** @internal */
+export const SetServingTenantRequest$outboundSchema: z.ZodType<
+  SetServingTenantRequest$Outbound,
+  z.ZodTypeDef,
+  SetServingTenantRequest
+> = z.object({
+  orgId: z.string(),
+  xOnBehalfOf: z.string().optional(),
+  servingTenantSet: components.ServingTenantSet$outboundSchema,
+}).transform((v) => {
+  return remap$(v, {
+    orgId: "org_id",
+    xOnBehalfOf: "X-On-Behalf-Of",
+    servingTenantSet: "ServingTenantSet",
+  });
+});
+
+export function setServingTenantRequestToJSON(
+  setServingTenantRequest: SetServingTenantRequest,
+): string {
+  return JSON.stringify(
+    SetServingTenantRequest$outboundSchema.parse(setServingTenantRequest),
+  );
+}
+export function setServingTenantRequestFromJSON(
+  jsonString: string,
+): SafeParseResult<SetServingTenantRequest, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => SetServingTenantRequest$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'SetServingTenantRequest' from JSON`,
+  );
+}

package/src/react-query/index.ts

@@ -323,6 +323,9 @@ export * from "./servingModelsRegister.js";
 export * from "./servingSecretsCreate.js";
 export * from "./servingSecretsDelete.js";
 export * from "./servingSecretsList.js";
+export * from "./servingTenantsGet.js";
+export * from "./servingTenantsList.js";
+export * from "./servingTenantsSet.js";
 export * from "./servingVolumesCreate.js";
 export * from "./servingVolumesDelete.js";
 export * from "./servingVolumesGet.js";

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

@@ -0,0 +1,76 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import {
+  QueryClient,
+  QueryFunctionContext,
+  QueryKey,
+} from "@tanstack/react-query";
+import { SDKCore } from "../core.js";
+import { servingTenantsGet } from "../funcs/servingTenantsGet.js";
+import { combineSignals } from "../lib/primitives.js";
+import { RequestOptions } from "../lib/sdks.js";
+import * as components from "../models/components/index.js";
+import * as operations from "../models/operations/index.js";
+import { unwrapAsync } from "../types/fp.js";
+export type ServingTenantsGetQueryData = components.ServingTenant;
+
+export function prefetchServingTenantsGet(
+  queryClient: QueryClient,
+  client$: SDKCore,
+  request: operations.GetServingTenantRequest,
+  options?: RequestOptions,
+): Promise<void> {
+  return queryClient.prefetchQuery({
+    ...buildServingTenantsGetQuery(
+      client$,
+      request,
+      options,
+    ),
+  });
+}
+
+export function buildServingTenantsGetQuery(
+  client$: SDKCore,
+  request: operations.GetServingTenantRequest,
+  options?: RequestOptions,
+): {
+  queryKey: QueryKey;
+  queryFn: (
+    context: QueryFunctionContext,
+  ) => Promise<ServingTenantsGetQueryData>;
+} {
+  return {
+    queryKey: queryKeyServingTenantsGet(request.orgId, {
+      xOnBehalfOf: request.xOnBehalfOf,
+    }),
+    queryFn: async function servingTenantsGetQueryFn(
+      ctx,
+    ): Promise<ServingTenantsGetQueryData> {
+      const sig = combineSignals(
+        ctx.signal,
+        options?.signal,
+        options?.fetchOptions?.signal,
+      );
+      const mergedOptions = {
+        ...options?.fetchOptions,
+        ...options,
+        signal: sig,
+      };
+
+      return unwrapAsync(servingTenantsGet(
+        client$,
+        request,
+        mergedOptions,
+      ));
+    },
+  };
+}
+
+export function queryKeyServingTenantsGet(
+  orgId: string,
+  parameters: { xOnBehalfOf?: string | undefined },
+): QueryKey {
+  return ["@meetkai/mka1", "tenants", "get", orgId, parameters];
+}