Fix #55: prevent crashing if no response is returned when finding ent…

…ities
NickSavage · Jan 4, 2025 · edc2d73 · edc2d73
1 parent 29569f0
commit edc2d73
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/go-backend/llms/embeddings.go b/go-backend/llms/embeddings.go
@@ -65,7 +65,6 @@ func ProcessQueue(c *models.LLMClient) {
 			}
 			continue
 		}
-		log.Printf("mainembeddings %v", request.Chunk)
 		err = StoreEmbeddings(
 			c.EmbeddingQueue.DB,
 			request.UserID,

diff --git a/go-backend/llms/entity.go b/go-backend/llms/entity.go
@@ -71,6 +71,9 @@ Return only valid JSON matching the specified structure.`
 			log.Printf("error getting completion: %v", err)
 			return []models.Entity{}, err
 		}
+		if len(resp.Choices) == 0 {
+			continue
+		}
 		content := resp.Choices[0].Message.Content
 		content = strings.TrimPrefix(content, "```json")
 		content = strings.TrimSuffix(content, "```")