Module model_manager (0.4.0)

API documentation for model_manager module.

Classes

AlloyDBModel

AlloyDBModel(
    model_id: str,
    model_request_url: typing.Optional[str],
    model_provider: str,
    model_type: str,
    model_qualified_name: typing.Optional[str],
    model_auth_type: typing.Optional[str],
    model_auth_id: typing.Optional[str],
    model_in_transform_fn: typing.Optional[str],
    model_out_transform_fn: typing.Optional[str],
    generate_headers_fn: typing.Optional[str],
)

A model endpoint registered in AlloyDB AI model endpoint management.

Fields use the create_model argument names, so asdict(model) without model_id can be passed back to it. model_in_transform_fn, model_out_transform_fn and generate_headers_fn are read from the input_transform_fn, output_transform_fn and header_gen_fn view columns.

AlloyDBModelManager

AlloyDBModelManager(
    key: object, engine: llama_index_alloydb_pg.engine.AlloyDBEngine
)

Register, inspect and drop AlloyDB AI model endpoints.

See https://cloud.google.com/alloydb/docs/ai/register-model-endpoint.