nucliadb-utils 5.1.2.post1390__py3-none-any.whl → 5.1.2.post1397__py3-none-any.whl

nucliadb_utils/audit/stream.py

@@ -34,21 +34,21 @@ from starlette.responses import Response, StreamingResponse
 from starlette.types import ASGIApp
 
 from nucliadb_protos.audit_pb2 import AuditField, AuditRequest, ChatContext, ClientType, RetrievedContext
-from nucliadb_protos.nodereader_pb2 import SearchRequest
-from nucliadb_protos.resources_pb2 import FieldID
-from nucliadb_utils import logger
-from nucliadb_utils.audit.audit import AuditStorage
-from nucliadb_utils.nats import get_traced_jetstream
-from nucliadb_utils.nuclia_usage.protos.kb_usage_pb2 import (
+from nucliadb_protos.kb_usage_pb2 import (
     ClientType as ClientTypeKbUsage,
 )
-from nucliadb_utils.nuclia_usage.protos.kb_usage_pb2 import (
+from nucliadb_protos.kb_usage_pb2 import (
     KBSource,
     Search,
     SearchType,
     Service,
     Storage,
 )
+from nucliadb_protos.nodereader_pb2 import SearchRequest
+from nucliadb_protos.resources_pb2 import FieldID
+from nucliadb_utils import logger
+from nucliadb_utils.audit.audit import AuditStorage
+from nucliadb_utils.nats import get_traced_jetstream
 from nucliadb_utils.nuclia_usage.utils.kb_usage_report import KbUsageReportUtility
 
 

nucliadb_utils/nuclia_usage/utils/kb_usage_report.py

@@ -26,7 +26,7 @@ from typing import Optional
 
 from nats.js.client import JetStreamContext
 
-from nucliadb_utils.nuclia_usage.protos.kb_usage_pb2 import (
+from nucliadb_protos.kb_usage_pb2 import (
     KBSource,
     KbUsage,
     Predict,

{nucliadb_utils-5.1.2.post1390.dist-info → nucliadb_utils-5.1.2.post1397.dist-info}/METADATA

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

{nucliadb_utils-5.1.2.post1390.dist-info → nucliadb_utils-5.1.2.post1397.dist-info}/RECORD

@@ -24,7 +24,7 @@ nucliadb_utils/aiopynecone/models.py,sha256=XkNIZx4bxdbVo9zYVn8IRp70q4DWUMWN79yb
 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=dNYpX3xdNGZswwITvlVH6I031a7r8tLgEIQIIsN1seU,13883
+nucliadb_utils/audit/stream.py,sha256=ww11OmqKZ8AOXVrN1sizIfEqJbEjkqldV1ts8od_6cY,13845
 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
@@ -37,13 +37,8 @@ nucliadb_utils/fastapi/openapi.py,sha256=b0pLuri0QuzQd0elDyOVXM42YYmES_cmT-jEfsQ
 nucliadb_utils/fastapi/run.py,sha256=n6vOX64QqF1I5n4UlKnpm_ZJ24rmwfRGi-J9YMGpZzA,3631
 nucliadb_utils/fastapi/versioning.py,sha256=pwiwuesJW1jElUUI3pI5kcxkigfGBvI64IL6QCBEWS8,3805
 nucliadb_utils/nuclia_usage/__init__.py,sha256=cp15ZcFnHvpcu_5-aK2A4uUyvuZVV_MJn4bIXMa20ks,835
-nucliadb_utils/nuclia_usage/protos/__init__.py,sha256=cp15ZcFnHvpcu_5-aK2A4uUyvuZVV_MJn4bIXMa20ks,835
-nucliadb_utils/nuclia_usage/protos/kb_usage_pb2.py,sha256=9pDy1QE4E6Z_q7x86_8FwHHLJYDSgmvt8LRD7NJiy88,5925
-nucliadb_utils/nuclia_usage/protos/kb_usage_pb2.pyi,sha256=lA1BiWVOQWioPHPxrfzKZSrj-zyfOlO_H2IEYYdyWtU,14567
-nucliadb_utils/nuclia_usage/protos/kb_usage_pb2_grpc.py,sha256=3XT17o832unN2OTBPXho5jj8mm0_8ChSr6m5Jds7SF8,1008
-nucliadb_utils/nuclia_usage/protos/kb_usage_pb2_grpc.pyi,sha256=sXLWrl7G0vtMXs49h-efrwKWc6A_TU420zF9xBMv_ng,926
 nucliadb_utils/nuclia_usage/utils/__init__.py,sha256=cp15ZcFnHvpcu_5-aK2A4uUyvuZVV_MJn4bIXMa20ks,835
-nucliadb_utils/nuclia_usage/utils/kb_usage_report.py,sha256=lTr9CMBpdk34KtkH5K8vFxJRtF6xZpslVMNg1SD03n8,3647
+nucliadb_utils/nuclia_usage/utils/kb_usage_report.py,sha256=F6dhQWhrWzl3qfAdvrqV08hnFKcvFlU8IWN2lzYX0vg,3628
 nucliadb_utils/storages/__init__.py,sha256=5Qc8AUWiJv9_JbGCBpAn88AIJhwDlm0OPQpg2ZdRL4U,872
 nucliadb_utils/storages/azure.py,sha256=egMDwLNIGSQyVevuySt2AswzFdNAcih05BbRg3-p8IU,16015
 nucliadb_utils/storages/exceptions.py,sha256=mm_wX4YRtp7u7enkk_4pMSlX5AQQuFbq4xLmupVDt3Y,2502
@@ -64,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.post1390.dist-info/METADATA,sha256=ggwKzVMKIL_dw8ODqi81YeEo5crNe0OPkGryEV-VhyY,2071
-nucliadb_utils-5.1.2.post1390.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
-nucliadb_utils-5.1.2.post1390.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
-nucliadb_utils-5.1.2.post1390.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-nucliadb_utils-5.1.2.post1390.dist-info/RECORD,,
+nucliadb_utils-5.1.2.post1397.dist-info/METADATA,sha256=WHH6e2ptnnPeX7Q-4gFGrbEsvqogZIn6igazAdYTni0,2071
+nucliadb_utils-5.1.2.post1397.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
+nucliadb_utils-5.1.2.post1397.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
+nucliadb_utils-5.1.2.post1397.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+nucliadb_utils-5.1.2.post1397.dist-info/RECORD,,

nucliadb_utils/nuclia_usage/protos/__init__.py

@@ -1,19 +0,0 @@
-# Copyright (C) 2021 Bosutech XXI S.L.
-#
-# nucliadb is offered under the AGPL v3.0 and as commercial software.
-# For commercial licensing, contact us at info@nuclia.com.
-#
-# AGPL:
-# This program is free software: you can redistribute it and/or modify
-# it under the terms of the GNU Affero General Public License as
-# published by the Free Software Foundation, either version 3 of the
-# License, or (at your option) any later version.
-#
-# This program is distributed in the hope that it will be useful,
-# but WITHOUT ANY WARRANTY; without even the implied warranty of
-# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
-# GNU Affero General Public License for more details.
-#
-# You should have received a copy of the GNU Affero General Public License
-# along with this program. If not, see <http://www.gnu.org/licenses/>.
-#

nucliadb_utils/nuclia_usage/protos/kb_usage_pb2.py

@@ -1,69 +0,0 @@
-# Copyright (C) 2021 Bosutech XXI S.L.
-#
-# nucliadb is offered under the AGPL v3.0 and as commercial software.
-# For commercial licensing, contact us at info@nuclia.com.
-#
-# AGPL:
-# This program is free software: you can redistribute it and/or modify
-# it under the terms of the GNU Affero General Public License as
-# published by the Free Software Foundation, either version 3 of the
-# License, or (at your option) any later version.
-#
-# This program is distributed in the hope that it will be useful,
-# but WITHOUT ANY WARRANTY; without even the implied warranty of
-# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
-# GNU Affero General Public License for more details.
-#
-# You should have received a copy of the GNU Affero General Public License
-# along with this program. If not, see <http://www.gnu.org/licenses/>.
-#
-# flake8: noqa
-# -*- coding: utf-8 -*-
-# Generated by the protocol buffer compiler.  DO NOT EDIT!
-# source: protos/kb_usage.proto
-# Protobuf Python Version: 4.25.1
-"""Generated protocol buffer code."""
-from google.protobuf import descriptor as _descriptor
-from google.protobuf import descriptor_pool as _descriptor_pool
-from google.protobuf import symbol_database as _symbol_database
-from google.protobuf.internal import builder as _builder
-# @@protoc_insertion_point(imports)
-
-_sym_db = _symbol_database.Default()
-
-
-from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
-
-
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x15protos/kb_usage.proto\x12\x08kb_usage\x1a\x1fgoogle/protobuf/timestamp.proto\"\xf2\x01\n\x07Process\x12$\n\x06\x63lient\x18\x01 \x01(\x0e\x32\x14.kb_usage.ClientType\x12\x1c\n\x14slow_processing_time\x18\x02 \x01(\x02\x12\x1b\n\x13pre_processing_time\x18\x03 \x01(\x02\x12\r\n\x05\x62ytes\x18\x04 \x01(\x04\x12\r\n\x05\x63hars\x18\x05 \x01(\r\x12\x15\n\rmedia_seconds\x18\x06 \x01(\r\x12\r\n\x05pages\x18\x07 \x01(\r\x12\x12\n\nparagraphs\x18\x08 \x01(\r\x12\x17\n\x0bmedia_files\x18\t \x01(\rB\x02\x18\x01\x12\x15\n\rnum_processed\x18\n \x01(\r\"\x95\x01\n\x07Storage\x12\x17\n\nparagraphs\x18\x01 \x01(\x04H\x00\x88\x01\x01\x12\x13\n\x06\x66ields\x18\x02 \x01(\x04H\x01\x88\x01\x01\x12\x16\n\tresources\x18\x03 \x01(\x04H\x02\x88\x01\x01\x12\x12\n\x05\x62ytes\x18\x04 \x01(\x04H\x03\x88\x01\x01\x42\r\n\x0b_paragraphsB\t\n\x07_fieldsB\x0c\n\n_resourcesB\x08\n\x06_bytes\"x\n\x06Search\x12$\n\x06\x63lient\x18\x01 \x01(\x0e\x32\x14.kb_usage.ClientType\x12\"\n\x04type\x18\x02 \x01(\x0e\x32\x14.kb_usage.SearchType\x12\x0e\n\x06tokens\x18\x03 \x01(\r\x12\x14\n\x0cnum_searches\x18\x04 \x01(\r\"\xa7\x01\n\x07Predict\x12$\n\x06\x63lient\x18\x01 \x01(\x0e\x32\x14.kb_usage.ClientType\x12#\n\x04type\x18\x02 \x01(\x0e\x32\x15.kb_usage.PredictType\x12\r\n\x05model\x18\x03 \x01(\t\x12\r\n\x05input\x18\x04 \x01(\r\x12\x0e\n\x06output\x18\x05 \x01(\r\x12\r\n\x05image\x18\x06 \x01(\r\x12\x14\n\x0cnum_predicts\x18\x07 \x01(\r\"\xed\x02\n\x07KbUsage\x12\"\n\x07service\x18\x01 \x01(\x0e\x32\x11.kb_usage.Service\x12-\n\ttimestamp\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x17\n\naccount_id\x18\x03 \x01(\tH\x00\x88\x01\x01\x12\x12\n\x05kb_id\x18\x04 \x01(\tH\x01\x88\x01\x01\x12%\n\tkb_source\x18\x05 \x01(\x0e\x32\x12.kb_usage.KBSource\x12$\n\tprocesses\x18\x06 \x03(\x0b\x32\x11.kb_usage.Process\x12#\n\x08predicts\x18\x07 \x03(\x0b\x32\x11.kb_usage.Predict\x12\"\n\x08searches\x18\x08 \x03(\x0b\x32\x10.kb_usage.Search\x12\'\n\x07storage\x18\t \x01(\x0b\x32\x11.kb_usage.StorageH\x02\x88\x01\x01\x42\r\n\x0b_account_idB\x08\n\x06_kb_idB\n\n\x08_storage\"9\n\x11KbUsageAggregated\x12$\n\tkb_usages\x18\x01 \x03(\x0b\x32\x11.kb_usage.KbUsage*\"\n\x08KBSource\x12\n\n\x06HOSTED\x10\x00\x12\n\n\x06ONPREM\x10\x01*5\n\x07Service\x12\x0b\n\x07PREDICT\x10\x00\x12\x0e\n\nPROCESSING\x10\x01\x12\r\n\tNUCLIA_DB\x10\x02*%\n\nSearchType\x12\n\n\x06SEARCH\x10\x00\x12\x0b\n\x07SUGGEST\x10\x01*\xa0\x01\n\x0bPredictType\x12\x0c\n\x08SENTENCE\x10\x00\x12\t\n\x05TOKEN\x10\x01\x12\x13\n\x0fQUESTION_ANSWER\x10\x02\x12\x0c\n\x08REPHRASE\x10\x03\x12\r\n\tSUMMARIZE\x10\x04\x12\x12\n\x0e\x45XTRACT_TABLES\x10\x05\x12\n\n\x06RERANK\x10\x06\x12\r\n\tRELATIONS\x10\x07\x12\n\n\x06SPEECH\x10\x08\x12\x0b\n\x07\x43\x41PTION\x10\t*j\n\nClientType\x12\x07\n\x03\x41PI\x10\x00\x12\x07\n\x03WEB\x10\x01\x12\n\n\x06WIDGET\x10\x02\x12\x0b\n\x07\x44\x45SKTOP\x10\x03\x12\r\n\tDASHBOARD\x10\x04\x12\x14\n\x10\x43HROME_EXTENSION\x10\x05\x12\x0c\n\x08INTERNAL\x10\x06\x62\x06proto3')
-
-_globals = globals()
-_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'protos.kb_usage_pb2', _globals)
-if _descriptor._USE_C_DESCRIPTORS == False:
-  DESCRIPTOR._options = None
-  _globals['_PROCESS'].fields_by_name['media_files']._options = None
-  _globals['_PROCESS'].fields_by_name['media_files']._serialized_options = b'\030\001'
-  _globals['_KBSOURCE']._serialized_start=1184
-  _globals['_KBSOURCE']._serialized_end=1218
-  _globals['_SERVICE']._serialized_start=1220
-  _globals['_SERVICE']._serialized_end=1273
-  _globals['_SEARCHTYPE']._serialized_start=1275
-  _globals['_SEARCHTYPE']._serialized_end=1312
-  _globals['_PREDICTTYPE']._serialized_start=1315
-  _globals['_PREDICTTYPE']._serialized_end=1475
-  _globals['_CLIENTTYPE']._serialized_start=1477
-  _globals['_CLIENTTYPE']._serialized_end=1583
-  _globals['_PROCESS']._serialized_start=69
-  _globals['_PROCESS']._serialized_end=311
-  _globals['_STORAGE']._serialized_start=314
-  _globals['_STORAGE']._serialized_end=463
-  _globals['_SEARCH']._serialized_start=465
-  _globals['_SEARCH']._serialized_end=585
-  _globals['_PREDICT']._serialized_start=588
-  _globals['_PREDICT']._serialized_end=755
-  _globals['_KBUSAGE']._serialized_start=758
-  _globals['_KBUSAGE']._serialized_end=1123
-  _globals['_KBUSAGEAGGREGATED']._serialized_start=1125
-  _globals['_KBUSAGEAGGREGATED']._serialized_end=1182
-# @@protoc_insertion_point(module_scope)

nucliadb_utils/nuclia_usage/protos/kb_usage_pb2.pyi

@@ -1,344 +0,0 @@
-# Copyright (C) 2021 Bosutech XXI S.L.
-#
-# nucliadb is offered under the AGPL v3.0 and as commercial software.
-# For commercial licensing, contact us at info@nuclia.com.
-#
-# AGPL:
-# This program is free software: you can redistribute it and/or modify
-# it under the terms of the GNU Affero General Public License as
-# published by the Free Software Foundation, either version 3 of the
-# License, or (at your option) any later version.
-#
-# This program is distributed in the hope that it will be useful,
-# but WITHOUT ANY WARRANTY; without even the implied warranty of
-# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
-# GNU Affero General Public License for more details.
-#
-# You should have received a copy of the GNU Affero General Public License
-# along with this program. If not, see <http://www.gnu.org/licenses/>.
-#
-# flake8: noqa
-"""
-@generated by mypy-protobuf.  Do not edit manually!
-isort:skip_file
-"""
-import builtins
-import collections.abc
-import google.protobuf.descriptor
-import google.protobuf.internal.containers
-import google.protobuf.internal.enum_type_wrapper
-import google.protobuf.message
-import google.protobuf.timestamp_pb2
-import sys
-import typing
-
-if sys.version_info >= (3, 10):
-    import typing as typing_extensions
-else:
-    import typing_extensions
-
-DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
-
-class _KBSource:
-    ValueType = typing.NewType("ValueType", builtins.int)
-    V: typing_extensions.TypeAlias = ValueType
-
-class _KBSourceEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_KBSource.ValueType], builtins.type):
-    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
-    HOSTED: _KBSource.ValueType  # 0
-    ONPREM: _KBSource.ValueType  # 1
-
-class KBSource(_KBSource, metaclass=_KBSourceEnumTypeWrapper): ...
-
-HOSTED: KBSource.ValueType  # 0
-ONPREM: KBSource.ValueType  # 1
-global___KBSource = KBSource
-
-class _Service:
-    ValueType = typing.NewType("ValueType", builtins.int)
-    V: typing_extensions.TypeAlias = ValueType
-
-class _ServiceEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_Service.ValueType], builtins.type):
-    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
-    PREDICT: _Service.ValueType  # 0
-    PROCESSING: _Service.ValueType  # 1
-    NUCLIA_DB: _Service.ValueType  # 2
-
-class Service(_Service, metaclass=_ServiceEnumTypeWrapper): ...
-
-PREDICT: Service.ValueType  # 0
-PROCESSING: Service.ValueType  # 1
-NUCLIA_DB: Service.ValueType  # 2
-global___Service = Service
-
-class _SearchType:
-    ValueType = typing.NewType("ValueType", builtins.int)
-    V: typing_extensions.TypeAlias = ValueType
-
-class _SearchTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_SearchType.ValueType], builtins.type):
-    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
-    SEARCH: _SearchType.ValueType  # 0
-    SUGGEST: _SearchType.ValueType  # 1
-
-class SearchType(_SearchType, metaclass=_SearchTypeEnumTypeWrapper): ...
-
-SEARCH: SearchType.ValueType  # 0
-SUGGEST: SearchType.ValueType  # 1
-global___SearchType = SearchType
-
-class _PredictType:
-    ValueType = typing.NewType("ValueType", builtins.int)
-    V: typing_extensions.TypeAlias = ValueType
-
-class _PredictTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_PredictType.ValueType], builtins.type):
-    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
-    SENTENCE: _PredictType.ValueType  # 0
-    TOKEN: _PredictType.ValueType  # 1
-    QUESTION_ANSWER: _PredictType.ValueType  # 2
-    REPHRASE: _PredictType.ValueType  # 3
-    SUMMARIZE: _PredictType.ValueType  # 4
-    EXTRACT_TABLES: _PredictType.ValueType  # 5
-    RERANK: _PredictType.ValueType  # 6
-    RELATIONS: _PredictType.ValueType  # 7
-    SPEECH: _PredictType.ValueType  # 8
-    CAPTION: _PredictType.ValueType  # 9
-
-class PredictType(_PredictType, metaclass=_PredictTypeEnumTypeWrapper): ...
-
-SENTENCE: PredictType.ValueType  # 0
-TOKEN: PredictType.ValueType  # 1
-QUESTION_ANSWER: PredictType.ValueType  # 2
-REPHRASE: PredictType.ValueType  # 3
-SUMMARIZE: PredictType.ValueType  # 4
-EXTRACT_TABLES: PredictType.ValueType  # 5
-RERANK: PredictType.ValueType  # 6
-RELATIONS: PredictType.ValueType  # 7
-SPEECH: PredictType.ValueType  # 8
-CAPTION: PredictType.ValueType  # 9
-global___PredictType = PredictType
-
-class _ClientType:
-    ValueType = typing.NewType("ValueType", builtins.int)
-    V: typing_extensions.TypeAlias = ValueType
-
-class _ClientTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_ClientType.ValueType], builtins.type):
-    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
-    API: _ClientType.ValueType  # 0
-    WEB: _ClientType.ValueType  # 1
-    WIDGET: _ClientType.ValueType  # 2
-    DESKTOP: _ClientType.ValueType  # 3
-    DASHBOARD: _ClientType.ValueType  # 4
-    CHROME_EXTENSION: _ClientType.ValueType  # 5
-    INTERNAL: _ClientType.ValueType  # 6
-
-class ClientType(_ClientType, metaclass=_ClientTypeEnumTypeWrapper): ...
-
-API: ClientType.ValueType  # 0
-WEB: ClientType.ValueType  # 1
-WIDGET: ClientType.ValueType  # 2
-DESKTOP: ClientType.ValueType  # 3
-DASHBOARD: ClientType.ValueType  # 4
-CHROME_EXTENSION: ClientType.ValueType  # 5
-INTERNAL: ClientType.ValueType  # 6
-global___ClientType = ClientType
-
-@typing_extensions.final
-class Process(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    CLIENT_FIELD_NUMBER: builtins.int
-    SLOW_PROCESSING_TIME_FIELD_NUMBER: builtins.int
-    PRE_PROCESSING_TIME_FIELD_NUMBER: builtins.int
-    BYTES_FIELD_NUMBER: builtins.int
-    CHARS_FIELD_NUMBER: builtins.int
-    MEDIA_SECONDS_FIELD_NUMBER: builtins.int
-    PAGES_FIELD_NUMBER: builtins.int
-    PARAGRAPHS_FIELD_NUMBER: builtins.int
-    MEDIA_FILES_FIELD_NUMBER: builtins.int
-    NUM_PROCESSED_FIELD_NUMBER: builtins.int
-    client: global___ClientType.ValueType
-    slow_processing_time: builtins.float
-    pre_processing_time: builtins.float
-    bytes: builtins.int
-    chars: builtins.int
-    media_seconds: builtins.int
-    pages: builtins.int
-    paragraphs: builtins.int
-    media_files: builtins.int
-    num_processed: builtins.int
-    def __init__(
-        self,
-        *,
-        client: global___ClientType.ValueType = ...,
-        slow_processing_time: builtins.float = ...,
-        pre_processing_time: builtins.float = ...,
-        bytes: builtins.int = ...,
-        chars: builtins.int = ...,
-        media_seconds: builtins.int = ...,
-        pages: builtins.int = ...,
-        paragraphs: builtins.int = ...,
-        media_files: builtins.int = ...,
-        num_processed: builtins.int = ...,
-    ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["bytes", b"bytes", "chars", b"chars", "client", b"client", "media_files", b"media_files", "media_seconds", b"media_seconds", "num_processed", b"num_processed", "pages", b"pages", "paragraphs", b"paragraphs", "pre_processing_time", b"pre_processing_time", "slow_processing_time", b"slow_processing_time"]) -> None: ...
-
-global___Process = Process
-
-@typing_extensions.final
-class Storage(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    PARAGRAPHS_FIELD_NUMBER: builtins.int
-    FIELDS_FIELD_NUMBER: builtins.int
-    RESOURCES_FIELD_NUMBER: builtins.int
-    BYTES_FIELD_NUMBER: builtins.int
-    paragraphs: builtins.int
-    fields: builtins.int
-    resources: builtins.int
-    bytes: builtins.int
-    def __init__(
-        self,
-        *,
-        paragraphs: builtins.int | None = ...,
-        fields: builtins.int | None = ...,
-        resources: builtins.int | None = ...,
-        bytes: builtins.int | None = ...,
-    ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["_bytes", b"_bytes", "_fields", b"_fields", "_paragraphs", b"_paragraphs", "_resources", b"_resources", "bytes", b"bytes", "fields", b"fields", "paragraphs", b"paragraphs", "resources", b"resources"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["_bytes", b"_bytes", "_fields", b"_fields", "_paragraphs", b"_paragraphs", "_resources", b"_resources", "bytes", b"bytes", "fields", b"fields", "paragraphs", b"paragraphs", "resources", b"resources"]) -> None: ...
-    @typing.overload
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["_bytes", b"_bytes"]) -> typing_extensions.Literal["bytes"] | None: ...
-    @typing.overload
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["_fields", b"_fields"]) -> typing_extensions.Literal["fields"] | None: ...
-    @typing.overload
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["_paragraphs", b"_paragraphs"]) -> typing_extensions.Literal["paragraphs"] | None: ...
-    @typing.overload
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["_resources", b"_resources"]) -> typing_extensions.Literal["resources"] | None: ...
-
-global___Storage = Storage
-
-@typing_extensions.final
-class Search(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    CLIENT_FIELD_NUMBER: builtins.int
-    TYPE_FIELD_NUMBER: builtins.int
-    TOKENS_FIELD_NUMBER: builtins.int
-    NUM_SEARCHES_FIELD_NUMBER: builtins.int
-    client: global___ClientType.ValueType
-    type: global___SearchType.ValueType
-    tokens: builtins.int
-    num_searches: builtins.int
-    def __init__(
-        self,
-        *,
-        client: global___ClientType.ValueType = ...,
-        type: global___SearchType.ValueType = ...,
-        tokens: builtins.int = ...,
-        num_searches: builtins.int = ...,
-    ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["client", b"client", "num_searches", b"num_searches", "tokens", b"tokens", "type", b"type"]) -> None: ...
-
-global___Search = Search
-
-@typing_extensions.final
-class Predict(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    CLIENT_FIELD_NUMBER: builtins.int
-    TYPE_FIELD_NUMBER: builtins.int
-    MODEL_FIELD_NUMBER: builtins.int
-    INPUT_FIELD_NUMBER: builtins.int
-    OUTPUT_FIELD_NUMBER: builtins.int
-    IMAGE_FIELD_NUMBER: builtins.int
-    NUM_PREDICTS_FIELD_NUMBER: builtins.int
-    client: global___ClientType.ValueType
-    type: global___PredictType.ValueType
-    model: builtins.str
-    input: builtins.int
-    output: builtins.int
-    image: builtins.int
-    num_predicts: builtins.int
-    def __init__(
-        self,
-        *,
-        client: global___ClientType.ValueType = ...,
-        type: global___PredictType.ValueType = ...,
-        model: builtins.str = ...,
-        input: builtins.int = ...,
-        output: builtins.int = ...,
-        image: builtins.int = ...,
-        num_predicts: builtins.int = ...,
-    ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["client", b"client", "image", b"image", "input", b"input", "model", b"model", "num_predicts", b"num_predicts", "output", b"output", "type", b"type"]) -> None: ...
-
-global___Predict = Predict
-
-@typing_extensions.final
-class KbUsage(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    SERVICE_FIELD_NUMBER: builtins.int
-    TIMESTAMP_FIELD_NUMBER: builtins.int
-    ACCOUNT_ID_FIELD_NUMBER: builtins.int
-    KB_ID_FIELD_NUMBER: builtins.int
-    KB_SOURCE_FIELD_NUMBER: builtins.int
-    PROCESSES_FIELD_NUMBER: builtins.int
-    PREDICTS_FIELD_NUMBER: builtins.int
-    SEARCHES_FIELD_NUMBER: builtins.int
-    STORAGE_FIELD_NUMBER: builtins.int
-    service: global___Service.ValueType
-    """Identifiers"""
-    @property
-    def timestamp(self) -> google.protobuf.timestamp_pb2.Timestamp: ...
-    account_id: builtins.str
-    kb_id: builtins.str
-    kb_source: global___KBSource.ValueType
-    @property
-    def processes(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___Process]:
-        """Data"""
-    @property
-    def predicts(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___Predict]: ...
-    @property
-    def searches(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___Search]: ...
-    @property
-    def storage(self) -> global___Storage: ...
-    def __init__(
-        self,
-        *,
-        service: global___Service.ValueType = ...,
-        timestamp: google.protobuf.timestamp_pb2.Timestamp | None = ...,
-        account_id: builtins.str | None = ...,
-        kb_id: builtins.str | None = ...,
-        kb_source: global___KBSource.ValueType = ...,
-        processes: collections.abc.Iterable[global___Process] | None = ...,
-        predicts: collections.abc.Iterable[global___Predict] | None = ...,
-        searches: collections.abc.Iterable[global___Search] | None = ...,
-        storage: global___Storage | None = ...,
-    ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["_account_id", b"_account_id", "_kb_id", b"_kb_id", "_storage", b"_storage", "account_id", b"account_id", "kb_id", b"kb_id", "storage", b"storage", "timestamp", b"timestamp"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["_account_id", b"_account_id", "_kb_id", b"_kb_id", "_storage", b"_storage", "account_id", b"account_id", "kb_id", b"kb_id", "kb_source", b"kb_source", "predicts", b"predicts", "processes", b"processes", "searches", b"searches", "service", b"service", "storage", b"storage", "timestamp", b"timestamp"]) -> None: ...
-    @typing.overload
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["_account_id", b"_account_id"]) -> typing_extensions.Literal["account_id"] | None: ...
-    @typing.overload
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["_kb_id", b"_kb_id"]) -> typing_extensions.Literal["kb_id"] | None: ...
-    @typing.overload
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["_storage", b"_storage"]) -> typing_extensions.Literal["storage"] | None: ...
-
-global___KbUsage = KbUsage
-
-@typing_extensions.final
-class KbUsageAggregated(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    KB_USAGES_FIELD_NUMBER: builtins.int
-    @property
-    def kb_usages(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___KbUsage]: ...
-    def __init__(
-        self,
-        *,
-        kb_usages: collections.abc.Iterable[global___KbUsage] | None = ...,
-    ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["kb_usages", b"kb_usages"]) -> None: ...
-
-global___KbUsageAggregated = KbUsageAggregated

nucliadb_utils/nuclia_usage/protos/kb_usage_pb2_grpc.py

@@ -1,23 +0,0 @@
-# Copyright (C) 2021 Bosutech XXI S.L.
-#
-# nucliadb is offered under the AGPL v3.0 and as commercial software.
-# For commercial licensing, contact us at info@nuclia.com.
-#
-# AGPL:
-# This program is free software: you can redistribute it and/or modify
-# it under the terms of the GNU Affero General Public License as
-# published by the Free Software Foundation, either version 3 of the
-# License, or (at your option) any later version.
-#
-# This program is distributed in the hope that it will be useful,
-# but WITHOUT ANY WARRANTY; without even the implied warranty of
-# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
-# GNU Affero General Public License for more details.
-#
-# You should have received a copy of the GNU Affero General Public License
-# along with this program. If not, see <http://www.gnu.org/licenses/>.
-#
-# flake8: noqa
-# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
-"""Client and server classes corresponding to protobuf-defined services."""
-import grpc

nucliadb_utils/nuclia_usage/protos/kb_usage_pb2_grpc.pyi

@@ -1,24 +0,0 @@
-# Copyright (C) 2021 Bosutech XXI S.L.
-#
-# nucliadb is offered under the AGPL v3.0 and as commercial software.
-# For commercial licensing, contact us at info@nuclia.com.
-#
-# AGPL:
-# This program is free software: you can redistribute it and/or modify
-# it under the terms of the GNU Affero General Public License as
-# published by the Free Software Foundation, either version 3 of the
-# License, or (at your option) any later version.
-#
-# This program is distributed in the hope that it will be useful,
-# but WITHOUT ANY WARRANTY; without even the implied warranty of
-# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
-# GNU Affero General Public License for more details.
-#
-# You should have received a copy of the GNU Affero General Public License
-# along with this program. If not, see <http://www.gnu.org/licenses/>.
-#
-# flake8: noqa
-"""
-@generated by mypy-protobuf.  Do not edit manually!
-isort:skip_file
-"""