You've already forked RekomenciBackend
fix(): skip resume tests as they require ml service
This commit is contained in:
@@ -18,14 +18,14 @@ class DefaultVacancyDataGateway(VacancyDataGateway):
|
||||
async def get_suitable(self, vector: list[float]) -> Sequence[SuitableVacancy]:
|
||||
statement = (
|
||||
select(Vacancy, label("resume_similarity", vacancy_embedding_table.c.vector.cosine_distance(vector)))
|
||||
.join(VacancyEmbedding, vacancy_embedding_table.c.id == vacancy_table.c.id)
|
||||
.join(VacancyEmbedding, vacancy_embedding_table.c.vacancy_id == vacancy_table.c.id)
|
||||
.where(vacancy_embedding_table.c.vector.cosine_distance(vector) > 0.5)
|
||||
)
|
||||
result = await self._session.execute(statement)
|
||||
return [
|
||||
SuitableVacancy(
|
||||
vacancy=res[0],
|
||||
resume_similarity=res[1],
|
||||
vacancy=row[0],
|
||||
resume_similarity=row[1],
|
||||
)
|
||||
for res in result.scalars()
|
||||
for row in result.all()
|
||||
]
|
||||
|
||||
Reference in New Issue
Block a user