Pārlūkot izejas kodu

fix dataset api retrieval model null handling (#18151)

Signed-off-by: kenwoodjw <blackxin55+@gmail.com>
tags/1.3.0
kenwoodjw pirms 6 mēnešiem
vecāks
revīzija
b247ef85bf
Revīzijas autora e-pasta adrese nav piesaistīta nevienam kontam
1 mainītis faili ar 3 papildinājumiem un 1 dzēšanām
  1. 3
    1
      api/controllers/service_api/dataset/dataset.py

+ 3
- 1
api/controllers/service_api/dataset/dataset.py Parādīt failu

@@ -139,7 +139,9 @@ class DatasetListApi(DatasetApiResource):
external_knowledge_id=args["external_knowledge_id"],
embedding_model_provider=args["embedding_model_provider"],
embedding_model_name=args["embedding_model"],
retrieval_model=RetrievalModel(**args["retrieval_model"]),
retrieval_model=RetrievalModel(**args["retrieval_model"])
if args["retrieval_model"] is not None
else None,
)
except services.errors.dataset.DatasetNameDuplicateError:
raise DatasetNameDuplicateError()

Notiek ielāde…
Atcelt
Saglabāt