aboba

Signed-off-by: ITQ <itq.dev@ya.ru>

src/template_project/application/resume/resume_embedding_pipeline.py

@@ -0,0 +1,71 @@
+from collections.abc import Callable
+
+from Levenshtein import ratio
+
+from template_project.application.common.unit_of_work import UnitOfWork
+from template_project.application.resume.data_gateway import ResumeDataGateway
+from template_project.application.resume.entity import Resume, ResumeEmbedding, ResumePrediction
+from template_project.application.resume.vector_generator import ResumeEmbeddingVectorGenerator
+
+
+def suitable_resumes_key(
+    resume: Resume,
+) -> Callable[[Resume], bool]:
+    def wrapper(suitable_resume: Resume) -> bool:
+        count_skills = 0
+        ratio_skill_sum = 0.0
+        for resum_key_skill in resume.key_skills:
+            for suitable_resume_key_skill in suitable_resume.key_skills:
+                ratio_skill = ratio(resum_key_skill, suitable_resume_key_skill)
+                if ratio_skill != 0:
+                    count_skills += 1
+                    ratio_skill_sum += ratio_skill
+
+        try:
+            matching_skills = ratio_skill_sum / count_skills
+        except ZeroDivisionError:
+            matching_skills = 0
+
+        return resume.experience_type == suitable_resume.experience_type and matching_skills >= 50
+
+    return wrapper
+
+
+class ResumeEmbeddingPipeline:
+    def __init__(
+        self,
+        unit_of_work: UnitOfWork,
+        resume_data_gateway: ResumeDataGateway,
+        vector_generator: ResumeEmbeddingVectorGenerator,
+    ) -> None:
+        self.unit_of_work = unit_of_work
+        self.resume_data_gateway = resume_data_gateway
+        self.vector_generator = vector_generator
+
+    async def run(
+        self,
+        resume: Resume,
+    ) -> ResumePrediction:
+        vector = await self.vector_generator.generate(
+            position=resume.position,
+            about_me=resume.about_me,
+            key_skills=resume.key_skills,
+        )
+        resume_embedding = ResumeEmbedding.factory(
+            resume_id=resume.id,
+            vector=vector,
+        )
+
+        suitable_resumes = await self.resume_data_gateway.get_suitable_resumes(resume_embedding.id)
+        suitable_resumes_filtered = sorted(
+            suitable_resumes,
+            key=suitable_resumes_key(resume),
+        )
+        suitable_resumes = suitable_resumes_filtered[:50]
+
+        # TODO: тут надо сделать отправку в ИИ
+
+        await self.unit_of_work.add(resume_embedding)
+        await self.unit_of_work.commit()
+
+        raise NotImplementedError