Skip to content

Commit 164e84c

Browse files
committed
fix: setting model
1 parent 0b4aa5e commit 164e84c

File tree

1 file changed

+17
-7
lines changed

1 file changed

+17
-7
lines changed

cmd/main.go

Lines changed: 17 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -179,7 +179,9 @@ func parseConfig(ctx context.Context) (bool, error) {
179179
fmt.Println(model)
180180
}
181181
fmt.Println()
182-
fmt.Printf("Currently selected model: %s\n", *cfg.Model)
182+
if cfg.Model != nil {
183+
fmt.Printf("Currently selected model: %s\n", *cfg.Model)
184+
}
183185
return true, nil
184186
}
185187

@@ -279,7 +281,9 @@ func cmd(ctx context.Context) error {
279281
out.Close()
280282
<-done
281283
for _, block := range blocks {
282-
runBlock(block)
284+
if err := runBlock(block); err != nil {
285+
return "", err
286+
}
283287
}
284288

285289
return response, nil
@@ -338,14 +342,11 @@ func cmd(ctx context.Context) error {
338342
func main() {
339343
flag.Parse()
340344

341-
ctx := context.Background()
342-
done, err := parseConfig(ctx)
345+
// read config first so we can use the right provider
346+
cfg, err := config.ReadConfig()
343347
if err != nil {
344348
panic(err)
345349
}
346-
if done {
347-
return
348-
}
349350

350351
switch cfg.Provider {
351352
case config.ProviderGroq:
@@ -356,6 +357,15 @@ func main() {
356357
log.Fatalf("unknown provider: %s", cfg.Provider)
357358
}
358359

360+
ctx := context.Background()
361+
done, err := parseConfig(ctx)
362+
if err != nil {
363+
panic(err)
364+
}
365+
if done {
366+
return
367+
}
368+
359369
if cfg.Record {
360370
closer, err := provider.NewCacheProvider(prov, ".cache/cache.json")
361371
if err != nil {

0 commit comments

Comments
 (0)