Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
38 changes: 33 additions & 5 deletions cmd/cli/store/store.go
Original file line number Diff line number Diff line change
Expand Up @@ -110,7 +110,13 @@ func (s *StoreMethods) Save(LLMConfig LLMProvider) error {
return err
}

return os.WriteFile(configPath, data, 0600)
err = os.WriteFile(configPath, data, 0600)
if err != nil {
return err
}

fmt.Printf("LLM provider %s saved successfully\n", LLMConfig.LLM.String())
return nil
}

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

return os.WriteFile(configPath, data, 0600)
err = os.WriteFile(configPath, data, 0600)
if err != nil {
return err
}

fmt.Printf("%s set as default\n", Model.String())
return nil
}

// DeleteModel removes the specified provider from the saved configuration.
Expand Down Expand Up @@ -280,7 +292,12 @@ func (s *StoreMethods) DeleteModel(Model types.LLMProvider) error {
if err != nil {
return err
}
return os.WriteFile(configPath, []byte("{}"), 0600)
err = os.WriteFile(configPath, []byte("{}"), 0600)
if err != nil {
return err
}
fmt.Printf("%s model deleted\n", Model.String())
return nil
}
} else {

Expand All @@ -301,7 +318,12 @@ func (s *StoreMethods) DeleteModel(Model types.LLMProvider) error {
if err != nil {
return err
}
return os.WriteFile(configPath, data, 0600)
err = os.WriteFile(configPath, data, 0600)
if err != nil {
return err
}
fmt.Printf("%s model deleted\n", Model.String())
return nil

}
}
Expand Down Expand Up @@ -357,6 +379,12 @@ func (s *StoreMethods) UpdateAPIKey(Model types.LLMProvider, APIKey string) erro
return err
}

return os.WriteFile(configPath, data, 0600)
err = os.WriteFile(configPath, data, 0600)
if err != nil {
return err
}

fmt.Printf("API key for %s updated successfully\n", Model.String())
return nil

}
Loading