Skip to content

Commit 7ec5a73

Browse files
committed
#98 adopt to universal text object
1 parent 934c591 commit 7ec5a73

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

text_extract_api/extract/strategies/minicpm_v.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44

55
import ollama
66

7+
from extract.extract_result import ExtractResult
78
from text_extract_api.extract.strategies.strategy import Strategy
89
from text_extract_api.files.file_formats.file_format import FileFormat
910
from text_extract_api.files.file_formats.image import ImageFileFormat
@@ -16,7 +17,7 @@ class MiniCPMVStrategy(Strategy):
1617
def name(cls) -> str:
1718
return "minicpm_v"
1819

19-
def extract_text(self, file_format: FileFormat, language: str = 'en') -> str:
20+
def extract_text(self, file_format: FileFormat, language: str = 'en') -> ExtractResult:
2021

2122
if (
2223
not isinstance(file_format, ImageFileFormat)
@@ -66,4 +67,4 @@ def extract_text(self, file_format: FileFormat, language: str = 'en') -> str:
6667

6768
print(response)
6869

69-
return extracted_text
70+
return ExtractResult.from_text(extracted_text)

0 commit comments

Comments
 (0)