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:
huangjh131 2024-08-01 18:41:23 +08:00 committed by GitHub
parent 25d7d94b89
commit 7ea9b8c419
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 8 additions and 5 deletions

View File

@ -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

View File

@ -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"""

View File

@ -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,