@crowdstrike/aidr 1.0.2

package/src/schemas/index.ts

@@ -0,0 +1,114 @@
+import * as v from 'valibot';
+
+const BaseAPIResponseSchema = {
+  request_id: v.string(),
+  request_time: v.string(),
+  response_time: v.string(),
+  status: v.string(),
+};
+
+export const ErrorCodeSchema = v.picklist([
+  'FieldRequired',
+  'InvalidString',
+  'InvalidNumber',
+  'InvalidInteger',
+  'InvalidObject',
+  'InvalidArray',
+  'InvalidNull',
+  'InvalidBool',
+  'BadFormat',
+  'BadFormatPangeaDuration',
+  'BadFormatDateTime',
+  'BadFormatTime',
+  'BadFormatDate',
+  'BadFormatEmail',
+  'BadFormatHostname',
+  'BadFormatIPv4',
+  'BadFormatIPv6',
+  'BadFormatIPAddress',
+  'BadFormatUUID',
+  'BadFormatURI',
+  'BadFormatURIReference',
+  'BadFormatIRI',
+  'BadFormatIRIReference',
+  'BadFormatJSONPointer',
+  'BadFormatRelativeJSONPointer',
+  'BadFormatRegex',
+  'BadFormatJSONPath',
+  'BadFormatBase64',
+  'DoesNotMatchPattern',
+  'DoesNotMatchPatternProperties',
+  'NotEnumMember',
+  'AboveMaxLength',
+  'BelowMinLength',
+  'AboveMaxItems',
+  'BelowMinItems',
+  'NotMultipleOf',
+  'NotWithinRange',
+  'UnexpectedProperty',
+  'InvalidPropertyName',
+  'AboveMaxProperties',
+  'BelowMinProperties',
+  'NotContains',
+  'ContainsTooMany',
+  'ContainsTooFew',
+  'ItemNotUnique',
+  'UnexpectedAdditionalItem',
+  'InvalidConst',
+  'IsDependentOn',
+  'IsTooBig',
+  'IsTooSmall',
+  'ShouldNotBeValid',
+  'NoUnevaluatedItems',
+  'NoUnevaluatedProperties',
+  'DoesNotExist',
+  'IsReadOnly',
+  'CannotAddToDefault',
+  'MustProvideOne',
+  'MutuallyExclusive',
+  'BadState',
+  'InaccessibleURI',
+  'ProviderDisabled',
+  'ConfigProjectMismatch',
+  'ConfigServiceMismatch',
+  'ConfigNotExist',
+]);
+
+const ErrorDetailSchema = v.object({
+  code: ErrorCodeSchema,
+  detail: v.string(),
+  source: v.string(),
+  path: v.optional(v.string()),
+});
+
+export const AcceptedResponseSchema = v.object({
+  ...BaseAPIResponseSchema,
+  status: v.literal('Accepted'),
+  result: v.object({
+    ttl_mins: v.number(),
+    retry_counter: v.number(),
+    location: v.string(),
+  }),
+});
+
+export const ErrorResponseSchema = v.object({
+  ...BaseAPIResponseSchema,
+  status: v.string(),
+  result: v.nullable(v.array(ErrorDetailSchema)),
+});
+
+export const SuccessResponseSchema = <T extends v.GenericSchema>(
+  resultSchema: T
+) =>
+  v.object({
+    ...BaseAPIResponseSchema,
+    status: v.literal('Success'),
+    result: resultSchema,
+  });
+
+export const APIResponseSchema = <T extends v.GenericSchema>(resultSchema: T) =>
+  v.union([
+    AcceptedResponseSchema,
+    SuccessResponseSchema(resultSchema),
+    ErrorResponseSchema,
+  ]);

package/src/services/ai-guard.ts

@@ -0,0 +1,27 @@
+import { Client } from '../client';
+import type { RequestOptions } from '../internal/request-options';
+import type { MaybeAcceptedResponse } from '../types';
+import type {
+  AidrPostV1GuardChatCompletionsResponse,
+  ChatCompletionsGuard,
+} from '../types/ai-guard';
+
+export class AIGuard extends Client {
+  serviceName = 'aiguard';
+
+  /**
+   * Analyze and redact content to avoid manipulation of the model, addition of
+   * malicious content, and other undesirable data transfers.
+   */
+  guardChatCompletions(
+    body: ChatCompletionsGuard,
+    options?: RequestOptions
+  ): Promise<
+    MaybeAcceptedResponse<AidrPostV1GuardChatCompletionsResponse['result']>
+  > {
+    return this.post('/v1/guard_chat_completions', {
+      body,
+      ...options,
+    });
+  }
+}