feat(ml): better multilingual search with nllb models (#13567)

2025-03-31 11:06:57 -04:00
parent 838a8dd9a6
commit 6789c2ac19
16 changed files with 301 additions and 18 deletions
@@ -78,12 +78,10 @@ export class SearchService extends BaseService {
    }

    const userIds = await this.getUserIdsToSearch(auth);
-
-    const embedding = await this.machineLearningRepository.encodeText(
-      machineLearning.urls,
-      dto.query,
-      machineLearning.clip,
-    );
+    const embedding = await this.machineLearningRepository.encodeText(machineLearning.urls, dto.query, {
+      modelName: machineLearning.clip.modelName,
+      language: dto.language,
+    });
    const page = dto.page ?? 1;
    const size = dto.size || 100;
    const { hasNextPage, items } = await this.searchRepository.searchSmart(