Skip to content

Commit

Permalink
Update Embeddings.kt (#480)
Browse files Browse the repository at this point in the history
  • Loading branch information
rafaparadela authored Oct 6, 2023
1 parent 2aa5c58 commit 4e71c37
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ interface Embeddings : LLM {
else
texts
.chunked(chunkSize ?: 400)
.parMap { createEmbeddings(EmbeddingRequest(name, texts, requestConfig.user.id)).data }
.parMap { createEmbeddings(EmbeddingRequest(name, it, requestConfig.user.id)).data }
.flatten()

suspend fun embedQuery(text: String, requestConfig: RequestConfig): List<Embedding> =
Expand Down

0 comments on commit 4e71c37

Please sign in to comment.