notion fix

This commit is contained in:
jyong
2025-08-07 11:13:02 +08:00
parent 218e778099
commit ca8f80ee33
3 changed files with 21 additions and 7 deletions

View File

@ -405,6 +405,7 @@ class RagPipelinePublishedDatasourceNodeRunApi(Resource):
parser = reqparse.RequestParser()
parser.add_argument("inputs", type=dict, required=True, nullable=False, location="json")
parser.add_argument("datasource_type", type=str, required=True, location="json")
parser.add_argument("credential_id", type=str, required=False, location="json")
args = parser.parse_args()
inputs = args.get("inputs")
@ -424,6 +425,7 @@ class RagPipelinePublishedDatasourceNodeRunApi(Resource):
account=current_user,
datasource_type=datasource_type,
is_published=False,
credential_id=args.get("credential_id"),
)
)
)
@ -448,6 +450,7 @@ class RagPipelineDraftDatasourceNodeRunApi(Resource):
parser = reqparse.RequestParser()
parser.add_argument("inputs", type=dict, required=True, nullable=False, location="json")
parser.add_argument("datasource_type", type=str, required=True, location="json")
parser.add_argument("credential_id", type=str, required=False, location="json")
args = parser.parse_args()
inputs = args.get("inputs")
@ -467,6 +470,7 @@ class RagPipelineDraftDatasourceNodeRunApi(Resource):
account=current_user,
datasource_type=datasource_type,
is_published=False,
credential_id=args.get("credential_id"),
)
)
)