nucliadb-utils 6.2.1.post2949__py3-none-any.whl → 6.2.1.post2954__py3-none-any.whl

nucliadb_utils/audit/audit.py

@@ -84,9 +84,9 @@ class AuditStorage:
         chat_context: List[ChatContext],
         retrieved_context: List[RetrievedContext],
         answer: Optional[str],
-        learning_id: str,
+        learning_id: Optional[str],
         status_code: int,
-        model: str,
+        model: Optional[str],
         rephrase_time: Optional[float] = None,
         generative_answer_time: Optional[float] = None,
         generative_answer_first_chunk_time: Optional[float] = None,

nucliadb_utils/audit/basic.py

@@ -85,9 +85,9 @@ class BasicAuditStorage(AuditStorage):
         chat_context: List[ChatContext],
         retrieved_context: List[RetrievedContext],
         answer: Optional[str],
-        learning_id: str,
+        learning_id: Optional[str],
         status_code: int,
-        model: str,
+        model: Optional[str],
         rephrase_time: Optional[float] = None,
         generative_answer_time: Optional[float] = None,
         generative_answer_first_chunk_time: Optional[float] = None,

nucliadb_utils/audit/stream.py

@@ -375,9 +375,9 @@ class StreamAuditStorage(AuditStorage):
         chat_context: List[ChatContext],
         retrieved_context: List[RetrievedContext],
         answer: Optional[str],
-        learning_id: str,
+        learning_id: Optional[str],
         status_code: int,
-        model: str,
+        model: Optional[str],
         rephrase_time: Optional[float] = None,
         generative_answer_time: Optional[float] = None,
         generative_answer_first_chunk_time: Optional[float] = None,
@@ -402,13 +402,15 @@ class StreamAuditStorage(AuditStorage):
         auditrequest.chat.question = question
         auditrequest.chat.chat_context.extend(chat_context)
         auditrequest.chat.retrieved_context.extend(retrieved_context)
-        auditrequest.chat.learning_id = learning_id
+        if learning_id is not None:
+            auditrequest.chat.learning_id = learning_id
         if rephrased_question is not None:
             auditrequest.chat.rephrased_question = rephrased_question
         if answer is not None:
             auditrequest.chat.answer = answer
         auditrequest.chat.status_code = status_code
-        auditrequest.chat.model = model
+        if model is not None:
+            auditrequest.chat.model = model
 
     def feedback(
         self,

{nucliadb_utils-6.2.1.post2949.dist-info → nucliadb_utils-6.2.1.post2954.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: nucliadb_utils
-Version: 6.2.1.post2949
+Version: 6.2.1.post2954
 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>=6.2.1.post2949
-Requires-Dist: nucliadb-telemetry>=6.2.1.post2949
+Requires-Dist: nucliadb-protos>=6.2.1.post2954
+Requires-Dist: nucliadb-telemetry>=6.2.1.post2954
 Provides-Extra: cache
 Requires-Dist: redis>=4.3.4; extra == "cache"
 Requires-Dist: orjson>=3.6.7; extra == "cache"

{nucliadb_utils-6.2.1.post2949.dist-info → nucliadb_utils-6.2.1.post2954.dist-info}/RECORD

@@ -22,9 +22,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=2kRG4MLyvuxwaZl1uFkRNfItHNE5r9BNxxoNKWt1pFU,3364
-nucliadb_utils/audit/basic.py,sha256=p9DB9wYaKJuzHjNkXTMcbdoIDQRVTFjbLmx7fwoRQFI,4012
-nucliadb_utils/audit/stream.py,sha256=p1CBs-35G5VCCABmv9n3z9K4Je30cEEbdBvSv_NG-lI,15139
+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=zT--ECfFNfhL22hhN4IKkPrCaNITKijmufXVoVWm_4c,15233
 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
@@ -59,8 +59,8 @@ nucliadb_utils/tests/indexing.py,sha256=YW2QhkhO9Q_8A4kKWJaWSvXvyQ_AiAwY1VylcfVQ
 nucliadb_utils/tests/local.py,sha256=fXIBasrvdaFJM-sw2wk1_oiFzBcm9O10iCyC-OiXwY8,1914
 nucliadb_utils/tests/nats.py,sha256=RWHjwqq5esuO7OFbP24yYX1cXnpPLcWJwDUdmwCpH28,1897
 nucliadb_utils/tests/s3.py,sha256=pl-RJFjA4MH6iXkqhsh5g8gDuEhrYu1nPZ-laxlrMlE,3704
-nucliadb_utils-6.2.1.post2949.dist-info/METADATA,sha256=JcZuBmz1u-pTUWuP4DB1U7T_HpQbKjAtciA_z_31-iE,2204
-nucliadb_utils-6.2.1.post2949.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-nucliadb_utils-6.2.1.post2949.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
-nucliadb_utils-6.2.1.post2949.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-nucliadb_utils-6.2.1.post2949.dist-info/RECORD,,
+nucliadb_utils-6.2.1.post2954.dist-info/METADATA,sha256=g4UqwRJKnLXp75z3CjGmvWbRCOU0_cDHqFDuaiSAUE4,2204
+nucliadb_utils-6.2.1.post2954.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+nucliadb_utils-6.2.1.post2954.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
+nucliadb_utils-6.2.1.post2954.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+nucliadb_utils-6.2.1.post2954.dist-info/RECORD,,