@cdot65/prisma-airs-sdk 0.5.1 → 0.6.0

package/dist/index.js

@@ -23,7 +23,7 @@ var MAX_NUMBER_OF_BATCH_SCAN_OBJECTS = 5;
 var MAX_CONNECTION_POOL_SIZE = 100;
 var MAX_NUMBER_OF_RETRIES = 5;
 var HTTP_FORCE_RETRY_STATUS_CODES = [500, 502, 503, 504];
-var SDK_VERSION = "0.5.1";
+var SDK_VERSION = "0.6.0";
 var USER_AGENT = `PAN-AIRS/${SDK_VERSION}-typescript-sdk`;
 var DEFAULT_MGMT_ENDPOINT = "https://api.sase.paloaltonetworks.com/aisec";
 var DEFAULT_TOKEN_ENDPOINT = "https://auth.apps.paloaltonetworks.com/oauth2/access_token";
@@ -1481,33 +1481,83 @@ var ValidationErrorSchema = z18.object({
 });
 var HTTPValidationErrorSchema = z18.object({ detail: z18.array(ValidationErrorSchema).optional() }).passthrough();
 var TargetBackgroundSchema = z18.object({
-  industry: z18.unknown().optional(),
-  use_case: z18.unknown().optional(),
-  competitors: z18.unknown().optional()
+  industry: z18.string().nullable().optional(),
+  use_case: z18.string().nullable().optional(),
+  competitors: z18.array(z18.string()).nullable().optional()
 }).passthrough();
 var TargetAdditionalContextSchema = z18.object({
-  base_model: z18.unknown().optional(),
-  core_architecture: z18.unknown().optional(),
-  system_prompt: z18.unknown().optional(),
-  languages_supported: z18.unknown().optional(),
-  banned_keywords: z18.unknown().optional(),
-  tools_accessible: z18.unknown().optional()
+  base_model: z18.string().nullable().optional(),
+  core_architecture: z18.string().nullable().optional(),
+  system_prompt: z18.string().nullable().optional(),
+  languages_supported: z18.array(z18.string()).nullable().optional(),
+  banned_keywords: z18.array(z18.string()).nullable().optional(),
+  tools_accessible: z18.array(z18.string()).nullable().optional()
 }).passthrough();
 var TargetMetadataSchema = z18.object({
   multi_turn: z18.boolean().optional(),
-  multi_turn_error_message: z18.unknown().optional(),
-  rate_limit: z18.unknown().optional(),
+  multi_turn_error_message: z18.string().nullable().optional(),
+  rate_limit: z18.number().int().nullable().optional(),
   rate_limit_enabled: z18.boolean().optional(),
-  rate_limit_error_code: z18.unknown().optional(),
-  rate_limit_error_json: z18.unknown().optional(),
-  rate_limit_error_message: z18.unknown().optional(),
+  rate_limit_error_code: z18.number().int().nullable().optional(),
+  rate_limit_error_json: z18.record(z18.unknown()).nullable().optional(),
+  rate_limit_error_message: z18.string().nullable().optional(),
   content_filter_enabled: z18.boolean().optional(),
-  content_filter_error_code: z18.unknown().optional(),
-  content_filter_error_json: z18.unknown().optional(),
-  content_filter_error_message: z18.unknown().optional(),
+  content_filter_error_code: z18.number().int().nullable().optional(),
+  content_filter_error_json: z18.record(z18.unknown()).nullable().optional(),
+  content_filter_error_message: z18.string().nullable().optional(),
   probe_message: z18.string().optional(),
   request_timeout: z18.number().optional()
 }).passthrough();
+var MultiTurnStatefulConfigSchema = z18.object({
+  type: z18.string().default("stateful"),
+  response_id_field: z18.string(),
+  request_id_field: z18.string()
+}).passthrough();
+var MultiTurnStatelessConfigSchema = z18.object({
+  type: z18.string().default("stateless"),
+  assistant_role: z18.string().nullable().optional()
+}).passthrough();
+var OpenAIConnectionParamsSchema = z18.object({
+  api_key: z18.string(),
+  model_name: z18.string()
+}).passthrough();
+var HuggingfaceConnectionParamsSchema = z18.object({
+  api_key: z18.string(),
+  model_name: z18.string()
+}).passthrough();
+var DatabricksConnectionParamsSchema = z18.object({
+  auth_type: z18.string(),
+  workspace_url: z18.string(),
+  model_name: z18.string(),
+  access_token: z18.string().nullable().optional(),
+  client_id: z18.string().nullable().optional(),
+  secret: z18.string().nullable().optional()
+}).passthrough();
+var BedrockAccessConnectionParamsSchema = z18.object({
+  access_id: z18.string(),
+  access_secret: z18.string(),
+  region: z18.string(),
+  model_id: z18.string(),
+  session_token: z18.string().nullable().optional()
+}).passthrough();
+var RestConnectionParamsSchema = z18.object({
+  api_endpoint: z18.string().nullable().optional(),
+  request_headers: z18.record(z18.unknown()).nullable().optional(),
+  request_json: z18.record(z18.unknown()).nullable().optional(),
+  response_json: z18.record(z18.unknown()).nullable().optional(),
+  response_key: z18.string().nullable().optional(),
+  target_connection_config: z18.unknown().nullable().optional(),
+  curl: z18.string().nullable().optional(),
+  multi_turn_config: z18.unknown().nullable().optional()
+}).passthrough();
+var StreamingConnectionParamsSchema = RestConnectionParamsSchema.extend({
+  response_stop_key: z18.string(),
+  response_stop_value: z18.string()
+}).passthrough();
+var ConnectionParamsSchema = z18.union([
+  StreamingConnectionParamsSchema,
+  RestConnectionParamsSchema
+]);
 var TargetJobRequestSchema = z18.object({
   uuid: z18.string(),
   version: z18.number().int().nullable().optional()
@@ -2233,9 +2283,9 @@ var CustomPromptSetVersionInfoSchema = z18.object({
   uuid: z18.string(),
   status: z18.string(),
   is_latest: z18.boolean(),
-  version: z18.unknown().optional(),
-  stats: z18.unknown().optional(),
-  snapshot_created_at: z18.unknown().optional()
+  version: z18.string().nullable().optional(),
+  stats: PromptSetStatsSchema.nullable().optional(),
+  snapshot_created_at: z18.string().nullable().optional()
 }).passthrough();
 var CustomPromptCreateRequestSchema = z18.object({
   prompt: z18.string(),
@@ -3839,12 +3889,15 @@ var RedTeamTargetsClient = class {
     this.numRetries = opts.numRetries;
   }
   /** Create a new target. */
-  async create(request) {
+  async create(request, opts) {
+    const params = {};
+    if (opts?.validate !== void 0) params.validate = String(opts.validate);
     const res = await managementHttpRequest({
       method: "POST",
       baseUrl: this.baseUrl,
       path: RED_TEAM_TARGET_PATH,
       body: request,
+      params: Object.keys(params).length > 0 ? params : void 0,
       oauthClient: this.oauthClient,
       numRetries: this.numRetries
     });
@@ -3884,18 +3937,21 @@ var RedTeamTargetsClient = class {
     return res.data;
   }
   /** Update a target. */
-  async update(uuid, request) {
+  async update(uuid, request, opts) {
     if (!isValidUuid(uuid)) {
       throw new AISecSDKException(
         `Invalid target uuid: ${uuid}`,
         "AISEC_USER_REQUEST_PAYLOAD_ERROR" /* USER_REQUEST_PAYLOAD_ERROR */
       );
     }
+    const params = {};
+    if (opts?.validate !== void 0) params.validate = String(opts.validate);
     const res = await managementHttpRequest({
       method: "PUT",
       baseUrl: this.baseUrl,
       path: `${RED_TEAM_TARGET_PATH}/${uuid}`,
       body: request,
+      params: Object.keys(params).length > 0 ? params : void 0,
       oauthClient: this.oauthClient,
       numRetries: this.numRetries
     });
@@ -4106,6 +4162,33 @@ var RedTeamCustomAttacksClient = class {
     });
     return res.data;
   }
+  /** Upload a CSV file of custom prompts for a prompt set. */
+  async uploadPromptsCsv(promptSetUuid, file) {
+    validateUuid(promptSetUuid, "prompt set uuid");
+    const token = await this.oauthClient.getToken();
+    const url = new URL(
+      `${this.baseUrl.replace(/\/+$/, "")}${RED_TEAM_CUSTOM_ATTACK_PATH}/upload-custom-prompts-csv`
+    );
+    url.searchParams.set("prompt_set_uuid", promptSetUuid);
+    const formData = new FormData();
+    formData.append("file", file);
+    const response = await fetch(url.toString(), {
+      method: "POST",
+      headers: {
+        Authorization: `Bearer ${token}`,
+        "User-Agent": USER_AGENT
+      },
+      body: formData
+    });
+    const text = await response.text();
+    if (!response.ok) {
+      throw new AISecSDKException(
+        `Upload failed (${response.status}): ${text}`,
+        "AISEC_SERVER_SIDE_ERROR" /* SERVER_SIDE_ERROR */
+      );
+    }
+    return text ? JSON.parse(text) : { message: "ok", status: 201 };
+  }
   // -----------------------------------------------------------------------
   // Prompt operations
   // -----------------------------------------------------------------------
@@ -4454,6 +4537,7 @@ export {
   AuthType,
   BEARER,
   BaseResponseSchema,
+  BedrockAccessConnectionParamsSchema,
   BrandSubCategory,
   Category,
   CategoryModelSchema,
@@ -4461,6 +4545,7 @@ export {
   ComplianceReportSchema,
   ComplianceSubCategory,
   ComplianceTechniqueSchema,
+  ConnectionParamsSchema,
   Content,
   CountByNameSchema,
   CountedQuotaEnum,
@@ -4496,6 +4581,7 @@ export {
   DSDetailResultSchema,
   DSResultMetadataSchema,
   DashboardOverviewResponseSchema,
+  DatabricksConnectionParamsSchema,
   DateRangeFilter,
   DeleteProfileConflictSchema,
   DeleteProfileResponseSchema,
@@ -4529,6 +4615,7 @@ export {
   HEADER_AUTH_TOKEN,
   HTTPValidationErrorSchema,
   HTTP_FORCE_RETRY_STATUS_CODES,
+  HuggingfaceConnectionParamsSchema,
   IODetectedSchema,
   JobAbortResponseSchema,
   JobCreateRequestSchema,
@@ -4600,7 +4687,10 @@ export {
   ModelSecurityRuleResponseSchema,
   ModelSecurityRulesClient,
   ModelSecurityScansClient,
+  MultiTurnStatefulConfigSchema,
+  MultiTurnStatelessConfigSchema,
   OAuthClient,
+  OpenAIConnectionParamsSchema,
   PAYLOAD_HASH,
   PolicySchema,
   PolicyType,
@@ -4658,6 +4748,7 @@ export {
   ResponseDetectedSchema,
   ResponseDetectionDetailsSchema,
   ResponseMode,
+  RestConnectionParamsSchema,
   RiskLevelSchema,
   RiskRating,
   RuleConfigurationSchema,
@@ -4714,6 +4805,7 @@ export {
   StreamIterationDataSchema,
   StreamListResponseSchema,
   StreamType,
+  StreamingConnectionParamsSchema,
   SubCategoryModelSchema,
   SubCategoryStatsSchema,
   TargetAdditionalContextSchema,