Merge pull request #1539 from vespa-engine/andreer/summaries

andreer/summaries
vespa-engine · Oct 25, 2024 · 32ed627 · 32ed627
2 parents d87a260 + 1792cb3
commit 32ed627
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/visual-retrieval-colpali/backend/vespa_app.py b/visual-retrieval-colpali/backend/vespa_app.py
@@ -318,12 +318,12 @@ async def get_suggestions(self, query: str) -> list:
         async with self.app.asyncio(connections=1) as session:
             start = time.perf_counter()
             yql = f'select questions from {self.VESPA_SCHEMA_NAME} where questions matches "{query}" limit 3'
-            print(yql)
             response: VespaQueryResponse = await session.query(
                 body={
                     "yql": yql,
                     "ranking": "unranked",
                     "presentation.timing": True,
+                    "presentation.summary": "suggestions",
                 },
             )
             assert response.is_successful(), response.json
@@ -337,8 +337,6 @@ async def get_suggestions(self, query: str) -> list:
                 if "root" in response.json and "children" in response.json["root"]
                 else []
             )
-            print(response.json)
-
             questions = [
                 result["fields"]["questions"]
                 for result in search_results

diff --git a/visual-retrieval-colpali/prepare_feed_deploy.py b/visual-retrieval-colpali/prepare_feed_deploy.py
@@ -723,6 +723,15 @@ def float_to_binary_embedding(float_query_embedding: dict) -> dict:
             ],
             from_disk=True,
         ),
+        DocumentSummary(
+            name="suggestions",
+            summary_fields=[
+                Summary(
+                    name="questions"
+                ),
+            ],
+            from_disk=True,
+        ),
     ],
 )