Module llm (0.4.0)

LLM models.

Classes

PaLM2TextEmbeddingGenerator

PaLM2TextEmbeddingGenerator(
    session: typing.Optional[bigframes.session.Session] = None,
    connection_name: typing.Optional[str] = None,
)

PaLM2 text embedding generator LLM model.

Parameters
NameDescription
session bigframes.Session or None

BQ session to create the model. If None, use the global default session.

connection_name str or None

connection to connect with remote service. str of the format <PROJECT_NUMBER/PROJECT_ID>.

PaLM2TextGenerator

PaLM2TextGenerator(
    session: typing.Optional[bigframes.session.Session] = None,
    connection_name: typing.Optional[str] = None,
)

PaLM2 text generator LLM model.

Parameters
NameDescription
session bigframes.Session or None

BQ session to create the model. If None, use the global default session.

connection_name str or None

connection to connect with remote service. str of the format <PROJECT_NUMBER/PROJECT_ID>.