caikit.runtime.service_generation
Submodules
Functions
|
Logic about whether it's okay to include this set of modules in service generation |
Handles the logic to create all the RPCs for inference |
|
Handles the logic to create all the RPCs for inference jobs |
|
|
Handles the logic to create all the RPCs for training |
Package Contents
- caikit.runtime.service_generation.assert_compatible(modules: List[str], previous_modules: List[str])[source]
Logic about whether it’s okay to include this set of modules in service generation
- Args:
modules: list of module IDs that we are considering in service generation previous_modules: list of module IDs that were supported in the previous service version
- Raises:
If a new service should not be built with this set of modules
- caikit.runtime.service_generation.create_inference_rpcs(modules: List[Type[caikit.core.ModuleBase]], caikit_config: aconfig.aconfig.Config = None) List[caikit.runtime.service_generation.rpcs.CaikitRPCBase][source]
Handles the logic to create all the RPCs for inference
- caikit.runtime.service_generation.create_job_inference_rpcs(modules: List[Type[caikit.core.ModuleBase]], caikit_config: aconfig.aconfig.Config = None) List[caikit.runtime.service_generation.rpcs.CaikitRPCBase][source]
Handles the logic to create all the RPCs for inference jobs
- caikit.runtime.service_generation.create_training_rpcs(modules: List[Type[caikit.core.ModuleBase]]) List[caikit.runtime.service_generation.rpcs.CaikitRPCBase][source]
Handles the logic to create all the RPCs for training