tbelway/immich-quadlet
1
0
Fork 0
forked from Cutlery/immich
Code Pull Requests Activity

feat(server): Fallback to text search if machine-learning is disabled (#2015)

Browse Source
This commit is contained in:
bo0tzz 2023-03-18 22:30:48 +01:00 committed by GitHub
parent d7bfab7b13
commit dd02f1025f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
server/libs/domain/src/search/search.service.ts
View File

@ -99,21 +99,18 @@ export class SearchService {
this.assertEnabled(); this.assertEnabled();
const query = dto.q || dto.query || '*'; const query = dto.q || dto.query || '*';
const strategy = dto.clip ? SearchStrategy.CLIP : SearchStrategy.TEXT; const strategy = dto.clip && MACHINE_LEARNING_ENABLED ? SearchStrategy.CLIP : SearchStrategy.TEXT;
const filters = { userId: authUser.id, ...dto }; const filters = { userId: authUser.id, ...dto };
let assets: SearchResult<AssetEntity>; let assets: SearchResult<AssetEntity>;
switch (strategy) { switch (strategy) {
case SearchStrategy.TEXT:
assets = await this.searchRepository.searchAssets(query, filters);
break;
case SearchStrategy.CLIP: case SearchStrategy.CLIP:
default:
if (!MACHINE_LEARNING_ENABLED) {
throw new BadRequestException('Machine Learning is disabled');
}
const clip = await this.machineLearning.encodeText(query); const clip = await this.machineLearning.encodeText(query);
assets = await this.searchRepository.vectorSearch(clip, filters); assets = await this.searchRepository.vectorSearch(clip, filters);
case SearchStrategy.TEXT:
default:
assets = await this.searchRepository.searchAssets(query, filters);
break;
} }
const albums = await this.searchRepository.searchAlbums(query, filters); const albums = await this.searchRepository.searchAlbums(query, filters);