AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-08-12 15:59:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix error in exception (#2694)

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

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Kevin Hu 2024-09-30 17:54:27 +08:00 committed by GitHub
parent d6db805885
commit 0a7654c747
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
rag/llm/chat_model.py
View File

@ -1340,7 +1340,7 @@ class GoogleChat(Base):
+ response["usage"]["output_tokens"], + response["usage"]["output_tokens"],
) )
except Exception as e: except Exception as e:
return ans + "\n**ERROR**: " + str(e), 0 return "\n**ERROR**: " + str(e), 0
else: else:
self.client._system_instruction = self.system self.client._system_instruction = self.system
if "max_tokens" in gen_conf: if "max_tokens" in gen_conf: