(modelId: string, providers: readonly ConfigOptionProvider[])
| 113 | } |
| 114 | |
| 115 | export function parseModelSelection(modelId: string, providers: readonly ConfigOptionProvider[]): ModelSelection { |
| 116 | const provider = providers.find((item) => modelId.startsWith(`${item.id}/`)) |
| 117 | if (provider) { |
| 118 | const modelID = modelId.slice(provider.id.length + 1) |
| 119 | if (provider.models[modelID]) { |
| 120 | return { model: { providerID: provider.id, modelID } } |
| 121 | } |
| 122 | |
| 123 | const separator = modelID.lastIndexOf("/") |
| 124 | if (separator > -1) { |
| 125 | const baseModelID = modelID.slice(0, separator) |
| 126 | const variant = modelID.slice(separator + 1) |
| 127 | if (provider.models[baseModelID]?.variants?.[variant]) { |
| 128 | return { model: { providerID: provider.id, modelID: baseModelID }, variant } |
| 129 | } |
| 130 | } |
| 131 | |
| 132 | return { model: { providerID: provider.id, modelID } } |
| 133 | } |
| 134 | |
| 135 | const separator = modelId.indexOf("/") |
| 136 | if (separator === -1) { |
| 137 | return { model: { providerID: modelId, modelID: "" } } |
| 138 | } |
| 139 | |
| 140 | return { |
| 141 | model: { |
| 142 | providerID: modelId.slice(0, separator), |
| 143 | modelID: modelId.slice(separator + 1), |
| 144 | }, |
| 145 | } |
| 146 | } |
| 147 | |
| 148 | export function formatCurrentModelId(input: { |
| 149 | model: ModelSelection["model"] |
no test coverage detected