You've already forked RekomenciBackend
не помнб
This commit is contained in:
@@ -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()
|
||||
Reference in New Issue
Block a user