youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-04 09:28:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: fix basedpyright/ruff errors

Browse Source
This commit is contained in:
Stream
2025-09-22 15:17:19 +08:00
parent e9313b9c1b
commit 394b7d09b8
11 changed files with 37 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
api/controllers/service_api/app/chatflow_memory.py
View File

@ -1,13 +1,9 @@
from flask_restx import Resource, reqparse
from sqlalchemy.orm import Session
from controllers.service_api import api
from controllers.service_api.wraps import FetchUserArg, WhereisUserArg, validate_app_token
from core.memory.entities import MemoryBlock
from core.workflow.entities.variable_pool import VariablePool
from libs.helper import uuid_value
from models import db
from models.chatflow_memory import ChatflowMemoryVariable
from services.chatflow_memory_service import ChatflowMemoryService
from services.workflow_service import WorkflowService
@ -31,6 +27,7 @@ class MemoryListApi(Resource):
result = [it for it in result if it.memory_id == memory_id]
return [it for it in result if it.end_user_visible]
class MemoryEditApi(Resource):
@validate_app_token(fetch_user_arg=FetchUserArg(fetch_from=WhereisUserArg.JSON, required=True))
def put(self, app_model):
@ -44,6 +41,8 @@ class MemoryEditApi(Resource):
update = args.get("update")
conversation_id = args.get("conversation_id")
node_id = args.get("node_id")
if not isinstance(update, str):
return {'error': 'Invalid update'}, 400
if not workflow:
return {'error': 'Workflow not found'}, 404
memory_spec = next((it for it in workflow.memory_blocks if it.id == args['id']), None)
@ -63,5 +62,6 @@ class MemoryEditApi(Resource):
)
return '', 204
api.add_resource(MemoryListApi, '/memories')
api.add_resource(MemoryEditApi, '/memory-edit')

9
api/controllers/web/chatflow_memory.py
View File

@ -1,14 +1,9 @@
from flask_restx import reqparse
from sqlalchemy.orm.session import Session
from sympy import false
from controllers.web import api
from controllers.web.wraps import WebApiResource
from core.memory.entities import MemoryBlock
from core.workflow.entities.variable_pool import VariablePool
from libs.helper import uuid_value
from models import db
from models.chatflow_memory import ChatflowMemoryVariable
from services.chatflow_memory_service import ChatflowMemoryService
from services.workflow_service import WorkflowService
@ -31,6 +26,7 @@ class MemoryListApi(WebApiResource):
result = [it for it in result if it.memory_id == memory_id]
return [it for it in result if it.end_user_visible]
class MemoryEditApi(WebApiResource):
def put(self, app_model):
parser = reqparse.RequestParser()
@ -43,6 +39,8 @@ class MemoryEditApi(WebApiResource):
update = args.get("update")
conversation_id = args.get("conversation_id")
node_id = args.get("node_id")
if not isinstance(update, str):
return {'error': 'Update must be a string'}, 400
if not workflow:
return {'error': 'Workflow not found'}, 404
memory_spec = next((it for it in workflow.memory_blocks if it.id == args['id']), None)
@ -64,5 +62,6 @@ class MemoryEditApi(WebApiResource):
)
return '', 204
api.add_resource(MemoryListApi, '/memories')
api.add_resource(MemoryEditApi, '/memory-edit')