[OND211-2329]: Fixed ruff checks.
This commit is contained in:
parent
8a9ada69dc
commit
b0ecda42fb
1 changed files with 12 additions and 31 deletions
|
|
@ -15,8 +15,6 @@
|
|||
#
|
||||
import json
|
||||
import logging
|
||||
import re
|
||||
import sys
|
||||
import time
|
||||
from functools import partial
|
||||
from typing import Any, Dict, Optional
|
||||
|
|
@ -27,24 +25,17 @@ from quart import (
|
|||
request,
|
||||
Response,
|
||||
)
|
||||
import trio
|
||||
|
||||
from agent.canvas import Canvas
|
||||
from agent.component import LLM
|
||||
from api.apps import current_user, login_required
|
||||
from api.common.permission_utils import has_permission
|
||||
from api.db import CanvasCategory, FileType
|
||||
from api.db import CanvasCategory
|
||||
from api.db.db_models import APIToken, Task
|
||||
from api.db.services.canvas_service import (
|
||||
API4ConversationService,
|
||||
CanvasTemplateService,
|
||||
UserCanvasService,
|
||||
)
|
||||
from functools import partial
|
||||
from quart import request, Response, make_response
|
||||
from agent.component import LLM
|
||||
from api.db import CanvasCategory
|
||||
from api.db.services.canvas_service import CanvasTemplateService, UserCanvasService, API4ConversationService
|
||||
from api.db.services.document_service import DocumentService
|
||||
from api.db.services.file_service import FileService
|
||||
from api.db.services.pipeline_operation_log_service import (
|
||||
|
|
@ -64,19 +55,9 @@ from api.utils.api_utils import (
|
|||
server_error_response,
|
||||
validate_request,
|
||||
)
|
||||
from api.utils.file_utils import filename_type, read_potential_broken_pdf
|
||||
from common import settings
|
||||
from common.constants import RetCode, StatusEnum
|
||||
from common.misc_utils import get_uuid
|
||||
from common.constants import RetCode
|
||||
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, \
|
||||
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
|
||||
from rag.utils.redis_conn import REDIS_CONN
|
||||
|
|
@ -112,7 +93,7 @@ async def save() -> Any:
|
|||
if not isinstance(req["dsl"], str):
|
||||
req["dsl"] = json.dumps(req["dsl"], ensure_ascii=False)
|
||||
req["dsl"] = json.loads(req["dsl"])
|
||||
|
||||
|
||||
# Validate shared_tenant_id if provided
|
||||
shared_tenant_id: Optional[str] = req.get("shared_tenant_id")
|
||||
if shared_tenant_id:
|
||||
|
|
@ -131,7 +112,7 @@ async def save() -> Any:
|
|||
message="You are not a member of the selected team",
|
||||
code=RetCode.PERMISSION_ERROR
|
||||
)
|
||||
|
||||
|
||||
cate = req.get("canvas_category", CanvasCategory.Agent)
|
||||
if "id" not in req:
|
||||
# Check create permission if sharing with team
|
||||
|
|
@ -279,8 +260,8 @@ async def rerun():
|
|||
doc["chunk_num"] = 0
|
||||
doc["token_num"] = 0
|
||||
DocumentService.clear_chunk_num_when_rerun(doc["id"])
|
||||
DocumentService.update_by_id(id, doc)
|
||||
TaskService.filter_delete([Task.doc_id == id])
|
||||
DocumentService.update_by_id(doc["id"], doc)
|
||||
TaskService.filter_delete([Task.doc_id == doc["id"]])
|
||||
|
||||
dsl = req["dsl"]
|
||||
dsl["path"] = [req["component_id"]]
|
||||
|
|
@ -338,7 +319,7 @@ async def upload(canvas_id):
|
|||
try:
|
||||
return get_json_result(data=FileService.upload_info(user_id, file, request.args.get("url")))
|
||||
except Exception as e:
|
||||
return server_error_response(e)
|
||||
return server_error_response(e)
|
||||
|
||||
|
||||
@manager.route('/input_form', methods=['GET']) # noqa: F821
|
||||
|
|
@ -383,7 +364,7 @@ async def debug():
|
|||
|
||||
if isinstance(component, LLM):
|
||||
component.set_debug_inputs(req["params"])
|
||||
component.invoke(**{k: o["value"] for k,o in req["params"].items()})
|
||||
component.invoke(**{k: o["value"] for k, o in req["params"].items()})
|
||||
outputs = component.output()
|
||||
for k in outputs.keys():
|
||||
if isinstance(outputs[k], partial):
|
||||
|
|
@ -495,7 +476,7 @@ async def test_db_connect():
|
|||
@login_required
|
||||
def getlistversion(canvas_id):
|
||||
try:
|
||||
versions =sorted([c.to_dict() for c in UserCanvasVersionService.list_by_canvas_id(canvas_id)], key=lambda x: x["update_time"]*-1)
|
||||
versions = sorted([c.to_dict() for c in UserCanvasVersionService.list_by_canvas_id(canvas_id)], key=lambda x: x["update_time"] * -1)
|
||||
return get_json_result(data=versions)
|
||||
except Exception as e:
|
||||
return get_data_error_result(message=f"Error getting history files: {e}")
|
||||
|
|
@ -504,7 +485,7 @@ def getlistversion(canvas_id):
|
|||
#api get version dsl of canvas
|
||||
@manager.route('/getversion/<version_id>', methods=['GET']) # noqa: F821
|
||||
@login_required
|
||||
def getversion( version_id):
|
||||
def getversion(version_id):
|
||||
try:
|
||||
e, version = UserCanvasVersionService.get_by_id(version_id)
|
||||
if version:
|
||||
|
|
@ -525,7 +506,7 @@ def list_canvas():
|
|||
desc = False
|
||||
else:
|
||||
desc = True
|
||||
owner_ids = [id for id in request.args.get("owner_ids", "").strip().split(",") if id]
|
||||
owner_ids = [owner_id for owner_id in request.args.get("owner_ids", "").strip().split(",") if owner_id]
|
||||
if not owner_ids:
|
||||
tenants = TenantService.get_joined_tenants_by_user_id(current_user.id)
|
||||
tenants = [m["tenant_id"] for m in tenants]
|
||||
|
|
@ -556,7 +537,7 @@ async def setting():
|
|||
code=RetCode.PERMISSION_ERROR
|
||||
)
|
||||
|
||||
e,flow = UserCanvasService.get_by_id(req["id"])
|
||||
e, flow = UserCanvasService.get_by_id(req["id"])
|
||||
if not e:
|
||||
return get_data_error_result(message="canvas not found.")
|
||||
flow = flow.to_dict()
|
||||
|
|
@ -566,7 +547,7 @@ async def setting():
|
|||
if value := req.get(key):
|
||||
flow[key] = value
|
||||
|
||||
num= UserCanvasService.update_by_id(req["id"], flow)
|
||||
num = UserCanvasService.update_by_id(req["id"], flow)
|
||||
return get_json_result(data=num)
|
||||
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue