retrieveConfig: de_RetrieveConfig()

in clients/client-bedrock/src/protocols/Aws_restJson1.ts [4304:4622]

• 211 lines of code
• 9 McCabe index (conditional complexity)

      retrieveConfig: de_RetrieveConfig(output.retrieveConfig, context),
    };
  }
  return { $unknown: Object.entries(output)[0] };
};

/**
 * deserializeAws_restJson1KnowledgeBaseRetrievalConfiguration
 */
const de_KnowledgeBaseRetrievalConfiguration = (
  output: any,
  context: __SerdeContext
): KnowledgeBaseRetrievalConfiguration => {
  return take(output, {
    vectorSearchConfiguration: (_: any) => de_KnowledgeBaseVectorSearchConfiguration(_, context),
  }) as any;
};

/**
 * deserializeAws_restJson1KnowledgeBaseRetrieveAndGenerateConfiguration
 */
const de_KnowledgeBaseRetrieveAndGenerateConfiguration = (
  output: any,
  context: __SerdeContext
): KnowledgeBaseRetrieveAndGenerateConfiguration => {
  return take(output, {
    generationConfiguration: (_: any) => de_GenerationConfiguration(_, context),
    knowledgeBaseId: __expectString,
    modelArn: __expectString,
    orchestrationConfiguration: _json,
    retrievalConfiguration: (_: any) => de_KnowledgeBaseRetrievalConfiguration(_, context),
  }) as any;
};

/**
 * deserializeAws_restJson1KnowledgeBaseVectorSearchConfiguration
 */
const de_KnowledgeBaseVectorSearchConfiguration = (
  output: any,
  context: __SerdeContext
): KnowledgeBaseVectorSearchConfiguration => {
  return take(output, {
    filter: (_: any) => de_RetrievalFilter(__expectUnion(_), context),
    numberOfResults: __expectInt32,
    overrideSearchType: __expectString,
  }) as any;
};

// de_LoggingConfig omitted.

/**
 * deserializeAws_restJson1MarketplaceModelEndpoint
 */
const de_MarketplaceModelEndpoint = (output: any, context: __SerdeContext): MarketplaceModelEndpoint => {
  return take(output, {
    createdAt: (_: any) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
    endpointArn: __expectString,
    endpointConfig: (_: any) => _json(__expectUnion(_)),
    endpointStatus: __expectString,
    endpointStatusMessage: __expectString,
    modelSourceIdentifier: __expectString,
    status: __expectString,
    statusMessage: __expectString,
    updatedAt: (_: any) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
  }) as any;
};

/**
 * deserializeAws_restJson1MarketplaceModelEndpointSummaries
 */
const de_MarketplaceModelEndpointSummaries = (
  output: any,
  context: __SerdeContext
): MarketplaceModelEndpointSummary[] => {
  const retVal = (output || [])
    .filter((e: any) => e != null)
    .map((entry: any) => {
      return de_MarketplaceModelEndpointSummary(entry, context);
    });
  return retVal;
};

/**
 * deserializeAws_restJson1MarketplaceModelEndpointSummary
 */
const de_MarketplaceModelEndpointSummary = (output: any, context: __SerdeContext): MarketplaceModelEndpointSummary => {
  return take(output, {
    createdAt: (_: any) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
    endpointArn: __expectString,
    modelSourceIdentifier: __expectString,
    status: __expectString,
    statusMessage: __expectString,
    updatedAt: (_: any) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
  }) as any;
};

/**
 * deserializeAws_restJson1ModelCopyJobSummaries
 */
const de_ModelCopyJobSummaries = (output: any, context: __SerdeContext): ModelCopyJobSummary[] => {
  const retVal = (output || [])
    .filter((e: any) => e != null)
    .map((entry: any) => {
      return de_ModelCopyJobSummary(entry, context);
    });
  return retVal;
};

/**
 * deserializeAws_restJson1ModelCopyJobSummary
 */
const de_ModelCopyJobSummary = (output: any, context: __SerdeContext): ModelCopyJobSummary => {
  return take(output, {
    creationTime: (_: any) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
    failureMessage: __expectString,
    jobArn: __expectString,
    sourceAccountId: __expectString,
    sourceModelArn: __expectString,
    sourceModelName: __expectString,
    status: __expectString,
    targetModelArn: __expectString,
    targetModelKmsKeyArn: __expectString,
    targetModelName: __expectString,
    targetModelTags: _json,
  }) as any;
};

// de_ModelCustomizationHyperParameters omitted.

/**
 * deserializeAws_restJson1ModelCustomizationJobSummaries
 */
const de_ModelCustomizationJobSummaries = (output: any, context: __SerdeContext): ModelCustomizationJobSummary[] => {
  const retVal = (output || [])
    .filter((e: any) => e != null)
    .map((entry: any) => {
      return de_ModelCustomizationJobSummary(entry, context);
    });
  return retVal;
};

/**
 * deserializeAws_restJson1ModelCustomizationJobSummary
 */
const de_ModelCustomizationJobSummary = (output: any, context: __SerdeContext): ModelCustomizationJobSummary => {
  return take(output, {
    baseModelArn: __expectString,
    creationTime: (_: any) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
    customModelArn: __expectString,
    customModelName: __expectString,
    customizationType: __expectString,
    endTime: (_: any) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
    jobArn: __expectString,
    jobName: __expectString,
    lastModifiedTime: (_: any) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
    status: __expectString,
    statusDetails: (_: any) => de_StatusDetails(_, context),
  }) as any;
};

// de_ModelCustomizationList omitted.

// de_ModelDataSource omitted.

/**
 * deserializeAws_restJson1ModelImportJobSummaries
 */
const de_ModelImportJobSummaries = (output: any, context: __SerdeContext): ModelImportJobSummary[] => {
  const retVal = (output || [])
    .filter((e: any) => e != null)
    .map((entry: any) => {
      return de_ModelImportJobSummary(entry, context);
    });
  return retVal;
};

/**
 * deserializeAws_restJson1ModelImportJobSummary
 */
const de_ModelImportJobSummary = (output: any, context: __SerdeContext): ModelImportJobSummary => {
  return take(output, {
    creationTime: (_: any) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
    endTime: (_: any) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
    importedModelArn: __expectString,
    importedModelName: __expectString,
    jobArn: __expectString,
    jobName: __expectString,
    lastModifiedTime: (_: any) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
    status: __expectString,
  }) as any;
};

// de_ModelInvocationJobInputDataConfig omitted.

// de_ModelInvocationJobOutputDataConfig omitted.

// de_ModelInvocationJobS3InputDataConfig omitted.

// de_ModelInvocationJobS3OutputDataConfig omitted.

/**
 * deserializeAws_restJson1ModelInvocationJobSummaries
 */
const de_ModelInvocationJobSummaries = (output: any, context: __SerdeContext): ModelInvocationJobSummary[] => {
  const retVal = (output || [])
    .filter((e: any) => e != null)
    .map((entry: any) => {
      return de_ModelInvocationJobSummary(entry, context);
    });
  return retVal;
};

/**
 * deserializeAws_restJson1ModelInvocationJobSummary
 */
const de_ModelInvocationJobSummary = (output: any, context: __SerdeContext): ModelInvocationJobSummary => {
  return take(output, {
    clientRequestToken: __expectString,
    endTime: (_: any) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
    inputDataConfig: (_: any) => _json(__expectUnion(_)),
    jobArn: __expectString,
    jobExpirationTime: (_: any) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
    jobName: __expectString,
    lastModifiedTime: (_: any) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
    message: __expectString,
    modelId: __expectString,
    outputDataConfig: (_: any) => _json(__expectUnion(_)),
    roleArn: __expectString,
    status: __expectString,
    submitTime: (_: any) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
    timeoutDurationInHours: __expectInt32,
    vpcConfig: _json,
  }) as any;
};

// de_ModelModalityList omitted.

// de_OrchestrationConfiguration omitted.

// de_OutputDataConfig omitted.

// de_PerformanceConfiguration omitted.

/**
 * deserializeAws_restJson1PromptRouterSummaries
 */
const de_PromptRouterSummaries = (output: any, context: __SerdeContext): PromptRouterSummary[] => {
  const retVal = (output || [])
    .filter((e: any) => e != null)
    .map((entry: any) => {
      return de_PromptRouterSummary(entry, context);
    });
  return retVal;
};

/**
 * deserializeAws_restJson1PromptRouterSummary
 */
const de_PromptRouterSummary = (output: any, context: __SerdeContext): PromptRouterSummary => {
  return take(output, {
    createdAt: (_: any) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
    description: __expectString,
    fallbackModel: _json,
    models: _json,
    promptRouterArn: __expectString,
    promptRouterName: __expectString,
    routingCriteria: (_: any) => de_RoutingCriteria(_, context),
    status: __expectString,
    type: __expectString,
    updatedAt: (_: any) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
  }) as any;
};

// de_PromptRouterTargetModel omitted.

// de_PromptRouterTargetModels omitted.

// de_PromptTemplate omitted.

/**
 * deserializeAws_restJson1ProvisionedModelSummaries
 */
const de_ProvisionedModelSummaries = (output: any, context: __SerdeContext): ProvisionedModelSummary[] => {
  const retVal = (output || [])
    .filter((e: any) => e != null)
    .map((entry: any) => {
      return de_ProvisionedModelSummary(entry, context);
    });
  return retVal;
};

/**
 * deserializeAws_restJson1ProvisionedModelSummary
 */
const de_ProvisionedModelSummary = (output: any, context: __SerdeContext): ProvisionedModelSummary => {
  return take(output, {
    commitmentDuration: __expectString,
    commitmentExpirationTime: (_: any) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
    creationTime: (_: any) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
    desiredModelArn: __expectString,
    desiredModelUnits: __expectInt32,
    foundationModelArn: __expectString,
    lastModifiedTime: (_: any) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
    modelArn: __expectString,
    modelUnits: __expectInt32,
    provisionedModelArn: __expectString,
    provisionedModelName: __expectString,
    status: __expectString,
  }) as any;
};

// de_QueryTransformationConfiguration omitted.

/**
 * deserializeAws_restJson1RAGConfig
 */
const de_RAGConfig = (output: any, context: __SerdeContext): RAGConfig => {
  if (output.knowledgeBaseConfig != null) {
    return {