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

fix(ChatKnowledge): Fix knowledge command load bug (#1767)

Browse Source 
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
dbgpt/app/knowledge/_cli/knowledge_client.py
View File

@ -125,7 +125,7 @@ def knowledge_init(
logger.info(f"Create space: {space}") logger.info(f"Create space: {space}")
client.space_add(space) client.space_add(space)
logger.info("Create space successfully") 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: if len(space_list) != 1:
raise Exception(f"List space {space.name} error") raise Exception(f"List space {space.name} error")
space = KnowledgeSpaceRequest(**space_list[0]) space = KnowledgeSpaceRequest(**space_list[0])
@ -146,7 +146,10 @@ def knowledge_init(
f"Document {filename} already exist in space {space.name}, overwrite it" f"Document {filename} already exist in space {space.name}, overwrite it"
) )
client.document_delete( 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( doc_id = client.document_upload(
space.name, filename, KnowledgeType.DOCUMENT.value, filename space.name, filename, KnowledgeType.DOCUMENT.value, filename

4
dbgpt/app/knowledge/request/request.py
View File

@ -16,12 +16,12 @@ class KnowledgeSpaceRequest(BaseModel):
"""name: knowledge space name""" """name: knowledge space name"""
"""vector_type: vector type""" """vector_type: vector type"""
id: int = None id: Optional[int] = None
name: str = None name: str = None
"""vector_type: vector type""" """vector_type: vector type"""
vector_type: str = None vector_type: str = None
"""vector_type: vector type""" """vector_type: vector type"""
domain_type: str = "normal" domain_type: str = "Normal"
"""desc: description""" """desc: description"""
desc: str = None desc: str = None
"""owner: owner""" """owner: owner"""

2
dbgpt/serve/rag/service/service.py
View File

@ -460,7 +460,7 @@ class Service(BaseService[KnowledgeSpaceEntity, SpaceServeRequest, SpaceServeRes
) )
knowledge = None knowledge = None
if not space.domain_type or ( if not space.domain_type or (
space.domain_type == BusinessFieldType.NORMAL.value space.domain_type.lower() == BusinessFieldType.NORMAL.value.lower()
): ):
knowledge = KnowledgeFactory.create( knowledge = KnowledgeFactory.create(
datasource=doc.content, datasource=doc.content,