nucliadb 6.8.1.post5003__py3-none-any.whl → 6.8.1.post5005__py3-none-any.whl

nucliadb/search/search/chat/ask.py

@@ -25,6 +25,7 @@ from typing import AsyncGenerator, Optional, Union, cast
 from nuclia_models.common.consumption import Consumption
 from nuclia_models.predict.generative_responses import (
     CitationsGenerativeResponse,
+    FootnoteCitationsGenerativeResponse,
     GenerativeChunk,
     JSONGenerativeResponse,
     MetaGenerativeResponse,
@@ -91,6 +92,7 @@ from nucliadb_models.search import (
     FindOptions,
     FindParagraph,
     FindRequest,
+    FootnoteCitationsAskResponseItem,
     GraphStrategy,
     JSONAskResponseItem,
     KnowledgeboxFindResults,
@@ -173,6 +175,7 @@ class AskResult:
         self._object: Optional[JSONGenerativeResponse] = None
         self._status: Optional[StatusGenerativeResponse] = None
         self._citations: Optional[CitationsGenerativeResponse] = None
+        self._footnote_citations: Optional[FootnoteCitationsGenerativeResponse] = None
         self._metadata: Optional[MetaGenerativeResponse] = None
         self._relations: Optional[Relations] = None
         self._consumption: Optional[Consumption] = None
@@ -303,6 +306,11 @@ class AskResult:
             yield CitationsAskResponseItem(
                 citations=self._citations.citations,
             )
+        # Stream out the footnote citations mapping
+        if self._footnote_citations is not None:
+            yield FootnoteCitationsAskResponseItem(
+                footnote_to_context=self._footnote_citations.footnote_to_context,
+            )
 
         # Stream out generic metadata about the answer
         if self._metadata is not None:
@@ -380,6 +388,10 @@ class AskResult:
         if self._citations is not None:
             citations = self._citations.citations
 
+        footnote_citations = {}
+        if self._footnote_citations is not None:
+            footnote_citations = self._footnote_citations.footnote_to_context
+
         answer_json = None
         if self._object is not None:
             answer_json = self._object.object
@@ -408,6 +420,7 @@ class AskResult:
             retrieval_best_matches=best_matches,
             prequeries=prequeries_results,
             citations=citations,
+            citation_footnote_to_context=footnote_citations,
             metadata=metadata,
             consumption=self._consumption,
             learning_id=self.nuclia_learning_id or "",
@@ -467,6 +480,8 @@ class AskResult:
                 self._status = item
             elif isinstance(item, CitationsGenerativeResponse):
                 self._citations = item
+            elif isinstance(item, FootnoteCitationsGenerativeResponse):
+                self._footnote_citations = item
             elif isinstance(item, MetaGenerativeResponse):
                 self._metadata = item
             elif isinstance(item, Consumption):

{nucliadb-6.8.1.post5003.dist-info → nucliadb-6.8.1.post5005.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nucliadb
-Version: 6.8.1.post5003
+Version: 6.8.1.post5005
 Summary: NucliaDB
 Author-email: Nuclia <nucliadb@nuclia.com>
 License-Expression: AGPL-3.0-or-later
@@ -19,13 +19,13 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3 :: Only
 Requires-Python: <4,>=3.9
 Description-Content-Type: text/markdown
-Requires-Dist: nucliadb-telemetry[all]>=6.8.1.post5003
-Requires-Dist: nucliadb-utils[cache,fastapi,storages]>=6.8.1.post5003
-Requires-Dist: nucliadb-protos>=6.8.1.post5003
-Requires-Dist: nucliadb-models>=6.8.1.post5003
-Requires-Dist: nidx-protos>=6.8.1.post5003
+Requires-Dist: nucliadb-telemetry[all]>=6.8.1.post5005
+Requires-Dist: nucliadb-utils[cache,fastapi,storages]>=6.8.1.post5005
+Requires-Dist: nucliadb-protos>=6.8.1.post5005
+Requires-Dist: nucliadb-models>=6.8.1.post5005
+Requires-Dist: nidx-protos>=6.8.1.post5005
 Requires-Dist: nucliadb-admin-assets>=1.0.0.post1224
-Requires-Dist: nuclia-models>=0.47.0
+Requires-Dist: nuclia-models>=0.50.0
 Requires-Dist: uvicorn[standard]
 Requires-Dist: argdantic
 Requires-Dist: aiohttp>=3.11.11

{nucliadb-6.8.1.post5003.dist-info → nucliadb-6.8.1.post5005.dist-info}/RECORD

@@ -267,7 +267,7 @@ nucliadb/search/search/shards.py,sha256=mc5DK-MoCv9AFhlXlOFHbPvetcyNDzTFOJ5rimK8
 nucliadb/search/search/summarize.py,sha256=3lLdwsM28W505bKvmK7JLXmz7kcjd8Hp70LQs391ofY,5087
 nucliadb/search/search/utils.py,sha256=ajRIXfdTF67dBVahQCXW-rSv6gJpUMPt3QhJrWqArTQ,2175
 nucliadb/search/search/chat/__init__.py,sha256=cp15ZcFnHvpcu_5-aK2A4uUyvuZVV_MJn4bIXMa20ks,835
-nucliadb/search/search/chat/ask.py,sha256=Akq5nCl-0P9Hcil2hzuOEIlhEA9IxJtlQo4dz54BG3s,40884
+nucliadb/search/search/chat/ask.py,sha256=A-XKXXMt9i2sw1thuztAVLG2hgP8eJwd3KXEBfA1CQI,41640
 nucliadb/search/search/chat/exceptions.py,sha256=Siy4GXW2L7oPhIR86H3WHBhE9lkV4A4YaAszuGGUf54,1356
 nucliadb/search/search/chat/images.py,sha256=PA8VWxT5_HUGfW1ULhKTK46UBsVyINtWWqEM1ulzX1E,3095
 nucliadb/search/search/chat/prompt.py,sha256=Bk69WVki5XIzXFYO6o1uQw1feHtuMwfEx_A1TT3piR0,54136
@@ -384,8 +384,8 @@ nucliadb/writer/tus/local.py,sha256=7jYa_w9b-N90jWgN2sQKkNcomqn6JMVBOVeDOVYJHto,
 nucliadb/writer/tus/s3.py,sha256=vu1BGg4VqJ_x2P1u2BxqPKlSfw5orT_a3R-Ln5oPUpU,8483
 nucliadb/writer/tus/storage.py,sha256=ToqwjoYnjI4oIcwzkhha_MPxi-k4Jk3Lt55zRwaC1SM,2903
 nucliadb/writer/tus/utils.py,sha256=MSdVbRsRSZVdkaum69_0wku7X3p5wlZf4nr6E0GMKbw,2556
-nucliadb-6.8.1.post5003.dist-info/METADATA,sha256=MMNHSJZnrs5r025lBiBPmfA9tVdqp0bmiatio-nGEpY,4158
-nucliadb-6.8.1.post5003.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nucliadb-6.8.1.post5003.dist-info/entry_points.txt,sha256=XqGfgFDuY3zXQc8ewXM2TRVjTModIq851zOsgrmaXx4,1268
-nucliadb-6.8.1.post5003.dist-info/top_level.txt,sha256=hwYhTVnX7jkQ9gJCkVrbqEG1M4lT2F_iPQND1fCzF80,20
-nucliadb-6.8.1.post5003.dist-info/RECORD,,
+nucliadb-6.8.1.post5005.dist-info/METADATA,sha256=fpf1lfmxPlWps30utgc-SIfOVnxaU70tLyvm1Iclxw8,4158
+nucliadb-6.8.1.post5005.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nucliadb-6.8.1.post5005.dist-info/entry_points.txt,sha256=XqGfgFDuY3zXQc8ewXM2TRVjTModIq851zOsgrmaXx4,1268
+nucliadb-6.8.1.post5005.dist-info/top_level.txt,sha256=hwYhTVnX7jkQ9gJCkVrbqEG1M4lT2F_iPQND1fCzF80,20
+nucliadb-6.8.1.post5005.dist-info/RECORD,,