(provider, requestedID string)
| 1325 | } |
| 1326 | |
| 1327 | func normalizeAgentAccountModelID(provider, requestedID string) string { |
| 1328 | switch provider { |
| 1329 | case "custom", "vllm": |
| 1330 | return normalizeCustomModel(requestedID) |
| 1331 | case "ollama": |
| 1332 | target := strings.TrimSpace(requestedID) |
| 1333 | if strings.HasPrefix(target, "ollama/") { |
| 1334 | return target |
| 1335 | } |
| 1336 | target = strings.TrimLeft(strings.TrimSpace(target), "/") |
| 1337 | if target == "" { |
| 1338 | return "" |
| 1339 | } |
| 1340 | return "ollama/" + target |
| 1341 | default: |
| 1342 | target := strings.TrimSpace(requestedID) |
| 1343 | if target == "" { |
| 1344 | return "" |
| 1345 | } |
| 1346 | prefix := poolModelPrefix(provider) |
| 1347 | if strings.Contains(target, "/") { |
| 1348 | parts := strings.SplitN(target, "/", 2) |
| 1349 | targetPrefix := parts[0] |
| 1350 | targetModel := strings.TrimSpace(parts[1]) |
| 1351 | if targetModel == "" { |
| 1352 | return strings.TrimSpace(target) |
| 1353 | } |
| 1354 | for _, item := range supportedProviderModelPrefixes(provider) { |
| 1355 | if item == targetPrefix { |
| 1356 | if prefix != "" { |
| 1357 | return prefix + "/" + targetModel |
| 1358 | } |
| 1359 | return strings.TrimSpace(target) |
| 1360 | } |
| 1361 | } |
| 1362 | return strings.TrimSpace(target) |
| 1363 | } |
| 1364 | target = strings.TrimLeft(strings.TrimSpace(target), "/") |
| 1365 | if prefix == "" { |
| 1366 | return target |
| 1367 | } |
| 1368 | return prefix + "/" + target |
| 1369 | } |
| 1370 | } |
| 1371 | |
| 1372 | func sanitizeAgentAccountModelInputs(values []string) []string { |
| 1373 | result := make([]string, 0, len(values)) |
no test coverage detected