make format

Author: ikkvix

kubechat/db/models.py

@@ -128,7 +128,7 @@ class Collection(models.Model):
     user = models.CharField(max_length=256)
     status = models.CharField(max_length=16, choices=CollectionStatus.choices)
     type = models.CharField(max_length=16, choices=CollectionType.choices)
-    question_status = models.CharField(max_length=16, choices=CollectionStatus.choices,default=CollectionStatus.ACTIVE)
+    question_status = models.CharField(max_length=16, choices=CollectionStatus.choices, default=CollectionStatus.ACTIVE)
     config = models.TextField()
     gmt_created = models.DateTimeField(auto_now_add=True)
     gmt_updated = models.DateTimeField(auto_now=True)

kubechat/tasks/index.py

@@ -16,15 +16,15 @@
 from config.vector_db import get_vector_db_connector
 from kubechat.context.full_text import insert_document, remove_document
 from kubechat.db.models import (
+    Collection,
+    CollectionStatus,
     Document,
     DocumentStatus,
     MessageFeedback,
     MessageFeedbackStatus,
     ProtectAction,
     Question,
     QuestionStatus,
-    Collection,
-    CollectionStatus,
 )
 from kubechat.source.base import get_source
 from kubechat.source.feishu.client import FeishuNoPermission, FeishuPermissionDenied

kubechat/views/main.py

@@ -8,8 +8,8 @@
 import redis.asyncio as redis
 import yaml
 from asgiref.sync import sync_to_async
-from celery.result import GroupResult
 from celery import chain, group
+from celery.result import GroupResult
 from django.contrib.auth.models import User
 from django.core.files.base import ContentFile
 from django.db import IntegrityError
@@ -80,9 +80,9 @@
     generate_questions,
     message_feedback,
     remove_index,
+    update_collection_status,
     update_index,
     update_index_for_question,
-    update_collection_status,
 )
 from kubechat.tasks.scan import delete_sync_documents_cron_job, update_sync_documents_cron_job
 from kubechat.tasks.sync_documents_task import get_sync_progress, sync_documents

readers/question_embedding.py

@@ -40,7 +40,7 @@ def load_data(self, **kwargs) -> list[str]:
         nodes: List[NodeWithEmbedding] = []
         for doc in docs:
             doc.text = doc.text.strip()
-            logger.info("generating questions for document: %s",file_name)
+            logger.info("generating questions for document: %s", file_name)
 
             # ignore page less than 30 characters
             text_size_threshold = 30
@@ -86,7 +86,7 @@ def load_data(self, **kwargs) -> list[str]:
                 continue
 
             self.questions.extend(questions)
-            logger.info("generating questions: %s",str(questions))
+            logger.info("generating questions: %s", str(questions))
 
             for q in questions:
                 node = Node(

readers/question_generator.py

@@ -1,6 +1,6 @@
 import logging
-from abc import ABC
 import re
+from abc import ABC
 
 from langchain import PromptTemplate