Skip to content

Commit e51743a

Browse files
K-coder05RayCrom2
andauthored
Removal of Duplicate Mentions of Mistral, Sglang and Inconsistencies in Fine-tuning Files (tensorzero#2940)
* removed mistral as an option in the model_options.ts schema despite client.ts not supporting it - inconsistency * removed filter that excluded Mistral from provider options * Removed duplicate reference of mistral and sglang * fixed formatting issue using Prettier --------- Co-authored-by: Raymond Cromwell <[email protected]>
1 parent e2de963 commit e51743a

File tree

3 files changed

+3
-7
lines changed

3 files changed

+3
-7
lines changed

tensorzero-core/tests/e2e/providers/common.rs

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8043,8 +8043,6 @@ pub async fn test_stop_sequences_inference_request_with_provider(
80438043
"deepseek",
80448044
"openrouter",
80458045
"openai",
8046-
"sglang",
8047-
"mistral",
80488046
"azure",
80498047
"groq",
80508048
"hyperbolic",

ui/app/routes/optimization/supervised-fine-tuning/ModelSelector.tsx

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -39,10 +39,8 @@ export function ModelSelector({
3939
const newModels = [...initialModels];
4040

4141
// Pick all providers from the schema
42-
// TODO: remove mistral filter once we support it.
43-
const providers = (
44-
ModelOptionSchema.shape.provider.options as ModelOption["provider"][]
45-
).filter((provider) => provider !== "mistral");
42+
const providers = ModelOptionSchema.shape.provider
43+
.options as ModelOption["provider"][];
4644

4745
providers.forEach((provider) => {
4846
const modelExists = initialModels.some(

ui/app/routes/optimization/supervised-fine-tuning/model_options.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ import { z } from "zod";
33
export const ModelOptionSchema = z.object({
44
displayName: z.string().nonempty("Model name is required"),
55
name: z.string().nonempty("Model name is required"),
6-
provider: z.enum(["openai", "fireworks", "mistral"]),
6+
provider: z.enum(["openai", "fireworks"]),
77
});
88

99
export type ModelOption = z.infer<typeof ModelOptionSchema>;

0 commit comments

Comments
 (0)