• Home
• ai.tock
• tock-bot-open-data
• 21.9.0
• source code

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

bot-open-data.21.9.0.source-code.OpenDataNlpListener.kt Maven / Gradle / Ivy

package ai.tock.bot.open.data

import ai.tock.bot.engine.dialog.Dialog
import ai.tock.bot.engine.dialog.EntityValue
import ai.tock.bot.engine.event.Event
import ai.tock.bot.engine.nlp.NlpListener
import ai.tock.bot.engine.user.UserTimeline
import ai.tock.bot.open.data.client.sncf.SncfOpenDataClient.findPlace
import ai.tock.bot.open.data.client.sncf.model.PlaceValue
import ai.tock.nlp.api.client.model.NlpResult
import ai.tock.nlp.entity.ValueResolverRepository

/**
 * Evaluate place entity values.
 */
object OpenDataNlpListener : NlpListener {

    init {
        //we add a new value type in order to manage open data api place
        //{"@type" : "place"} will look better in mongo than {"@type" : "ai.tock.bot.open.data.client.sncf.model.Place"}
        ValueResolverRepository.registerType(PlaceValue::class)
    }

    override fun evaluateEntities(
        userTimeline: UserTimeline,
        dialog: Dialog,
        event: Event,
        nlpResult: NlpResult
    ): List {
        //evaluate localities
        return nlpResult
            .entities
            .filter { it.entity.entityType == locationEntity.entityType }
            .mapNotNull { entityValue ->
                findPlace(nlpResult.entityTextContent(entityValue))?.let { EntityValue(entityValue.entity, it) }
            }
    }
}