Module llm (0.2.0)

LLM models.

Classes

PaLM2TextEmbeddingGenerator

PaLM2TextEmbeddingGenerator(
    session: bigframes.session.Session, connection_name: str
)

PaLM2 text embedding generator LLM model.

Parameters
NameDescription
session BigQuery Session

BQ session to create the model

connection_name str

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

PaLM2TextGenerator

PaLM2TextGenerator(session: bigframes.session.Session, connection_name: str)

PaLM2 text generator LLM model.

Parameters
NameDescription
session BigQuery Session

BQ session to create the model

connection_name str

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