nucliadb 6.5.1.post4507__py3-none-any.whl → 6.5.1.post4520__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (6) hide show
  1. nucliadb/search/search/predict_proxy.py +2 -3
  2. {nucliadb-6.5.1.post4507.dist-info → nucliadb-6.5.1.post4520.dist-info}/METADATA +6 -6
  3. {nucliadb-6.5.1.post4507.dist-info → nucliadb-6.5.1.post4520.dist-info}/RECORD +6 -6
  4. {nucliadb-6.5.1.post4507.dist-info → nucliadb-6.5.1.post4520.dist-info}/WHEEL +0 -0
  5. {nucliadb-6.5.1.post4507.dist-info → nucliadb-6.5.1.post4520.dist-info}/entry_points.txt +0 -0
  6. {nucliadb-6.5.1.post4507.dist-info → nucliadb-6.5.1.post4520.dist-info}/top_level.txt +0 -0
nucliadb/search/search/predict_proxy.py CHANGED
@@ -171,14 +171,13 @@ async def chat_streaming_generator(
171
171
  user_query: str,
172
172
  is_json: bool,
173
173
  ):
174
- stream = predict_response.content.iter_any()
175
174
  first = True
176
175
  status_code = AnswerStatusCode.ERROR.value
177
176
  text_answer = ""
178
177
  json_object = None
179
178
  metrics = AskMetrics()
180
179
  with metrics.time(PREDICT_ANSWER_METRIC):
181
- async for chunk in stream:
180
+ async for chunk in predict_response.content:
182
181
  if first:
183
182
  metrics.record_first_chunk_yielded()
184
183
  first = False
@@ -204,7 +203,7 @@ async def chat_streaming_generator(
204
203
 
205
204
  if is_json is False and chunk: # Ensure chunk is not empty before decoding
206
205
  # If response is text the status_code comes at the last chunk of data
207
- status_code = chunk.decode()
206
+ status_code = chunk.decode().split(".")[-1]
208
207
 
209
208
  audit_predict_proxy_endpoint(
210
209
  headers=predict_response.headers,
{nucliadb-6.5.1.post4507.dist-info → nucliadb-6.5.1.post4520.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: nucliadb
3
- Version: 6.5.1.post4507
3
+ Version: 6.5.1.post4520
4
4
  Summary: NucliaDB
5
5
  Author-email: Nuclia <nucliadb@nuclia.com>
6
6
  License-Expression: AGPL-3.0-or-later
@@ -19,11 +19,11 @@ Classifier: Programming Language :: Python :: 3.12
19
19
  Classifier: Programming Language :: Python :: 3 :: Only
20
20
  Requires-Python: <4,>=3.9
21
21
  Description-Content-Type: text/markdown
22
- Requires-Dist: nucliadb-telemetry[all]>=6.5.1.post4507
23
- Requires-Dist: nucliadb-utils[cache,fastapi,storages]>=6.5.1.post4507
24
- Requires-Dist: nucliadb-protos>=6.5.1.post4507
25
- Requires-Dist: nucliadb-models>=6.5.1.post4507
26
- Requires-Dist: nidx-protos>=6.5.1.post4507
22
+ Requires-Dist: nucliadb-telemetry[all]>=6.5.1.post4520
23
+ Requires-Dist: nucliadb-utils[cache,fastapi,storages]>=6.5.1.post4520
24
+ Requires-Dist: nucliadb-protos>=6.5.1.post4520
25
+ Requires-Dist: nucliadb-models>=6.5.1.post4520
26
+ Requires-Dist: nidx-protos>=6.5.1.post4520
27
27
  Requires-Dist: nucliadb-admin-assets>=1.0.0.post1224
28
28
  Requires-Dist: nuclia-models>=0.24.2
29
29
  Requires-Dist: uvicorn[standard]
{nucliadb-6.5.1.post4507.dist-info → nucliadb-6.5.1.post4520.dist-info}/RECORD RENAMED
@@ -255,7 +255,7 @@ nucliadb/search/search/merge.py,sha256=XiRBsxhYPshPV7lZXD-9E259KZOPIf4I2tKosY0lP
255
255
  nucliadb/search/search/metrics.py,sha256=3I6IN0qDSmqIvUaWJmT3rt-Jyjs6LcvnKI8ZqCiuJPY,3501
256
256
  nucliadb/search/search/paragraphs.py,sha256=pNAEiYqJGGUVcEf7xf-PFMVqz0PX4Qb-WNG-_zPGN2o,7799
257
257
  nucliadb/search/search/pgcatalog.py,sha256=O_nRjSJf1Qc-XorVwcNlsDOftzy_zQLLfagkjU4YmSA,16718
258
- nucliadb/search/search/predict_proxy.py,sha256=cuD_sfM3RLdEoQaanRz0CflO6nKVGGKPzoFA17shb_w,8647
258
+ nucliadb/search/search/predict_proxy.py,sha256=T4T6c1ZAbBCgPLy-6JoSlixy-0bAh6z0rNH1bMMy3bg,8631
259
259
  nucliadb/search/search/query.py,sha256=0qIQdt548L3jtKOyKo06aGJ73SLBxAW3N38_Hc1M3Uw,11528
260
260
  nucliadb/search/search/rank_fusion.py,sha256=xZtXhbmKb_56gs73u6KkFm2efvTATOSMmpOV2wrAIqE,9613
261
261
  nucliadb/search/search/rerankers.py,sha256=E2J1QdKAojqbhHM3KAyaOXKf6tJyETUxKs4tf_BEyqk,7472
@@ -375,8 +375,8 @@ nucliadb/writer/tus/local.py,sha256=7jYa_w9b-N90jWgN2sQKkNcomqn6JMVBOVeDOVYJHto,
375
375
  nucliadb/writer/tus/s3.py,sha256=vF0NkFTXiXhXq3bCVXXVV-ED38ECVoUeeYViP8uMqcU,8357
376
376
  nucliadb/writer/tus/storage.py,sha256=ToqwjoYnjI4oIcwzkhha_MPxi-k4Jk3Lt55zRwaC1SM,2903
377
377
  nucliadb/writer/tus/utils.py,sha256=MSdVbRsRSZVdkaum69_0wku7X3p5wlZf4nr6E0GMKbw,2556
378
- nucliadb-6.5.1.post4507.dist-info/METADATA,sha256=mXI1_xVFcW5gf0mv4h3PdH3RYyaqAUTt6UlvPEU2gb4,4158
379
- nucliadb-6.5.1.post4507.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
380
- nucliadb-6.5.1.post4507.dist-info/entry_points.txt,sha256=XqGfgFDuY3zXQc8ewXM2TRVjTModIq851zOsgrmaXx4,1268
381
- nucliadb-6.5.1.post4507.dist-info/top_level.txt,sha256=hwYhTVnX7jkQ9gJCkVrbqEG1M4lT2F_iPQND1fCzF80,20
382
- nucliadb-6.5.1.post4507.dist-info/RECORD,,
378
+ nucliadb-6.5.1.post4520.dist-info/METADATA,sha256=4zLZ2kYpWRVlz8HgcOR7TZ9WIIgBCyd-w6wDuMJU_ww,4158
379
+ nucliadb-6.5.1.post4520.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
380
+ nucliadb-6.5.1.post4520.dist-info/entry_points.txt,sha256=XqGfgFDuY3zXQc8ewXM2TRVjTModIq851zOsgrmaXx4,1268
381
+ nucliadb-6.5.1.post4520.dist-info/top_level.txt,sha256=hwYhTVnX7jkQ9gJCkVrbqEG1M4lT2F_iPQND1fCzF80,20
382
+ nucliadb-6.5.1.post4520.dist-info/RECORD,,