added feature #887

This commit is contained in:
sarfaraz siddiqui
2024-03-31 03:50:11 +05:30
parent d18598bc33
commit afecae3786
2 changed files with 18 additions and 2 deletions

View File

@@ -207,7 +207,9 @@ def stream():
prompt_id = data["prompt_id"]
else:
prompt_id = 'default'
if 'chunks' in data:
if 'selectedDocs' in data and data['selectedDocs'] is None:
chunks = 0
elif 'chunks' in data:
chunks = int(data["chunks"])
else:
chunks = 2