nucliadb 6.4.0.post4241__py3-none-any.whl → 6.4.0.post4271__py3-none-any.whl

nucliadb/search/api/v1/find.py

@@ -34,6 +34,7 @@ from nucliadb.search.api.v1.utils import fastapi_query
 from nucliadb.search.search import cache
 from nucliadb.search.search.exceptions import InvalidQueryError
 from nucliadb.search.search.find import find
+from nucliadb.search.search.metrics import Metrics
 from nucliadb.search.search.utils import maybe_log_request_payload, min_score_from_query_params
 from nucliadb_models.common import FieldTypeName
 from nucliadb_models.configuration import FindConfig
@@ -231,10 +232,13 @@ async def _find_endpoint(
     try:
         maybe_log_request_payload(kbid, "/find", item)
         with cache.request_caches():
+            metrics = Metrics("find")
             results, incomplete, _ = await find(
-                kbid, item, x_ndb_client, x_nucliadb_user, x_forwarded_for
+                kbid, item, x_ndb_client, x_nucliadb_user, x_forwarded_for, metrics
             )
             response.status_code = 206 if incomplete else 200
+            if item.debug:
+                results.metrics = metrics.to_dict()
             return results
     except KnowledgeBoxNotFound:
         return HTTPClientError(status_code=404, detail="Knowledge Box not found")

nucliadb/search/predict.py

@@ -22,6 +22,7 @@ import json
 import logging
 import os
 import random
+from dataclasses import dataclass
 from enum import Enum
 from typing import Any, AsyncGenerator, Optional
 from unittest.mock import AsyncMock, Mock
@@ -108,7 +109,7 @@ RERANK = "/rerank"
 
 NUCLIA_LEARNING_ID_HEADER = "NUCLIA-LEARNING-ID"
 NUCLIA_LEARNING_MODEL_HEADER = "NUCLIA-LEARNING-MODEL"
-
+NUCLIA_LEARNING_CHAT_HISTORY_HEADER = "NUCLIA-LEARNING-CHAT-HISTORY"
 
 predict_observer = metrics.Observer(
     "predict_engine",
@@ -139,6 +140,12 @@ class AnswerStatusCode(str, Enum):
         }[self]
 
 
+@dataclass
+class RephraseResponse:
+    rephrased_query: str
+    use_chat_history: Optional[bool]
+
+
 async def start_predict_engine():
     if nuclia_settings.dummy_predict:
         predict_util = DummyPredictEngine()
@@ -267,7 +274,7 @@ class PredictEngine:
         return await func(**request_args)
 
     @predict_observer.wrap({"type": "rephrase"})
-    async def rephrase_query(self, kbid: str, item: RephraseModel) -> str:
+    async def rephrase_query(self, kbid: str, item: RephraseModel) -> RephraseResponse:
         try:
             self.check_nua_key_is_configured_for_onprem()
         except NUAKeyMissingError:
@@ -477,9 +484,9 @@ class DummyPredictEngine(PredictEngine):
         response.headers = {NUCLIA_LEARNING_ID_HEADER: DUMMY_LEARNING_ID}
         return response
 
-    async def rephrase_query(self, kbid: str, item: RephraseModel) -> str:
+    async def rephrase_query(self, kbid: str, item: RephraseModel) -> RephraseResponse:
         self.calls.append(("rephrase_query", item))
-        return DUMMY_REPHRASE_QUERY
+        return RephraseResponse(rephrased_query=DUMMY_REPHRASE_QUERY, use_chat_history=None)
 
     async def chat_query_ndjson(
         self, kbid: str, item: ChatModel
@@ -624,7 +631,7 @@ def get_chat_ndjson_generator(
 
 async def _parse_rephrase_response(
     resp: aiohttp.ClientResponse,
-) -> str:
+) -> RephraseResponse:
     """
     Predict api is returning a json payload that is a string with the following format:
     <rephrased_query><status_code>
@@ -632,12 +639,15 @@ async def _parse_rephrase_response(
     it will raise an exception if the status code is not 0
     """
     content = await resp.json()
+
     if content.endswith("0"):
-        return content[:-1]
+        content = content[:-1]
     elif content.endswith("-1"):
         raise RephraseError(content[:-2])
     elif content.endswith("-2"):
         raise RephraseMissingContextError(content[:-2])
-    else:
-        # bw compatibility
-        return content
+
+    use_chat_history = None
+    if NUCLIA_LEARNING_CHAT_HISTORY_HEADER in resp.headers:
+        use_chat_history = resp.headers[NUCLIA_LEARNING_CHAT_HISTORY_HEADER] == "true"
+    return RephraseResponse(rephrased_query=content, use_chat_history=use_chat_history)

nucliadb/search/search/chat/ask.py

@@ -62,7 +62,7 @@ from nucliadb.search.search.exceptions import (
     InvalidQueryError,
 )
 from nucliadb.search.search.graph_strategy import get_graph_results
-from nucliadb.search.search.metrics import RAGMetrics
+from nucliadb.search.search.metrics import AskMetrics, Metrics
 from nucliadb.search.search.query_parser.fetcher import Fetcher
 from nucliadb.search.search.query_parser.parsers.ask import fetcher_for_ask, parse_ask
 from nucliadb.search.search.rank_fusion import WeightedCombSum
@@ -140,7 +140,7 @@ class AskResult:
         prompt_context: PromptContext,
         prompt_context_order: PromptContextOrder,
         auditor: ChatAuditor,
-        metrics: RAGMetrics,
+        metrics: AskMetrics,
         best_matches: list[RetrievalMatch],
         debug_chat_model: Optional[ChatModel],
     ):
@@ -155,7 +155,7 @@ class AskResult:
         self.debug_chat_model = debug_chat_model
         self.prompt_context_order = prompt_context_order
         self.auditor: ChatAuditor = auditor
-        self.metrics: RAGMetrics = metrics
+        self.metrics: AskMetrics = metrics
         self.best_matches: list[RetrievalMatch] = best_matches
 
         # Computed from the predict chat answer stream
@@ -264,18 +264,11 @@ class AskResult:
             audit_answer = self._answer_text.encode("utf-8")
         else:
             audit_answer = json.dumps(self._object.object).encode("utf-8")
-
-        try:
-            rephrase_time = self.metrics.elapsed("rephrase")
-        except KeyError:
-            # Not all ask requests have a rephrase step
-            rephrase_time = None
-
         self.auditor.audit(
             text_answer=audit_answer,
-            generative_answer_time=self.metrics.elapsed("stream_predict_answer"),
+            generative_answer_time=self.metrics["stream_predict_answer"],
             generative_answer_first_chunk_time=self.metrics.get_first_chunk_time() or 0,
-            rephrase_time=rephrase_time,
+            rephrase_time=self.metrics.get("rephrase"),
             status_code=self.status_code,
         )
 
@@ -317,7 +310,8 @@ class AskResult:
                         self.prompt_context, self.prompt_context_order
                     ),
                     "predict_request": predict_request,
-                }
+                },
+                metrics=self.metrics.dump(),
             )
 
     async def json(self) -> str:
@@ -382,6 +376,9 @@ class AskResult:
             response.prompt_context = sorted_prompt_context
             if self.debug_chat_model:
                 response.predict_request = self.debug_chat_model.model_dump(mode="json")
+            response.debug = {
+                "metrics": self.metrics.dump(),
+            }
         return response.model_dump_json(exclude_none=True, by_alias=True)
 
     async def get_relations_results(self) -> Relations:
@@ -481,7 +478,7 @@ async def ask(
     origin: str,
     resource: Optional[str] = None,
 ) -> AskResult:
-    metrics = RAGMetrics()
+    metrics = AskMetrics()
     chat_history = ask_request.chat_history or []
     user_context = ask_request.extra_context or []
     user_query = ask_request.query
@@ -491,14 +488,21 @@ async def ask(
     if len(chat_history) > 0 or len(user_context) > 0:
         try:
             with metrics.time("rephrase"):
-                rephrased_query = await rephrase_query(
+                rephrase_response = await rephrase_query(
                     kbid,
                     chat_history=chat_history,
                     query=user_query,
                     user_id=user_id,
                     user_context=user_context,
                     generative_model=ask_request.generative_model,
+                    chat_history_relevance_threshold=ask_request.chat_history_relevance_threshold,
                 )
+                rephrased_query = rephrase_response.rephrased_query
+                if rephrase_response.use_chat_history is False:
+                    # Ignored if the question is not relevant enough with the chat history
+                    logger.info("Chat history was ignored for this request")
+                    chat_history = []
+
         except RephraseMissingContextError:
             logger.info("Failed to rephrase ask query, using original")
 
@@ -515,12 +519,6 @@ async def ask(
             resource=resource,
         )
     except NoRetrievalResultsError as err:
-        try:
-            rephrase_time = metrics.elapsed("rephrase")
-        except KeyError:
-            # Not all ask requests have a rephrase step
-            rephrase_time = None
-
         maybe_audit_chat(
             kbid=kbid,
             user_id=user_id,
@@ -528,7 +526,7 @@ async def ask(
             origin=origin,
             generative_answer_time=0,
             generative_answer_first_chunk_time=0,
-            rephrase_time=rephrase_time,
+            rephrase_time=metrics.get("rephrase"),
             user_query=user_query,
             rephrased_query=rephrased_query,
             retrieval_rephrase_query=err.main_query.rephrased_query if err.main_query else None,
@@ -709,7 +707,7 @@ async def retrieval_step(
     client_type: NucliaDBClientType,
     user_id: str,
     origin: str,
-    metrics: RAGMetrics,
+    metrics: Metrics,
     resource: Optional[str] = None,
 ) -> RetrievalResults:
     """
@@ -745,7 +743,7 @@ async def retrieval_in_kb(
     client_type: NucliaDBClientType,
     user_id: str,
     origin: str,
-    metrics: RAGMetrics,
+    metrics: Metrics,
 ) -> RetrievalResults:
     prequeries = parse_prequeries(ask_request)
     graph_strategy = parse_graph_strategy(ask_request)
@@ -812,7 +810,7 @@ async def retrieval_in_resource(
     client_type: NucliaDBClientType,
     user_id: str,
     origin: str,
-    metrics: RAGMetrics,
+    metrics: Metrics,
 ) -> RetrievalResults:
     if any(strategy.name == "full_resource" for strategy in ask_request.rag_strategies):
         # Retrieval is not needed if we are chatting on a specific resource and the full_resource strategy is enabled

nucliadb/search/search/chat/query.py

@@ -27,13 +27,13 @@ from nidx_protos.nodereader_pb2 import (
 
 from nucliadb.common.models_utils import to_proto
 from nucliadb.search import logger
-from nucliadb.search.predict import AnswerStatusCode
+from nucliadb.search.predict import AnswerStatusCode, RephraseResponse
 from nucliadb.search.requesters.utils import Method, node_query
 from nucliadb.search.search.chat.exceptions import NoRetrievalResultsError
 from nucliadb.search.search.exceptions import IncompleteFindResultsError
 from nucliadb.search.search.find import find
 from nucliadb.search.search.merge import merge_relations_results
-from nucliadb.search.search.metrics import RAGMetrics
+from nucliadb.search.search.metrics import Metrics
 from nucliadb.search.search.query_parser.models import ParsedQuery, Query, RelationQuery, UnitRetrieval
 from nucliadb.search.search.query_parser.parsers.unit_retrieval import convert_retrieval_to_proto
 from nucliadb.search.settings import settings
@@ -71,7 +71,8 @@ async def rephrase_query(
     user_id: str,
     user_context: list[str],
     generative_model: Optional[str] = None,
-) -> str:
+    chat_history_relevance_threshold: Optional[float] = None,
+) -> RephraseResponse:
     predict = get_predict()
     req = RephraseModel(
         question=query,
@@ -79,6 +80,7 @@ async def rephrase_query(
         user_id=user_id,
         user_context=user_context,
         generative_model=generative_model,
+        chat_history_relevance_threshold=chat_history_relevance_threshold,
     )
     return await predict.rephrase_query(kbid, req)
 
@@ -91,7 +93,7 @@ async def get_find_results(
     ndb_client: NucliaDBClientType,
     user: str,
     origin: str,
-    metrics: RAGMetrics = RAGMetrics(),
+    metrics: Metrics,
     prequeries_strategy: Optional[PreQueriesStrategy] = None,
 ) -> tuple[KnowledgeboxFindResults, Optional[list[PreQueryResult]], ParsedQuery]:
     prequeries_results = None
@@ -108,7 +110,7 @@ async def get_find_results(
                     x_ndb_client=ndb_client,
                     x_nucliadb_user=user,
                     x_forwarded_for=origin,
-                    metrics=metrics,
+                    metrics=metrics.child_span("prefilters"),
                 )
                 prefilter_matching_resources = {
                     resource
@@ -133,8 +135,7 @@ async def get_find_results(
                     x_ndb_client=ndb_client,
                     x_nucliadb_user=user,
                     x_forwarded_for=origin,
-                    generative_model=item.generative_model,
-                    metrics=metrics,
+                    metrics=metrics.child_span("prequeries"),
                 )
 
         prequeries_results = (prefilter_queries_results or []) + (queries_results or [])
@@ -147,7 +148,7 @@ async def get_find_results(
             ndb_client,
             user,
             origin,
-            metrics=metrics,
+            metrics=metrics.child_span("main_query"),
         )
     return main_results, prequeries_results, query_parser
 
@@ -223,7 +224,7 @@ async def run_main_query(
     ndb_client: NucliaDBClientType,
     user: str,
     origin: str,
-    metrics: RAGMetrics = RAGMetrics(),
+    metrics: Metrics,
 ) -> tuple[KnowledgeboxFindResults, ParsedQuery]:
     find_request = find_request_from_ask_request(item, query)
 
@@ -455,8 +456,7 @@ async def run_prequeries(
     x_ndb_client: NucliaDBClientType,
     x_nucliadb_user: str,
     x_forwarded_for: str,
-    generative_model: Optional[str] = None,
-    metrics: RAGMetrics = RAGMetrics(),
+    metrics: Metrics,
 ) -> list[PreQueryResult]:
     """
     Runs simultaneous find requests for each prequery and returns the merged results according to the normalized weights.
@@ -464,23 +464,22 @@ async def run_prequeries(
     results: list[PreQueryResult] = []
     max_parallel_prequeries = asyncio.Semaphore(settings.prequeries_max_parallel)
 
-    async def _prequery_find(
-        prequery: PreQuery,
-    ):
+    async def _prequery_find(prequery: PreQuery, index: int):
         async with max_parallel_prequeries:
+            prequery_id = prequery.id or f"prequery-{index}"
             find_results, _, _ = await find(
                 kbid,
                 prequery.request,
                 x_ndb_client,
                 x_nucliadb_user,
                 x_forwarded_for,
-                metrics=metrics,
+                metrics=metrics.child_span(prequery_id),
             )
             return prequery, find_results
 
     ops = []
-    for prequery in prequeries:
-        ops.append(asyncio.create_task(_prequery_find(prequery)))
+    for idx, prequery in enumerate(prequeries):
+        ops.append(asyncio.create_task(_prequery_find(prequery, idx)))
     ops_results = await asyncio.gather(*ops)
     for prequery, find_results in ops_results:
         results.append((prequery, find_results))

nucliadb/search/search/find.py

@@ -34,7 +34,7 @@ from nucliadb.search.search.hydrator import (
     TextBlockHydrationOptions,
 )
 from nucliadb.search.search.metrics import (
-    RAGMetrics,
+    Metrics,
 )
 from nucliadb.search.search.query_parser.models import ParsedQuery
 from nucliadb.search.search.query_parser.parsers import parse_find
@@ -64,7 +64,7 @@ async def find(
     x_ndb_client: NucliaDBClientType,
     x_nucliadb_user: str,
     x_forwarded_for: str,
-    metrics: RAGMetrics = RAGMetrics(),
+    metrics: Metrics,
 ) -> tuple[KnowledgeboxFindResults, bool, ParsedQuery]:
     external_index_manager = await get_external_index_manager(kbid=kbid)
     if external_index_manager is not None:
@@ -85,7 +85,7 @@ async def _index_node_retrieval(
     x_ndb_client: NucliaDBClientType,
     x_nucliadb_user: str,
     x_forwarded_for: str,
-    metrics: RAGMetrics = RAGMetrics(),
+    metrics: Metrics,
 ) -> tuple[KnowledgeboxFindResults, bool, ParsedQuery]:
     audit = get_audit()
     start_time = time()
@@ -104,7 +104,7 @@ async def _index_node_retrieval(
             rephrased_query,
         ) = await legacy_convert_retrieval_to_proto(parsed)
 
-    with metrics.time("node_query"):
+    with metrics.time("index_search"):
         results, query_incomplete_results, queried_shards = await node_query(
             kbid, Method.SEARCH, pb_query
         )
@@ -142,8 +142,8 @@ async def _index_node_retrieval(
     search_results.shards = queried_shards
     search_results.autofilters = autofilters
 
-    ndb_time = metrics.elapsed("node_query") + metrics.elapsed("results_merge")
-    if metrics.elapsed("node_query") > settings.slow_node_query_log_threshold:
+    ndb_time = metrics["index_search"] + metrics["results_merge"]
+    if metrics["index_search"] > settings.slow_node_query_log_threshold:
         logger.warning(
             "Slow nidx query",
             extra={
@@ -152,7 +152,7 @@ async def _index_node_retrieval(
                 "client": x_ndb_client,
                 "query": item.model_dump_json(),
                 "time": search_time,
-                "durations": metrics.steps(),
+                "metrics": metrics.to_dict(),
             },
         )
     elif ndb_time > settings.slow_find_log_threshold:
@@ -164,7 +164,7 @@ async def _index_node_retrieval(
                 "client": x_ndb_client,
                 "query": item.model_dump_json(),
                 "time": search_time,
-                "durations": metrics.steps(),
+                "metrics": metrics.to_dict(),
             },
         )
 

nucliadb/search/search/graph_strategy.py

@@ -43,7 +43,7 @@ from nucliadb.search.search.find_merge import (
     hydrate_and_rerank,
 )
 from nucliadb.search.search.hydrator import ResourceHydrationOptions, TextBlockHydrationOptions
-from nucliadb.search.search.metrics import RAGMetrics
+from nucliadb.search.search.metrics import Metrics
 from nucliadb.search.search.rerankers import (
     Reranker,
     RerankingOptions,
@@ -305,8 +305,8 @@ async def get_graph_results(
     origin: str,
     graph_strategy: GraphStrategy,
     text_block_reranker: Reranker,
+    metrics: Metrics,
     generative_model: Optional[str] = None,
-    metrics: RAGMetrics = RAGMetrics(),
     shards: Optional[list[str]] = None,
 ) -> tuple[KnowledgeboxFindResults, FindRequest]:
     relations = Relations(entities={})

nucliadb/search/search/metrics.py

@@ -19,7 +19,7 @@
 #
 import contextlib
 import time
-from typing import Optional
+from typing import Any, Optional, Union
 
 from nucliadb_telemetry import metrics
 
@@ -58,27 +58,55 @@ rag_histogram = metrics.Histogram(
     buckets=buckets,
 )
 
+MetricsData = dict[str, Union[int, float]]
 
-class RAGMetrics:
-    def __init__(self: "RAGMetrics"):
-        self.global_start = time.monotonic()
-        self._start_times: dict[str, float] = {}
-        self._end_times: dict[str, float] = {}
-        self.first_chunk_yielded_at: Optional[float] = None
+
+class Metrics:
+    def __init__(self: "Metrics", id: str):
+        self.id = id
+        self.child_spans: list[Metrics] = []
+        self._metrics: MetricsData = {}
 
     @contextlib.contextmanager
     def time(self, step: str):
-        self._start(step)
+        start_time = time.monotonic()
         try:
             yield
         finally:
-            self._end(step)
+            elapsed = time.monotonic() - start_time
+            self._metrics[step] = elapsed
+            rag_histogram.observe(elapsed, labels={"step": step})
+
+    def child_span(self, id: str) -> "Metrics":
+        child_span = Metrics(id)
+        self.child_spans.append(child_span)
+        return child_span
+
+    def set(self, key: str, value: Union[int, float]):
+        self._metrics[key] = value
+
+    def get(self, key: str) -> Optional[Union[int, float]]:
+        return self._metrics.get(key)
+
+    def to_dict(self) -> MetricsData:
+        return self._metrics
 
-    def steps(self) -> dict[str, float]:
-        return {step: self.elapsed(step) for step in self._end_times.keys()}
+    def dump(self) -> dict[str, Any]:
+        result = {}
+        for child in self.child_spans:
+            result.update(child.dump())
+        result[self.id] = self.to_dict()
+        return result
 
-    def elapsed(self, step: str) -> float:
-        return self._end_times[step] - self._start_times[step]
+    def __getitem__(self, key: str) -> Union[int, float]:
+        return self._metrics[key]
+
+
+class AskMetrics(Metrics):
+    def __init__(self: "AskMetrics"):
+        super().__init__(id="ask")
+        self.global_start = time.monotonic()
+        self.first_chunk_yielded_at: Optional[float] = None
 
     def record_first_chunk_yielded(self):
         self.first_chunk_yielded_at = time.monotonic()
@@ -88,11 +116,3 @@ class RAGMetrics:
         if self.first_chunk_yielded_at is None:
             return None
         return self.first_chunk_yielded_at - self.global_start
-
-    def _start(self, step: str):
-        self._start_times[step] = time.monotonic()
-
-    def _end(self, step: str):
-        self._end_times[step] = time.monotonic()
-        elapsed = self.elapsed(step)
-        rag_histogram.observe(elapsed, labels={"step": step})

{nucliadb-6.4.0.post4241.dist-info → nucliadb-6.4.0.post4271.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nucliadb
-Version: 6.4.0.post4241
+Version: 6.4.0.post4271
 Summary: NucliaDB
 Author-email: Nuclia <nucliadb@nuclia.com>
 License: AGPL
@@ -20,11 +20,11 @@ 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.4.0.post4241
-Requires-Dist: nucliadb-utils[cache,fastapi,storages]>=6.4.0.post4241
-Requires-Dist: nucliadb-protos>=6.4.0.post4241
-Requires-Dist: nucliadb-models>=6.4.0.post4241
-Requires-Dist: nidx-protos>=6.4.0.post4241
+Requires-Dist: nucliadb-telemetry[all]>=6.4.0.post4271
+Requires-Dist: nucliadb-utils[cache,fastapi,storages]>=6.4.0.post4271
+Requires-Dist: nucliadb-protos>=6.4.0.post4271
+Requires-Dist: nucliadb-models>=6.4.0.post4271
+Requires-Dist: nidx-protos>=6.4.0.post4271
 Requires-Dist: nucliadb-admin-assets>=1.0.0.post1224
 Requires-Dist: nuclia-models>=0.24.2
 Requires-Dist: uvicorn[standard]

{nucliadb-6.4.0.post4241.dist-info → nucliadb-6.4.0.post4271.dist-info}/RECORD

@@ -204,7 +204,7 @@ nucliadb/search/__init__.py,sha256=tnypbqcH4nBHbGpkINudhKgdLKpwXQCvDtPchUlsyY4,1
 nucliadb/search/app.py,sha256=-WEX1AZRA8R_9aeOo9ovOTwjXW_7VfwWN7N2ccSoqXg,3387
 nucliadb/search/lifecycle.py,sha256=hiylV-lxsAWkqTCulXBg0EIfMQdejSr8Zar0L_GLFT8,2218
 nucliadb/search/openapi.py,sha256=t3Wo_4baTrfPftg2BHsyLWNZ1MYn7ZRdW7ht-wFOgRs,1016
-nucliadb/search/predict.py,sha256=BYkKL2-3-MNT8JnE7y7XTEMKMnynUm2y4VJZP1jRjdQ,22987
+nucliadb/search/predict.py,sha256=__0qwIU2CIRYRTYsbG9zZEjXXrxNe8puZWYJIyOT6dg,23492
 nucliadb/search/predict_models.py,sha256=ZAe0dneUsPmV9uBar57cCFADCGOrYDsJHuqKlA5zWag,5937
 nucliadb/search/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nucliadb/search/run.py,sha256=aFb-CXRi_C8YMpP_ivNj8KW1BYhADj88y8K9Lr_nUPI,1402
@@ -215,7 +215,7 @@ nucliadb/search/api/v1/__init__.py,sha256=DH16OYnw9jQ38OpKlmdXeoq2j40ZPXZRtGvClK
 nucliadb/search/api/v1/ask.py,sha256=b4tz33HNsfT5DXv_2DMc_jirnFsHuobreWkbAKkzj5o,5337
 nucliadb/search/api/v1/catalog.py,sha256=Nw4wIj4AjGp-p64FFVQFN4v2LFcV3A0UJIxfo3_XGmY,7670
 nucliadb/search/api/v1/feedback.py,sha256=kNLc4dHz2SXHzV0PwC1WiRAwY88fDptPcP-kO0q-FrQ,2620
-nucliadb/search/api/v1/find.py,sha256=JFbGDRFBHBTApYR1qHp9RngbE_QDb96fXORMdjcN6lg,10654
+nucliadb/search/api/v1/find.py,sha256=C4sTGFRS9tQFF8v1zhnHQvnExJoGDYi78bZTRfwhGrc,10831
 nucliadb/search/api/v1/graph.py,sha256=ItVpzJbqfDLjoIo2fTb2mKGCM1Z34sx7CBb3gNmj6IQ,4274
 nucliadb/search/api/v1/knowledgebox.py,sha256=e9xeLPUqnQTx33i4A8xuV93ENvtJGrpjPlLRbGJtAI8,8415
 nucliadb/search/api/v1/predict_proxy.py,sha256=Q03ZTvWp7Sq0x71t5Br4LHxTiYsRd6-GCb4YuKqhynM,3131
@@ -237,14 +237,14 @@ nucliadb/search/search/cut.py,sha256=ytY0_GY7ocNjfxTb4aosxEp4ZfhQNDP--JkhEMGD298
 nucliadb/search/search/exceptions.py,sha256=klGLgAGGrXcSGix_W6418ZBMqDchAIGjN77ofkOScEI,1039
 nucliadb/search/search/fetch.py,sha256=eiljOKim-4OOEZn-3fyVZSYxztCH156BXYdqlIwVdN4,6181
 nucliadb/search/search/filters.py,sha256=1MkHlJjAQqoRCj7e5cEzK2HvBxGLE17I_omsjiklbtw,6476
-nucliadb/search/search/find.py,sha256=WTc_0HdbaU0Mwkmlf9s4pCmTuU0hz1jetBjIpNLXEEM,7982
+nucliadb/search/search/find.py,sha256=i1auc8visRakBwbbZGhyQgXNAmsaAVheisYi2xGjdKY,7925
 nucliadb/search/search/find_merge.py,sha256=c-7IlfjfdmWAvQOyM7IO3bKS1EQpnR4oi6pN6mwrQKw,19815
 nucliadb/search/search/graph_merge.py,sha256=y5V7X-BhjHsKDXE69tzQLIIKGm4XuaFrZXw0odcHVNM,3402
-nucliadb/search/search/graph_strategy.py,sha256=JsV-i9PGhekCAzmGpqeueQIitJb7fWCihIwUf76Q3pU,32912
+nucliadb/search/search/graph_strategy.py,sha256=zYfi1df982ZYOFtYSksnHEJvQn-ZZsCIFSruVZP_934,32891
 nucliadb/search/search/hydrator.py,sha256=-R37gCrGxkyaiHQalnTWHNG_FCx11Zucd7qA1vQCxuw,6985
 nucliadb/search/search/ingestion_agents.py,sha256=NeJr4EEX-bvFFMGvXOOwLv8uU7NuQ-ntJnnrhnKfMzY,3174
 nucliadb/search/search/merge.py,sha256=Abg9YblQJvH2jDvXVT45MNxaIpNa7TTpsiUSJqb3NDc,23307
-nucliadb/search/search/metrics.py,sha256=HJVQPLOIwLuc733G4keqEgx1-Dcg97hyWGKZQTojiSE,2973
+nucliadb/search/search/metrics.py,sha256=3I6IN0qDSmqIvUaWJmT3rt-Jyjs6LcvnKI8ZqCiuJPY,3501
 nucliadb/search/search/paragraphs.py,sha256=pNAEiYqJGGUVcEf7xf-PFMVqz0PX4Qb-WNG-_zPGN2o,7799
 nucliadb/search/search/pgcatalog.py,sha256=s_J98fsX_RuFXwpejpkGqG-tD9ELuzz4YQ6U3ew5h2g,9313
 nucliadb/search/search/predict_proxy.py,sha256=JwgBeEg1j4LnCjPCvTUrnmOd9LceJAt3iAu4m9cmJBo,3390
@@ -255,11 +255,11 @@ nucliadb/search/search/shards.py,sha256=mc5DK-MoCv9AFhlXlOFHbPvetcyNDzTFOJ5rimK8
 nucliadb/search/search/summarize.py,sha256=ksmYPubEQvAQgfPdZHfzB_rR19B2ci4IYZ6jLdHxZo8,4996
 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=jYOGh2rySV4aFx_D2KlNVbPXHBsbkcy0Ve-eBS7CSYc,37611
+nucliadb/search/search/chat/ask.py,sha256=4cpeWC7Q4NI30vrOeq22N0Vw0PQS0Ko9b47R5ISlgUw,37833
 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=Jnja-Ss7skgnnDY8BymVfdeYsFPnIQFL8tEvcRXTKUE,47356
-nucliadb/search/search/chat/query.py,sha256=IdVPeKLUbq4hWJ81LePWdUrljeyehnIXg-Ars-37msQ,16878
+nucliadb/search/search/chat/query.py,sha256=6v6twBUTWfUUzklVV6xqJSYPkAshnIrBH9wbTcjQvkI,17063
 nucliadb/search/search/query_parser/__init__.py,sha256=cp15ZcFnHvpcu_5-aK2A4uUyvuZVV_MJn4bIXMa20ks,835
 nucliadb/search/search/query_parser/exceptions.py,sha256=szAOXUZ27oNY-OSa9t2hQ5HHkQQC0EX1FZz_LluJHJE,1224
 nucliadb/search/search/query_parser/fetcher.py,sha256=SkvBRDfSKmuz-QygNKLAU4AhZhhDo1dnOZmt1zA28RA,16851
@@ -368,8 +368,8 @@ nucliadb/writer/tus/local.py,sha256=7jYa_w9b-N90jWgN2sQKkNcomqn6JMVBOVeDOVYJHto,
 nucliadb/writer/tus/s3.py,sha256=vF0NkFTXiXhXq3bCVXXVV-ED38ECVoUeeYViP8uMqcU,8357
 nucliadb/writer/tus/storage.py,sha256=ToqwjoYnjI4oIcwzkhha_MPxi-k4Jk3Lt55zRwaC1SM,2903
 nucliadb/writer/tus/utils.py,sha256=MSdVbRsRSZVdkaum69_0wku7X3p5wlZf4nr6E0GMKbw,2556
-nucliadb-6.4.0.post4241.dist-info/METADATA,sha256=twJEtMMEkGzMqaXb2hvjgRRgPmvl4ESDTJuUw3ObLuk,4223
-nucliadb-6.4.0.post4241.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
-nucliadb-6.4.0.post4241.dist-info/entry_points.txt,sha256=XqGfgFDuY3zXQc8ewXM2TRVjTModIq851zOsgrmaXx4,1268
-nucliadb-6.4.0.post4241.dist-info/top_level.txt,sha256=hwYhTVnX7jkQ9gJCkVrbqEG1M4lT2F_iPQND1fCzF80,20
-nucliadb-6.4.0.post4241.dist-info/RECORD,,
+nucliadb-6.4.0.post4271.dist-info/METADATA,sha256=DKjN0N70XgUrzoOshpgsNU9xMWIweEdjtQgziYbveU0,4223
+nucliadb-6.4.0.post4271.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
+nucliadb-6.4.0.post4271.dist-info/entry_points.txt,sha256=XqGfgFDuY3zXQc8ewXM2TRVjTModIq851zOsgrmaXx4,1268
+nucliadb-6.4.0.post4271.dist-info/top_level.txt,sha256=hwYhTVnX7jkQ9gJCkVrbqEG1M4lT2F_iPQND1fCzF80,20
+nucliadb-6.4.0.post4271.dist-info/RECORD,,

{nucliadb-6.4.0.post4241.dist-info → nucliadb-6.4.0.post4271.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.3.1)
+Generator: setuptools (80.4.0)
 Root-Is-Purelib: true
 Tag: py3-none-any