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

feat:Add Knowledge Process Workflow (#2210)

Browse Source
This commit is contained in:
Aries-ckt
2024-12-18 11:16:30 +08:00
committed by GitHub
parent 3745d7411d
commit b05febbf77
23 changed files with 7217 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
dbgpt/serve/rag/api/schemas.py
View File

@@ -96,6 +96,8 @@ class ChunkServeResponse(BaseModel):
content: Optional[str] = Field(None, description="chunk content")
meta_info: Optional[str] = Field(None, description="chunk meta info")
questions: Optional[str] = Field(None, description="chunk questions")
gmt_created: Optional[str] = Field(None, description="chunk create time")
gmt_modified: Optional[str] = Field(None, description="chunk modify time")
class KnowledgeSyncRequest(BaseModel):

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

@@ -323,7 +323,7 @@ class Service(BaseService[KnowledgeSpaceEntity, SpaceServeRequest, SpaceServeRes
update_chunk = self._chunk_dao.get_one({"document_id": entity.id})
if update_chunk:
update_chunk.doc_name = request.doc_name
self._chunk_dao.update_chunk(update_chunk)
self._chunk_dao.update({"id": update_chunk.id}, update_chunk)
if len(request.questions) == 0:
request.questions = ""
questions = [