@@ -143,20 +143,21 @@ def get_request_params_api():
143
143
required = True ,
144
144
description = '知识库id' ),
145
145
]
146
+
146
147
@staticmethod
147
148
def get_request_body_api ():
148
149
return openapi .Schema (
149
150
type = openapi .TYPE_OBJECT ,
150
151
required = ['source_url_list' ],
151
152
properties = {
152
- 'source_url_list' : openapi .Schema (type = openapi .TYPE_ARRAY , title = "文档地址列表" , description = "文档地址列表" ,
153
+ 'source_url_list' : openapi .Schema (type = openapi .TYPE_ARRAY , title = "文档地址列表" ,
154
+ description = "文档地址列表" ,
153
155
items = openapi .Schema (type = openapi .TYPE_STRING )),
154
156
'selector' : openapi .Schema (type = openapi .TYPE_STRING , title = "选择器" , description = "选择器" )
155
157
}
156
158
)
157
159
158
160
159
-
160
161
class DocumentInstanceSerializer (ApiMixin , serializers .Serializer ):
161
162
name = serializers .CharField (required = True ,
162
163
error_messages = ErrMessage .char ("文档名称" ),
@@ -396,8 +397,8 @@ def get_query_set(self):
396
397
query_set = query_set .annotate (
397
398
reversed_status = Reverse ('status' ),
398
399
task_type_status = Substr ('reversed_status' , TaskType (task_type ).value ,
399
- TaskType ( task_type ). value ),
400
- ).filter (task_type_status__in = [ State (status ).value ] ).values ('id' )
400
+ 1 ),
401
+ ).filter (task_type_status = State (status ).value ).values ('id' )
401
402
else :
402
403
if status != State .SUCCESS .value :
403
404
query_set = query_set .filter (status__icontains = status )
@@ -674,15 +675,15 @@ def cancel(self, instance, with_valid=True):
674
675
ListenerManagement .update_status (QuerySet (Paragraph ).annotate (
675
676
reversed_status = Reverse ('status' ),
676
677
task_type_status = Substr ('reversed_status' , TaskType (instance .get ('type' )).value ,
677
- TaskType ( instance . get ( 'type' )). value ),
678
+ 1 ),
678
679
).filter (task_type_status__in = [State .PENDING .value , State .STARTED .value ]).filter (
679
680
document_id = document_id ).values ('id' ),
680
681
TaskType (instance .get ('type' )),
681
682
State .REVOKE )
682
683
ListenerManagement .update_status (QuerySet (Document ).annotate (
683
684
reversed_status = Reverse ('status' ),
684
685
task_type_status = Substr ('reversed_status' , TaskType (instance .get ('type' )).value ,
685
- TaskType ( instance . get ( 'type' )). value ),
686
+ 1 ),
686
687
).filter (task_type_status__in = [State .PENDING .value , State .STARTED .value ]).filter (
687
688
id = document_id ).values ('id' ),
688
689
TaskType (instance .get ('type' )),
0 commit comments