.coveragerc
.gitignore
.isort.cfg
.pre-commit-config.yaml
.prettierignore
.prettierrc.yaml
.readthedocs.yaml
.whitesource
CODEOWNERS
CONTRIBUTING.md
LICENSE
README.md
SECURITY.md
caikit-overview.png
code-of-conduct.md
pyproject.toml
releases.md
setup_requirements.txt
tox.ini
.github/dependabot.yml
.github/pull_request_template.md
.github/ISSUE_TEMPLATE/bug_report.md
.github/ISSUE_TEMPLATE/feature_request.md
.github/ISSUE_TEMPLATE/user_story.md
.github/workflows/build-library.yml
.github/workflows/lint-code.yml
.github/workflows/publish-library.yml
caikit/__init__.py
caikit/_version.py
caikit/version.py
caikit.egg-info/PKG-INFO
caikit.egg-info/SOURCES.txt
caikit.egg-info/dependency_links.txt
caikit.egg-info/entry_points.txt
caikit.egg-info/requires.txt
caikit.egg-info/top_level.txt
caikit/config/__init__.py
caikit/config/config.py
caikit/config/config.yml
caikit/core/__init__.py
caikit/core/model_manager.py
caikit/core/registries.py
caikit/core/task.py
caikit/core/augmentors/__init__.py
caikit/core/augmentors/base.py
caikit/core/augmentors/merged_augmentor.py
caikit/core/augmentors/schemes/__init__.py
caikit/core/augmentors/schemes/always_selection_scheme.py
caikit/core/augmentors/schemes/base.py
caikit/core/augmentors/schemes/random_multi_selection_scheme.py
caikit/core/augmentors/schemes/random_single_selection_scheme.py
caikit/core/data_model/__init__.py
caikit/core/data_model/base.py
caikit/core/data_model/dataobject.py
caikit/core/data_model/enums.py
caikit/core/data_model/json_dict.py
caikit/core/data_model/package.py
caikit/core/data_model/producer.py
caikit/core/data_model/timestamp.py
caikit/core/data_model/training_status.py
caikit/core/data_model/data_backends/__init__.py
caikit/core/data_model/data_backends/base.py
caikit/core/data_model/data_backends/dict_backend.py
caikit/core/data_model/protobufs/__init__.py
caikit/core/data_model/streams/__init__.py
caikit/core/data_model/streams/converter.py
caikit/core/data_model/streams/csv_column_formatter.py
caikit/core/data_model/streams/data_stream.py
caikit/core/data_model/streams/multipart_decoder.py
caikit/core/data_model/streams/resolver.py
caikit/core/data_model/streams/validator.py
caikit/core/exceptions/__init__.py
caikit/core/exceptions/caikit_core_exception.py
caikit/core/exceptions/error_handler.py
caikit/core/exceptions/validation_error.py
caikit/core/model_management/__init__.py
caikit/core/model_management/factories.py
caikit/core/model_management/local_model_finder.py
caikit/core/model_management/local_model_initializer.py
caikit/core/model_management/local_model_trainer.py
caikit/core/model_management/model_finder_base.py
caikit/core/model_management/model_initializer_base.py
caikit/core/model_management/model_trainer_base.py
caikit/core/model_management/multi_model_finder.py
caikit/core/model_management/multi_model_initializer.py
caikit/core/module_backends/__init__.py
caikit/core/module_backends/backend_types.py
caikit/core/module_backends/base.py
caikit/core/module_backends/local_backend.py
caikit/core/modules/__init__.py
caikit/core/modules/base.py
caikit/core/modules/config.py
caikit/core/modules/decorator.py
caikit/core/modules/loader.py
caikit/core/modules/meta.py
caikit/core/modules/saver.py
caikit/core/signature_parsing/__init__.py
caikit/core/signature_parsing/docstrings.py
caikit/core/signature_parsing/module_signature.py
caikit/core/signature_parsing/parsers.py
caikit/core/toolkit/__init__.py
caikit/core/toolkit/compatibility.py
caikit/core/toolkit/error_handler.py
caikit/core/toolkit/factory.py
caikit/core/toolkit/fileio.py
caikit/core/toolkit/logging.py
caikit/core/toolkit/name_tools.py
caikit/core/toolkit/quality_evaluation.py
caikit/core/toolkit/reversible_hasher.py
caikit/core/toolkit/serializers.py
caikit/core/toolkit/sync_to_async.py
caikit/core/toolkit/wip_decorator.py
caikit/core/toolkit/concurrency/__init__.py
caikit/core/toolkit/concurrency/destroyable.py
caikit/core/toolkit/concurrency/destroyable_process.py
caikit/core/toolkit/concurrency/destroyable_thread.py
caikit/core/toolkit/concurrency/pickling_exception.py
caikit/core/toolkit/errors/__init__.py
caikit/interfaces/__init__.py
caikit/interfaces/common/__init__.py
caikit/interfaces/common/data_model/__init__.py
caikit/interfaces/common/data_model/file.py
caikit/interfaces/common/data_model/primitive_sequences.py
caikit/interfaces/common/data_model/producer.py
caikit/interfaces/common/data_model/remote.py
caikit/interfaces/common/data_model/stream_sources.py
caikit/interfaces/common/data_model/vectors.py
caikit/interfaces/nlp/__init__.py
caikit/interfaces/nlp/tasks.py
caikit/interfaces/nlp/data_model/__init__.py
caikit/interfaces/nlp/data_model/classification.py
caikit/interfaces/nlp/data_model/embedding_vectors.py
caikit/interfaces/nlp/data_model/package.py
caikit/interfaces/nlp/data_model/reranker.py
caikit/interfaces/nlp/data_model/sentence_similarity.py
caikit/interfaces/nlp/data_model/text.py
caikit/interfaces/nlp/data_model/text_generation.py
caikit/interfaces/runtime/__init__.py
caikit/interfaces/runtime/data_model/__init__.py
caikit/interfaces/runtime/data_model/info.py
caikit/interfaces/runtime/data_model/training_management.py
caikit/interfaces/ts/__init__.py
caikit/interfaces/ts/tasks.py
caikit/interfaces/ts/data_model/__init__.py
caikit/interfaces/ts/data_model/_single_timeseries.py
caikit/interfaces/ts/data_model/package.py
caikit/interfaces/ts/data_model/time_types.py
caikit/interfaces/ts/data_model/timeseries.py
caikit/interfaces/ts/data_model/timeseries_evaluation.py
caikit/interfaces/ts/data_model/backends/__init__.py
caikit/interfaces/ts/data_model/backends/_spark_backends.py
caikit/interfaces/ts/data_model/backends/base.py
caikit/interfaces/ts/data_model/backends/dfcache.py
caikit/interfaces/ts/data_model/backends/pandas_backends.py
caikit/interfaces/ts/data_model/backends/spark_util.py
caikit/interfaces/ts/data_model/backends/util.py
caikit/interfaces/ts/data_model/toolkit/__init__.py
caikit/interfaces/ts/data_model/toolkit/optional_dependencies.py
caikit/interfaces/ts/data_model/toolkit/sparkconf.py
caikit/interfaces/vision/__init__.py
caikit/interfaces/vision/data_model/__init__.py
caikit/interfaces/vision/data_model/image.py
caikit/interfaces/vision/data_model/package.py
caikit/interfaces/vision/data_model/backends/__init__.py
caikit/interfaces/vision/data_model/backends/image_pil_backend.py
caikit/runtime/__init__.py
caikit/runtime/__main__.py
caikit/runtime/dump_services.py
caikit/runtime/grpc_server.py
caikit/runtime/names.py
caikit/runtime/server_base.py
caikit/runtime/service_factory.py
caikit/runtime/train.py
caikit/runtime/client/__init__.py
caikit/runtime/client/remote_config.py
caikit/runtime/client/remote_model_finder.py
caikit/runtime/client/remote_model_initializer.py
caikit/runtime/client/remote_module_base.py
caikit/runtime/client/utils.py
caikit/runtime/http_server/__init__.py
caikit/runtime/http_server/__main__.py
caikit/runtime/http_server/http_server.py
caikit/runtime/http_server/pydantic_wrapper.py
caikit/runtime/http_server/request_aborter.py
caikit/runtime/http_server/utils.py
caikit/runtime/interceptors/__init__.py
caikit/runtime/interceptors/caikit_runtime_server_wrapper.py
caikit/runtime/metrics/__init__.py
caikit/runtime/metrics/rpc_meter.py
caikit/runtime/model_management/__init__.py
caikit/runtime/model_management/batcher.py
caikit/runtime/model_management/loaded_model.py
caikit/runtime/model_management/model_loader.py
caikit/runtime/model_management/model_manager.py
caikit/runtime/model_management/model_sizer.py
caikit/runtime/protobufs/README.md
caikit/runtime/protobufs/__init__.py
caikit/runtime/protobufs/model_mesh_pb2.py
caikit/runtime/protobufs/model_mesh_pb2_grpc.py
caikit/runtime/protobufs/model_runtime_pb2.py
caikit/runtime/protobufs/model_runtime_pb2_grpc.py
caikit/runtime/protobufs/process_pb2.py
caikit/runtime/protobufs/process_pb2_grpc.py
caikit/runtime/protobufs/protos/model-mesh.proto
caikit/runtime/protobufs/protos/model-runtime.proto
caikit/runtime/protobufs/protos/process.proto
caikit/runtime/service_generation/__init__.py
caikit/runtime/service_generation/compatibility_checker.py
caikit/runtime/service_generation/create_service.py
caikit/runtime/service_generation/data_stream_source.py
caikit/runtime/service_generation/protoable.py
caikit/runtime/service_generation/rpcs.py
caikit/runtime/service_generation/type_helpers.py
caikit/runtime/servicers/__init__.py
caikit/runtime/servicers/global_predict_servicer.py
caikit/runtime/servicers/global_train_servicer.py
caikit/runtime/servicers/info_servicer.py
caikit/runtime/servicers/model_runtime_servicer.py
caikit/runtime/servicers/model_train_servicer.py
caikit/runtime/servicers/training_management_servicer.py
caikit/runtime/types/__init__.py
caikit/runtime/types/aborted_exception.py
caikit/runtime/types/caikit_runtime_exception.py
caikit/runtime/utils/__init__.py
caikit/runtime/utils/import_util.py
caikit/runtime/utils/servicer_util.py
caikit/runtime/work_management/__init__.py
caikit/runtime/work_management/abortable_context.py
caikit/runtime/work_management/rpc_aborter.py
caikit_health_probe/__init__.py
caikit_health_probe/__main__.py
docs/Makefile
docs/make.bat
docs/adrs/001-module.md
docs/adrs/010-data-model-definition.md
docs/adrs/015-runtime-service-generation.md
docs/adrs/018-shared-backends.md
docs/adrs/019-loader-stack.md
docs/adrs/020-model-management-abstractions.md
docs/adrs/021-http-server.md
docs/adrs/022-pluggable-runtime-io.md
docs/adrs/023-info-service-endpoint.md
docs/adrs/024-remote-module-invocation.md
docs/adrs/README.md
docs/architecture_club/04-25-23.md
docs/architecture_club/README.md
docs/source/conf.py
docs/source/index.rst
examples/sample_lib/README.md
examples/sample_lib/client.py
examples/sample_lib/client_proto.py
examples/sample_lib/requirements.txt
examples/sample_lib/start_runtime_with_sample_lib.py
examples/sample_lib/generated/__init__.py
examples/text-sentiment/README.md
examples/text-sentiment/client.py
examples/text-sentiment/requirements.txt
examples/text-sentiment/start_runtime.py
examples/text-sentiment/models/text_sentiment/config.yml
examples/text-sentiment/text_sentiment/__init__.py
examples/text-sentiment/text_sentiment/config.yml
examples/text-sentiment/text_sentiment/data_model/__init__.py
examples/text-sentiment/text_sentiment/data_model/classification.py
examples/text-sentiment/text_sentiment/runtime_model/__init__.py
examples/text-sentiment/text_sentiment/runtime_model/hf_module.py
scripts/check_deps.sh
scripts/dsconverter.py
scripts/fmt.sh
scripts/test_dsconverter.py
tests/__init__.py
tests/base.py
tests/conftest.py
tests/data_model_helpers.py
tests/config/__init__.py
tests/config/test_configs.py
tests/core/__init__.py
tests/core/helpers.py
tests/core/test_imports.py
tests/core/test_model_manager.py
tests/core/test_no_write_permissions.py
tests/core/test_task.py
tests/core/augmentors/test_augmentor_base.py
tests/core/augmentors/test_merged_augmentors.py
tests/core/data_model/__init__.py
tests/core/data_model/test_base.py
tests/core/data_model/test_dataobject.py
tests/core/data_model/test_enums.py
tests/core/data_model/test_json_dict.py
tests/core/data_model/test_producer.py
tests/core/data_model/test_timestamp.py
tests/core/data_model/data_backends/__init__.py
tests/core/data_model/data_backends/test_dict_backend.py
tests/core/data_model/streams/test_converter.py
tests/core/data_model/streams/test_csv_column_formatter.py
tests/core/data_model/streams/test_data_stream.py
tests/core/data_model/streams/test_multipart_decoder.py
tests/core/data_model/streams/test_resolver.py
tests/core/data_model/streams/test_validator.py
tests/core/model_management/__init__.py
tests/core/model_management/test_factories.py
tests/core/model_management/test_local_model_trainer.py
tests/core/model_management/test_multi_model_finder.py
tests/core/model_management/test_multi_model_initializer.py
tests/core/module_backends/__init__.py
tests/core/module_backends/test_backend_types.py
tests/core/modules/__init__.py
tests/core/modules/test_module.py
tests/core/modules/test_module_metadata.py
tests/core/signature_parsing/__init__.py
tests/core/signature_parsing/test_docstrings.py
tests/core/signature_parsing/test_parsers.py
tests/core/toolkit/__init__.py
tests/core/toolkit/test_compatibility.py
tests/core/toolkit/test_error_handler.py
tests/core/toolkit/test_factory.py
tests/core/toolkit/test_fileio.py
tests/core/toolkit/test_quality_evaluation.py
tests/core/toolkit/test_reversible_hasher.py
tests/core/toolkit/test_serializers.py
tests/core/toolkit/test_sync_to_async.py
tests/core/toolkit/test_wip_decorator.py
tests/core/toolkit/concurrency/__init__.py
tests/core/toolkit/concurrency/test_destroyable_process.py
tests/core/toolkit/concurrency/test_destroyable_thread.py
tests/core/toolkit/concurrency/test_exception_pickler.py
tests/examples/__init__.py
tests/examples/shared.py
tests/examples/test_examples.py
tests/fixtures/__init__.py
tests/fixtures/dummy_dataset.json
tests/fixtures/dummy_module.zip
tests/fixtures/dummy_module_no_nesting.zip
tests/fixtures/fixtures.py
tests/fixtures/invalid.zip
tests/fixtures/linux.txt
tests/fixtures/sample.csv
tests/fixtures/config/config.yml
tests/fixtures/data/sample_image.png
tests/fixtures/data_stream_inputs/README.md
tests/fixtures/data_stream_inputs/control_chars.jsonl
tests/fixtures/data_stream_inputs/multi_part_csv.txt
tests/fixtures/data_stream_inputs/multi_part_json.txt
tests/fixtures/data_stream_inputs/multi_part_json_with_content_header.txt
tests/fixtures/data_stream_inputs/sample.csv
tests/fixtures/data_stream_inputs/sample.json
tests/fixtures/data_stream_inputs/sample.jsonl
tests/fixtures/data_stream_inputs/sample.txt
tests/fixtures/data_stream_inputs/sample_w_header.csv
tests/fixtures/data_stream_inputs/sample_csv_collection/a.csv
tests/fixtures/data_stream_inputs/sample_csv_collection/b.csv
tests/fixtures/data_stream_inputs/sample_csv_collection/c.csv
tests/fixtures/data_stream_inputs/sample_json_collection/a.json
tests/fixtures/data_stream_inputs/sample_json_collection/b.json
tests/fixtures/data_stream_inputs/sample_json_collection/c.json
tests/fixtures/data_stream_inputs/sample_jsonl_collection/a.jsonl
tests/fixtures/data_stream_inputs/sample_jsonl_collection/b.jsonl
tests/fixtures/data_stream_inputs/sample_jsonl_collection/c.jsonl
tests/fixtures/data_stream_inputs/sample_txt_collection/a.txt
tests/fixtures/data_stream_inputs/sample_txt_collection/b.txt
tests/fixtures/data_stream_inputs/sample_txt_collection/c.txt
tests/fixtures/dummy_module/config.yml
tests/fixtures/dummy_module/data.json
tests/fixtures/dummy_module/data.pkl
tests/fixtures/dummy_module_backend/config.yml
tests/fixtures/dummy_module_foo/config.yml
tests/fixtures/dummy_module_no_id/config.yml
tests/fixtures/dummy_module_singleton/config.yml
tests/fixtures/dummy_module_singleton/data.json
tests/fixtures/dummy_module_singleton/data.pkl
tests/fixtures/dummy_resource/config.yml
tests/fixtures/dummy_streaming_module/config.yml
tests/fixtures/models/bad_archive.zip
tests/fixtures/models/bad_model.zip
tests/fixtures/models/foo_archive.zip
tests/fixtures/models/bar/config.yml
tests/fixtures/models/box/config.yml
tests/fixtures/models/foo/config.yml
tests/fixtures/models/multi/config.yml
tests/fixtures/models/primitive/config.yml
tests/fixtures/sample_lib/__init__.py
tests/fixtures/sample_lib/config.yml
tests/fixtures/sample_lib/data_model/__init__.py
tests/fixtures/sample_lib/data_model/sample.py
tests/fixtures/sample_lib/modules/__init__.py
tests/fixtures/sample_lib/modules/file_processing/__init__.py
tests/fixtures/sample_lib/modules/file_processing/bounding_box_module.py
tests/fixtures/sample_lib/modules/geospatial/__init__.py
tests/fixtures/sample_lib/modules/geospatial/geo_streaming_module.py
tests/fixtures/sample_lib/modules/multi_task/__init__.py
tests/fixtures/sample_lib/modules/multi_task/multi_task_module.py
tests/fixtures/sample_lib/modules/other_task/__init__.py
tests/fixtures/sample_lib/modules/other_task/other_implementation.py
tests/fixtures/sample_lib/modules/sample_task/__init__.py
tests/fixtures/sample_lib/modules/sample_task/composite_module.py
tests/fixtures/sample_lib/modules/sample_task/inner_module.py
tests/fixtures/sample_lib/modules/sample_task/list_implementation.py
tests/fixtures/sample_lib/modules/sample_task/primitive_party_implementation.py
tests/fixtures/sample_lib/modules/sample_task/sample_implementation.py
tests/fixtures/sample_lib/modules/sample_task/streaming_module.py
tests/fixtures/sample_module/config.yml
tests/interfaces/__init__.py
tests/interfaces/common/test_producer.py
tests/interfaces/common/test_vectors.py
tests/interfaces/nlp/__init__.py
tests/interfaces/nlp/test_classification.py
tests/interfaces/nlp/test_nlp_tasks.py
tests/interfaces/nlp/test_reranker.py
tests/interfaces/nlp/test_text.py
tests/interfaces/nlp/test_text_generation.py
tests/interfaces/ts/__init__.py
tests/interfaces/ts/helpers.py
tests/interfaces/ts/data_model/__init__.py
tests/interfaces/ts/data_model/test_single_timeseries.py
tests/interfaces/ts/data_model/test_timeseries.py
tests/interfaces/ts/data_model/test_timeseries_evaluation.py
tests/interfaces/ts/data_model/util.py
tests/interfaces/vision/test_image.py
tests/runtime/__init__.py
tests/runtime/conftest.py
tests/runtime/test_caikit_health_probe.py
tests/runtime/test_dump_services.py
tests/runtime/test_grpc_server.py
tests/runtime/test_service_factory.py
tests/runtime/test_train.py
tests/runtime/client/__init__.py
tests/runtime/client/test_remote_model_finder.py
tests/runtime/client/test_remote_model_initializer.py
tests/runtime/generated/__init__.py
tests/runtime/http_server/test_http_server.py
tests/runtime/http_server/test_pydantic_wrapper.py
tests/runtime/http_server/test_request_aborter.py
tests/runtime/http_server/test_utils.py
tests/runtime/interceptors/__init__.py
tests/runtime/interceptors/test_caikit_runtime_server_wrapper.py
tests/runtime/metrics/__init__.py
tests/runtime/metrics/test_rpc_meter.py
tests/runtime/model_management/__init__.py
tests/runtime/model_management/test_batcher.py
tests/runtime/model_management/test_model_loader.py
tests/runtime/model_management/test_model_manager.py
tests/runtime/model_management/test_model_sizer.py
tests/runtime/service_generation/__init__.py
tests/runtime/service_generation/test_create_service.py
tests/runtime/service_generation/test_data_stream_plugins.py
tests/runtime/service_generation/test_data_stream_source.py
tests/runtime/service_generation/test_protoable.py
tests/runtime/service_generation/test_rpcs.py
tests/runtime/service_generation/test_type_helpers.py
tests/runtime/servicers/__init__.py
tests/runtime/servicers/test_global_predict_servicer_impl.py
tests/runtime/servicers/test_global_train_servicer_impl.py
tests/runtime/servicers/test_model_runtime_servicer_impl.py
tests/runtime/servicers/test_model_train_servicer_impl.py
tests/runtime/servicers/test_servicer_impl_subprocess.py
tests/runtime/servicers/test_training_management_servicer.py
tests/runtime/types/test_caikit_runtime_exception.py
tests/runtime/utils/__init__.py
tests/runtime/utils/test_import_util.py
tests/runtime/utils/test_servicer_util.py
tests/runtime/work_management/__init__.py
tests/runtime/work_management/test_abortable_context.py
tests/runtime/work_management/test_call_aborter.py