mirror of https://github.com/langgenius/dify.git
59 lines
2.3 KiB
Python
59 lines
2.3 KiB
Python
from flask_restful import reqparse
|
|
from sqlalchemy.orm.session import Session
|
|
|
|
from controllers.web import api
|
|
from controllers.web.wraps import WebApiResource
|
|
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
|
|
|
|
|
|
class MemoryListApi(WebApiResource):
|
|
def get(self, app_model):
|
|
parser = reqparse.RequestParser()
|
|
parser.add_argument("conversation_id", required=False, type=uuid_value, location="args")
|
|
args = parser.parse_args()
|
|
conversation_id = args.get("conversation_id")
|
|
|
|
result = ChatflowMemoryService.get_persistent_memories(app_model)
|
|
if conversation_id:
|
|
result = [*result, *ChatflowMemoryService.get_session_memories(app_model, conversation_id)]
|
|
|
|
return [it for it in result if it.end_user_visible]
|
|
|
|
class MemoryEditApi(WebApiResource):
|
|
def put(self, app_model):
|
|
parser = reqparse.RequestParser()
|
|
parser.add_argument('id', type=str, required=True)
|
|
parser.add_argument('node_id', type=str, required=False)
|
|
parser.add_argument('update', type=str, required=True)
|
|
args = parser.parse_args()
|
|
workflow = WorkflowService().get_published_workflow(app_model)
|
|
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)
|
|
if not memory_spec:
|
|
return {'error': 'Memory not found'}, 404
|
|
if not memory_spec.end_user_editable:
|
|
return {'error': 'Memory not editable'}, 403
|
|
with Session(db.engine) as session:
|
|
session.merge(
|
|
ChatflowMemoryVariable(
|
|
tenant_id=app_model.tenant_id,
|
|
app_id=app_model.id,
|
|
node_id=args['node_id'],
|
|
memory_id=args['id'],
|
|
name=memory_spec.name,
|
|
value=args['update'],
|
|
scope=memory_spec.scope,
|
|
term=memory_spec.term,
|
|
)
|
|
)
|
|
session.commit()
|
|
return '', 204
|
|
|
|
api.add_resource(MemoryListApi, '/memories')
|
|
api.add_resource(MemoryEditApi, '/memory-edit')
|