Method save

apps/knowledge/serializers/paragraph.py:133–186

(self, instance: Dict, with_valid=True, with_embedding=True, embedding_by_problem=None)

        @transaction.atomic
        def save(self, instance: Dict, with_valid=True, with_embedding=True, embedding_by_problem=None):
            if with_valid:
                self.is_valid()
                ProblemInstanceSerializer(data=instance).is_valid(raise_exception=True)
            problem = (
                QuerySet(Problem)
                .filter(knowledge_id=self.data.get("knowledge_id"), content=instance.get("content"))
                .first()
            )
            if problem is None:
                problem = Problem(
                    id=uuid.uuid7(), knowledge_id=self.data.get("knowledge_id"), content=instance.get("content")
                )
                problem.save()
            if (
                QuerySet(ProblemParagraphMapping)
                .filter(
                    knowledge_id=self.data.get("knowledge_id"),
                    problem_id=problem.id,
                    paragraph_id=self.data.get("paragraph_id"),
                )
                .exists()
            ):
                raise AppApiException(500, _("Already associated, please do not associate again"))
            problem_paragraph_mapping = ProblemParagraphMapping(
                id=uuid.uuid7(),
                problem_id=problem.id,
                document_id=self.data.get("document_id"),
                paragraph_id=self.data.get("paragraph_id"),
                knowledge_id=self.data.get("knowledge_id"),
            )
            problem_paragraph_mapping.save()
            model_id = get_embedding_model_id_by_knowledge_id(self.data.get("knowledge_id"))
            if with_embedding:
                embedding_by_problem_task(
                    {
                        "text": problem.content,
                        "is_active": True,
                        "source_type": SourceType.PROBLEM,
                        "source_id": problem_paragraph_mapping.id,
                        "document_id": self.data.get("document_id"),
                        "paragraph_id": self.data.get("paragraph_id"),
                        "knowledge_id": self.data.get("knowledge_id"),
                    },
                    model_id,
                )

            return ProblemSerializers.Operate(
                data={
                    "workspace_id": self.data.get("workspace_id"),
                    "knowledge_id": self.data.get("knowledge_id"),
                    "problem_id": problem.id,
                }
            ).one(with_valid=True)

    class Operate(serializers.Serializer):
        workspace_id = serializers.CharField(required=True, label=_("workspace id"))