26 lines
648 B
TypeScript
26 lines
648 B
TypeScript
import { parseModelString } from "../../tools/delegate-task/model-string-parser"
|
|
|
|
export function buildRetryModelPayload(
|
|
model: string,
|
|
): { model: { providerID: string; modelID: string }; variant?: string } | undefined {
|
|
const parsedModel = parseModelString(model)
|
|
if (!parsedModel) {
|
|
return undefined
|
|
}
|
|
|
|
return parsedModel.variant
|
|
? {
|
|
model: {
|
|
providerID: parsedModel.providerID,
|
|
modelID: parsedModel.modelID,
|
|
},
|
|
variant: parsedModel.variant,
|
|
}
|
|
: {
|
|
model: {
|
|
providerID: parsedModel.providerID,
|
|
modelID: parsedModel.modelID,
|
|
},
|
|
}
|
|
}
|