nucliadb-utils 5.0.1.post1139__py3-none-any.whl → 5.0.1.post1141__py3-none-any.whl

nucliadb_utils/audit/audit.py

@@ -21,11 +21,7 @@ from typing import List, Optional
 
 from google.protobuf.timestamp_pb2 import Timestamp
 
-from nucliadb_protos.audit_pb2 import (
-    AuditField,
-    AuditRequest,
-    ChatContext,
-)
+from nucliadb_protos.audit_pb2 import AuditField, AuditRequest, ChatContext, RetrievedContext
 from nucliadb_protos.nodereader_pb2 import SearchRequest
 from nucliadb_protos.resources_pb2 import FieldID
 
@@ -85,7 +81,8 @@ class AuditStorage:
         origin: str,
         question: str,
         rephrased_question: Optional[str],
-        context: List[ChatContext],
+        chat_context: List[ChatContext],
+        retrieved_context: List[RetrievedContext],
         answer: Optional[str],
         learning_id: str,
         rephrase_time: Optional[float] = None,

nucliadb_utils/audit/basic.py

@@ -21,11 +21,7 @@ from typing import List, Optional
 
 from google.protobuf.timestamp_pb2 import Timestamp
 
-from nucliadb_protos.audit_pb2 import (
-    AuditField,
-    AuditRequest,
-    ChatContext,
-)
+from nucliadb_protos.audit_pb2 import AuditField, AuditRequest, ChatContext, RetrievedContext
 from nucliadb_protos.nodereader_pb2 import SearchRequest
 from nucliadb_protos.resources_pb2 import FieldID
 from nucliadb_protos.writer_pb2 import BrokerMessage
@@ -86,7 +82,8 @@ class BasicAuditStorage(AuditStorage):
         origin: str,
         question: str,
         rephrased_question: Optional[str],
-        context: List[ChatContext],
+        chat_context: List[ChatContext],
+        retrieved_context: List[RetrievedContext],
         answer: Optional[str],
         learning_id: str,
         rephrase_time: Optional[float] = None,

nucliadb_utils/audit/stream.py

@@ -33,12 +33,7 @@ from starlette.middleware.base import BaseHTTPMiddleware, RequestResponseEndpoin
 from starlette.responses import Response, StreamingResponse
 from starlette.types import ASGIApp
 
-from nucliadb_protos.audit_pb2 import (
-    AuditField,
-    AuditRequest,
-    ChatContext,
-    ClientType,
-)
+from nucliadb_protos.audit_pb2 import AuditField, AuditRequest, ChatContext, ClientType, RetrievedContext
 from nucliadb_protos.nodereader_pb2 import SearchRequest
 from nucliadb_protos.resources_pb2 import FieldID
 from nucliadb_utils import logger
@@ -380,7 +375,8 @@ class StreamAuditStorage(AuditStorage):
         origin: str,
         question: str,
         rephrased_question: Optional[str],
-        context: List[ChatContext],
+        chat_context: List[ChatContext],
+        retrieved_context: List[RetrievedContext],
         answer: Optional[str],
         learning_id: str,
         rephrase_time: Optional[float] = None,
@@ -405,7 +401,8 @@ class StreamAuditStorage(AuditStorage):
             auditrequest.generative_answer_first_chunk_time = generative_answer_first_chunk_time
         auditrequest.type = AuditRequest.CHAT
         auditrequest.chat.question = question
-        auditrequest.chat.context.extend(context)
+        auditrequest.chat.chat_context.extend(chat_context)
+        auditrequest.chat.retrieved_context.extend(retrieved_context)
         auditrequest.chat.learning_id = learning_id
         if rephrased_question is not None:
             auditrequest.chat.rephrased_question = rephrased_question

{nucliadb_utils-5.0.1.post1139.dist-info → nucliadb_utils-5.0.1.post1141.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nucliadb_utils
-Version: 5.0.1.post1139
+Version: 5.0.1.post1141
 Home-page: https://nuclia.com
 License: BSD
 Classifier: Development Status :: 4 - Beta
@@ -24,8 +24,8 @@ Requires-Dist: PyNaCl
 Requires-Dist: pyjwt>=2.4.0
 Requires-Dist: memorylru>=1.1.2
 Requires-Dist: mrflagly>=0.2.9
-Requires-Dist: nucliadb-protos>=5.0.1.post1139
-Requires-Dist: nucliadb-telemetry>=5.0.1.post1139
+Requires-Dist: nucliadb-protos>=5.0.1.post1141
+Requires-Dist: nucliadb-telemetry>=5.0.1.post1141
 Provides-Extra: cache
 Requires-Dist: redis>=4.3.4; extra == "cache"
 Requires-Dist: orjson>=3.6.7; extra == "cache"

{nucliadb_utils-5.0.1.post1139.dist-info → nucliadb_utils-5.0.1.post1141.dist-info}/RECORD

@@ -22,9 +22,9 @@ nucliadb_utils/aiopynecone/client.py,sha256=3DGJuHXO9Vm83atMLD4fd2FTz8SAr8RvPCGU
 nucliadb_utils/aiopynecone/exceptions.py,sha256=yfUXIADRuhPb2rNynoFRlpQ6QINm1EkmvT_BbZ7dv8I,3536
 nucliadb_utils/aiopynecone/models.py,sha256=ketK2IYLWiwFZ76rnJmwfcuopFJrCAtCUszdTSurm_Q,3236
 nucliadb_utils/audit/__init__.py,sha256=cp15ZcFnHvpcu_5-aK2A4uUyvuZVV_MJn4bIXMa20ks,835
-nucliadb_utils/audit/audit.py,sha256=Tbi8poUP6BKZy4ZsaFaeTaQ1K5ys8GXym5ps389-XHE,2966
-nucliadb_utils/audit/basic.py,sha256=1dIWwRWXCb_q3TZObHIMJYOllnbPxullpuqEFaWSUR8,3577
-nucliadb_utils/audit/stream.py,sha256=VpwSrfPU95sqBUgMiWBX0cOC6yjuBXuh5f3uRGFJdwc,13672
+nucliadb_utils/audit/audit.py,sha256=0RoJxK_SesKD9cbNMqCtpT4ylznFxEcHsetyh1eRlEk,3023
+nucliadb_utils/audit/basic.py,sha256=g-790vVvua6OrUkHGBjvgz2TaZt21dWoZw3XiYptt_E,3634
+nucliadb_utils/audit/stream.py,sha256=mHZD-iX5t7N5dd6BS4j0XUkQ8G6rmBXub6GngfIOcx0,13805
 nucliadb_utils/cache/__init__.py,sha256=itSI7dtTwFP55YMX4iK7JzdMHS5CQVUiB1XzQu4UBh8,833
 nucliadb_utils/cache/exceptions.py,sha256=Zu-O_-0-yctOEgoDGI92gPzWfBMRrpiAyESA62ld6MA,975
 nucliadb_utils/cache/nats.py,sha256=-AjCfkFgKVdJUlGR0hT9JDSNkPVFg4S6w9eW-ZIcXPM,7037
@@ -64,8 +64,8 @@ nucliadb_utils/tests/indexing.py,sha256=YW2QhkhO9Q_8A4kKWJaWSvXvyQ_AiAwY1VylcfVQ
 nucliadb_utils/tests/local.py,sha256=7nuP8EFUAiA8ZH50R1iPV9EUXBySQxOanVm3Zht_e0g,1835
 nucliadb_utils/tests/nats.py,sha256=xqpww4jZjTKY9oPGlJdDJG67L3FIBQsa9qDHxILR8r8,7687
 nucliadb_utils/tests/s3.py,sha256=IdMxK_cNdSHLvO1u8BwsKFzD87Hk1MVPDZ57zx6h-rA,3656
-nucliadb_utils-5.0.1.post1139.dist-info/METADATA,sha256=Y53E9HwIMf8HaURu5LqsQ9xr4YAwX-T57tSbuJ3_xF8,2071
-nucliadb_utils-5.0.1.post1139.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
-nucliadb_utils-5.0.1.post1139.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
-nucliadb_utils-5.0.1.post1139.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-nucliadb_utils-5.0.1.post1139.dist-info/RECORD,,
+nucliadb_utils-5.0.1.post1141.dist-info/METADATA,sha256=K-62J0W9ahmctvyty7ELeoJ39F39lwtkZje7odWeREg,2071
+nucliadb_utils-5.0.1.post1141.dist-info/WHEEL,sha256=HiCZjzuy6Dw0hdX5R3LCFPDmFS4BWl8H-8W39XfmgX4,91
+nucliadb_utils-5.0.1.post1141.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
+nucliadb_utils-5.0.1.post1141.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+nucliadb_utils-5.0.1.post1141.dist-info/RECORD,,

{nucliadb_utils-5.0.1.post1139.dist-info → nucliadb_utils-5.0.1.post1141.dist-info}/WHEEL

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