|
56 | 56 |
|
57 | 57 | _BQML_ENDPOINT_TYPE_MAPPING = MappingProxyType(
|
58 | 58 | {
|
59 |
| - llm._TEXT_GENERATOR_BISON_ENDPOINT: llm.PaLM2TextGenerator, |
60 |
| - llm._TEXT_GENERATOR_BISON_32K_ENDPOINT: llm.PaLM2TextGenerator, |
61 |
| - llm._EMBEDDING_GENERATOR_GECKO_ENDPOINT: llm.PaLM2TextEmbeddingGenerator, |
62 |
| - llm._EMBEDDING_GENERATOR_GECKO_MULTILINGUAL_ENDPOINT: llm.PaLM2TextEmbeddingGenerator, |
63 |
| - llm._GEMINI_PRO_ENDPOINT: llm.GeminiTextGenerator, |
64 | 59 | llm._GEMINI_1P5_PRO_PREVIEW_ENDPOINT: llm.GeminiTextGenerator,
|
65 | 60 | llm._GEMINI_1P5_PRO_FLASH_PREVIEW_ENDPOINT: llm.GeminiTextGenerator,
|
66 | 61 | llm._GEMINI_1P5_PRO_001_ENDPOINT: llm.GeminiTextGenerator,
|
67 | 62 | llm._GEMINI_1P5_PRO_002_ENDPOINT: llm.GeminiTextGenerator,
|
68 | 63 | llm._GEMINI_1P5_FLASH_001_ENDPOINT: llm.GeminiTextGenerator,
|
69 | 64 | llm._GEMINI_1P5_FLASH_002_ENDPOINT: llm.GeminiTextGenerator,
|
70 | 65 | llm._GEMINI_2_FLASH_EXP_ENDPOINT: llm.GeminiTextGenerator,
|
| 66 | + llm._GEMINI_2_FLASH_001_ENDPOINT: llm.GeminiTextGenerator, |
| 67 | + llm._GEMINI_2_FLASH_LITE_001_ENDPOINT: llm.GeminiTextGenerator, |
71 | 68 | llm._CLAUDE_3_HAIKU_ENDPOINT: llm.Claude3TextGenerator,
|
72 | 69 | llm._CLAUDE_3_SONNET_ENDPOINT: llm.Claude3TextGenerator,
|
73 | 70 | llm._CLAUDE_3_5_SONNET_ENDPOINT: llm.Claude3TextGenerator,
|
@@ -95,8 +92,6 @@ def from_bq(
|
95 | 92 | imported.TensorFlowModel,
|
96 | 93 | imported.ONNXModel,
|
97 | 94 | imported.XGBoostModel,
|
98 |
| - llm.PaLM2TextGenerator, |
99 |
| - llm.PaLM2TextEmbeddingGenerator, |
100 | 95 | llm.Claude3TextGenerator,
|
101 | 96 | llm.TextEmbeddingGenerator,
|
102 | 97 | llm.MultimodalEmbeddingGenerator,
|
|
0 commit comments