19 lines
737 B
Python
19 lines
737 B
Python
from django.core.management import BaseCommand
|
|
from vacancies.main.vector_store import vector_store
|
|
from vacancies.main.models import CustomerCV, RecommendedVacancy
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = "Generates new recommended vacancies"
|
|
|
|
def handle(self, *args, **options):
|
|
customer_cvs = CustomerCV.objects.all()
|
|
for customer_cv in customer_cvs:
|
|
documents = vector_store.similarity_search(customer_cv.content, k=10)
|
|
for document in documents:
|
|
RecommendedVacancy.objects.get_or_create(
|
|
customer=customer_cv.customer,
|
|
vacancy_id=document.metadata["vacancy_id"],
|
|
defaults=dict(is_shown=False),
|
|
)
|