perf(prebuilt): only format the call_model error message when raising

2026-11-08 15:12:33 +01:00 · 2026-11-07 15:48:22 -05:00
1 changed files with 5 additions and 5 deletions
@@ -638,15 +638,15 @@ def create_react_agent(
            messages = (
                _get_state_value(state, "llm_input_messages")
            ) or _get_state_value(state, "messages")
-            error_msg = f"Expected input to call_model to have 'llm_input_messages' or 'messages' key, but got {state}"
+            expected_keys = "'llm_input_messages' or 'messages' key"
        else:
            messages = _get_state_value(state, "messages")
-            error_msg = (
-                f"Expected input to call_model to have 'messages' key, but got {state}"
-            )
+            expected_keys = "'messages' key"

        if messages is None:
-            raise ValueError(error_msg)
+            raise ValueError(
+                f"Expected input to call_model to have {expected_keys}, but got {state}"
+            )

        _validate_chat_history(messages)
        # we're passing messages under `messages` key, as this is expected by the prompt