style: isort + black
This commit is contained in:
@@ -439,19 +439,19 @@ class PostRepository(DatabaseConnection):
|
||||
f"Получен контент опубликованного поста: {len(post_content)} элементов для published_message_id={published_message_id}"
|
||||
)
|
||||
return post_content
|
||||
|
||||
|
||||
# ============================================
|
||||
# Методы для работы с ML-скорингом
|
||||
# ============================================
|
||||
|
||||
|
||||
async def update_ml_scores(self, message_id: int, ml_scores_json: str) -> bool:
|
||||
"""
|
||||
Обновляет ML-скоры для поста.
|
||||
|
||||
|
||||
Args:
|
||||
message_id: ID сообщения в группе модерации
|
||||
ml_scores_json: JSON строка со скорами
|
||||
|
||||
|
||||
Returns:
|
||||
True если обновлено успешно
|
||||
"""
|
||||
@@ -461,16 +461,18 @@ class PostRepository(DatabaseConnection):
|
||||
self.logger.info(f"ML-скоры обновлены для message_id={message_id}")
|
||||
return True
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка обновления ML-скоров для message_id={message_id}: {e}")
|
||||
self.logger.error(
|
||||
f"Ошибка обновления ML-скоров для message_id={message_id}: {e}"
|
||||
)
|
||||
return False
|
||||
|
||||
|
||||
async def get_ml_scores_by_message_id(self, message_id: int) -> Optional[str]:
|
||||
"""
|
||||
Получает ML-скоры для поста.
|
||||
|
||||
|
||||
Args:
|
||||
message_id: ID сообщения
|
||||
|
||||
|
||||
Returns:
|
||||
JSON строка со скорами или None
|
||||
"""
|
||||
@@ -479,14 +481,14 @@ class PostRepository(DatabaseConnection):
|
||||
if rows and rows[0][0]:
|
||||
return rows[0][0]
|
||||
return None
|
||||
|
||||
|
||||
async def get_post_text_by_message_id(self, message_id: int) -> Optional[str]:
|
||||
"""
|
||||
Получает текст поста по message_id.
|
||||
|
||||
|
||||
Args:
|
||||
message_id: ID сообщения
|
||||
|
||||
|
||||
Returns:
|
||||
Текст поста или None
|
||||
"""
|
||||
@@ -495,14 +497,14 @@ class PostRepository(DatabaseConnection):
|
||||
if rows and rows[0][0]:
|
||||
return rows[0][0]
|
||||
return None
|
||||
|
||||
|
||||
async def get_approved_posts_texts(self, limit: int = 1000) -> List[str]:
|
||||
"""
|
||||
Получает тексты опубликованных постов для обучения RAG.
|
||||
|
||||
|
||||
Args:
|
||||
limit: Максимальное количество постов
|
||||
|
||||
|
||||
Returns:
|
||||
Список текстов
|
||||
"""
|
||||
@@ -519,14 +521,14 @@ class PostRepository(DatabaseConnection):
|
||||
texts = [row[0] for row in rows if row[0]]
|
||||
self.logger.info(f"Получено {len(texts)} опубликованных постов для обучения")
|
||||
return texts
|
||||
|
||||
|
||||
async def get_declined_posts_texts(self, limit: int = 1000) -> List[str]:
|
||||
"""
|
||||
Получает тексты отклоненных постов для обучения RAG.
|
||||
|
||||
|
||||
Args:
|
||||
limit: Максимальное количество постов
|
||||
|
||||
|
||||
Returns:
|
||||
Список текстов
|
||||
"""
|
||||
@@ -543,4 +545,3 @@ class PostRepository(DatabaseConnection):
|
||||
texts = [row[0] for row in rows if row[0]]
|
||||
self.logger.info(f"Получено {len(texts)} отклоненных постов для обучения")
|
||||
return texts
|
||||
|
||||
|
||||
Reference in New Issue
Block a user