nucliadb-utils 6.3.5.post4033__py3-none-any.whl → 6.3.5.post4041__py3-none-any.whl

nucliadb_utils/audit/audit.py

@@ -70,6 +70,7 @@ class AuditStorage:
         search: SearchRequest,
         timeit: float,
         resources: int,
+        retrieval_rephrased_question: Optional[str] = None,
     ):
         raise NotImplementedError
 
@@ -81,6 +82,7 @@ class AuditStorage:
         origin: str,
         question: str,
         rephrased_question: Optional[str],
+        retrieval_rephrased_question: Optional[str],
         chat_context: List[ChatContext],
         retrieved_context: List[RetrievedContext],
         answer: Optional[str],

nucliadb_utils/audit/basic.py

@@ -71,6 +71,7 @@ class BasicAuditStorage(AuditStorage):
         search: SearchRequest,
         timeit: float,
         resources: int,
+        retrieval_rephrased_question: Optional[str] = None,
     ):
         logger.debug(f"SEARCH {kbid} {user} {origin} ''{search}'' {timeit} {resources}")
 
@@ -82,6 +83,7 @@ class BasicAuditStorage(AuditStorage):
         origin: str,
         question: str,
         rephrased_question: Optional[str],
+        retrieval_rephrased_question: Optional[str],
         chat_context: List[ChatContext],
         retrieved_context: List[RetrievedContext],
         answer: Optional[str],

nucliadb_utils/audit/stream.py

@@ -343,6 +343,7 @@ class StreamAuditStorage(AuditStorage):
         search: SearchRequest,
         timeit: float,
         resources: int,
+        retrieval_rephrased_question: Optional[str] = None,
     ):
         context = get_request_context()
         if context is None:
@@ -361,7 +362,8 @@ class StreamAuditStorage(AuditStorage):
             auditrequest.type = AuditRequest.CHAT
         else:
             auditrequest.type = AuditRequest.SEARCH
-
+        if retrieval_rephrased_question is not None:
+            auditrequest.retrieval_rephrased_question = retrieval_rephrased_question
         trace_id = get_trace_id()
         self.kb_usage_utility.send_kb_usage(
             service=Service.NUCLIA_DB,
@@ -390,6 +392,7 @@ class StreamAuditStorage(AuditStorage):
         origin: str,
         question: str,
         rephrased_question: Optional[str],
+        retrieval_rephrased_question: Optional[str],
         chat_context: List[ChatContext],
         retrieved_context: List[RetrievedContext],
         answer: Optional[str],
@@ -416,6 +419,8 @@ class StreamAuditStorage(AuditStorage):
             auditrequest.generative_answer_time = generative_answer_time
         if generative_answer_first_chunk_time is not None:
             auditrequest.generative_answer_first_chunk_time = generative_answer_first_chunk_time
+        if retrieval_rephrased_question is not None:
+            auditrequest.retrieval_rephrased_question = retrieval_rephrased_question
         auditrequest.type = AuditRequest.CHAT
         auditrequest.chat.question = question
         auditrequest.chat.chat_context.extend(chat_context)

{nucliadb_utils-6.3.5.post4033.dist-info → nucliadb_utils-6.3.5.post4041.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nucliadb_utils
-Version: 6.3.5.post4033
+Version: 6.3.5.post4041
 Summary: NucliaDB util library
 Author-email: Nuclia <nucliadb@nuclia.com>
 License: AGPL
@@ -27,8 +27,8 @@ Requires-Dist: nats-py[nkeys]>=2.6.0
 Requires-Dist: PyNaCl
 Requires-Dist: pyjwt>=2.4.0
 Requires-Dist: mrflagly>=0.2.9
-Requires-Dist: nucliadb-protos>=6.3.5.post4033
-Requires-Dist: nucliadb-telemetry>=6.3.5.post4033
+Requires-Dist: nucliadb-protos>=6.3.5.post4041
+Requires-Dist: nucliadb-telemetry>=6.3.5.post4041
 Provides-Extra: cache
 Requires-Dist: redis>=4.3.4; extra == "cache"
 Requires-Dist: orjson>=3.6.7; extra == "cache"

{nucliadb_utils-6.3.5.post4033.dist-info → nucliadb_utils-6.3.5.post4041.dist-info}/RECORD

@@ -21,9 +21,9 @@ nucliadb_utils/aiopynecone/client.py,sha256=MPyHnDXwhukJr7U3CJh7BpsekfSuOkyM4g5b
 nucliadb_utils/aiopynecone/exceptions.py,sha256=fUErx3ceKQK1MUbOnYcZhIzpNe8UVAptZE9JIRDLXDE,4000
 nucliadb_utils/aiopynecone/models.py,sha256=XkNIZx4bxdbVo9zYVn8IRp70q4DWUMWN79ybGloFj2Q,3492
 nucliadb_utils/audit/__init__.py,sha256=cp15ZcFnHvpcu_5-aK2A4uUyvuZVV_MJn4bIXMa20ks,835
-nucliadb_utils/audit/audit.py,sha256=VDUqcSToEoz5m7iG-4omjTlsBSEJGcp0_xl51idnU3Q,3384
-nucliadb_utils/audit/basic.py,sha256=pVf1oKp7T_oTDCyt8QbSsfC_uWjF66ooF6rqY-SWSrc,4032
-nucliadb_utils/audit/stream.py,sha256=SovPj5iTzXka2cg4NnwAZwnOYljhsT6ynhDC2OsRVPc,16035
+nucliadb_utils/audit/audit.py,sha256=m5Mkm4m1W4O91t8jsVQocy57hqYSjboCyFw8P1-g-7A,3497
+nucliadb_utils/audit/basic.py,sha256=trH4IgLnb-VRj2vrRL2686HL7z9Zcdb6bZ7I47Usrt8,4145
+nucliadb_utils/audit/stream.py,sha256=b1W_5XK_8uc9tO1zcEp_sYAm5pG7-ODYbX8zqzMHXUk,16423
 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
@@ -57,7 +57,7 @@ nucliadb_utils/tests/gcs.py,sha256=MBMzn_UHU5SU6iILuCsB5zU4umhNcaCw_MKrxZhwvOc,4
 nucliadb_utils/tests/local.py,sha256=cxIfPrKuqs5Ef0nbrVYQQAH2mwc4E0iD9bC2sWegS-c,1934
 nucliadb_utils/tests/nats.py,sha256=RWHjwqq5esuO7OFbP24yYX1cXnpPLcWJwDUdmwCpH28,1897
 nucliadb_utils/tests/s3.py,sha256=DACUh3HvgH3BchKFZ9R7RFUzsrg3v9A-cxTcXx4nmvA,3734
-nucliadb_utils-6.3.5.post4033.dist-info/METADATA,sha256=oBnO7AC58x7i7McBWy9kJPfvOF1AOZmrCJZIzan2Sdw,2205
-nucliadb_utils-6.3.5.post4033.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-nucliadb_utils-6.3.5.post4033.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
-nucliadb_utils-6.3.5.post4033.dist-info/RECORD,,
+nucliadb_utils-6.3.5.post4041.dist-info/METADATA,sha256=vy3CPGxOdaBgojQZCmk_jECYT1XUmMOFqeL8n3F9r1Y,2205
+nucliadb_utils-6.3.5.post4041.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+nucliadb_utils-6.3.5.post4041.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
+nucliadb_utils-6.3.5.post4041.dist-info/RECORD,,