Skip to content

Commit

Permalink
Remove VertexAI GCP SDK API and deprecate unsupported models.
Browse files Browse the repository at this point in the history
We fully turn to REST API.

Note that the deprecated model (mainly old palm models) is not supported with SDK API anymore either.

PiperOrigin-RevId: 702202962
  • Loading branch information
Langfun Authors committed Dec 3, 2024
1 parent 120866b commit c47d506
Show file tree
Hide file tree
Showing 5 changed files with 45 additions and 769 deletions.
1 change: 0 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -140,7 +140,6 @@ If you want to customize your installation, you can select specific features usi
| all | All Langfun features. |
| llm | All supported LLMs. |
| llm-google | All supported Google-powered LLMs. |
| llm-google-vertexai | LLMs powered by Google Cloud VertexAI |
| llm-google-genai | LLMs powered by Google Generative AI API |
| mime | All MIME supports. |
| mime-auto | Automatic MIME type detection. |
Expand Down
8 changes: 1 addition & 7 deletions langfun/core/llms/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -120,25 +120,19 @@
from langfun.core.llms.groq import GroqWhisper_Large_v3Turbo

from langfun.core.llms.vertexai import VertexAI
from langfun.core.llms.vertexai import VertexAIRest
from langfun.core.llms.vertexai import VertexAIRestGemini1_5
from langfun.core.llms.vertexai import VertexAIGemini1_5
from langfun.core.llms.vertexai import VertexAIGeminiPro1_5
from langfun.core.llms.vertexai import VertexAIGeminiPro1_5_Latest
from langfun.core.llms.vertexai import VertexAIGeminiPro1_5_001
from langfun.core.llms.vertexai import VertexAIGeminiPro1_5_002
from langfun.core.llms.vertexai import VertexAIGeminiPro1_5_0514
from langfun.core.llms.vertexai import VertexAIGeminiPro1_5_0409
from langfun.core.llms.vertexai import VertexAIGeminiFlash1_5_Latest
from langfun.core.llms.vertexai import VertexAIGeminiFlash1_5
from langfun.core.llms.vertexai import VertexAIGeminiFlash1_5_001
from langfun.core.llms.vertexai import VertexAIGeminiFlash1_5_002
from langfun.core.llms.vertexai import VertexAIGeminiFlash1_5_0514
from langfun.core.llms.vertexai import VertexAIGeminiPro1
from langfun.core.llms.vertexai import VertexAIGeminiPro1Vision
from langfun.core.llms.vertexai import VertexAIPalm2
from langfun.core.llms.vertexai import VertexAIPalm2_32K
from langfun.core.llms.vertexai import VertexAICustom
from langfun.core.llms.vertexai import VertexAIEndpoint


# LLaMA C++ models.
Expand Down
Loading

0 comments on commit c47d506

Please sign in to comment.