Module llm (0.6.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

Name

Description

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.