Skip to content

Commit

Permalink
remove file size check for sdk API (#3066)
Browse files Browse the repository at this point in the history
### What problem does this PR solve?

### Type of change

- [ ] Bug Fix (non-breaking change which fixes an issue)
- [ ] New Feature (non-breaking change which adds functionality)
- [ ] Documentation Update
- [x] Refactoring
- [ ] Performance Improvement
- [ ] Other (please describe):
  • Loading branch information
KevinHuSh authored Oct 28, 2024
1 parent e56bd77 commit d133cc0
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 21 deletions.
22 changes: 2 additions & 20 deletions api/apps/sdk/doc.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,14 +46,6 @@
from rag.utils.storage_factory import STORAGE_IMPL
import os

MAXIMUM_OF_UPLOADING_FILES = 256

MAXIMUM_OF_UPLOADING_FILES = 256

MAXIMUM_OF_UPLOADING_FILES = 256

MAXIMUM_OF_UPLOADING_FILES = 256


@manager.route('/datasets/<dataset_id>/documents', methods=['POST'])
@token_required
Expand All @@ -66,21 +58,11 @@ def upload(dataset_id, tenant_id):
if file_obj.filename == '':
return get_result(
retmsg='No file selected!', retcode=RetCode.ARGUMENT_ERROR)
# total size
total_size = 0
for file_obj in file_objs:
file_obj.seek(0, os.SEEK_END)
total_size += file_obj.tell()
file_obj.seek(0)
MAX_TOTAL_FILE_SIZE=10*1024*1024
if total_size > MAX_TOTAL_FILE_SIZE:
return get_result(
retmsg=f'Total file size exceeds 10MB limit! ({total_size / (1024 * 1024):.2f} MB)',
retcode=RetCode.ARGUMENT_ERROR)

e, kb = KnowledgebaseService.get_by_id(dataset_id)
if not e:
raise LookupError(f"Can't find the dataset with ID {dataset_id}!")
err, files= FileService.upload_document(kb, file_objs, tenant_id)
err, files = FileService.upload_document(kb, file_objs, tenant_id)
if err:
return get_result(
retmsg="\n".join(err), retcode=RetCode.SERVER_ERROR)
Expand Down
2 changes: 1 addition & 1 deletion intergrations/chatgpt-on-wechat/plugins/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -52,6 +52,6 @@ This file must be configured to point to your RAGFlow instance, with the `ragflo
Before you can use this plugin, ensure the following are in place:

1. You have installed and configured [ChatGPT-on-WeChat](https://github.com/zhayujie/chatgpt-on-wechat).
2. You have deployed and are running the [RAGFlow](https://github.com/jina-ai/ragflow) server.
2. You have deployed and are running the [RAGFlow](https://github.com/infiniflow/ragflow) server.

Make sure both `config.json` files (ChatGPT-on-WeChat and RAGFlow Chat Plugin) are correctly set up as per the examples above.

0 comments on commit d133cc0

Please sign in to comment.