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

fix:create document return id

Browse Source 
create document return id
This commit is contained in:
aries_ckt 2023-06-30 10:39:47 +08:00
parent 4c4c028b21
commit d0a0d3bbdf
3 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
pilot/openapi/knowledge/knowledge_controller.py
View File

@ -56,10 +56,10 @@ def space_list(request: KnowledgeSpaceRequest):
def document_add(space_name: str, request: KnowledgeDocumentRequest): def document_add(space_name: str, request: KnowledgeDocumentRequest):
print(f"/document/add params: {space_name}, {request}") print(f"/document/add params: {space_name}, {request}")
try: try:
knowledge_space_service.create_knowledge_document( return Result.succ(knowledge_space_service.create_knowledge_document(
space=space_name, request=request space=space_name, request=request
) ))
return Result.succ([]) # return Result.succ([])
except Exception as e: except Exception as e:
return Result.faild(code="E000X", msg=f"document add error {e}") return Result.faild(code="E000X", msg=f"document add error {e}")
@ -106,10 +106,10 @@ async def document_upload(
KNOWLEDGE_UPLOAD_ROOT_PATH, space_name, doc_file.filename KNOWLEDGE_UPLOAD_ROOT_PATH, space_name, doc_file.filename
), ),
) )
knowledge_space_service.create_knowledge_document( return Result.succ(knowledge_space_service.create_knowledge_document(
space=space_name, request=request space=space_name, request=request
) ))
return Result.succ([]) # return Result.succ([])
return Result.faild(code="E000X", msg=f"doc_file is None") return Result.faild(code="E000X", msg=f"doc_file is None")
except Exception as e: except Exception as e:
return Result.faild(code="E000X", msg=f"document add error {e}") return Result.faild(code="E000X", msg=f"document add error {e}")

3
pilot/openapi/knowledge/knowledge_document_dao.py
View File

@ -56,8 +56,9 @@ class KnowledgeDocumentDao:
) )
session.add(knowledge_document) session.add(knowledge_document)
session.commit() session.commit()
doc_id = knowledge_document.id
session.close() session.close()
return doc_id
def get_knowledge_documents(self, query, page=1, page_size=20): def get_knowledge_documents(self, query, page=1, page_size=20):
session = self.Session() session = self.Session()

3
pilot/openapi/knowledge/knowledge_service.py
View File

@ -78,8 +78,7 @@ class KnowledgeService:
content=request.content, content=request.content,
result="", result="",
) )
knowledge_document_dao.create_knowledge_document(document) return knowledge_document_dao.create_knowledge_document(document)
return True
"""get knowledge space""" """get knowledge space"""