feat: support EMBEDDING_MAX_BATCH_SIZE (#311)

2025-07-30 15:31:54 +08:00
parent f93f26fc48
commit bb74272385
9 changed files with 85 additions and 71 deletions
--- a/backend/infra/impl/document/searchstore/vikingdb/vk_test.go
+++ b/backend/infra/impl/document/searchstore/vikingdb/vk_test.go
@@ -179,7 +179,7 @@ func TestBuiltinEmbeddingIntegration(t *testing.T) {
 		Model:      os.Getenv("OPENAI_EMBEDDING_MODEL"),
 		Dimensions: ptr.Of(1024),
 	}
-	emb, err := wrap.NewOpenAIEmbedder(ctx, embConfig, 1024)
+	emb, err := wrap.NewOpenAIEmbedder(ctx, embConfig, 1024, 100)
 	assert.NoError(t, err)

 	cfg := &ManagerConfig{