github-mirrors/immich
1
0
Fork 0
mirror of https://github.com/immich-app/immich.git synced 2026-02-13 12:27:56 +03:00
Code Issues Packages Projects Releases Wiki Activity

chore(server): change upsert signature for search repo (#8210)

Browse Source 
* upsert embedding

* remove unused imports
This commit is contained in:
Mert
2024-03-23 14:37:06 -04:00
committed by GitHub
parent 787eebcf1e
commit b07a565e34
4 changed files with 4 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
server/src/services/smart-info.service.spec.ts
View File

@@ -104,7 +104,6 @@ describe(SmartInfoService.name, () => {
});
it('should save the returned objects', async () => {
searchMock.upsert.mockResolvedValue();
machineMock.encodeImage.mockResolvedValue([0.01, 0.02, 0.03]);
await sut.handleEncodeClip({ id: asset.id });
@@ -114,12 +113,7 @@ describe(SmartInfoService.name, () => {
{ imagePath: 'path/to/resize.ext' },
{ enabled: true, modelName: 'ViT-B-32__openai' },
);
expect(searchMock.upsert).toHaveBeenCalledWith(
{
assetId: 'asset-1',
},
[0.01, 0.02, 0.03],
);
expect(searchMock.upsert).toHaveBeenCalledWith('asset-1', [0.01, 0.02, 0.03]);
});
});

2
server/src/services/smart-info.service.ts
View File

@@ -98,7 +98,7 @@ export class SmartInfoService {
await this.databaseRepository.wait(DatabaseLock.CLIPDimSize);
}
await this.repository.upsert({ assetId: asset.id }, clipEmbedding);
await this.repository.upsert(asset.id, clipEmbedding);
return JobStatus.SUCCESS;
}