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": if self.file_type == "url":
embedding = URLEmbedding( embedding = URLEmbedding(
file_path=self.file_path, file_path=self.file_path,
model_name=self.model_name,
vector_store_config=self.vector_store_config, vector_store_config=self.vector_store_config,
) )
return embedding return embedding

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):