fix type issue w/ mistral referencechunk

UKGovernmentBEIS · Nov 18, 2024 · c998c95 · c998c95
1 parent 589d86e
commit c998c95
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/inspect_ai/model/_providers/mistral.py b/src/inspect_ai/model/_providers/mistral.py
@@ -9,6 +9,7 @@
     ImageURL,
     ImageURLChunk,
     Mistral,
+    ReferenceChunk,
     TextChunk,
 )
 from mistralai.models import (
@@ -351,7 +352,9 @@ def completion_content(content: str | list[ContentChunk]) -> str | list[Content]
 
 
 def completion_content_chunk(content: ContentChunk) -> Content:
-    if isinstance(content, TextChunk):
+    if isinstance(content, ReferenceChunk):
+        raise TypeError("ReferenceChunk content is not supported by Inspect.")
+    elif isinstance(content, TextChunk):
         return ContentText(text=content.text)
     else:
         if isinstance(content.image_url, str):