|  | @@ -8,6 +8,7 @@ from typing import Optional
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  from flask_login import current_user
 | 
	
		
			
				|  |  |  from sqlalchemy import func
 | 
	
		
			
				|  |  | +from werkzeug.exceptions import NotFound
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  from configs import dify_config
 | 
	
		
			
				|  |  |  from core.errors.error import LLMBadRequestError, ProviderTokenNotInitError
 | 
	
	
		
			
				|  | @@ -975,6 +976,8 @@ class DocumentService:
 | 
	
		
			
				|  |  |      ):
 | 
	
		
			
				|  |  |          DatasetService.check_dataset_model_setting(dataset)
 | 
	
		
			
				|  |  |          document = DocumentService.get_document(dataset.id, document_data["original_document_id"])
 | 
	
		
			
				|  |  | +        if document is None:
 | 
	
		
			
				|  |  | +            raise NotFound("Document not found")
 | 
	
		
			
				|  |  |          if document.display_status != "available":
 | 
	
		
			
				|  |  |              raise ValueError("Document is not available")
 | 
	
		
			
				|  |  |          # update document name
 |