Skip to content

Commit fd3cbc8

Browse files
committed
fix(generate): use 2nd part for gemini thinking models
1 parent ba18aaf commit fd3cbc8

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

bigcodebench/provider/google.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,10 @@ def codegen(
4343
for candidate in ret.candidates:
4444
parts = candidate.content.parts
4545
if parts:
46-
outputs.append(parts[0].text)
46+
if "-thinking-" in self.name:
47+
outputs.append(parts[1].text)
48+
else:
49+
outputs.append(parts[0].text)
4750
else:
4851
print("Empty response!")
4952
outputs.append("")

0 commit comments

Comments
 (0)