nucliadb-utils 5.1.2.post1522__py3-none-any.whl → 5.1.2.post1530__py3-none-any.whl

nucliadb_utils/audit/audit.py

@@ -86,6 +86,7 @@ class AuditStorage:
         answer: Optional[str],
         learning_id: str,
         status_code: int,
+        model: 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

@@ -87,6 +87,7 @@ class BasicAuditStorage(AuditStorage):
         answer: Optional[str],
         learning_id: str,
         status_code: int,
+        model: 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

@@ -86,6 +86,11 @@ class AuditMiddleware(BaseHTTPMiddleware):
         token = request_context_var.set(context)
         context.audit_request.time.FromDatetime(datetime.now(tz=timezone.utc))
         context.audit_request.trace_id = get_trace_id()
+
+        if request.url.path.split("/")[-1] in ("ask", "search", "find"):
+            body = (await request.body()).decode()
+            context.audit_request.user_request = body
+
         response = await call_next(request)
 
         if isinstance(response, StreamingResponse):
@@ -380,6 +385,7 @@ class StreamAuditStorage(AuditStorage):
         answer: Optional[str],
         learning_id: str,
         status_code: int,
+        model: str,
         rephrase_time: Optional[float] = None,
         generative_answer_time: Optional[float] = None,
         generative_answer_first_chunk_time: Optional[float] = None,
@@ -410,6 +416,7 @@ class StreamAuditStorage(AuditStorage):
         if answer is not None:
             auditrequest.chat.answer = answer
         auditrequest.chat.status_code = status_code
+        auditrequest.chat.model = model
 
     def feedback(
         self,

{nucliadb_utils-5.1.2.post1522.dist-info → nucliadb_utils-5.1.2.post1530.dist-info}/METADATA

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

{nucliadb_utils-5.1.2.post1522.dist-info → nucliadb_utils-5.1.2.post1530.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=-mkN8gkTc6QPvbJtKxhILtSPLS9SH0augctx5HpYWWM,3306
-nucliadb_utils/audit/basic.py,sha256=BoXeJgG-TpFNcOl2Yjt-RRMOEMwmKCaHDopZiNSirWE,3954
-nucliadb_utils/audit/stream.py,sha256=KOrp15cDsdu9OGxPMiRBL23Y69BzpqWhZ32f6hDhP9U,14667
+nucliadb_utils/audit/audit.py,sha256=hH1L5XeRjTO896wMyINqiI5dkgGD_h8kbMsRs2HZ8Ts,3326
+nucliadb_utils/audit/basic.py,sha256=qKA7YAesuhRfW3iQXInI_GVm32az-WI6UPdS8QKQ_Fw,3974
+nucliadb_utils/audit/stream.py,sha256=FhadnIj7h1IFC4KgEIfgNf3ka3hFMc70yhyX76htWaY,14907
 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.post1522.dist-info/METADATA,sha256=G67xsDs1_jrITTIA4P2HDarErTovHRhIqWV-r6gKFU8,2071
-nucliadb_utils-5.1.2.post1522.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-nucliadb_utils-5.1.2.post1522.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
-nucliadb_utils-5.1.2.post1522.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-nucliadb_utils-5.1.2.post1522.dist-info/RECORD,,
+nucliadb_utils-5.1.2.post1530.dist-info/METADATA,sha256=T7qpwzUyoNjDeCLPBEA0Zvktie7KSj3l3o5fiVKFuH0,2071
+nucliadb_utils-5.1.2.post1530.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+nucliadb_utils-5.1.2.post1530.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
+nucliadb_utils-5.1.2.post1530.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+nucliadb_utils-5.1.2.post1530.dist-info/RECORD,,