Merge pull request #108 from samgozman/107-runtimeboundserror-runtime…

…-error fix: bound error
samgozman · Feb 24, 2024 · 7966016 · 7966016
2 parents 4cded13 + 6be9119
commit 7966016
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/composer/composer.go b/composer/composer.go
@@ -81,6 +81,10 @@ func (c *Composer) Compose(ctx context.Context, news journalist.NewsList) ([]*Co
 		return nil, newError(err, errlvl.WARN, "Compose", "OpenAiClient.CreateChatCompletion")
 	}
 
+	if len(resp.Choices) == 0 {
+		return nil, newError(errors.New("empty response"), errlvl.WARN, "Compose", "OpenAiClient.CreateChatCompletion")
+	}
+
 	matches, err := aiJSONStringFixer(resp.Choices[0].Message.Content)
 	if err != nil {
 		return nil, newError(err, errlvl.ERROR, "Compose", "aiJSONStringFixer")
@@ -152,6 +156,10 @@ func (c *Composer) Summarise(ctx context.Context, headlines []*Headline, headlin
 		return nil, newError(err, errlvl.WARN, "Summarise", "OpenAiClient.CreateChatCompletion")
 	}
 
+	if len(resp.Choices) == 0 {
+		return nil, newError(errors.New("empty response"), errlvl.WARN, "Summarise", "OpenAiClient.CreateChatCompletion")
+	}
+
 	matches, err := aiJSONStringFixer(resp.Choices[0].Message.Content)
 	if err != nil {
 		return nil, newError(err, errlvl.ERROR, "Summarise", "aiJSONStringFixer")
@@ -196,6 +204,10 @@ func (c *Composer) Filter(ctx context.Context, news journalist.NewsList) (journa
 		return nil, newError(err, errlvl.WARN, "Filter", "TogetherAIClient.CreateChatCompletion")
 	}
 
+	if len(resp.Choices) == 0 {
+		return nil, newError(errors.New("empty response"), errlvl.WARN, "Filter", "TogetherAIClient.CreateChatCompletion")
+	}
+
 	matches, err := aiJSONStringFixer(resp.Choices[0].Text)
 	if err != nil {
 		return nil, newError(err, errlvl.ERROR, "Filter", "aiJSONStringFixer")