nucliadb 6.4.0.post4227__py3-none-any.whl → 6.4.0.post4265__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/api/v1/predict_proxy.py

@@ -21,7 +21,7 @@ import json
 from typing import Union
 
 from fastapi import Request
-from fastapi.responses import JSONResponse, StreamingResponse
+from fastapi.responses import Response, StreamingResponse
 from fastapi_versioning import version
 
 from nucliadb.common.datamanagers.exceptions import KnowledgeBoxNotFound
@@ -58,14 +58,19 @@ async def predict_proxy_endpoint(
     request: Request,
     kbid: str,
     endpoint: PredictProxiedEndpoints,
-) -> Union[JSONResponse, StreamingResponse, HTTPClientError]:
+) -> Union[Response, StreamingResponse, HTTPClientError]:
     try:
         payload = await request.json()
     except json.JSONDecodeError:
         payload = None
     try:
         return await predict_proxy(
-            kbid, endpoint, request.method, params=request.query_params, json=payload
+            kbid,
+            endpoint,
+            request.method,
+            params=request.query_params,
+            json=payload,
+            headers=dict(request.headers),
         )
     except KnowledgeBoxNotFound:
         return HTTPClientError(status_code=404, detail="Knowledge box not found")

nucliadb/search/predict.py

@@ -262,7 +262,7 @@ class PredictEngine:
         jitter=backoff.random_jitter,
         max_tries=MAX_TRIES,
     )
-    async def make_request(self, method: str, **request_args):
+    async def make_request(self, method: str, **request_args) -> aiohttp.ClientResponse:
         func = getattr(self.session, method.lower())
         return await func(**request_args)
 
@@ -311,8 +311,8 @@ class PredictEngine:
             timeout=None,
         )
         await self.check_response(kbid, resp, expected_status=200)
-        ident = resp.headers.get(NUCLIA_LEARNING_ID_HEADER)
-        model = resp.headers.get(NUCLIA_LEARNING_MODEL_HEADER)
+        ident = resp.headers.get(NUCLIA_LEARNING_ID_HEADER) or "unknown"
+        model = resp.headers.get(NUCLIA_LEARNING_MODEL_HEADER) or "unknown"
         return ident, model, get_chat_ndjson_generator(resp)
 
     @predict_observer.wrap({"type": "query"})
@@ -471,7 +471,9 @@ class DummyPredictEngine(PredictEngine):
 
     async def make_request(self, method: str, **request_args):
         response = Mock(status=200)
-        response.json = AsyncMock(return_value={"foo": "bar"})
+        json_data = {"foo": "bar"}
+        response.json = AsyncMock(return_value=json_data)
+        response.read = AsyncMock(return_value=json.dumps(json_data).encode("utf-8"))
         response.headers = {NUCLIA_LEARNING_ID_HEADER: DUMMY_LEARNING_ID}
         return response
 

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
@@ -515,12 +512,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 +519,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 +700,7 @@ async def retrieval_step(
     client_type: NucliaDBClientType,
     user_id: str,
     origin: str,
-    metrics: RAGMetrics,
+    metrics: Metrics,
     resource: Optional[str] = None,
 ) -> RetrievalResults:
     """
@@ -745,7 +736,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 +803,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

@@ -33,7 +33,7 @@ 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
@@ -91,7 +91,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 +108,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 +133,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 +146,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 +222,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 +454,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 +462,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/search/search/predict_proxy.py

@@ -21,7 +21,7 @@ from enum import Enum
 from typing import Any, Optional, Union
 
 from fastapi.datastructures import QueryParams
-from fastapi.responses import JSONResponse, StreamingResponse
+from fastapi.responses import Response, StreamingResponse
 
 from nucliadb.common import datamanagers
 from nucliadb.search.predict import PredictEngine
@@ -42,20 +42,25 @@ class PredictProxiedEndpoints(str, Enum):
     REMI = "remi"
 
 
+ALLOWED_HEADERS = [
+    "Accept",  # To allow 'application/x-ndjson' on the /chat endpoint
+]
+
+
 async def predict_proxy(
     kbid: str,
     endpoint: PredictProxiedEndpoints,
     method: str,
     params: QueryParams,
     json: Optional[Any] = None,
-) -> Union[JSONResponse, StreamingResponse]:
+    headers: dict[str, str] = {},
+) -> Union[Response, StreamingResponse]:
     if not await exists_kb(kbid=kbid):
         raise datamanagers.exceptions.KnowledgeBoxNotFound()
 
     predict: PredictEngine = get_predict()
-
-    # Add KB configuration headers
-    headers = predict.get_predict_headers(kbid)
+    predict_headers = predict.get_predict_headers(kbid)
+    user_headers = {k: v for k, v in headers.items() if k.capitalize() in ALLOWED_HEADERS}
 
     # Proxy the request to predict API
     predict_response = await predict.make_request(
@@ -63,22 +68,24 @@ async def predict_proxy(
         url=predict.get_predict_url(endpoint, kbid),
         json=json,
         params=params,
-        headers=headers,
+        headers={**user_headers, **predict_headers},
     )
 
     # Proxy the response back to the client
     status_code = predict_response.status
-    response: Union[JSONResponse, StreamingResponse]
+    media_type = predict_response.headers.get("Content-Type")
+    response: Union[Response, StreamingResponse]
     if predict_response.headers.get("Transfer-Encoding") == "chunked":
         response = StreamingResponse(
             content=predict_response.content.iter_any(),
             status_code=status_code,
-            media_type=predict_response.headers.get("Content-Type"),
+            media_type=media_type,
         )
     else:
-        response = JSONResponse(
-            content=await predict_response.json(),
+        response = Response(
+            content=await predict_response.read(),
             status_code=status_code,
+            media_type=media_type,
         )
     nuclia_learning_id = predict_response.headers.get("NUCLIA-LEARNING-ID")
     if nuclia_learning_id:

{nucliadb-6.4.0.post4227.dist-info → nucliadb-6.4.0.post4265.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nucliadb
-Version: 6.4.0.post4227
+Version: 6.4.0.post4265
 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.post4227
-Requires-Dist: nucliadb-utils[cache,fastapi,storages]>=6.4.0.post4227
-Requires-Dist: nucliadb-protos>=6.4.0.post4227
-Requires-Dist: nucliadb-models>=6.4.0.post4227
-Requires-Dist: nidx-protos>=6.4.0.post4227
+Requires-Dist: nucliadb-telemetry[all]>=6.4.0.post4265
+Requires-Dist: nucliadb-utils[cache,fastapi,storages]>=6.4.0.post4265
+Requires-Dist: nucliadb-protos>=6.4.0.post4265
+Requires-Dist: nucliadb-models>=6.4.0.post4265
+Requires-Dist: nidx-protos>=6.4.0.post4265
 Requires-Dist: nucliadb-admin-assets>=1.0.0.post1224
 Requires-Dist: nuclia-models>=0.24.2
 Requires-Dist: uvicorn[standard]

{nucliadb-6.4.0.post4227.dist-info → nucliadb-6.4.0.post4265.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=VJr5Itx8FE7CZIGYcP-fRgd2YGxAnP9Qj9NxiwWiwcc,22819
+nucliadb/search/predict.py,sha256=BYkKL2-3-MNT8JnE7y7XTEMKMnynUm2y4VJZP1jRjdQ,22987
 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,10 +215,10 @@ 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=QrGzo0hKjtmyGZ6pjlJHYAh4hxwVUIOTcVcerRCw7eE,3047
+nucliadb/search/api/v1/predict_proxy.py,sha256=Q03ZTvWp7Sq0x71t5Br4LHxTiYsRd6-GCb4YuKqhynM,3131
 nucliadb/search/api/v1/router.py,sha256=mtT07rBZcVfpa49doaw9b1tj3sdi3qLH0gn9Io6NYM0,988
 nucliadb/search/api/v1/search.py,sha256=Or-mUvmBAyh0Y55NqTYNXe_BWR0lLLaTSL2ChjJaE2M,12402
 nucliadb/search/api/v1/suggest.py,sha256=Em7ApddZNHMHjL_ZfXmUIVUk504f58J96JlxJXnIxaM,6438
@@ -237,17 +237,17 @@ 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=IFI3v_ODz2_UU1XZnyaD391fE7-2C0npSmj_HmDvzS4,3123
+nucliadb/search/search/predict_proxy.py,sha256=JwgBeEg1j4LnCjPCvTUrnmOd9LceJAt3iAu4m9cmJBo,3390
 nucliadb/search/search/query.py,sha256=-gvKsyGmKYpsoEVzKkq3HJUMcs_3LD3TYUueOcJsTec,11511
 nucliadb/search/search/rank_fusion.py,sha256=xZtXhbmKb_56gs73u6KkFm2efvTATOSMmpOV2wrAIqE,9613
 nucliadb/search/search/rerankers.py,sha256=PvhExUb8zZYghiFHRgGotw6h6bU--Rft09wE8arvtAw,7424
@@ -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=tkU431_tZpFQ8Au9RpGrHO78D0vCKE-VaBT5BNTL0pA,37393
 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=yK7J8vwJT7gKeBPn21uYqQw0tEJJ_HRtGLSt2bnJEuQ,16895
 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.post4227.dist-info/METADATA,sha256=eZVBfu03hI3bpE7r2yLzoiPcUtqH82BsKHO51-dFfN0,4223
-nucliadb-6.4.0.post4227.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
-nucliadb-6.4.0.post4227.dist-info/entry_points.txt,sha256=XqGfgFDuY3zXQc8ewXM2TRVjTModIq851zOsgrmaXx4,1268
-nucliadb-6.4.0.post4227.dist-info/top_level.txt,sha256=hwYhTVnX7jkQ9gJCkVrbqEG1M4lT2F_iPQND1fCzF80,20
-nucliadb-6.4.0.post4227.dist-info/RECORD,,
+nucliadb-6.4.0.post4265.dist-info/METADATA,sha256=DdUvK3SLFsfNpySAdkMKFq9721wSRZ-R0n7fFHGEpxQ,4223
+nucliadb-6.4.0.post4265.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
+nucliadb-6.4.0.post4265.dist-info/entry_points.txt,sha256=XqGfgFDuY3zXQc8ewXM2TRVjTModIq851zOsgrmaXx4,1268
+nucliadb-6.4.0.post4265.dist-info/top_level.txt,sha256=hwYhTVnX7jkQ9gJCkVrbqEG1M4lT2F_iPQND1fCzF80,20
+nucliadb-6.4.0.post4265.dist-info/RECORD,,

{nucliadb-6.4.0.post4227.dist-info → nucliadb-6.4.0.post4265.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