Skip to content

Commit b42413e

Browse files
fix: fixed validation of parsed llm output
1 parent b4124d0 commit b42413e

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

src/rai_bench/rai_bench/vlm_benchmark/interfaces.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -191,8 +191,11 @@ def get_structured_output_from_messages(
191191
)
192192

193193
parsed = validated_message.parsed
194-
if isinstance(parsed, self.structured_output):
195-
return parsed
194+
expected_output_type = self.structured_output
195+
parsed_valid_output = expected_output_type.model_validate(
196+
parsed.model_dump()
197+
)
198+
return parsed_valid_output
196199
except ValidationError:
197200
continue
198201
return None

0 commit comments

Comments
 (0)