Fixed bug in list models

Author: richardwooding

README.md

@@ -42,6 +42,11 @@ promptsLibraryDir: <directory to load prompts, defaults to $HOME/prompts-library
 You will need to install the gcloud sdk before using VertexAI.
 You will need a project on Google Cloud which gives your user access to Vertex AI.
 
+You need to login with:
+`gcloud auth login`
+
+Before using.
+
 ### Sample configuration
 
 ```yaml

client/openai/openaiclient.go

@@ -109,6 +109,9 @@ func (c *OpenAiClient) Models(models chan<- client.LanguageModel) error {
 	for _, languageModel := range response.Data {
 		models <- languageModel
 	}
+
+	close(models)
+
 	return nil
 }
 

cmd/listModels.go

@@ -29,7 +29,10 @@ var listModelsCmd = &cobra.Command{
 		f := bufio.NewWriter(os.Stdout)
 		defer f.Flush()
 		models := make(chan client.LanguageModel)
-		err := chatGPT.Models(models)
+		var err error
+		go func() {
+			err = chatGPT.Models(models)
+		}()
 		for model := range models {
 			fmt.Fprintln(os.Stdout, model)
 		}