Fix tool_choice being dropped in inspect_model_request

UKGovernmentBEIS · Mar 7, 2025 · 8af7553 · 8af7553
1 parent db5f359
commit 8af7553
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/inspect_ai/solver/_bridge/patch.py b/src/inspect_ai/solver/_bridge/patch.py
@@ -114,6 +114,8 @@ async def inspect_model_request(
             )
         )
 
+    tool_choice = json_data.get("tool_choice", None)
+
     # resolve model
     if model_name == "inspect":
         model = get_model()
@@ -123,6 +125,7 @@ async def inspect_model_request(
     output = await model.generate(
         input=input,
         tools=inspect_tools,
+        tool_choice=tool_choice,
         config=generate_config_from_openai(options),
     )