mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-07-30 15:21:02 +00:00
fix(ChatKnowledge): Fix knowledge command load bug (#1767)
Co-authored-by: huangjh131 <14678508+huangjh131@user.noreply.gitee.com> Co-authored-by: Fangyin Cheng <staneyffer@gmail.com>
This commit is contained in:
parent
25d7d94b89
commit
7ea9b8c419
@ -125,7 +125,7 @@ def knowledge_init(
|
||||
logger.info(f"Create space: {space}")
|
||||
client.space_add(space)
|
||||
logger.info("Create space successfully")
|
||||
space_list = client.space_list(KnowledgeSpaceRequest(name=space.name))
|
||||
space_list = client.space_list(space)
|
||||
if len(space_list) != 1:
|
||||
raise Exception(f"List space {space.name} error")
|
||||
space = KnowledgeSpaceRequest(**space_list[0])
|
||||
@ -146,7 +146,10 @@ def knowledge_init(
|
||||
f"Document {filename} already exist in space {space.name}, overwrite it"
|
||||
)
|
||||
client.document_delete(
|
||||
space.name, KnowledgeDocumentRequest(doc_name=filename)
|
||||
space.name,
|
||||
KnowledgeDocumentRequest(
|
||||
doc_name=filename, doc_type=KnowledgeType.DOCUMENT.value
|
||||
),
|
||||
)
|
||||
doc_id = client.document_upload(
|
||||
space.name, filename, KnowledgeType.DOCUMENT.value, filename
|
||||
|
@ -16,12 +16,12 @@ class KnowledgeSpaceRequest(BaseModel):
|
||||
"""name: knowledge space name"""
|
||||
|
||||
"""vector_type: vector type"""
|
||||
id: int = None
|
||||
id: Optional[int] = None
|
||||
name: str = None
|
||||
"""vector_type: vector type"""
|
||||
vector_type: str = None
|
||||
"""vector_type: vector type"""
|
||||
domain_type: str = "normal"
|
||||
domain_type: str = "Normal"
|
||||
"""desc: description"""
|
||||
desc: str = None
|
||||
"""owner: owner"""
|
||||
|
@ -460,7 +460,7 @@ class Service(BaseService[KnowledgeSpaceEntity, SpaceServeRequest, SpaceServeRes
|
||||
)
|
||||
knowledge = None
|
||||
if not space.domain_type or (
|
||||
space.domain_type == BusinessFieldType.NORMAL.value
|
||||
space.domain_type.lower() == BusinessFieldType.NORMAL.value.lower()
|
||||
):
|
||||
knowledge = KnowledgeFactory.create(
|
||||
datasource=doc.content,
|
||||
|
Loading…
Reference in New Issue
Block a user