Split ingress keys into text/image protocols with Replicate model listing.
CI / docker (push) Successful in 3m38s
CI / docker (push) Successful in 3m38s
Add model input defaults (including prompt), workflow model editing, ingress API docs, and fix img2img image upload filenames for signed URLs. Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
@@ -0,0 +1,32 @@
|
||||
package model
|
||||
|
||||
import "testing"
|
||||
|
||||
func TestSplitIngressOwnerName(t *testing.T) {
|
||||
owner, name := SplitIngressOwnerName("luminary/flux-sdxl", "comfy-local")
|
||||
if owner != "luminary" || name != "flux-sdxl" {
|
||||
t.Fatalf("got %q/%q", owner, name)
|
||||
}
|
||||
owner, name = SplitIngressOwnerName("plain-model", "comfy-local")
|
||||
if owner != "comfy-local" || name != "plain-model" {
|
||||
t.Fatalf("got %q/%q", owner, name)
|
||||
}
|
||||
}
|
||||
|
||||
func TestMatchesIngressRef(t *testing.T) {
|
||||
entry := ModelEntry{
|
||||
Enabled: true,
|
||||
Alias: "luminary/flux-txt2img",
|
||||
ModelID: "internal-flux",
|
||||
}
|
||||
provider := "comfy-ui"
|
||||
if !entry.MatchesIngressRef("luminary/flux-txt2img", provider) {
|
||||
t.Fatal("expected ingress id / owner/name match")
|
||||
}
|
||||
if !entry.MatchesIngressRef("flux-txt2img", provider) {
|
||||
t.Fatal("expected short name match")
|
||||
}
|
||||
if entry.MatchesIngressRef("other/flux-txt2img", provider) {
|
||||
t.Fatal("unexpected owner mismatch match")
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user