github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-17 15:10:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(app): new ui app module support

Browse Source
This commit is contained in:
yhjun1026
2024-08-08 17:47:02 +08:00
parent 98d85ec11d
commit f3403cfc10
2 changed files with 9 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
dbgpt/app/knowledge/api.py
View File

@@ -81,6 +81,7 @@ def space_list(request: KnowledgeSpaceRequest):
try: try:
return Result.succ(knowledge_space_service.get_knowledge_space(request)) return Result.succ(knowledge_space_service.get_knowledge_space(request))
except Exception as e: except Exception as e:
logger.exception(f"Space list error!{str(e)}")
return Result.failed(code="E000X", msg=f"space list error {e}") return Result.failed(code="E000X", msg=f"space list error {e}")

18
dbgpt/app/knowledge/document_db.py
View File

@@ -240,17 +240,17 @@ class KnowledgeDocumentDao(BaseDao):
# 构建一个查询，聚合每个知识空间的文档数量 # 构建一个查询，聚合每个知识空间的文档数量
counts_query = ( counts_query = (
session.query( session.query(
KnowledgeDocumentEntity.space_id, KnowledgeDocumentEntity.id,
func.count(KnowledgeDocumentEntity.id).label("document_count"), func.count(KnowledgeDocumentEntity.id).label("document_count"),
) )
.filter(KnowledgeDocumentEntity.space_id.in_(space_ids)) .filter(KnowledgeDocumentEntity.id.in_(space_ids))
.group_by(KnowledgeDocumentEntity.space_id) .group_by(KnowledgeDocumentEntity.id)
) )
# 执行查询并获取结果 # 执行查询并获取结果
results = counts_query.all() results = counts_query.all()
# 将结果转换为字典 # 将结果转换为字典
docs_count = {result.space_id: result.document_count for result in results} docs_count = {result.id: result.document_count for result in results}
session.close() session.close()
return docs_count return docs_count
@@ -263,7 +263,7 @@ class KnowledgeDocumentDao(BaseDao):
) )
if query.space_id is not None: if query.space_id is not None:
knowledge_documents = knowledge_documents.filter( knowledge_documents = knowledge_documents.filter(
KnowledgeDocumentEntity.space_id == query.space_id KnowledgeDocumentEntity.id == query.space_id
) )
if query.doc_name is not None: if query.doc_name is not None:
knowledge_documents = knowledge_documents.filter( knowledge_documents = knowledge_documents.filter(
@@ -290,10 +290,8 @@ class KnowledgeDocumentDao(BaseDao):
knowledge_documents = session.query(KnowledgeDocumentEntity) knowledge_documents = session.query(KnowledgeDocumentEntity)
if space is not None: if space is not None:
knowledge_documents.filter(KnowledgeDocumentEntity.space == space).filter( knowledge_documents.filter(KnowledgeDocumentEntity.space == space).filter(
KnowledgeDocumentEntity.space_id == None KnowledgeDocumentEntity.id == None
).update( ).update({KnowledgeDocumentEntity.id: space_id}, synchronize_session=False)
{KnowledgeDocumentEntity.space_id: space_id}, synchronize_session=False
)
session.commit() session.commit()
session.close() session.close()
@@ -319,7 +317,7 @@ class KnowledgeDocumentDao(BaseDao):
) )
if query.space_id is not None: if query.space_id is not None:
knowledge_documents = knowledge_documents.filter( knowledge_documents = knowledge_documents.filter(
KnowledgeDocumentEntity.space_id == query.space_id KnowledgeDocumentEntity.id == query.space_id
) )
if query.space is not None: if query.space is not None:
knowledge_documents = knowledge_documents.filter( knowledge_documents = knowledge_documents.filter(