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

Merge branch 'dev' into llm_fxp

Browse Source
This commit is contained in:
yhjun1026 2023-06-06 14:40:37 +08:00
commit 49746b0247
2 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pilot/source_embedding/knowledge_embedding.py
View File

@ -49,7 +49,6 @@ class KnowledgeEmbedding:
if self.file_type == "url":
embedding = URLEmbedding(
file_path=self.file_path,
model_name=self.model_name,
vector_store_config=self.vector_store_config,
)
return embedding

3
pilot/source_embedding/pdf_embedding.py
View File

@ -15,12 +15,11 @@ CFG = Config()
class PDFEmbedding(SourceEmbedding):
"""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."""
super().__init__(file_path, vector_store_config)
self.file_path = file_path
self.vector_store_config = vector_store_config
self.encoding = encoding
@register
def read(self):