AI & Agent Dev Bug Sandbox Back to Radar

HuggingFacePipeline._Generate Rejects Valid Translation Tasks Due To Substring Check

The `_generate` method uses `self.pipeline.task in "translation"` instead of an equality or prefix check, causing valid transformers translation tasks like `translation_en_to_fr` to raise ValueError. It also incorrectly matches empty string.

highConfidence 95%Langchain-Huggingface Affected V1.2.2

Origin Analysis

The condition `elif self.pipeline.task in "translation":` performs a substring check against the string `"translation"`. Since `"translation_en_to_fr"` is not a substring of `"translation"`, it falls through to the else branch and raises ValueError. Additionally, any string that is a substring of `"translation"` (including the empty string) will erroneously enter the translation branch.

Run the provided minimal example with a MagicMock pipeline whose `task` attribute is set to `"translation_en_to_fr"`. Calling `_generate` raises ValueError: Got invalid task translation_en_to_fr.

Fixing Code Block

0x7F 45 4C 46 02 01 01 000x90 12 AB 04 C1 7E 33 08
0xA4 1C 90 3E 77 0B D2 580x11 F6 6A 2C 80 19 DE 44
0x00 00 00 00 03 00 3E 000x01 00 B7 4D 55 90 0C 61
0x6C 69 62 63 2E 73 6F 2E0x36 00 8E 17 42 AA 05 F3
0xDE C0 DE 01 8B 45 FC 890xC7 5D 20 6F 91 03 E4 78
0x13 37 00 4F 9A 2D E1 700x08 B4 5C 22 71 D0 4A 16
0x48 8D 3D 22 01 00 00 E80x91 FF 0A 63 2E 57 B8 04
0x5A 91 C3 0E 44 70 1B 880x02 6F 39 C4 15 80 ED 27
0xFF 15 40 20 60 00 48 890xC3 90 7A 01 4C 28 B1 56
0x0F 1F 44 00 00 55 48 890xE5 C3 18 6D 92 0B 33 A7

Edge Case Audit

This monkey patch overrides the complete `_generate` method. If the original method included additional behavior (e.g., batch processing, stop sequence handling, or newer task types), those features are not replicated here and may be lost. The extraction key for `image-text-to-text` is assumed to be `generated_text`; if the actual pipeline output uses a different key, that branch must be adjusted. Apply this patch only as a temporary fix until an official release includes the correction. To rollback, remove the assignment `HuggingFacePipeline._generate = _generate` and restart the Python process or reload the module.