File size: 819 Bytes
3943768 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
--- /home/jon/miniconda3/envs/h2ogpt/lib/python3.10/site-packages/langchain_google_genai/chat_models.py 2024-07-25 17:02:46.040222538 -0700
+++ /home/jon/chat_models.py 2024-07-25 17:01:48.722952945 -0700
@@ -550,7 +550,10 @@
for candidate in response.candidates:
generation_info = {}
if candidate.finish_reason:
- generation_info["finish_reason"] = candidate.finish_reason.name
+ if hasattr(candidate.finish_reason, 'name'):
+ generation_info["finish_reason"] = candidate.finish_reason.name
+ else:
+ generation_info["finish_reason"] = 'unknown'
generation_info["safety_ratings"] = [
proto.Message.to_dict(safety_rating, use_integers_for_enums=False)
for safety_rating in candidate.safety_ratings
|