chk
/
AppFlowy-Cloud
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

chore: specify chunk size for search and index

This commit is contained in:
Bartosz Sypytkowski 2024-06-18 11:45:33 +02:00
parent 0e50650aa8
commit d9354202cd
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
services/appflowy-indexer/src/indexer.rs
View File

@ -138,7 +138,7 @@ impl PostgresIndexer {
.embeddings(EmbeddingRequest { .embeddings(EmbeddingRequest {
input: EmbeddingInput::StringArray(inputs), input: EmbeddingInput::StringArray(inputs),
model: EmbeddingsModel::TextEmbedding3Small.to_string(), model: EmbeddingsModel::TextEmbedding3Small.to_string(),
chunk_size: 0, chunk_size: 2000,
encoding_format: EmbeddingEncodingFormat::Float, encoding_format: EmbeddingEncodingFormat::Float,
dimensions: 1536, dimensions: 1536,
}) })

2
src/biz/search/ops.rs
View File

@ -24,7 +24,7 @@ pub async fn search_document(
.embeddings(EmbeddingRequest { .embeddings(EmbeddingRequest {
input: EmbeddingInput::String(request.query.clone()), input: EmbeddingInput::String(request.query.clone()),
model: EmbeddingsModel::TextEmbedding3Small.to_string(), model: EmbeddingsModel::TextEmbedding3Small.to_string(),
chunk_size: 0, chunk_size: 500,
encoding_format: EmbeddingEncodingFormat::Float, encoding_format: EmbeddingEncodingFormat::Float,
dimensions: 1536, dimensions: 1536,
}) })