youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-28 06:28:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix knowledge permission update

Browse Source
This commit is contained in:
jyong
2024-10-11 19:13:19 +08:00
parent 8456e6379d
commit 33d0904981
5 changed files with 43 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
api/controllers/console/datasets/external.py
View File

@ -13,7 +13,7 @@ from libs.login import login_required
from services.dataset_service import DatasetService
from services.external_knowledge_service import ExternalDatasetService
from services.hit_testing_service import HitTestingService
from services.knowledge_service import ExternalDatasetServiceTest
from services.knowledge_service import ExternalDatasetTestService
def _validate_name(name):
@ -234,16 +234,21 @@ class ExternalKnowledgeHitTestingApi(Resource):
class BedrockRetrievalApi(Resource):
# url : <your-endpoint>/retrieval
# this api is only for internal testing
def post(self):
parser = reqparse.RequestParser()
parser.add_argument("retrieval_setting", nullable=False, required=True, type=dict, location="json")
parser.add_argument("query", nullable=False, required=True, type=str, )
parser.add_argument(
"query",
nullable=False,
required=True,
type=str,
)
parser.add_argument("knowledge_id", nullable=False, required=True, type=str)
args = parser.parse_args()
# Call the knowledge retrieval service
result = ExternalDatasetServiceTest.knowledge_retrieval(
result = ExternalDatasetTestService.knowledge_retrieval(
args["retrieval_setting"], args["query"], args["knowledge_id"]
)
return result, 200
@ -254,4 +259,5 @@ api.add_resource(ExternalDatasetCreateApi, "/datasets/external")
api.add_resource(ExternalApiTemplateListApi, "/datasets/external-knowledge-api")
api.add_resource(ExternalApiTemplateApi, "/datasets/external-knowledge-api/<uuid:external_knowledge_api_id>")
api.add_resource(ExternalApiUseCheckApi, "/datasets/external-knowledge-api/<uuid:external_knowledge_api_id>/use-check")
api.add_resource(BedrockRetrievalApi, "/datasets/retrieval")
# this api is only for internal test
api.add_resource(BedrockRetrievalApi, "/test/retrieval")