Skip to content

Commit b4ad44f

Browse files
authored
Merge pull request #12 from adeeshperera/systematic-logging
fix #121 : add systematic logging for store operations
2 parents 12b8d33 + 7d01100 commit b4ad44f

1 file changed

Lines changed: 33 additions & 5 deletions

File tree

cmd/cli/store/store.go

Lines changed: 33 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -110,7 +110,13 @@ func (s *StoreMethods) Save(LLMConfig LLMProvider) error {
110110
return err
111111
}
112112

113-
return os.WriteFile(configPath, data, 0600)
113+
err = os.WriteFile(configPath, data, 0600)
114+
if err != nil {
115+
return err
116+
}
117+
118+
fmt.Printf("LLM provider %s saved successfully\n", LLMConfig.LLM.String())
119+
return nil
114120
}
115121

116122
// DefaultLLMKey returns the currently selected default LLM provider, if any.
@@ -240,7 +246,13 @@ func ChangeDefault(Model types.LLMProvider) error {
240246
return err
241247
}
242248

243-
return os.WriteFile(configPath, data, 0600)
249+
err = os.WriteFile(configPath, data, 0600)
250+
if err != nil {
251+
return err
252+
}
253+
254+
fmt.Printf("%s set as default\n", Model.String())
255+
return nil
244256
}
245257

246258
// DeleteModel removes the specified provider from the saved configuration.
@@ -280,7 +292,12 @@ func (s *StoreMethods) DeleteModel(Model types.LLMProvider) error {
280292
if err != nil {
281293
return err
282294
}
283-
return os.WriteFile(configPath, []byte("{}"), 0600)
295+
err = os.WriteFile(configPath, []byte("{}"), 0600)
296+
if err != nil {
297+
return err
298+
}
299+
fmt.Printf("%s model deleted\n", Model.String())
300+
return nil
284301
}
285302
} else {
286303

@@ -301,7 +318,12 @@ func (s *StoreMethods) DeleteModel(Model types.LLMProvider) error {
301318
if err != nil {
302319
return err
303320
}
304-
return os.WriteFile(configPath, data, 0600)
321+
err = os.WriteFile(configPath, data, 0600)
322+
if err != nil {
323+
return err
324+
}
325+
fmt.Printf("%s model deleted\n", Model.String())
326+
return nil
305327

306328
}
307329
}
@@ -357,6 +379,12 @@ func (s *StoreMethods) UpdateAPIKey(Model types.LLMProvider, APIKey string) erro
357379
return err
358380
}
359381

360-
return os.WriteFile(configPath, data, 0600)
382+
err = os.WriteFile(configPath, data, 0600)
383+
if err != nil {
384+
return err
385+
}
386+
387+
fmt.Printf("API key for %s updated successfully\n", Model.String())
388+
return nil
361389

362390
}

0 commit comments

Comments
 (0)