chore: import upstream snapshot with attribution

test/python/testpipeline/testllm/testgenerator.py

@@ -0,0 +1,24 @@
+"""
+Generator module tests
+"""
+
+import unittest
+
+from txtai.pipeline import Generator
+
+
+class TestGenerator(unittest.TestCase):
+    """
+    Sequences tests.
+    """
+
+    def testGeneration(self):
+        """
+        Test text pipeline generation
+        """
+
+        model = Generator("hf-internal-testing/tiny-random-gpt2")
+        start = "Hello, how are"
+
+        # Test that text is generated
+        self.assertIsNotNone(model(start))