[OND211-2329]: Fixed import issues and ruff checks.
This commit is contained in:
parent
0647abc46f
commit
477c72ca3a
3 changed files with 9 additions and 21 deletions
|
|
@ -52,19 +52,13 @@ from api.db.services.user_canvas_version import UserCanvasVersionService
|
|||
from api.utils.api_utils import (
|
||||
get_data_error_result,
|
||||
get_json_result,
|
||||
request_json,
|
||||
get_request_json,
|
||||
server_error_response,
|
||||
validate_request,
|
||||
)
|
||||
from common import settings
|
||||
from common.constants import RetCode, StatusEnum
|
||||
from common.misc_utils import get_uuid
|
||||
from api.utils.api_utils import get_json_result, server_error_response, validate_request, get_data_error_result, \
|
||||
get_request_json
|
||||
from agent.canvas import Canvas
|
||||
from peewee import MySQLDatabase, PostgresqlDatabase
|
||||
from api.db.db_models import APIToken, Task
|
||||
import time
|
||||
|
||||
from rag.flow.pipeline import Pipeline
|
||||
from rag.nlp import search
|
||||
|
|
@ -97,7 +91,7 @@ async def rm():
|
|||
@validate_request("dsl", "title")
|
||||
@login_required
|
||||
async def save() -> Any:
|
||||
req: Dict[str, Any] = await request_json()
|
||||
req: Dict[str, Any] = await get_request_json()
|
||||
if not isinstance(req["dsl"], str):
|
||||
req["dsl"] = json.dumps(req["dsl"], ensure_ascii=False)
|
||||
req["dsl"] = json.loads(req["dsl"])
|
||||
|
|
@ -292,7 +286,7 @@ def cancel(task_id):
|
|||
@validate_request("id")
|
||||
@login_required
|
||||
async def reset():
|
||||
req = await request_json()
|
||||
req = await get_request_json()
|
||||
# Check update permission (to reset the canvas)
|
||||
if not UserCanvasService.accessible(req["id"], current_user.id, required_permission="update"):
|
||||
return get_json_result(
|
||||
|
|
@ -354,7 +348,7 @@ def input_form():
|
|||
@validate_request("id", "component_id", "params")
|
||||
@login_required
|
||||
async def debug():
|
||||
req = await request_json()
|
||||
req = await get_request_json()
|
||||
# Check read permission (to debug the canvas)
|
||||
if not UserCanvasService.accessible(req["id"], current_user.id, required_permission="read"):
|
||||
return get_json_result(
|
||||
|
|
|
|||
|
|
@ -30,7 +30,7 @@ from api.db.services.file_service import FileService
|
|||
from api.db.services.pipeline_operation_log_service import PipelineOperationLogService
|
||||
from api.db.services.task_service import TaskService, GRAPH_RAPTOR_FAKE_DOC_ID
|
||||
from api.db.services.user_service import TenantService, UserTenantService
|
||||
from api.utils.api_utils import get_error_data_result, server_error_response, get_data_error_result, validate_request, not_allowed_parameters, request_json
|
||||
from api.utils.api_utils import get_error_data_result, server_error_response, get_data_error_result, validate_request, not_allowed_parameters, get_request_json
|
||||
from api.db import VALID_FILE_TYPES, UserTenantRole
|
||||
from api.db.services.knowledgebase_service import KnowledgebaseService
|
||||
from api.db.db_models import File
|
||||
|
|
@ -49,7 +49,7 @@ from api.apps import login_required, current_user
|
|||
@login_required
|
||||
@validate_request("name")
|
||||
async def create() -> Any:
|
||||
req: Dict[str, Any] = await request_json()
|
||||
req: Dict[str, Any] = await get_request_json()
|
||||
|
||||
# Validate shared_tenant_id if provided
|
||||
shared_tenant_id: Optional[str] = req.get("shared_tenant_id")
|
||||
|
|
@ -308,7 +308,7 @@ async def list_kbs():
|
|||
@login_required
|
||||
@validate_request("kb_id")
|
||||
async def rm():
|
||||
req = await request_json()
|
||||
req = await get_request_json()
|
||||
# Check if user has delete permission
|
||||
if not KnowledgebaseService.accessible4deletion(req["kb_id"], current_user.id):
|
||||
return get_json_result(
|
||||
|
|
|
|||
|
|
@ -31,15 +31,12 @@ from api.common.permission_utils import (
|
|||
get_user_permissions as get_member_permissions,
|
||||
update_user_permissions as update_member_permissions,
|
||||
)
|
||||
from api.db import FileType, UserTenantRole
|
||||
from api.db.db_models import UserTenant
|
||||
from api.db import FileType
|
||||
from api.db.services.file_service import FileService
|
||||
from api.db.services.llm_service import get_init_tenant_llm
|
||||
from api.db.services.tenant_llm_service import TenantLLMService
|
||||
from api.db.services.user_service import (
|
||||
TenantService,
|
||||
UserService,
|
||||
UserTenantService,
|
||||
)
|
||||
from api.utils.api_utils import (
|
||||
get_data_error_result,
|
||||
|
|
@ -52,10 +49,7 @@ from common import settings
|
|||
from common.constants import RetCode, StatusEnum
|
||||
from common.misc_utils import get_uuid
|
||||
from common.time_utils import delta_seconds
|
||||
from api.utils.api_utils import get_data_error_result, get_json_result, get_request_json, server_error_response, validate_request
|
||||
from api.utils.web_utils import send_invite_email
|
||||
from common import settings
|
||||
from api.apps import smtp_mail_server, login_required, current_user
|
||||
from api.utils.api_utils import get_request_json
|
||||
|
||||
# manager = Blueprint("tenant", __name__)
|
||||
def is_team_admin_or_owner(tenant_id: str, user_id: str) -> bool:
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue