github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-10 20:52:33 +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):
print(f"/document/add params: {space_name}, {request}")
try:
knowledge_space_service.create_knowledge_document(
return Result.succ(knowledge_space_service.create_knowledge_document(
space=space_name, request=request
)
return Result.succ([])
))
# return Result.succ([])
except Exception as 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_space_service.create_knowledge_document(
return Result.succ(knowledge_space_service.create_knowledge_document(
space=space_name, request=request
)
return Result.succ([])
))
# return Result.succ([])
return Result.faild(code="E000X", msg=f"doc_file is None")
except Exception as 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.commit()
doc_id = knowledge_document.id
session.close()
return doc_id
def get_knowledge_documents(self, query, page=1, page_size=20):
session = self.Session()

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

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