не помнб

This commit is contained in:
ivankirpichnikov
2025-11-22 18:15:29 +03:00
parent 995141a200
commit effbcfbc2d
17 changed files with 210 additions and 54 deletions
@@ -0,0 +1,24 @@
from collections.abc import Sequence
from typing import override
from sqlalchemy import select
from sqlalchemy.ext.asyncio import AsyncSession
from template_project.adapters.data_gateways.tables import vacancy_embedding_table, vacancy_table
from template_project.application.vacancy.data_gateway import VacancyDataGateway
from template_project.application.vacancy.entity import Vacancy, VacancyEmbedding
class DefaultVacancyDataGateway(VacancyDataGateway):
def __init__(self, session: AsyncSession) -> None:
self._session = session
@override
async def get_suitable(self, vector: list[float]) -> Sequence[Vacancy]:
statement = (
select(Vacancy)
.join(VacancyEmbedding, vacancy_embedding_table.c.id == vacancy_table.c.id)
.where(vacancy_embedding_table.c.vector.cosine_distance(vector) > 0.5)
)
result = await self._session.execute(statement)
return result.scalars().all()