aiq.cli.register_workflow#

Functions#

register_telemetry_exporter(config_type)

Register a workflow with optional framework_wrappers for automatic profiler hooking.

register_logging_method(config_type)

register_front_end(config_type)

Register a front end which is responsible for hosting a workflow.

register_function(config_type[, framework_wrappers])

Register a workflow with optional framework_wrappers for automatic profiler hooking.

register_llm_provider(config_type)

register_llm_client(config_type, wrapper_type)

register_embedder_provider(config_type)

register_embedder_client(config_type, wrapper_type)

register_evaluator(config_type)

register_memory(config_type)

register_retriever_provider(config_type)

register_retriever_client(config_type, wrapper_type)

register_tool_wrapper(wrapper_type)

register_registry_handler(config_type)

Module Contents#

register_telemetry_exporter(
config_type: type[aiq.cli.type_registry.TelemetryExporterConfigT],
)#

Register a workflow with optional framework_wrappers for automatic profiler hooking.

register_logging_method(
config_type: type[aiq.cli.type_registry.LoggingMethodConfigT],
)#
register_front_end(
config_type: type[aiq.data_models.front_end.FrontEndConfigT],
)#

Register a front end which is responsible for hosting a workflow.

register_function(
config_type: type[aiq.data_models.function.FunctionConfigT],
framework_wrappers: list[aiq.builder.framework_enum.LLMFrameworkEnum | str] | None = None,
)#

Register a workflow with optional framework_wrappers for automatic profiler hooking.

register_llm_provider(
config_type: type[aiq.data_models.llm.LLMBaseConfigT],
)#
register_llm_client(
config_type: type[aiq.data_models.llm.LLMBaseConfigT],
wrapper_type: aiq.builder.framework_enum.LLMFrameworkEnum | str,
)#
register_embedder_provider(
config_type: type[aiq.data_models.embedder.EmbedderBaseConfigT],
)#
register_embedder_client(
config_type: type[aiq.data_models.embedder.EmbedderBaseConfigT],
wrapper_type: aiq.builder.framework_enum.LLMFrameworkEnum | str,
)#
register_evaluator(
config_type: type[aiq.data_models.evaluator.EvaluatorBaseConfigT],
)#
register_memory(
config_type: type[aiq.data_models.memory.MemoryBaseConfigT],
)#
register_retriever_provider(
config_type: type[aiq.data_models.retriever.RetrieverBaseConfigT],
)#
register_retriever_client(
config_type: type[aiq.data_models.retriever.RetrieverBaseConfigT],
wrapper_type: aiq.builder.framework_enum.LLMFrameworkEnum | str | None,
)#
register_tool_wrapper(
wrapper_type: aiq.builder.framework_enum.LLMFrameworkEnum | str,
)#
register_registry_handler(
config_type: type[aiq.data_models.registry_handler.RegistryHandlerBaseConfigT],
)#