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

update to external knowledge api

Browse Source
This commit is contained in:
jyong
2024-09-26 16:38:53 +08:00
parent 5fa86074ed
commit 611f0fb3f6
2 changed files with 21 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
api/controllers/console/datasets/test_external.py
View File

@ -14,16 +14,11 @@ class TestExternalApi(Resource):
def post(self):
parser = reqparse.RequestParser()
parser.add_argument(
"top_k",
"retrieval_setting",
nullable=False,
required=True,
type=int,
)
parser.add_argument(
"score_threshold",
nullable=False,
required=True,
type=float,
type=dict,
location="json"
)
parser.add_argument(
"query",
@ -32,14 +27,14 @@ class TestExternalApi(Resource):
type=str,
)
parser.add_argument(
"external_knowledge_id",
"knowledge_id",
nullable=False,
required=True,
type=str,
)
args = parser.parse_args()
result = ExternalDatasetService.test_external_knowledge_retrieval(
args["top_k"], args["score_threshold"], args["query"], args["external_knowledge_id"]
args["retrieval_setting"], args["query"], args["knowledge_id"]
)
return result, 200