|
38 | 38 | selectedModel string |
39 | 39 | defaultModel string |
40 | 40 | availableModels []string |
| 41 | + modelVendor string |
41 | 42 | ) |
42 | 43 |
|
43 | 44 | func main() { |
@@ -98,6 +99,7 @@ func init() { |
98 | 99 | runCmd.Flags().StringVar(&selectedModel, "selected-model", "", "Selected model for this job") |
99 | 100 | runCmd.Flags().StringVar(&defaultModel, "default-model", "", "Default model for this engine") |
100 | 101 | runCmd.Flags().StringSliceVar(&availableModels, "available-model", nil, "Available model for this engine (repeatable)") |
| 102 | + runCmd.Flags().StringVar(&modelVendor, "model-vendor", "", "Model vendor for filtering (e.g. Anthropic, OpenAI)") |
101 | 103 |
|
102 | 104 | _ = runCmd.MarkFlagRequired("repo") |
103 | 105 | } |
@@ -164,6 +166,7 @@ func runEngine(cmd *cobra.Command, args []string) error { |
164 | 166 | SelectedModel: selectedModel, |
165 | 167 | DefaultModel: defaultModel, |
166 | 168 | AvailableModels: availableModels, |
| 169 | + ModelVendor: modelVendor, |
167 | 170 | } |
168 | 171 |
|
169 | 172 | prNumber := setup.PRNumber |
@@ -287,6 +290,7 @@ func runEngine(cmd *cobra.Command, args []string) error { |
287 | 290 | env.SelectedModel = selectedModel |
288 | 291 | env.DefaultModel = defaultModel |
289 | 292 | env.AvailableModels = availableModels |
| 293 | + env.ModelVendor = modelVendor |
290 | 294 | } |
291 | 295 |
|
292 | 296 | result := runner.Run(ctx, command, env, runner.Options{WorkingDir: workingDir}, runnerCallbacks) |
|
0 commit comments