Browse Source

训练语料(标准问题添加语料配置)

liangxunge 2 months ago
parent
commit
0ac77366b8

+ 12 - 0
api/controllers/console/intention.py

@@ -301,6 +301,12 @@ class IntentionCorpusListApi(Resource):
             help="type is required. Question must be between 1 to 40 characters.",
         )
         parser.add_argument(
+            "question_config",
+            nullable=True,
+            required=False,
+            location="json",
+        )
+        parser.add_argument(
             "intention_id",
             nullable=False,
             required=True,
@@ -333,6 +339,12 @@ class IntentionCorpusApi(Resource):
             location="json",
         )
         parser.add_argument(
+            "question_config",
+            nullable=True,
+            required=False,
+            location="json",
+        )
+        parser.add_argument(
             "intention_id",
             nullable=True,
             required=False,

+ 2 - 0
api/fields/intention_fields.py

@@ -30,6 +30,7 @@ intention_keyword_detail_fields = {
 intention_corpus_fields = {
     "id": fields.String,
     "question": fields.String,
+    "question_config": fields.String,
     "intention_id": fields.String,
     "created_by": fields.String,
     "created_at": TimestampField,
@@ -58,6 +59,7 @@ intention_page_fields = {
 intention_corpus_detail_fields = {
     "id": fields.String,
     "question": fields.String,
+    "question_config": fields.String,
     "intention": fields.Nested(intention_page_fields),
     "similarity_questions": fields.List(fields.Nested(intention_corpus_similarity_question_fields)),
 }

+ 1 - 0
api/models/intention.py

@@ -117,6 +117,7 @@ class IntentionCorpus(db.Model):
 
     id = db.Column(StringUUID, nullable=False, server_default=db.text("uuid_generate_v4()"))
     question = db.Column(db.String(255), nullable=False)
+    question_config = db.Column(db.JSON, nullable=False)
     intention_id = db.Column(StringUUID, nullable=False)
     created_by = db.Column(StringUUID, nullable=False)
     created_at = db.Column(db.DateTime, nullable=False, server_default=func.current_timestamp())

+ 7 - 0
api/services/intention_service.py

@@ -325,6 +325,10 @@ class IntentionCorpusService:
             created_by=current_user.id,
             created_at=datetime.now(),
         )
+
+        if "question_config" in args:
+            intention_corpus.question_config = args["question_config"]
+
         db.session.add(intention_corpus)
         db.session.commit()
         return intention_corpus
@@ -347,6 +351,9 @@ class IntentionCorpusService:
                 raise IntentionCorpusQuestionDuplicateError(f"IntentionCorpus with question {question} already exists.")
             intention_corpus.question = question
 
+        if "question_config" in args:
+            intention_corpus.question_config = args["question_config"]
+
         if "intention_id" in args:
             intention_id = args["intention_id"]
             intention = IntentionService.get_intention(intention_id)