Skip to content

Commit 243316b

Browse files
models-bot[bot]Truefoundry Models Bot
andauthored
feat(databricks): update model YAMLs [bot] (#1760)
Co-authored-by: Truefoundry Models Bot <truefoundry-models-bot@users.noreply.github.com>
1 parent 31bd567 commit 243316b

6 files changed

Lines changed: 25 additions & 4 deletions

providers/databricks/databricks-bge-large-en.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ model: databricks-bge-large-en
1616
provisioning: serverless
1717
removeParams:
1818
- max_tokens
19+
- reasoning_effort
1920
status: active
2021
supportedModes:
2122
- embedding

providers/databricks/databricks-claude-opus-4-1.yaml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,8 @@ modalities:
2020
mode: chat
2121
model: databricks-claude-opus-4-1
2222
provisioning: serverless
23+
removeParams:
24+
- reasoning_effort
2325
status: active
2426
supportedModes:
2527
- chat

providers/databricks/databricks-claude-opus-4-5.yaml

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,15 @@
11
costs:
2-
- input_cost_per_token: 0.000005
2+
- cache_creation_input_token_cost: 6.25e-6
3+
cache_read_input_token_cost: 5e-7
4+
input_cost_per_token: 0.000005
35
output_cost_per_token: 0.000025
46
region: "*"
57
features:
68
- function_calling
79
- tool_choice
810
- assistant_prefill
11+
- prompt_caching
12+
- system_messages
913
limits:
1014
context_window: 200000
1115
max_input_tokens: 200000
@@ -20,6 +24,9 @@ modalities:
2024
mode: chat
2125
model: databricks-claude-opus-4-5
2226
provisioning: serverless
27+
sources:
28+
- https://www.databricks.com/product/pricing/proprietary-foundation-model-serving
29+
- https://learn.microsoft.com/en-us/azure/databricks/machine-learning/foundation-model-apis/limits
2330
status: active
2431
supportedModes:
2532
- chat

providers/databricks/databricks-gemma-3-12b.yaml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,8 @@ modalities:
1919
mode: chat
2020
model: databricks-gemma-3-12b
2121
provisioning: serverless
22+
removeParams:
23+
- reasoning_effort
2224
sources:
2325
- https://docs.databricks.com/aws/en/machine-learning/foundation-model-apis/limits
2426
- https://docs.databricks.com/aws/en/machine-learning/model-serving/function-calling

providers/databricks/databricks-gpt-5-mini.yaml

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,17 @@
11
costs:
2-
- input_cost_per_token: 2.5e-7
2+
- cache_creation_input_token_cost: 2.5e-7
3+
cache_read_input_token_cost: 2.5e-8
4+
input_cost_per_token: 2.5e-7
35
output_cost_per_token: 0.000002
46
region: "*"
57
features:
68
- function_calling
79
- tool_choice
10+
- structured_output
11+
- json_output
12+
- prompt_caching
13+
- system_messages
14+
- cache_control
815
limits:
916
context_window: 400000
1017
max_input_tokens: 272000
@@ -32,6 +39,8 @@ provisioning: serverless
3239
sources:
3340
- https://docs.databricks.com/aws/en/machine-learning/foundation-model-apis/limits
3441
- https://docs.databricks.com/aws/en/machine-learning/model-serving/function-calling
42+
- https://docs.databricks.com/aws/en/machine-learning/model-serving/structured-outputs
43+
- https://www.databricks.com/product/pricing/proprietary-foundation-model-serving
3544
status: active
3645
supportedModes:
3746
- chat

providers/databricks/databricks-llama-4-maverick.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,6 @@ features:
77
- tool_choice
88
- structured_output
99
limits:
10-
context_window: 128000
11-
max_input_tokens: 128000
1210
max_output_tokens: 8192
1311
max_tokens: 8192
1412
modalities:
@@ -20,6 +18,8 @@ modalities:
2018
mode: chat
2119
model: databricks-llama-4-maverick
2220
provisioning: serverless
21+
removeParams:
22+
- reasoning_effort
2323
sources:
2424
- https://docs.databricks.com/aws/en/machine-learning/model-serving/score-foundation-models
2525
- https://docs.databricks.com/aws/en/machine-learning/foundation-model-apis/limits

0 commit comments

Comments
 (0)