diff --git a/packages/proxy/schema/index.ts b/packages/proxy/schema/index.ts index 464a3820..5a90c697 100644 --- a/packages/proxy/schema/index.ts +++ b/packages/proxy/schema/index.ts @@ -590,6 +590,9 @@ export const AvailableEndpointTypes: { [name: string]: ModelEndpointType[] } = { "grok-2-1212": ["xAI"], "grok-vision-beta": ["xAI"], "grok-beta": ["xAI"], + "databricks-gemini-3-1-pro": ["databricks"], + "databricks-gemini-2-5-pro": ["databricks"], + "databricks-gemini-2-5-flash": ["databricks"], "grok-4.20-0309-non-reasoning": ["xAI"], "accounts/fireworks/models/deepseek-v4-flash": ["fireworks"], "accounts/fireworks/models/glm-5": ["fireworks"], diff --git a/packages/proxy/schema/model_list.json b/packages/proxy/schema/model_list.json index 57d7b976..9881ba8c 100644 --- a/packages/proxy/schema/model_list.json +++ b/packages/proxy/schema/model_list.json @@ -9215,6 +9215,36 @@ "databricks" ] }, + "databricks-gemini-3-1-pro": { + "format": "openai", + "flavor": "chat", + "multimodal": true, + "displayName": "Gemini 3.1 Pro", + "max_input_tokens": 1000000, + "available_providers": [ + "databricks" + ] + }, + "databricks-gemini-2-5-pro": { + "format": "openai", + "flavor": "chat", + "multimodal": true, + "displayName": "Gemini 2.5 Pro", + "max_input_tokens": 1000000, + "available_providers": [ + "databricks" + ] + }, + "databricks-gemini-2-5-flash": { + "format": "openai", + "flavor": "chat", + "multimodal": true, + "displayName": "Gemini 2.5 Flash", + "max_input_tokens": 1000000, + "available_providers": [ + "databricks" + ] + }, "text-block": { "format": "js", "flavor": "completion",