diff --git a/vacancies/main/management/commands/generate_recommended_vacancies.py b/vacancies/main/management/commands/generate_recommended_vacancies.py index 51f2bb5..977cece 100644 --- a/vacancies/main/management/commands/generate_recommended_vacancies.py +++ b/vacancies/main/management/commands/generate_recommended_vacancies.py @@ -1,6 +1,6 @@ import asyncio from django.core.management import BaseCommand -from vacancies.main.vector_store import search_similarities +from vacancies.main.vector_store import search_similarities, client from vacancies.main.models import CustomerCV, RecommendedVacancy from vacancies.main.bot import application from telegram import InlineKeyboardButton, InlineKeyboardMarkup @@ -21,6 +21,11 @@ class Command(BaseCommand): query_filter = Filter(must_not=[HasIdCondition(has_id=recommended_vacancy_ids)]) search_result_id = search_similarities(query_filter, customer_cv.id) + vacancy_content = client.retrieve( + collection_name="vacancies", + ids=[search_result_id], + )[0].payload["content"] + recommendation = RecommendedVacancy.objects.create( customer=customer_cv.customer, vacancy_id=search_result_id, @@ -28,7 +33,7 @@ class Command(BaseCommand): asyncio.run(application.bot.send_message( chat_id=recommendation.customer.chat_id, - text=recommendation.vacancy.content, + text=vacancy_content, reply_markup=InlineKeyboardMarkup([[ InlineKeyboardButton("Откликнуться", url=recommendation.vacancy.link), ]]),