nucliadb-utils 6.7.2.post4911__py3-none-any.whl → 6.8.0.post4918__py3-none-any.whl

nucliadb_utils/audit/audit.py

@@ -86,12 +86,14 @@ class AuditStorage:
         chat_context: List[ChatContext],
         retrieved_context: List[RetrievedContext],
         answer: Optional[str],
+        reasoning: Optional[str],
         learning_id: Optional[str],
         status_code: int,
         model: Optional[str],
         rephrase_time: Optional[float] = None,
         generative_answer_time: Optional[float] = None,
         generative_answer_first_chunk_time: Optional[float] = None,
+        generative_reasoning_first_chunk_time: Optional[float] = None,
     ):
         raise NotImplementedError
 

nucliadb_utils/audit/basic.py

@@ -87,12 +87,14 @@ class BasicAuditStorage(AuditStorage):
         chat_context: List[ChatContext],
         retrieved_context: List[RetrievedContext],
         answer: Optional[str],
+        reasoning: Optional[str],
         learning_id: Optional[str],
         status_code: int,
         model: Optional[str],
         rephrase_time: Optional[float] = None,
         generative_answer_time: Optional[float] = None,
         generative_answer_first_chunk_time: Optional[float] = None,
+        generative_reasoning_first_chunk_time: Optional[float] = None,
     ):
         logger.debug(f"CHAT {kbid} {user} {origin}")
 

nucliadb_utils/audit/stream.py

@@ -425,12 +425,14 @@ class StreamAuditStorage(AuditStorage):
         chat_context: List[ChatContext],
         retrieved_context: List[RetrievedContext],
         answer: Optional[str],
+        reasoning: Optional[str],
         learning_id: Optional[str],
         status_code: int,
         model: Optional[str],
         rephrase_time: Optional[float] = None,
         generative_answer_time: Optional[float] = None,
         generative_answer_first_chunk_time: Optional[float] = None,
+        generative_reasoning_first_chunk_time: Optional[float] = None,
     ):
         rcontext = get_request_context()
         if rcontext is None:
@@ -448,6 +450,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 generative_reasoning_first_chunk_time is not None:
+            auditrequest.generative_reasoning_first_chunk_time = generative_reasoning_first_chunk_time
 
         if retrieval_rephrased_question is not None:
             auditrequest.retrieval_rephrased_question = retrieval_rephrased_question
@@ -462,6 +466,9 @@ class StreamAuditStorage(AuditStorage):
             auditrequest.chat.rephrased_question = rephrased_question
         if answer is not None:
             auditrequest.chat.answer = answer
+        if reasoning is not None:
+            auditrequest.chat.reasoning = reasoning
+
         auditrequest.chat.status_code = status_code
         if model is not None:
             auditrequest.chat.model = model

{nucliadb_utils-6.7.2.post4911.dist-info → nucliadb_utils-6.8.0.post4918.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nucliadb_utils
-Version: 6.7.2.post4911
+Version: 6.8.0.post4918
 Summary: NucliaDB util library
 Author-email: Nuclia <nucliadb@nuclia.com>
 License-Expression: AGPL-3.0-or-later
@@ -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.7.2.post4911
-Requires-Dist: nucliadb-telemetry>=6.7.2.post4911
+Requires-Dist: nucliadb-protos>=6.8.0.post4918
+Requires-Dist: nucliadb-telemetry>=6.8.0.post4918
 Provides-Extra: cache
 Requires-Dist: redis>=4.3.4; extra == "cache"
 Requires-Dist: orjson>=3.6.7; extra == "cache"

{nucliadb_utils-6.7.2.post4911.dist-info → nucliadb_utils-6.8.0.post4918.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=QNrohgWcWkO0exDOi-U7wp1YY7s8ezJqJESWGx4KnYM,3493
-nucliadb_utils/audit/basic.py,sha256=4NX3IkZtTM_F2jD9TGE8-1eoS-hc-_zWwFGYVahGNSs,4141
-nucliadb_utils/audit/stream.py,sha256=NPQAP6UNS1--YynGMOP2PfoXcYt0SC0_LqRtXq11tWk,17313
+nucliadb_utils/audit/audit.py,sha256=xmJJiAGG8rPGADwD9gXN9-QJ80GeGvqmY-kCwEf6PiQ,3598
+nucliadb_utils/audit/basic.py,sha256=fcCYvoFSGVbbB8cSCnm95bN2rf1AAeuWhGfh5no0S-Y,4246
+nucliadb_utils/audit/stream.py,sha256=A0PK-uHW-gXxrDsggbWwbcf2x1CWfP2rYf60wSixg0k,17670
 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=3aV25374fsA_oOr9ZqLDxhNn700O7CIagJ-0Y-ojwqE,7294
@@ -57,7 +57,7 @@ nucliadb_utils/tests/gcs.py,sha256=MBMzn_UHU5SU6iILuCsB5zU4umhNcaCw_MKrxZhwvOc,4
 nucliadb_utils/tests/local.py,sha256=z9E11_ol1mu7N8Y6PkjKl-WMPPMl7JqQbDj3uhVa1A0,1933
 nucliadb_utils/tests/nats.py,sha256=RWHjwqq5esuO7OFbP24yYX1cXnpPLcWJwDUdmwCpH28,1897
 nucliadb_utils/tests/s3.py,sha256=kz9ULxrAYLVslZ59I8dtweZ9DJz5R8Ioy2XYrveZzHw,3829
-nucliadb_utils-6.7.2.post4911.dist-info/METADATA,sha256=F-eH9NAHvGhqwRA49mBz7pllorkmqJJa4t8nRhiDyRk,2180
-nucliadb_utils-6.7.2.post4911.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nucliadb_utils-6.7.2.post4911.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
-nucliadb_utils-6.7.2.post4911.dist-info/RECORD,,
+nucliadb_utils-6.8.0.post4918.dist-info/METADATA,sha256=xBCt7TlYRhVeqSODpoBJQPxiIkxYyrFWeaypyWPY2so,2180
+nucliadb_utils-6.8.0.post4918.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nucliadb_utils-6.8.0.post4918.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
+nucliadb_utils-6.8.0.post4918.dist-info/RECORD,,