diff --git a/backend/modules/brain/knowledge_brain_qa.py b/backend/modules/brain/knowledge_brain_qa.py index 39afdb6a4..aef2fe072 100644 --- a/backend/modules/brain/knowledge_brain_qa.py +++ b/backend/modules/brain/knowledge_brain_qa.py @@ -181,6 +181,8 @@ class KnowledgeBrainQA(BaseModel, QAInterface): brain_id=brain_id, chat_id=chat_id, streaming=streaming, + max_input=self.max_input, + max_tokens=self.max_tokens, **kwargs, )