20 lines
530 B
Markdown
20 lines
530 B
Markdown
|
```python title="Python"
|
||
|
|
from kreuzberg import ExtractionResult, register_post_processor
|
||
|
|
|
||
|
|
class PdfOnlyProcessor:
|
||
|
|
def name(self) -> str:
|
||
|
|
return "pdf-only-processor"
|
||
|
|
|
||
|
|
def version(self) -> str:
|
||
|
|
return "1.0.0"
|
||
|
|
|
||
|
|
def process(self, result: ExtractionResult) -> ExtractionResult:
|
||
|
|
return result
|
||
|
|
|
||
|
|
def should_process(self, result: ExtractionResult) -> bool:
|
||
|
|
return result.mime_type == "application/pdf"
|
||
|
|
|
||
|
|
processor: PdfOnlyProcessor = PdfOnlyProcessor()
|
||
|
|
register_post_processor(processor)
|
||
|
|
```
|