working on level links
[gargantext.git] / src / Gargantext / Text / Search.hs
index a7102767187c28a819ba295cd77fe80bd068aeb9..834b3d0e3bbb164a1bbac69b107559e34e679aa2 100644 (file)
@@ -28,14 +28,14 @@ import Data.Ix
 import Data.Text (Text)
 
 import Gargantext.Prelude
-import Gargantext.Text.Ngrams
-import Gargantext.Text.Ngrams.Stem as ST
+import Gargantext.Text.Terms.Mono (monoTexts)
+import Gargantext.Text.Terms.Mono.Stem as ST
 import Gargantext.Text.Parsers.CSV
 
 type DocId = Int
 
 type DocSearchEngine = SearchEngine
-                         Doc
+                         CsvGargV3
                          DocId
                          DocField
                          NoFeatures
@@ -48,18 +48,18 @@ initialDocSearchEngine :: DocSearchEngine
 initialDocSearchEngine =
     initSearchEngine docSearchConfig defaultSearchRankParameters
 
-docSearchConfig :: SearchConfig Doc DocId DocField NoFeatures
+docSearchConfig :: SearchConfig CsvGargV3 DocId DocField NoFeatures
 docSearchConfig =
     SearchConfig {
       documentKey           = d_docId,
-      extractDocumentTerms  = extractTokens,
+      extractDocumentTerms  = extractTerms,
       transformQueryTerm    = normaliseQueryToken,
       documentFeatureValue  = const noFeatures
   }
   where
-    extractTokens :: Doc -> DocField -> [Text]
-    extractTokens doc TitleField       = monograms (d_title doc)
-    extractTokens doc AbstractField    = monograms (d_abstract doc)
+    extractTerms :: CsvGargV3 -> DocField -> [Text]
+    extractTerms doc TitleField       = monoTexts (d_title doc)
+    extractTerms doc AbstractField    = monoTexts (d_abstract doc)
 
     normaliseQueryToken :: Text -> DocField -> Text
     normaliseQueryToken tok =