red-future/rag-local
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

1

Browse Source
This commit is contained in:
admin
2026-08-05 15:35:55 +08:00
parent 20c1805efb
commit dc767d83f9
11 changed files with 239 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files
kb/controller/document_controller.go
+11 -2
View File
@@ -47,6 +47,14 @@ func (c *document) List(ctx context.Context, req *dto.ListDocumentReq) (*dto.Lis
}, nil
}
func (c *document) Detail(ctx context.Context, req *dto.GetDocumentDetailReq) (res *dto.GetDocumentDetailRes, err error) {
doc, err := service.DocumentService.Detail(ctx, req.Id)
if err != nil {
return nil, err
}
return &dto.GetDocumentDetailRes{Document: doc}, nil
}
func (c *document) Delete(ctx context.Context, req *dto.DeleteDocumentReq) (*dto.DeleteDocumentRes, error) {
if err := service.DocumentService.Delete(ctx, req.Id); err != nil {
return nil, err
@@ -55,8 +63,9 @@ func (c *document) Delete(ctx context.Context, req *dto.DeleteDocumentReq) (*dto
}
func (c *document) Reembed(ctx context.Context, req *dto.ReembedDocumentReq) (*dto.ReembedDocumentRes, error) {
if err := service.DocumentService.Reembed(ctx, req.Id); err != nil {
n, err := service.DocumentService.Reembed(ctx, req.Id)
if err != nil {
return nil, err
}
return &dto.ReembedDocumentRes{}, nil
return &dto.ReembedDocumentRes{Count: n}, nil
}