Merge pull request #381 from vespa-engine/tgm/remove-query-module

Tgm/remove query module

Author: thigm85

setup.py

@@ -30,6 +30,7 @@ def get_target_version():
         "cryptography",
         "aiohttp",
         "tenacity",
+        "learntorank"
     ],
     extras_require={
         "ml": ["transformers", "torch", "tensorflow", "tensorflow_ranking", "keras_tuner"],

vespa/application.py

@@ -20,7 +20,7 @@
 from time import sleep
 
 from vespa.io import VespaQueryResponse, VespaResponse
-from vespa.query import QueryModel
+from learntorank.query import QueryModel
 from vespa.package import ApplicationPackage
 
 retry_strategy = Retry(

vespa/gallery.py

@@ -12,7 +12,7 @@
     QueryProfileType,
     QueryTypeField,
 )
-from vespa.query import QueryModel, AND, RankProfile as Ranking
+from learntorank.query import QueryModel, AND, Ranking
 
 
 class TextSearch(ApplicationPackage):
@@ -58,7 +58,7 @@ def __init__(
             name=name,
             schema=[schema],
             default_query_model=QueryModel(
-                name="and_bm25", match_phase=AND(), rank_profile=Ranking(name="bm25")
+                name="and_bm25", match_phase=AND(), ranking=Ranking(name="bm25")
             ),
         )
 

vespa/package.py

@@ -9,7 +9,7 @@
 from io import BytesIO
 
 from vespa.json_serialization import ToJson, FromJson
-from vespa.query import QueryModel
+from learntorank.query import QueryModel
 
 
 class HNSW(ToJson, FromJson["HNSW"]):