♻️ Refactor Sort backend imports (utils) #1037

Author: Phinease

backend/services/file_management_service.py

@@ -1,21 +1,24 @@
-# Concurrency control
 import asyncio
 import json
 import logging
 import os
-import uuid
 from io import BytesIO
 from pathlib import Path
 from typing import List, Optional, AsyncGenerator
 
 import httpx
-import requests
 from fastapi import UploadFile
 
 from agents.preprocess_manager import preprocess_manager
 from consts.const import UPLOAD_FOLDER, MAX_CONCURRENT_UPLOADS, DATA_PROCESS_SERVICE
-from database.attachment_db import upload_fileobj, get_file_url, get_content_type, get_file_stream, delete_file, \
+from database.attachment_db import (
+    upload_fileobj,
+    get_file_url,
+    get_content_type,
+    get_file_stream,
+    delete_file,
     list_files
+)
 from utils.attachment_utils import convert_image_to_text, convert_long_text_to_text
 from utils.prompt_template_utils import get_file_processing_messages_template
 from utils.file_management_utils import save_upload_file

backend/services/image_service.py

@@ -7,6 +7,7 @@
 
 logger = logging.getLogger("image_service")
 
+
 async def proxy_image_impl(decoded_url: str):
     # Create session to call the data processing service
     async with aiohttp.ClientSession() as session:

backend/services/memory_config_service.py

@@ -2,12 +2,12 @@
 from typing import Dict, List, Union
 
 from consts.const import (
-	MEMORY_SWITCH_KEY,
-	MEMORY_AGENT_SHARE_KEY,
-	DISABLE_AGENT_ID_KEY,
-	DISABLE_USERAGENT_ID_KEY,
-	DEFAULT_MEMORY_SWITCH_KEY,
-	DEFAULT_MEMORY_AGENT_SHARE_KEY,
+    MEMORY_SWITCH_KEY,
+    MEMORY_AGENT_SHARE_KEY,
+    DISABLE_AGENT_ID_KEY,
+    DISABLE_USERAGENT_ID_KEY,
+    DEFAULT_MEMORY_SWITCH_KEY,
+    DEFAULT_MEMORY_AGENT_SHARE_KEY,
 )
 from consts.model import MemoryAgentShareMode
 from database.memory_config_db import (
@@ -204,4 +204,4 @@ def build_memory_context(user_id: str, tenant_id: str, agent_id: str | int) -> M
         tenant_id=tenant_id,
         user_id=user_id,
         agent_id=str(agent_id),
-    )
+    )

backend/services/northbound_service.py

@@ -30,7 +30,6 @@
     update_conversation_title as update_conversation_title_service,
 )
 
-
 logger = logging.getLogger("northbound_service")
 
 

backend/services/tenant_config_service.py

@@ -4,7 +4,6 @@
 from database.knowledge_db import get_knowledge_info_by_knowledge_ids, get_knowledge_ids_by_index_names
 from database.tenant_config_db import get_tenant_config_info, insert_config, delete_config_by_tenant_config_id
 
-
 logger = logging.getLogger("tenant_config_service")
 
 

backend/utils/attachment_utils.py

@@ -1,13 +1,12 @@
 from typing import Union, BinaryIO
 
-from utils.config_utils import tenant_config_manager, get_model_name_from_config
-from utils.prompt_template_utils import get_analyze_file_prompt_template
 from jinja2 import Template, StrictUndefined
-
-from nexent.core.models.openai_vlm import OpenAIVLModel
-from nexent.core.models.openai_long_context_model import OpenAILongContextModel
 from nexent.core import MessageObserver
+from nexent.core.models.openai_long_context_model import OpenAILongContextModel
+from nexent.core.models.openai_vlm import OpenAIVLModel
 
+from utils.config_utils import get_model_name_from_config, tenant_config_manager
+from utils.prompt_template_utils import get_analyze_file_prompt_template
 
 
 def convert_image_to_text(query: str, image_input: Union[str, BinaryIO], tenant_id: str, language: str = 'zh'):

backend/utils/auth_utils.py

@@ -1,16 +1,17 @@
-import logging
-import os
 import hashlib
 import hmac
+import logging
+import os
 import time
 from datetime import datetime, timedelta
 from typing import Optional, Tuple
 
-from fastapi import Request
-from consts.const import DEFAULT_USER_ID, DEFAULT_TENANT_ID, IS_SPEED_MODE
-from consts.exceptions import LimitExceededError, SignatureValidationError, UnauthorizedError
 import jwt
+from fastapi import Request
 from supabase import create_client
+
+from consts.const import DEFAULT_TENANT_ID, DEFAULT_USER_ID, IS_SPEED_MODE
+from consts.exceptions import LimitExceededError, SignatureValidationError, UnauthorizedError
 from database.user_tenant_db import get_user_tenant_by_user_id
 
 # Get Supabase configuration

backend/utils/config_utils.py

@@ -1,11 +1,18 @@
 import json
-import time
 import logging
-
+import time
 from typing import Dict, Any
+
 from sqlalchemy.sql import func
-from database.tenant_config_db import get_all_configs_by_tenant_id, insert_config, delete_config_by_tenant_config_id, update_config_by_tenant_config_id_and_data, get_single_config_info
+
 from database.model_management_db import get_model_by_model_id
+from database.tenant_config_db import (
+    delete_config_by_tenant_config_id,
+    get_all_configs_by_tenant_id,
+    get_single_config_info,
+    insert_config,
+    update_config_by_tenant_config_id_and_data,
+)
 
 logger = logging.getLogger("config_utils")
 

backend/utils/file_management_utils.py

@@ -1,13 +1,13 @@
 import logging
 import os
+import traceback
 from pathlib import Path
 from typing import List
-import traceback
 
 import aiofiles
 import httpx
-from fastapi import UploadFile
 import requests
+from fastapi import UploadFile
 
 from consts.const import DATA_PROCESS_SERVICE
 from consts.model import ProcessParams

backend/utils/langchain_utils.py

@@ -1,7 +1,7 @@
-import os
-import logging
 import importlib.util
-from typing import Callable, List, Any, Tuple, Optional
+import logging
+import os
+from typing import Any, Callable, List, Optional, Tuple
 
 logger = logging.getLogger(__name__)