Browse Source
make gemini robust (#3012)
### What problem does this PR solve?
#3003
### Type of change
- [x] Bug Fix (non-breaking change which fixes an issue)
tags/v0.13.0
Kevin Hu
1 year ago
No account linked to committer's email address
|
|
|
@@ -780,10 +780,11 @@ class GeminiChat(Base): |
|
|
|
ans += resp.text |
|
|
|
yield ans |
|
|
|
|
|
|
|
yield response._chunks[-1].usage_metadata.total_token_count |
|
|
|
except Exception as e: |
|
|
|
yield ans + "\n**ERROR**: " + str(e) |
|
|
|
|
|
|
|
yield response._chunks[-1].usage_metadata.total_token_count |
|
|
|
yield 0 |
|
|
|
|
|
|
|
|
|
|
|
class GroqChat: |