cmd/sketch: allow using other models in safe mode
diff --git a/cmd/sketch/main.go b/cmd/sketch/main.go
index 1fdac7a..a85594b 100644
--- a/cmd/sketch/main.go
+++ b/cmd/sketch/main.go
@@ -141,11 +141,10 @@
return dumpDistFilesystem(flagArgs.dumpDist)
}
- // Claude and Gemini are supported in container mode
- // TODO: finish support--thread through API keys, add server support
- isContainerSupported := ant.IsClaudeModel(flagArgs.modelName) || flagArgs.modelName == "gemini"
- if !isContainerSupported && (!flagArgs.unsafe || flagArgs.skabandAddr != "") {
- return fmt.Errorf("only claude and gemini are supported in safe mode right now, use -unsafe -skaband-addr=''")
+ // Only Claude and Gemini have skaband support, for now.
+ hasSkabandSupport := flagArgs.modelName == "gemini" || ant.IsClaudeModel(flagArgs.modelName)
+ if !hasSkabandSupport && flagArgs.skabandAddr != "" {
+ return fmt.Errorf("only claude and gemini are supported by skaband, use -skaband-addr='' for other models")
}
if err := flagArgs.experimentFlag.Process(); err != nil {