index.ts

const setSessionEntities = (conv: Conv) => {
  // Create sets for each type of entity type
  const trackTitles = new Set<string>()
  const trackGenres = new Set<string>()
  const trackTags = new Set<string>()

  conv.data.json.tracks.forEach(track => {
    trackTitles.add(track.track_title)
    track.track_genre.forEach(genre => {
      trackGenres.add(genre)
    })
    track.tags.forEach(tag => {
      trackTags.add(tag)
    })
  })

  conv.sessionEntities.add({
    name: 'track',
    entities: Array.from(trackTitles).map(title => {
      return {
        value: title,
        synonyms: [title]
      }
    })
  })
  conv.sessionEntities.add({
    name: 'genre',
    entities: Array.from(trackGenres).map(genre => {
      return {
        value: genre,
        synonyms: [genre]
      }
    })
  })
  conv.sessionEntities.add({
    name: 'tag',
    entities: Array.from(trackTags).map(tag => {
      return {
        value: tag,
        synonyms: [tag]
      }
    })
  })
}