ReferencePython SDK ReferenceNemoguardrailsTracingAdapters

nemoguardrails.tracing.adapters.registry

View as Markdown

Module Contents

Classes

NameDescription
LogAdapterRegistry-

Functions

NameDescription
register_log_adapterRegister an embedding provider.

API

class nemoguardrails.tracing.adapters.registry.LogAdapterRegistry()

Bases: Registry

nemoguardrails.tracing.adapters.registry.LogAdapterRegistry.validate(
    name: str,
    item: typing.Type
) -> None

Validate the item to be registered. Raises: TypeError: If an item is not an instance of InteractionLogAdapter.

nemoguardrails.tracing.adapters.registry.register_log_adapter(
    model: typing.Type,
    name: typing.Optional[str] = None
)

Register an embedding provider.

Parameters:

model
Type[EmbeddingModel]

The embedding model class.

name
strDefaults to None

The name of the embedding engine.

Raises:

  • ValueError: If the engine name is not provided and the model does not have an engine name.
  • TypeError: If the model is not an instance of EmbeddingModel.
  • ValueError: If the model does not have ‘encode’ or ‘encode_async’ methods.