added grok 2 to proxy (#138)

Co-authored-by: Adrian Barbir <[email protected]>
braintrustdata · Jan 30, 2025 · 01b1467 · 01b1467
1 parent b8e96de
commit 01b1467
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/packages/proxy/schema/index.ts b/packages/proxy/schema/index.ts
@@ -226,6 +226,7 @@ export const AvailableEndpointTypes: { [name: string]: ModelEndpointType[] } = {
   "amazon.nova-lite-v1:0": ["bedrock"],
   "amazon.nova-micro-v1:0": ["bedrock"],
   "grok-beta": ["xAI"],
+  "grok-2-latest": ["xAI"],
 };
 
 export function getModelEndpointTypes(model: string): ModelEndpointType[] {

diff --git a/packages/proxy/schema/models.ts b/packages/proxy/schema/models.ts
@@ -1095,6 +1095,14 @@ export const AvailableModels: { [name: string]: ModelSpec } = {
     displayName: "Grok Beta",
   },
 
+  "grok-2-latest": {
+    format: "openai",
+    flavor: "chat",
+    input_cost_per_mil_tokens: 2,
+    output_cost_per_mil_tokens: 10,
+    displayName: "Grok 2 Latest",
+  },
+
   // BEDROCK MODELS
   "amazon.nova-pro-v1:0": {
     format: "openai",