nucliadb-utils 5.1.2.post1427__py3-none-any.whl → 5.1.2.post1445__py3-none-any.whl

nucliadb_utils/audit/audit.py

@@ -105,3 +105,16 @@ class AuditStorage:
 
     def delete_kb(self, kbid: str):
         raise NotImplementedError
+
+    def feedback(
+        self,
+        kbid: str,
+        user: str,
+        client_type: int,
+        origin: str,
+        learning_id: str,
+        good: bool,
+        task: int,
+        feedback: Optional[str],
+    ):
+        raise NotImplementedError

nucliadb_utils/audit/basic.py

@@ -106,3 +106,16 @@ class BasicAuditStorage(AuditStorage):
 
     def delete_kb(self, kbid: str):
         logger.debug(f"DELETE_KB {kbid}")
+
+    def feedback(
+        self,
+        kbid: str,
+        user: str,
+        client_type: int,
+        origin: str,
+        learning_id: str,
+        good: bool,
+        task: int,
+        feedback: Optional[str],
+    ):
+        logger.debug(f"FEEDBACK {kbid} {user} {client_type} {origin}")

nucliadb_utils/audit/stream.py

@@ -410,3 +410,31 @@ class StreamAuditStorage(AuditStorage):
         if answer is not None:
             auditrequest.chat.answer = answer
         auditrequest.chat.status_code = status_code
+
+    def feedback(
+        self,
+        kbid: str,
+        user: str,
+        client_type: int,
+        origin: str,
+        learning_id: str,
+        good: bool,
+        task: int,
+        feedback: Optional[str],
+    ):
+        rcontext = get_request_context()
+        if rcontext is None:
+            return
+
+        auditrequest = rcontext.audit_request
+
+        auditrequest.origin = origin
+        auditrequest.client_type = client_type  # type: ignore
+        auditrequest.userid = user
+        auditrequest.kbid = kbid
+
+        auditrequest.feedback.learning_id = learning_id
+        auditrequest.feedback.good = good
+        auditrequest.feedback.task = task  # type: ignore
+        if feedback is not None:
+            auditrequest.feedback.feedback = feedback

{nucliadb_utils-5.1.2.post1427.dist-info → nucliadb_utils-5.1.2.post1445.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nucliadb_utils
-Version: 5.1.2.post1427
+Version: 5.1.2.post1445
 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.1.2.post1427
-Requires-Dist: nucliadb-telemetry>=5.1.2.post1427
+Requires-Dist: nucliadb-protos>=5.1.2.post1445
+Requires-Dist: nucliadb-telemetry>=5.1.2.post1445
 Provides-Extra: cache
 Requires-Dist: redis>=4.3.4; extra == "cache"
 Requires-Dist: orjson>=3.6.7; extra == "cache"

{nucliadb_utils-5.1.2.post1427.dist-info → nucliadb_utils-5.1.2.post1445.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=25Tzwmh0l8e4MvYGlmhHNt9awSb-gDAUdEjvtqFODnc,3049
-nucliadb_utils/audit/basic.py,sha256=_iZl7u5QB-Vl0ar5ZpmxppeFV0OdNHdcs6o0_W32eEw,3660
-nucliadb_utils/audit/stream.py,sha256=ww11OmqKZ8AOXVrN1sizIfEqJbEjkqldV1ts8od_6cY,13845
+nucliadb_utils/audit/audit.py,sha256=-mkN8gkTc6QPvbJtKxhILtSPLS9SH0augctx5HpYWWM,3306
+nucliadb_utils/audit/basic.py,sha256=BoXeJgG-TpFNcOl2Yjt-RRMOEMwmKCaHDopZiNSirWE,3954
+nucliadb_utils/audit/stream.py,sha256=Mr4G06_mBYUGP0OxnCdGz0maRKqa3MnlQz5PvHl_KyQ,14617
 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=7nuP8EFUAiA8ZH50R1iPV9EUXBySQxOanVm3Zht_e0g,1835
 nucliadb_utils/tests/nats.py,sha256=xqpww4jZjTKY9oPGlJdDJG67L3FIBQsa9qDHxILR8r8,7687
 nucliadb_utils/tests/s3.py,sha256=IdMxK_cNdSHLvO1u8BwsKFzD87Hk1MVPDZ57zx6h-rA,3656
-nucliadb_utils-5.1.2.post1427.dist-info/METADATA,sha256=CRgr6u1fkOUXX0-JCBYGkJqcrXDFQH4H4_DuB-TMHxA,2071
-nucliadb_utils-5.1.2.post1427.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
-nucliadb_utils-5.1.2.post1427.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
-nucliadb_utils-5.1.2.post1427.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-nucliadb_utils-5.1.2.post1427.dist-info/RECORD,,
+nucliadb_utils-5.1.2.post1445.dist-info/METADATA,sha256=SSFeJEfdB2ZL1ZxAevoHZu858ck5ABM3C9mJ_-Nuc74,2071
+nucliadb_utils-5.1.2.post1445.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
+nucliadb_utils-5.1.2.post1445.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
+nucliadb_utils-5.1.2.post1445.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+nucliadb_utils-5.1.2.post1445.dist-info/RECORD,,