From e0ce7a2e3bc1986251b201374e6a7e4f1fafaec6 Mon Sep 17 00:00:00 2001
From: davidmezzetti <561939+davidmezzetti@users.noreply.github.com>
Date: Fri, 10 Sep 2021 14:13:26 -0400
Subject: [PATCH] Update package imports, closes #41

---
 setup.py                            | 4 ++--
 src/python/paperai/highlights.py    | 2 +-
 src/python/paperai/index.py         | 2 +-
 src/python/paperai/query.py         | 2 +-
 src/python/paperai/report/common.py | 2 +-
 src/python/paperai/vectors.py       | 2 +-
 6 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/setup.py b/setup.py
index c2bcf68..388a2ae 100644
--- a/setup.py
+++ b/setup.py
@@ -5,7 +5,7 @@
     DESCRIPTION = f.read()
 
 setup(name="paperai",
-      version="1.9.0",
+      version="1.10.0",
       author="NeuML",
       description="AI-powered literature discovery and review engine for medical/scientific papers",
       long_description=DESCRIPTION,
@@ -32,7 +32,7 @@
           "networkx>=2.4",
           "PyYAML>=5.3",
           "regex>=2020.5.14",
-          "txtai[api,similarity]>=3.2.0",
+          "txtai[api,similarity]>=3.3.0",
           "txtmarker>=1.0.0"
       ],
       classifiers=[
diff --git a/src/python/paperai/highlights.py b/src/python/paperai/highlights.py
index 85a8b00..13805d8 100644
--- a/src/python/paperai/highlights.py
+++ b/src/python/paperai/highlights.py
@@ -6,7 +6,7 @@
 
 import networkx
 
-from txtai.tokenizer import Tokenizer
+from txtai.pipeline import Tokenizer
 
 class Highlights(object):
     """
diff --git a/src/python/paperai/index.py b/src/python/paperai/index.py
index b084648..a0729af 100644
--- a/src/python/paperai/index.py
+++ b/src/python/paperai/index.py
@@ -10,7 +10,7 @@
 import yaml
 
 from txtai.embeddings import Embeddings
-from txtai.tokenizer import Tokenizer
+from txtai.pipeline import Tokenizer
 
 from .models import Models
 
diff --git a/src/python/paperai/query.py b/src/python/paperai/query.py
index c8bde19..0d43748 100644
--- a/src/python/paperai/query.py
+++ b/src/python/paperai/query.py
@@ -9,7 +9,7 @@
 import html2markdown
 import mdv
 
-from txtai.tokenizer import Tokenizer
+from txtai.pipeline import Tokenizer
 
 from .highlights import Highlights
 from .models import Models
diff --git a/src/python/paperai/report/common.py b/src/python/paperai/report/common.py
index c8c59b2..2c2c301 100644
--- a/src/python/paperai/report/common.py
+++ b/src/python/paperai/report/common.py
@@ -4,7 +4,7 @@
 
 import regex as re
 
-from txtai.extractor import Extractor
+from txtai.pipeline import Extractor
 
 from ..index import Index
 from ..query import Query
diff --git a/src/python/paperai/vectors.py b/src/python/paperai/vectors.py
index dbaac82..7d96a93 100644
--- a/src/python/paperai/vectors.py
+++ b/src/python/paperai/vectors.py
@@ -8,8 +8,8 @@
 import sys
 import tempfile
 
+from txtai.pipeline import Tokenizer
 from txtai.vectors import WordVectors
-from txtai.tokenizer import Tokenizer
 
 from .models import Models