github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-13 14:06:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: pdf problem

Browse Source
This commit is contained in:
aries-ckt 2023-06-06 11:50:56 +08:00
parent c752c9de08
commit b4e31c8ea9
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pilot/source_embedding/pdf_embedding.py
View File

@ -15,12 +15,11 @@ CFG = Config()
class PDFEmbedding(SourceEmbedding): class PDFEmbedding(SourceEmbedding):
"""pdf embedding for read pdf document.""" """pdf embedding for read pdf document."""
def __init__(self, file_path, vector_store_config, encoding): def __init__(self, file_path, vector_store_config):
"""Initialize with pdf path.""" """Initialize with pdf path."""
super().__init__(file_path, vector_store_config) super().__init__(file_path, vector_store_config)
self.file_path = file_path self.file_path = file_path
self.vector_store_config = vector_store_config self.vector_store_config = vector_store_config
self.encoding = encoding
@register @register
def read(self): def read(self):