import { SearchClient, AzureKeyCredential } from '@azure/search-documents'

// An example schema for documents in the index
interface Hotel {
  hotelId?: string
  hotelName?: string | null
  description?: string | null
  descriptionVector?: number[] | null
  parkingIncluded?: boolean | null
  lastRenovationDate?: Date | null
  rating?: number | null
  rooms?: ({
    beds?: number | null
    description?: string | null
  } | null)[]
}

const client = new SearchClient<Hotel>('<endpoint>', '<indexName>', new AzureKeyCredential('<apiKey>'))

async function main() {
  const select = ['hotelId', 'hotelName', 'rooms/beds'] as const
  const searchResults = await client.search('wifi -luxury', {
    // Only fields in Hotel can be added to this array.
    // TS will complain if one is misspelled.
    select,
  })

  for await (const result of searchResults.results) {
    // result.document has hotelId, hotelName, and rating.
    // this example from the README works as expected
    console.log(result.document.hotelName)
    console.log(result.document.hotelId)
  }

  // however, the following does not: 
  const suggestResults = await client.suggest('wifi', 'sg', {
    select,
  })

  for await (const result of suggestResults.results) {
    console.log(result.document.hotelName) // causes TS error
    console.log(result.document.hotelId) // causes TS error
  }
}

main()