--- /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 | |
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 | |