mirror of
https://github.com/facebook/duckling.git
synced 2024-12-23 04:02:35 +03:00
ab0ad0256e
Summary: * Locales support for the library, following `<Lang>_<Region>` with ISO 639-1 code for `<Lang>` and ISO 3166-1 alpha-2 code for `<Region>` (#33) * `Locale` opaque type (composite of `Lang` and `Region`) with `makeLocale` smart constructor to only allow valid `(Lang, Region)` combinations * API: `Context`'s `lang` parameter has been replaced by `locale`, with optional `Region` and backward compatibility. * `Rules/<Lang>.hs` exposes - `langRules`: cross-locale rules for `<Lang>`, from `<Dimension>/<Lang>/Rules.hs` - `localeRules`: locale-specific rules, from `<Dimension>/<Lang>/<Region>/Rules.hs` - `defaultRules`: `langRules` + specific rules from select locales to ensure backward-compatibility * Corpus, tests & classifiers - 1 classifier per locale, with default classifier (`<Lang>_XX`) when no locale provided (backward-compatible) - Default classifiers are built on existing corpus - Locale classifiers are built on - `<Dimension>/<Lang>/Corpus.hs` exposes a common `corpus` to all locales of `<Lang>` - `<Dimension>/<Lang>/<Region>/Corpus.hs` exposes `allExamples`: a list of examples specific to the locale (following `<Dimension>/<Lang>/<Region>/Rules.hs`). - Locale classifiers use the language corpus extended with the locale examples as training set. - Locale examples need to use the same `Context` (i.e. reference time) as the language corpus. - For backward compatibility, `<Dimension>/<Lang>/Corpus.hs` can expose also `defaultCorpus`, which is `corpus` augmented with specific examples. This is controlled by `getDefaultCorpusForLang` in `Duckling.Ranking.Generate`. - Tests run against each classifier to make sure runtime works as expected. * MM/DD (en_US) vs DD/MM (en_GB) example to illustrate Reviewed By: JonCoens, blandinw Differential Revision: D6038096 fbshipit-source-id: f29c28d
69 lines
1.9 KiB
Haskell
69 lines
1.9 KiB
Haskell
-- Copyright (c) 2016-present, Facebook, Inc.
|
|
-- All rights reserved.
|
|
--
|
|
-- This source code is licensed under the BSD-style license found in the
|
|
-- LICENSE file in the root directory of this source tree. An additional grant
|
|
-- of patent rights can be found in the PATENTS file in the same directory.
|
|
|
|
|
|
{-# LANGUAGE NoRebindableSyntax #-}
|
|
|
|
-- | Everything needed to run Duckling.
|
|
|
|
module Duckling.Core
|
|
( Context(..)
|
|
, Region(..)
|
|
, Dimension(..)
|
|
, Entity(..)
|
|
, Lang(..)
|
|
, Locale
|
|
, Some(..)
|
|
, fromName
|
|
, makeLocale
|
|
, toName
|
|
|
|
-- Duckling API
|
|
, parse
|
|
, supportedDimensions
|
|
, allLocales
|
|
|
|
-- Reference time builders
|
|
, currentReftime
|
|
, fromZonedTime
|
|
, makeReftime
|
|
) where
|
|
|
|
import Data.HashMap.Strict (HashMap)
|
|
import Data.Maybe
|
|
import Data.Text (Text)
|
|
import Data.Time
|
|
import Data.Time.LocalTime.TimeZone.Series
|
|
import Prelude
|
|
import qualified Data.HashMap.Strict as HashMap
|
|
|
|
import Duckling.Api
|
|
import Duckling.Dimensions.Types
|
|
import Duckling.Locale
|
|
import Duckling.Resolve
|
|
import Duckling.Types
|
|
|
|
-- | Builds a `DucklingTime` for timezone `tz` at `utcTime`.
|
|
-- If no `series` found for `tz`, uses UTC.
|
|
makeReftime :: HashMap Text TimeZoneSeries -> Text -> UTCTime -> DucklingTime
|
|
makeReftime series tz utcTime = DucklingTime $ ZoneSeriesTime ducklingTime tzs
|
|
where
|
|
tzs = HashMap.lookupDefault (TimeZoneSeries utc []) tz series
|
|
ducklingTime = toUTC $ utcToLocalTime' tzs utcTime
|
|
|
|
-- | Builds a `DucklingTime` for timezone `tz` at current time.
|
|
-- If no `series` found for `tz`, uses UTC.
|
|
currentReftime :: HashMap Text TimeZoneSeries -> Text -> IO DucklingTime
|
|
currentReftime series tz = do
|
|
utcNow <- getCurrentTime
|
|
return $ makeReftime series tz utcNow
|
|
|
|
-- | Builds a `DucklingTime` from a `ZonedTime`.
|
|
fromZonedTime :: ZonedTime -> DucklingTime
|
|
fromZonedTime (ZonedTime localTime timeZone) = DucklingTime $
|
|
ZoneSeriesTime (toUTC localTime) (TimeZoneSeries timeZone [])
|