Skip to content

Commit b01a3f9

Browse files
committed
cmd/sketch: allow using other models in safe mode
1 parent 9386009 commit b01a3f9

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

cmd/sketch/main.go

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -141,11 +141,10 @@ func run() error {
141141
return dumpDistFilesystem(flagArgs.dumpDist)
142142
}
143143

144-
// Claude and Gemini are supported in container mode
145-
// TODO: finish support--thread through API keys, add server support
146-
isContainerSupported := ant.IsClaudeModel(flagArgs.modelName) || flagArgs.modelName == "gemini"
147-
if !isContainerSupported && (!flagArgs.unsafe || flagArgs.skabandAddr != "") {
148-
return fmt.Errorf("only claude and gemini are supported in safe mode right now, use -unsafe -skaband-addr=''")
144+
// Only Claude and Gemini have skaband support, for now.
145+
hasSkabandSupport := flagArgs.modelName == "gemini" || ant.IsClaudeModel(flagArgs.modelName)
146+
if !hasSkabandSupport && flagArgs.skabandAddr != "" {
147+
return fmt.Errorf("only claude and gemini are supported by skaband, use -skaband-addr='' for other models")
149148
}
150149

151150
if err := flagArgs.experimentFlag.Process(); err != nil {

0 commit comments

Comments
 (0)