mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-07-17 21:27:23 +08:00
21 lines
605 B
Go
21 lines
605 B
Go
|
package component
|
||
|
|
|
||
|
|
import "strings"
|
||
|
|
|
||
|
|
// splitCompositeLLMID extracts the provider driver and bare model id from the
|
||
|
|
// canvas llm_id convention:
|
||
|
|
// - "model@provider" -> ("model", "provider", true)
|
||
|
|
// - "model@instance@provider" -> ("model", "provider", true)
|
||
|
|
// - bare "model" -> ("model", "", false)
|
||
|
|
func splitCompositeLLMID(s string) (modelName, driver string, hasDriver bool) {
|
||
|
|
parts := strings.Split(strings.TrimSpace(s), "@")
|
||
|
|
switch len(parts) {
|
||
|
|
case 2:
|
||
|
|
return parts[0], parts[1], true
|
||
|
|
case 3:
|
||
|
|
return parts[0], parts[2], true
|
||
|
|
default:
|
||
|
|
return s, "", false
|
||
|
|
}
|
||
|
|
}
|