File tree Expand file tree Collapse file tree 1 file changed +5
-1
lines changed
Expand file tree Collapse file tree 1 file changed +5
-1
lines changed Original file line number Diff line number Diff line change @@ -11,6 +11,7 @@ import (
1111 "maps"
1212 _ "net/http/pprof"
1313 "path/filepath"
14+ "slices"
1415 "sync"
1516 "time"
1617
@@ -51,6 +52,8 @@ type Config struct {
5152 FixedRepository string `json:"repo"`
5253 // Use this LLM model (for testing, if empty use workflow-default model).
5354 Model string `json:"model"`
55+ // Names of workflows to serve (all if not set, mainly for testing/local experimentation).
56+ Workflows []string `json:"workflows"`
5457}
5558
5659func main () {
@@ -175,7 +178,8 @@ func (s *Server) poll(ctx context.Context) (bool, error) {
175178 CodeRevision : prog .GitRevision ,
176179 }
177180 for _ , flow := range aflow .Flows {
178- if s .modelOverQuota (flow ) {
181+ if len (s .cfg .Workflows ) != 0 && ! slices .Contains (s .cfg .Workflows , flow .Name ) ||
182+ s .modelOverQuota (flow ) {
179183 continue
180184 }
181185 req .Workflows = append (req .Workflows , dashapi.AIWorkflow {
You can’t perform that action at this time.
0 commit comments