using the chunkSize value provided in the opts

Simplou · May 24, 2024 · b13a10b · b13a10b
1 parent 631b64d
commit b13a10b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/embedding.go b/embedding.go
@@ -77,8 +77,8 @@ type ChunkTextOpts struct {
 
 // ChunkText splits the input text into chunks of specified size.
 func ChunkText(opts ChunkTextOpts) []string {
-	var chunkSize int
-	if opts.ChunkSize <= 0 {
+	chunkSize := opts.ChunkSize
+	if chunkSize <= 0 {
 		chunkSize = 512
 	}