Add LLamaCloud support (#5)

avvertix · web-flow · commit e9f1248100b2 · 2025-02-25T16:22:11.000+01:00
diff --git a/src/DocumentProcessor.php b/src/DocumentProcessor.php
@@ -17,4 +17,11 @@ enum DocumentProcessor: string
      * Uses https://github.com/pymupdf/PyMuPDF as document processor to extract text
      */
     case PYMUPDF = 'pymupdf';
+
+    /**
+     * The LLama Parse processor
+     *
+     * Uses LLamaCloud https://cloud.llamaindex.ai/ as document processor to extract text
+     */
+    case LLAMAPARSE = 'llama';
 }
diff --git a/tests/ParseProcessorSelectionTest.php b/tests/ParseProcessorSelectionTest.php
@@ -68,3 +68,34 @@
 
     $mockClient->assertSentCount(1);
 });
+
+test('llamaparse can be selected as processor', function () {
+    $mockClient = MockClient::global([
+        ExtractTextRequest::class => MockResponse::fixture('extract-text-empty'),
+    ]);
+
+    $connector = new ParseConnector('fake', 'http://localhost:5002');
+    $connector->withMockClient($mockClient);
+
+    $connector->parse(
+        url: 'http://localhost/empty.pdf',
+        options: new ParseOption(DocumentProcessor::LLAMAPARSE),
+    );
+
+    $mockClient->assertSent(ExtractTextRequest::class);
+
+    $mockClient->assertSent(function (Request $request, Response $response) {
+        if (! $request instanceof ExtractTextRequest) {
+            return false;
+        }
+
+        /** @var array */
+        $body = $request->body()->all();
+
+        return $body['url'] === 'http://localhost/empty.pdf'
+            && $body['mime_type'] === 'application/pdf'
+            && $body['driver'] === 'llama';
+    });
+
+    $mockClient->assertSentCount(1);
+});
