|
|
|
@@ -302,6 +302,8 @@ class DatasetInitApi(Resource): |
|
|
|
"doc_language", type=str, default="English", required=False, nullable=False, location="json" |
|
|
|
) |
|
|
|
parser.add_argument("retrieval_model", type=dict, required=False, nullable=False, location="json") |
|
|
|
parser.add_argument("embedding_model", type=str, required=False, nullable=True, location="json") |
|
|
|
parser.add_argument("embedding_model_provider", type=str, required=False, nullable=True, location="json") |
|
|
|
args = parser.parse_args() |
|
|
|
|
|
|
|
# The role of the current user in the ta table must be admin, owner, or editor, or dataset_operator |
|
|
|
@@ -309,6 +311,8 @@ class DatasetInitApi(Resource): |
|
|
|
raise Forbidden() |
|
|
|
|
|
|
|
if args["indexing_technique"] == "high_quality": |
|
|
|
if args["embedding_model"] is None or args["embedding_model_provider"] is None: |
|
|
|
raise ValueError("embedding model and embedding model provider are required for high quality indexing.") |
|
|
|
try: |
|
|
|
model_manager = ModelManager() |
|
|
|
model_manager.get_default_model_instance( |