mirror of
https://github.com/charmbracelet/crush.git
synced 2025-08-02 05:20:46 +03:00
39 lines
1.5 KiB
Go
39 lines
1.5 KiB
Go
package models
|
|
|
|
const (
|
|
ProviderVertexAI InferenceProvider = "vertexai"
|
|
|
|
// Models
|
|
VertexAIGemini25Flash ModelID = "vertexai.gemini-2.5-flash"
|
|
VertexAIGemini25 ModelID = "vertexai.gemini-2.5"
|
|
)
|
|
|
|
var VertexAIGeminiModels = map[ModelID]Model{
|
|
VertexAIGemini25Flash: {
|
|
ID: VertexAIGemini25Flash,
|
|
Name: "VertexAI: Gemini 2.5 Flash",
|
|
Provider: ProviderVertexAI,
|
|
APIModel: "gemini-2.5-flash-preview-04-17",
|
|
CostPer1MIn: GeminiModels[Gemini25Flash].CostPer1MIn,
|
|
CostPer1MInCached: GeminiModels[Gemini25Flash].CostPer1MInCached,
|
|
CostPer1MOut: GeminiModels[Gemini25Flash].CostPer1MOut,
|
|
CostPer1MOutCached: GeminiModels[Gemini25Flash].CostPer1MOutCached,
|
|
ContextWindow: GeminiModels[Gemini25Flash].ContextWindow,
|
|
DefaultMaxTokens: GeminiModels[Gemini25Flash].DefaultMaxTokens,
|
|
SupportsAttachments: true,
|
|
},
|
|
VertexAIGemini25: {
|
|
ID: VertexAIGemini25,
|
|
Name: "VertexAI: Gemini 2.5 Pro",
|
|
Provider: ProviderVertexAI,
|
|
APIModel: "gemini-2.5-pro-preview-03-25",
|
|
CostPer1MIn: GeminiModels[Gemini25].CostPer1MIn,
|
|
CostPer1MInCached: GeminiModels[Gemini25].CostPer1MInCached,
|
|
CostPer1MOut: GeminiModels[Gemini25].CostPer1MOut,
|
|
CostPer1MOutCached: GeminiModels[Gemini25].CostPer1MOutCached,
|
|
ContextWindow: GeminiModels[Gemini25].ContextWindow,
|
|
DefaultMaxTokens: GeminiModels[Gemini25].DefaultMaxTokens,
|
|
SupportsAttachments: true,
|
|
},
|
|
}
|