@cdot65/prisma-airs-sdk 0.5.2 → 0.6.1

package/dist/index.cjs

@@ -42,6 +42,7 @@ __export(index_exports, {
   AuthType: () => AuthType,
   BEARER: () => BEARER,
   BaseResponseSchema: () => BaseResponseSchema,
+  BedrockAccessConnectionParamsSchema: () => BedrockAccessConnectionParamsSchema,
   BrandSubCategory: () => BrandSubCategory,
   Category: () => Category,
   CategoryModelSchema: () => CategoryModelSchema,
@@ -49,6 +50,7 @@ __export(index_exports, {
   ComplianceReportSchema: () => ComplianceReportSchema,
   ComplianceSubCategory: () => ComplianceSubCategory,
   ComplianceTechniqueSchema: () => ComplianceTechniqueSchema,
+  ConnectionParamsSchema: () => ConnectionParamsSchema,
   Content: () => Content,
   CountByNameSchema: () => CountByNameSchema,
   CountedQuotaEnum: () => CountedQuotaEnum,
@@ -84,6 +86,7 @@ __export(index_exports, {
   DSDetailResultSchema: () => DSDetailResultSchema,
   DSResultMetadataSchema: () => DSResultMetadataSchema,
   DashboardOverviewResponseSchema: () => DashboardOverviewResponseSchema,
+  DatabricksConnectionParamsSchema: () => DatabricksConnectionParamsSchema,
   DateRangeFilter: () => DateRangeFilter,
   DeleteProfileConflictSchema: () => DeleteProfileConflictSchema,
   DeleteProfileResponseSchema: () => DeleteProfileResponseSchema,
@@ -117,6 +120,7 @@ __export(index_exports, {
   HEADER_AUTH_TOKEN: () => HEADER_AUTH_TOKEN,
   HTTPValidationErrorSchema: () => HTTPValidationErrorSchema,
   HTTP_FORCE_RETRY_STATUS_CODES: () => HTTP_FORCE_RETRY_STATUS_CODES,
+  HuggingfaceConnectionParamsSchema: () => HuggingfaceConnectionParamsSchema,
   IODetectedSchema: () => IODetectedSchema,
   JobAbortResponseSchema: () => JobAbortResponseSchema,
   JobCreateRequestSchema: () => JobCreateRequestSchema,
@@ -188,7 +192,10 @@ __export(index_exports, {
   ModelSecurityRuleResponseSchema: () => ModelSecurityRuleResponseSchema,
   ModelSecurityRulesClient: () => ModelSecurityRulesClient,
   ModelSecurityScansClient: () => ModelSecurityScansClient,
+  MultiTurnStatefulConfigSchema: () => MultiTurnStatefulConfigSchema,
+  MultiTurnStatelessConfigSchema: () => MultiTurnStatelessConfigSchema,
   OAuthClient: () => OAuthClient,
+  OpenAIConnectionParamsSchema: () => OpenAIConnectionParamsSchema,
   PAYLOAD_HASH: () => PAYLOAD_HASH,
   PolicySchema: () => PolicySchema,
   PolicyType: () => PolicyType,
@@ -246,6 +253,7 @@ __export(index_exports, {
   ResponseDetectedSchema: () => ResponseDetectedSchema,
   ResponseDetectionDetailsSchema: () => ResponseDetectionDetailsSchema,
   ResponseMode: () => ResponseMode,
+  RestConnectionParamsSchema: () => RestConnectionParamsSchema,
   RiskLevelSchema: () => RiskLevelSchema,
   RiskRating: () => RiskRating,
   RuleConfigurationSchema: () => RuleConfigurationSchema,
@@ -302,6 +310,7 @@ __export(index_exports, {
   StreamIterationDataSchema: () => StreamIterationDataSchema,
   StreamListResponseSchema: () => StreamListResponseSchema,
   StreamType: () => StreamType,
+  StreamingConnectionParamsSchema: () => StreamingConnectionParamsSchema,
   SubCategoryModelSchema: () => SubCategoryModelSchema,
   SubCategoryStatsSchema: () => SubCategoryStatsSchema,
   TargetAdditionalContextSchema: () => TargetAdditionalContextSchema,
@@ -362,7 +371,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.2";
+var SDK_VERSION = "0.6.1";
 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";
@@ -1820,33 +1829,83 @@ var ValidationErrorSchema = import_zod18.z.object({
 });
 var HTTPValidationErrorSchema = import_zod18.z.object({ detail: import_zod18.z.array(ValidationErrorSchema).optional() }).passthrough();
 var TargetBackgroundSchema = import_zod18.z.object({
-  industry: import_zod18.z.unknown().optional(),
-  use_case: import_zod18.z.unknown().optional(),
-  competitors: import_zod18.z.unknown().optional()
+  industry: import_zod18.z.string().nullable().optional(),
+  use_case: import_zod18.z.string().nullable().optional(),
+  competitors: import_zod18.z.array(import_zod18.z.string()).nullable().optional()
 }).passthrough();
 var TargetAdditionalContextSchema = import_zod18.z.object({
-  base_model: import_zod18.z.unknown().optional(),
-  core_architecture: import_zod18.z.unknown().optional(),
-  system_prompt: import_zod18.z.unknown().optional(),
-  languages_supported: import_zod18.z.unknown().optional(),
-  banned_keywords: import_zod18.z.unknown().optional(),
-  tools_accessible: import_zod18.z.unknown().optional()
+  base_model: import_zod18.z.string().nullable().optional(),
+  core_architecture: import_zod18.z.string().nullable().optional(),
+  system_prompt: import_zod18.z.string().nullable().optional(),
+  languages_supported: import_zod18.z.array(import_zod18.z.string()).nullable().optional(),
+  banned_keywords: import_zod18.z.array(import_zod18.z.string()).nullable().optional(),
+  tools_accessible: import_zod18.z.array(import_zod18.z.string()).nullable().optional()
 }).passthrough();
 var TargetMetadataSchema = import_zod18.z.object({
   multi_turn: import_zod18.z.boolean().optional(),
-  multi_turn_error_message: import_zod18.z.unknown().optional(),
-  rate_limit: import_zod18.z.unknown().optional(),
+  multi_turn_error_message: import_zod18.z.string().nullable().optional(),
+  rate_limit: import_zod18.z.number().int().nullable().optional(),
   rate_limit_enabled: import_zod18.z.boolean().optional(),
-  rate_limit_error_code: import_zod18.z.unknown().optional(),
-  rate_limit_error_json: import_zod18.z.unknown().optional(),
-  rate_limit_error_message: import_zod18.z.unknown().optional(),
+  rate_limit_error_code: import_zod18.z.number().int().nullable().optional(),
+  rate_limit_error_json: import_zod18.z.record(import_zod18.z.unknown()).nullable().optional(),
+  rate_limit_error_message: import_zod18.z.string().nullable().optional(),
   content_filter_enabled: import_zod18.z.boolean().optional(),
-  content_filter_error_code: import_zod18.z.unknown().optional(),
-  content_filter_error_json: import_zod18.z.unknown().optional(),
-  content_filter_error_message: import_zod18.z.unknown().optional(),
+  content_filter_error_code: import_zod18.z.number().int().nullable().optional(),
+  content_filter_error_json: import_zod18.z.record(import_zod18.z.unknown()).nullable().optional(),
+  content_filter_error_message: import_zod18.z.string().nullable().optional(),
   probe_message: import_zod18.z.string().optional(),
   request_timeout: import_zod18.z.number().optional()
 }).passthrough();
+var MultiTurnStatefulConfigSchema = import_zod18.z.object({
+  type: import_zod18.z.string().default("stateful"),
+  response_id_field: import_zod18.z.string(),
+  request_id_field: import_zod18.z.string()
+}).passthrough();
+var MultiTurnStatelessConfigSchema = import_zod18.z.object({
+  type: import_zod18.z.string().default("stateless"),
+  assistant_role: import_zod18.z.string().nullable().optional()
+}).passthrough();
+var OpenAIConnectionParamsSchema = import_zod18.z.object({
+  api_key: import_zod18.z.string(),
+  model_name: import_zod18.z.string()
+}).passthrough();
+var HuggingfaceConnectionParamsSchema = import_zod18.z.object({
+  api_key: import_zod18.z.string(),
+  model_name: import_zod18.z.string()
+}).passthrough();
+var DatabricksConnectionParamsSchema = import_zod18.z.object({
+  auth_type: import_zod18.z.string(),
+  workspace_url: import_zod18.z.string(),
+  model_name: import_zod18.z.string(),
+  access_token: import_zod18.z.string().nullable().optional(),
+  client_id: import_zod18.z.string().nullable().optional(),
+  secret: import_zod18.z.string().nullable().optional()
+}).passthrough();
+var BedrockAccessConnectionParamsSchema = import_zod18.z.object({
+  access_id: import_zod18.z.string(),
+  access_secret: import_zod18.z.string(),
+  region: import_zod18.z.string(),
+  model_id: import_zod18.z.string(),
+  session_token: import_zod18.z.string().nullable().optional()
+}).passthrough();
+var RestConnectionParamsSchema = import_zod18.z.object({
+  api_endpoint: import_zod18.z.string().nullable().optional(),
+  request_headers: import_zod18.z.record(import_zod18.z.unknown()).nullable().optional(),
+  request_json: import_zod18.z.record(import_zod18.z.unknown()).nullable().optional(),
+  response_json: import_zod18.z.record(import_zod18.z.unknown()).nullable().optional(),
+  response_key: import_zod18.z.string().nullable().optional(),
+  target_connection_config: import_zod18.z.unknown().nullable().optional(),
+  curl: import_zod18.z.string().nullable().optional(),
+  multi_turn_config: import_zod18.z.unknown().nullable().optional()
+}).passthrough();
+var StreamingConnectionParamsSchema = RestConnectionParamsSchema.extend({
+  response_stop_key: import_zod18.z.string(),
+  response_stop_value: import_zod18.z.string()
+}).passthrough();
+var ConnectionParamsSchema = import_zod18.z.union([
+  StreamingConnectionParamsSchema,
+  RestConnectionParamsSchema
+]);
 var TargetJobRequestSchema = import_zod18.z.object({
   uuid: import_zod18.z.string(),
   version: import_zod18.z.number().int().nullable().optional()
@@ -2572,9 +2631,9 @@ var CustomPromptSetVersionInfoSchema = import_zod18.z.object({
   uuid: import_zod18.z.string(),
   status: import_zod18.z.string(),
   is_latest: import_zod18.z.boolean(),
-  version: import_zod18.z.unknown().optional(),
-  stats: import_zod18.z.unknown().optional(),
-  snapshot_created_at: import_zod18.z.unknown().optional()
+  version: import_zod18.z.string().nullable().optional(),
+  stats: PromptSetStatsSchema.nullable().optional(),
+  snapshot_created_at: import_zod18.z.string().nullable().optional()
 }).passthrough();
 var CustomPromptCreateRequestSchema = import_zod18.z.object({
   prompt: import_zod18.z.string(),
@@ -2779,7 +2838,13 @@ async function managementHttpRequest(opts) {
       const url = new URL(`${stripped}${path}`);
       if (params) {
         for (const [key, value] of Object.entries(params)) {
-          url.searchParams.set(key, value);
+          if (Array.isArray(value)) {
+            for (const v of value) {
+              url.searchParams.append(key, v);
+            }
+          } else {
+            url.searchParams.set(key, value);
+          }
         }
       }
       const headers = {
@@ -3367,13 +3432,23 @@ var ModelSecurityScansClient = class {
 };
 
 // src/model-security/security-groups-client.ts
-function buildListParams2(opts) {
+function buildGroupListParams(opts) {
   const params = {};
   if (opts?.skip !== void 0) params.skip = String(opts.skip);
   if (opts?.limit !== void 0) params.limit = String(opts.limit);
-  if (opts?.sort_by !== void 0) params.sort_by = opts.sort_by;
-  if (opts?.sort_direction !== void 0) params.sort_direction = opts.sort_direction;
-  if (opts?.search !== void 0) params.search = opts.search;
+  if (opts?.sort_field !== void 0) params.sort_field = opts.sort_field;
+  if (opts?.sort_dir !== void 0) params.sort_dir = opts.sort_dir;
+  if (opts?.source_types !== void 0) params.source_types = opts.source_types;
+  if (opts?.search_query !== void 0) params.search_query = opts.search_query;
+  if (opts?.enabled_rules !== void 0) params.enabled_rules = opts.enabled_rules;
+  return params;
+}
+function buildRuleInstanceListParams(opts) {
+  const params = {};
+  if (opts?.skip !== void 0) params.skip = String(opts.skip);
+  if (opts?.limit !== void 0) params.limit = String(opts.limit);
+  if (opts?.security_rule_uuid !== void 0) params.security_rule_uuid = opts.security_rule_uuid;
+  if (opts?.state !== void 0) params.state = opts.state;
   return params;
 }
 var ModelSecurityGroupsClient = class {
@@ -3411,7 +3486,7 @@ var ModelSecurityGroupsClient = class {
       method: "GET",
       baseUrl: this.baseUrl,
       path: MODEL_SEC_SECURITY_GROUPS_PATH,
-      params: buildListParams2(opts),
+      params: buildGroupListParams(opts),
       oauthClient: this.oauthClient,
       numRetries: this.numRetries
     });
@@ -3497,7 +3572,7 @@ var ModelSecurityGroupsClient = class {
       method: "GET",
       baseUrl: this.baseUrl,
       path: `${MODEL_SEC_SECURITY_GROUPS_PATH}/${securityGroupUuid}/rule-instances`,
-      params: buildListParams2(opts),
+      params: buildRuleInstanceListParams(opts),
       oauthClient: this.oauthClient,
       numRetries: this.numRetries
     });
@@ -3564,13 +3639,12 @@ var ModelSecurityGroupsClient = class {
 };
 
 // src/model-security/security-rules-client.ts
-function buildListParams3(opts) {
+function buildRuleListParams(opts) {
   const params = {};
   if (opts?.skip !== void 0) params.skip = String(opts.skip);
   if (opts?.limit !== void 0) params.limit = String(opts.limit);
-  if (opts?.sort_by !== void 0) params.sort_by = opts.sort_by;
-  if (opts?.sort_direction !== void 0) params.sort_direction = opts.sort_direction;
-  if (opts?.search !== void 0) params.search = opts.search;
+  if (opts?.source_type !== void 0) params.source_type = opts.source_type;
+  if (opts?.search_query !== void 0) params.search_query = opts.search_query;
   return params;
 }
 var ModelSecurityRulesClient = class {
@@ -3592,7 +3666,7 @@ var ModelSecurityRulesClient = class {
       method: "GET",
       baseUrl: this.baseUrl,
       path: MODEL_SEC_SECURITY_RULES_PATH,
-      params: buildListParams3(opts),
+      params: buildRuleListParams(opts),
       oauthClient: this.oauthClient,
       numRetries: this.numRetries
     });
@@ -3702,7 +3776,7 @@ var ModelSecurityClient = class {
 };
 
 // src/red-team/scans-client.ts
-function buildListParams4(opts) {
+function buildListParams2(opts) {
   const params = {};
   if (opts?.skip !== void 0) params.skip = String(opts.skip);
   if (opts?.limit !== void 0) params.limit = String(opts.limit);
@@ -3734,7 +3808,7 @@ var RedTeamScansClient = class {
   }
   /** List red team scan jobs with optional filters. */
   async list(opts) {
-    const params = buildListParams4(opts);
+    const params = buildListParams2(opts);
     if (opts?.status !== void 0) params.status = opts.status;
     if (opts?.job_type !== void 0) params.job_type = opts.job_type;
     if (opts?.target_id !== void 0) params.target_id = opts.target_id;
@@ -3790,7 +3864,7 @@ var RedTeamScansClient = class {
 };
 
 // src/red-team/reports-client.ts
-function buildListParams5(opts) {
+function buildListParams3(opts) {
   const params = {};
   if (opts?.skip !== void 0) params.skip = String(opts.skip);
   if (opts?.limit !== void 0) params.limit = String(opts.limit);
@@ -3819,7 +3893,7 @@ var RedTeamReportsClient = class {
   /** List attacks for a static scan. */
   async listAttacks(jobId, opts) {
     validateJobId(jobId);
-    const params = buildListParams5(opts);
+    const params = buildListParams3(opts);
     if (opts?.status !== void 0) params.status = opts.status;
     if (opts?.severity !== void 0) params.severity = opts.severity;
     if (opts?.category !== void 0) params.category = opts.category;
@@ -3948,7 +4022,7 @@ var RedTeamReportsClient = class {
   /** List goals for a dynamic scan. */
   async listGoals(jobId, opts) {
     validateJobId(jobId);
-    const params = buildListParams5(opts);
+    const params = buildListParams3(opts);
     if (opts?.goal_type !== void 0) params.goal_type = opts.goal_type;
     const res = await managementHttpRequest({
       method: "GET",
@@ -3973,7 +4047,7 @@ var RedTeamReportsClient = class {
       method: "GET",
       baseUrl: this.baseUrl,
       path: `${RED_TEAM_REPORT_DYNAMIC_PATH}/${jobId}/goal/${goalId}/list-streams`,
-      params: buildListParams5(opts),
+      params: buildListParams3(opts),
       oauthClient: this.oauthClient,
       numRetries: this.numRetries
     });
@@ -4029,7 +4103,7 @@ var RedTeamReportsClient = class {
 };
 
 // src/red-team/custom-attack-reports-client.ts
-function buildListParams6(opts) {
+function buildListParams4(opts) {
   const params = {};
   if (opts?.skip !== void 0) params.skip = String(opts.skip);
   if (opts?.limit !== void 0) params.limit = String(opts.limit);
@@ -4089,7 +4163,7 @@ var RedTeamCustomAttackReportsClient = class {
       method: "GET",
       baseUrl: this.baseUrl,
       path: `${RED_TEAM_CUSTOM_ATTACKS_REPORT_PATH}/report/${jobId}/prompt-set/${promptSetId}/prompts`,
-      params: buildListParams6(opts),
+      params: buildListParams4(opts),
       oauthClient: this.oauthClient,
       numRetries: this.numRetries
     });
@@ -4120,7 +4194,7 @@ var RedTeamCustomAttackReportsClient = class {
       method: "GET",
       baseUrl: this.baseUrl,
       path: `${RED_TEAM_CUSTOM_ATTACKS_REPORT_PATH}/job/${jobId}/list-custom-attacks`,
-      params: buildListParams6(opts),
+      params: buildListParams4(opts),
       oauthClient: this.oauthClient,
       numRetries: this.numRetries
     });
@@ -4159,7 +4233,7 @@ var RedTeamCustomAttackReportsClient = class {
 };
 
 // src/red-team/targets-client.ts
-function buildListParams7(opts) {
+function buildListParams5(opts) {
   const params = {};
   if (opts?.skip !== void 0) params.skip = String(opts.skip);
   if (opts?.limit !== void 0) params.limit = String(opts.limit);
@@ -4178,12 +4252,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
     });
@@ -4191,7 +4268,7 @@ var RedTeamTargetsClient = class {
   }
   /** List targets with optional filters. */
   async list(opts) {
-    const params = buildListParams7(opts);
+    const params = buildListParams5(opts);
     if (opts?.target_type !== void 0) params.target_type = opts.target_type;
     if (opts?.status !== void 0) params.status = opts.status;
     if (opts?.active !== void 0) params.active = String(opts.active);
@@ -4223,18 +4300,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
     });
@@ -4307,7 +4387,7 @@ var RedTeamTargetsClient = class {
 };
 
 // src/red-team/custom-attacks-client.ts
-function buildListParams8(opts) {
+function buildListParams6(opts) {
   const params = {};
   if (opts?.skip !== void 0) params.skip = String(opts.skip);
   if (opts?.limit !== void 0) params.limit = String(opts.limit);
@@ -4347,7 +4427,7 @@ var RedTeamCustomAttacksClient = class {
   }
   /** List custom prompt sets. */
   async listPromptSets(opts) {
-    const params = buildListParams8(opts);
+    const params = buildListParams6(opts);
     if (opts?.active !== void 0) params.active = String(opts.active);
     if (opts?.archive !== void 0) params.archive = String(opts.archive);
     const res = await managementHttpRequest({
@@ -4445,6 +4525,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
   // -----------------------------------------------------------------------
@@ -4463,7 +4570,7 @@ var RedTeamCustomAttacksClient = class {
   /** List prompts in a prompt set. */
   async listPrompts(promptSetUuid, opts) {
     validateUuid(promptSetUuid, "prompt set uuid");
-    const params = buildListParams8(opts);
+    const params = buildListParams6(opts);
     if (opts?.active !== void 0) params.active = String(opts.active);
     const res = await managementHttpRequest({
       method: "GET",
@@ -4584,7 +4691,7 @@ var RedTeamCustomAttacksClient = class {
 };
 
 // src/red-team/client.ts
-function buildListParams9(opts) {
+function buildListParams7(opts) {
   const params = {};
   if (opts?.skip !== void 0) params.skip = String(opts.skip);
   if (opts?.limit !== void 0) params.limit = String(opts.limit);
@@ -4723,7 +4830,7 @@ var RedTeamClient = class {
       method: "GET",
       baseUrl: this.dataEndpoint,
       path: `${RED_TEAM_ERROR_LOG_PATH}/${jobId}`,
-      params: buildListParams9(opts),
+      params: buildListParams7(opts),
       oauthClient: this.oauthClient,
       numRetries: this.numRetries
     });
@@ -4794,6 +4901,7 @@ var RedTeamClient = class {
   AuthType,
   BEARER,
   BaseResponseSchema,
+  BedrockAccessConnectionParamsSchema,
   BrandSubCategory,
   Category,
   CategoryModelSchema,
@@ -4801,6 +4909,7 @@ var RedTeamClient = class {
   ComplianceReportSchema,
   ComplianceSubCategory,
   ComplianceTechniqueSchema,
+  ConnectionParamsSchema,
   Content,
   CountByNameSchema,
   CountedQuotaEnum,
@@ -4836,6 +4945,7 @@ var RedTeamClient = class {
   DSDetailResultSchema,
   DSResultMetadataSchema,
   DashboardOverviewResponseSchema,
+  DatabricksConnectionParamsSchema,
   DateRangeFilter,
   DeleteProfileConflictSchema,
   DeleteProfileResponseSchema,
@@ -4869,6 +4979,7 @@ var RedTeamClient = class {
   HEADER_AUTH_TOKEN,
   HTTPValidationErrorSchema,
   HTTP_FORCE_RETRY_STATUS_CODES,
+  HuggingfaceConnectionParamsSchema,
   IODetectedSchema,
   JobAbortResponseSchema,
   JobCreateRequestSchema,
@@ -4940,7 +5051,10 @@ var RedTeamClient = class {
   ModelSecurityRuleResponseSchema,
   ModelSecurityRulesClient,
   ModelSecurityScansClient,
+  MultiTurnStatefulConfigSchema,
+  MultiTurnStatelessConfigSchema,
   OAuthClient,
+  OpenAIConnectionParamsSchema,
   PAYLOAD_HASH,
   PolicySchema,
   PolicyType,
@@ -4998,6 +5112,7 @@ var RedTeamClient = class {
   ResponseDetectedSchema,
   ResponseDetectionDetailsSchema,
   ResponseMode,
+  RestConnectionParamsSchema,
   RiskLevelSchema,
   RiskRating,
   RuleConfigurationSchema,
@@ -5054,6 +5169,7 @@ var RedTeamClient = class {
   StreamIterationDataSchema,
   StreamListResponseSchema,
   StreamType,
+  StreamingConnectionParamsSchema,
   SubCategoryModelSchema,
   SubCategoryStatsSchema,
   TargetAdditionalContextSchema,