Module llm (0.13.0)

LLM models.

Classes

PaLM2TextEmbeddingGenerator

  PaLM2TextEmbeddingGenerator 
 ( 
 model_name 
 : 
 typing 
 . 
 Literal 
 [ 
 "textembedding-gecko" 
 , 
 "textembedding-gecko-multilingual" 
 ] 
 = 
 "textembedding-gecko" 
 , 
 session 
 : 
 typing 
 . 
 Optional 
 [ 
 bigframes 
 . 
 session 
 . 
 Session 
 ] 
 = 
 None 
 , 
 connection_name 
 : 
 typing 
 . 
 Optional 
 [ 
 str 
 ] 
 = 
 None 
 , 
 )

PaLM2 text embedding generator LLM model.

Parameters

Name

Description

model_name

str, Default to "textembedding-gecko"

The model for text embedding. “textembedding-gecko” returns model embeddings for text inputs. "textembedding-gecko-multilingual" returns model embeddings for text inputs which support over 100 languages Default to "textembedding-gecko".

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 
 ( 
 model_name 
 : 
 typing 
 . 
 Literal 
 [ 
 "text-bison" 
 , 
 "text-bison-32k" 
 ] 
 = 
 "text-bison" 
 , 
 session 
 : 
 typing 
 . 
 Optional 
 [ 
 bigframes 
 . 
 session 
 . 
 Session 
 ] 
 = 
 None 
 , 
 connection_name 
 : 
 typing 
 . 
 Optional 
 [ 
 str 
 ] 
 = 
 None 
 , 
 )

PaLM2 text generator LLM model.