fix: Gemini HARM_CATEGORY_JAILBREAK and Anthropic tool_result blocks

instructor/providers/anthropic/utils.py

@@ -9,8 +9,8 @@
 from textwrap import dedent
 from typing import Any, TypedDict, Union
 
 from pydantic import ValidationError
 from ...core.exceptions import ValidationError as InstructorValidationError
 
 from ...mode import Mode
 from ...processing.schema import generate_anthropic_schema
@@ -156,11 +156,7 @@
     tool_use_id = None
     for content in response.content:
         assistant_content.append(content.model_dump())  # type: ignore
-        if (
-            content.type == "tool_use"
-            and isinstance(exception, (ValidationError, InstructorValidationError))
-            and content.name == exception.title
-        ):
+        if content.type == "tool_use":
             tool_use_id = content.id
 
     reask_msgs = [{"role": "assistant", "content": assistant_content}]  # type: ignore

instructor/providers/gemini/utils.py

@@ -248,10 +248,15 @@ def update_genai_kwargs(
 
     # Filter out image related harm categories which are not
     # supported for text based models
+    # Exclude JAILBREAK category as it's only for Vertex AI, not google.genai
+    excluded_categories = {HarmCategory.HARM_CATEGORY_UNSPECIFIED}
+    if hasattr(HarmCategory, 'HARM_CATEGORY_JAILBREAK'):
+        excluded_categories.add(HarmCategory.HARM_CATEGORY_JAILBREAK)
+
     supported_categories = [
         c
         for c in HarmCategory
-        if c != HarmCategory.HARM_CATEGORY_UNSPECIFIED
+        if c not in excluded_categories
         and not c.name.startswith("HARM_CATEGORY_IMAGE_")
     ]
 

tests/llm/test_genai/test_utils.py

@@ -34,10 +34,15 @@ def test_update_genai_kwargs_safety_settings():
     """Test that safety settings are properly configured."""
     from google.genai.types import HarmCategory, HarmBlockThreshold
 
+    # Exclude JAILBREAK category as it's only for Vertex AI, not google.genai
+    excluded_categories = {HarmCategory.HARM_CATEGORY_UNSPECIFIED}
+    if hasattr(HarmCategory, 'HARM_CATEGORY_JAILBREAK'):
+        excluded_categories.add(HarmCategory.HARM_CATEGORY_JAILBREAK)
+
     supported_categories = [
         c
         for c in HarmCategory
-        if c != HarmCategory.HARM_CATEGORY_UNSPECIFIED
+        if c not in excluded_categories
         and not c.name.startswith("HARM_CATEGORY_IMAGE_")
     ]
 
@@ -65,10 +70,15 @@ def test_update_genai_kwargs_with_custom_safety_settings():
     """Test that custom safety settings are properly handled."""
     from google.genai.types import HarmCategory, HarmBlockThreshold
 
+    # Exclude JAILBREAK category as it's only for Vertex AI, not google.genai
+    excluded_categories = {HarmCategory.HARM_CATEGORY_UNSPECIFIED}
+    if hasattr(HarmCategory, 'HARM_CATEGORY_JAILBREAK'):
+        excluded_categories.add(HarmCategory.HARM_CATEGORY_JAILBREAK)
+
     supported_categories = [
         c
         for c in HarmCategory
-        if c != HarmCategory.HARM_CATEGORY_UNSPECIFIED
+        if c not in excluded_categories
         and not c.name.startswith("HARM_CATEGORY_IMAGE_")
     ]