pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2026-02-07 02:55:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix: Raptor: [Bug]: **ERROR**: Unknown field for GenerationConfig: max_tokens (#8331)

Browse Source 
### What problem does this PR solve?

https://github.com/infiniflow/ragflow/issues/8324

### Type of change
- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Stephen Hu
2025-06-18 16:40:57 +08:00
committed by GitHub
parent e470645efd
commit 35034fed73
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
rag/llm/chat_model.py
View File

@ -1217,7 +1217,7 @@ class GeminiChat(Base):
def _clean_conf(self, gen_conf): def _clean_conf(self, gen_conf):
for k in list(gen_conf.keys()): for k in list(gen_conf.keys()):
if k not in ["temperature", "top_p", "max_tokens"]: if k not in ["temperature", "top_p"]:
del gen_conf[k] del gen_conf[k]
return gen_conf return gen_conf