コード例 #1
0
    def setUp(self):
        self.factory = RequestFactory()

        self.owner, self.owner_token = create_user_and_token("owner")
        self.user, self.user_token = create_user_and_token("user")

        self.repository = Repository.objects.create(
            owner=self.owner,
            name="Repository 1",
            slug="repo",
            language=languages.LANGUAGE_EN,
        )

        self.example_1 = RepositoryExample.objects.create(
            repository_version_language=self.repository.current_version(),
            text="hi",
            intent="greet",
        )

        self.example_translated = RepositoryTranslatedExample.objects.create(
            original_example=self.example_1,
            language=languages.LANGUAGE_PT,
            text="oi")

        self.entity_1 = RepositoryExampleEntity.objects.create(
            repository_example=self.example_1, start=0, end=0, entity="hi")
        self.entity_1.entity.set_group("greet")
        self.entity_1.entity.save()
コード例 #2
0
    def setUp(self):
        self.factory = RequestFactory()

        self.owner, self.owner_token = create_user_and_token("owner")
        self.user, self.user_token = create_user_and_token("user")

        self.repository = Repository.objects.create(
            owner=self.owner.repository_owner,
            name="Repository 1",
            slug="repo",
            language=languages.LANGUAGE_EN,
        )

        self.example_intent_1 = RepositoryIntent.objects.create(
            text="greet",
            repository_version=self.repository.current_version().
            repository_version,
        )

        self.example_1 = RepositoryExample.objects.create(
            repository_version_language=self.repository.current_version(),
            text="hi",
            intent=self.example_intent_1,
        )

        self.version = RepositoryVersion.objects.create(
            name=self.repository.current_version().repository_version.name,
            last_update=self.repository.current_version(
            ).repository_version.last_update,
            is_default=False,
            repository=self.repository.current_version(
            ).repository_version.repository,
            created_by=self.repository.current_version().repository_version.
            created_by,
        )

        self.version_language = RepositoryVersionLanguage.objects.create(
            language=self.repository.current_version().language,
            training_started_at=self.repository.current_version(
            ).training_started_at,
            training_end_at=self.repository.current_version().training_end_at,
            failed_at=self.repository.current_version().failed_at,
            use_analyze_char=self.repository.current_version(
            ).use_analyze_char,
            use_name_entities=self.repository.current_version(
            ).use_name_entities,
            use_competing_intents=self.repository.current_version(
            ).use_competing_intents,
            algorithm=self.repository.current_version().algorithm,
            repository_version=self.version,
            training_log=self.repository.current_version().training_log,
            last_update=self.repository.current_version().last_update,
            total_training_end=self.repository.current_version().
            total_training_end,
        )
        self.version_language.update_trainer(
            self.repository.current_version().get_bot_data.bot_data,
            self.repository.current_version().get_bot_data.rasa_version,
        )
コード例 #3
0
    def setUp(self):
        self.factory = RequestFactory()

        self.owner, self.owner_token = create_user_and_token("owner")

        self.repository = Repository.objects.create(
            owner=self.owner,
            name="Testing",
            slug="test",
            language=languages.LANGUAGE_EN,
        )

        self.example_intent_1 = RepositoryIntent.objects.create(
            text="bias",
            repository_version=self.repository.current_version().
            repository_version,
        )
        self.example = RepositoryExample.objects.create(
            repository_version_language=self.repository.current_version(),
            text="hi",
            intent=self.example_intent_1,
        )

        self.repository_auth = RepositoryAuthorization.objects.create(
            user=self.owner, repository=self.repository, role=3)
コード例 #4
0
    def setUp(self):
        self.factory = RequestFactory()
        self.owner, self.owner_token = create_user_and_token("owner")
        self.user, self.user_token = create_user_and_token("user")

        self.repository = Repository.objects.create(
            owner=self.owner,
            name="Repository 1",
            slug="repo",
            language=languages.LANGUAGE_EN,
        )

        self.repository_2 = Repository.objects.create(
            owner=self.owner,
            name="Repository 2",
            slug="repo2",
            language=languages.LANGUAGE_EN,
        )

        self.knowledge_base_1 = QAKnowledgeBase.objects.create(
            repository=self.repository, title="Testando knowledge"
        )

        self.knowledge_base_2 = QAKnowledgeBase.objects.create(
            repository=self.repository_2, title="Testando knowledge"
        )

        self.context_1 = QAtext.objects.create(
            knowledge_base=self.knowledge_base_1,
            text="Lorem ipsum dolor sit amet, consectetur adipisicing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua. Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat. Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur. Excepteur sint occaecat cupidatat non proident, sunt in culpa qui officia deserunt mollit anim id est laborum.",
            language=languages.LANGUAGE_PT_BR,
        )

        self.context_2 = QAtext.objects.create(
            knowledge_base=self.knowledge_base_1,
            text="teste 2",
            language=languages.LANGUAGE_EN,
        )

        self.context_3 = QAtext.objects.create(
            knowledge_base=self.knowledge_base_2,
            text="teste 3",
            language=languages.LANGUAGE_PT_BR,
        )
        self.repository_auth = RepositoryAuthorization.objects.create(
            user=self.owner, repository=self.repository, role=3
        )
コード例 #5
0
    def setUp(self):
        self.factory = RequestFactory()

        self.owner, self.owner_token = create_user_and_token("owner")
        self.user, self.user_token = create_user_and_token("user")

        self.repository = Repository.objects.create(
            owner=self.owner,
            name="Repository 1",
            slug="repo",
            language=languages.LANGUAGE_EN,
        )

        self.example_1 = RepositoryExample.objects.create(
            repository_version_language=self.repository.current_version(),
            text="hi",
            intent="greet",
        )
コード例 #6
0
    def setUp(self):
        self.user, self.user_token = create_user_and_token("user")

        self.organization = Organization.objects.create(name="Weni")

        self.organization.organization_authorizations.create(
            user=self.user, role=OrganizationAuthorization.ROLE_ADMIN)

        super().setUp()

        self.user_permission_stub = (
            authentication_pb2_grpc.UserPermissionControllerStub(self.channel))
        self.user_stub = authentication_pb2_grpc.UserControllerStub(
            self.channel)
コード例 #7
0
ファイル: tests.py プロジェクト: Ilhasoft/bothub-engine
    def setUp(self):
        super().setUp()
        self.stub = repository_pb2_grpc.RepositoryControllerStub(self.channel)

        self.user, self.user_token = create_user_and_token("user")

        self.organization = Organization.objects.create(name="Weni")

        self.organization.organization_authorizations.create(
            user=self.user, role=OrganizationAuthorization.ROLE_ADMIN)

        self.repository = Repository.objects.create(
            name="Repository",
            language=languages.LANGUAGE_PT,
            owner=self.organization.repository_owner,
        )

        self.repository_authorization = RepositoryAuthorization.objects.create(
            user=self.organization.repository_owner,
            repository=self.repository,
            role=RepositoryAuthorization.LEVEL_ADMIN,
        )
コード例 #8
0
    def setUp(self):
        self.factory = RequestFactory()
        self.owner, self.owner_token = create_user_and_token("owner")
        self.user, self.user_token = create_user_and_token("user")

        self.repository = Repository.objects.create(
            owner=self.owner,
            name="Repository 1",
            slug="repo",
            language=languages.LANGUAGE_EN,
        )
        self.example_intent_1 = RepositoryIntent.objects.create(
            text="greet",
            repository_version=self.repository.current_version().
            repository_version,
        )
        self.example_intent_2 = RepositoryIntent.objects.create(
            text="farewell",
            repository_version=self.repository.current_version().
            repository_version,
        )
        self.example_1 = RepositoryExample.objects.create(
            repository_version_language=self.repository.current_version(),
            text="hi",
            intent=self.example_intent_1,
        )
        entity_1 = RepositoryExampleEntity.objects.create(
            repository_example=self.example_1, start=0, end=0, entity="hi")
        entity_1.entity.set_group("greet")
        entity_1.entity.save()
        self.example_2 = RepositoryExample.objects.create(
            repository_version_language=self.repository.current_version(),
            text="hello",
            intent=self.example_intent_1,
        )
        self.example_3 = RepositoryExample.objects.create(
            repository_version_language=self.repository.current_version(),
            text="bye",
            intent=self.example_intent_2,
        )
        self.example_4 = RepositoryExample.objects.create(
            repository_version_language=self.repository.current_version(),
            text="bye bye",
            intent=self.example_intent_2,
        )

        self.repository_2 = Repository.objects.create(
            owner=self.owner,
            name="Repository 2",
            slug="repo2",
            language=languages.LANGUAGE_EN,
        )
        self.example2_intent_1 = RepositoryIntent.objects.create(
            text="greet",
            repository_version=self.repository_2.current_version().
            repository_version,
        )
        self.example_5 = RepositoryExample.objects.create(
            repository_version_language=self.repository_2.current_version(),
            text="hi",
            intent=self.example2_intent_1,
        )
        self.example_6 = RepositoryExample.objects.create(
            repository_version_language=self.repository_2.current_version(
                languages.LANGUAGE_PT),
            text="oi",
            intent=self.example2_intent_1,
        )
        self.translation_6 = RepositoryTranslatedExample.objects.create(
            original_example=self.example_6,
            language=languages.LANGUAGE_EN,
            text="hi")
コード例 #9
0
 def setUp(self):
     self.factory = RequestFactory()
     self.owner, self.owner_token = create_user_and_token("owner")
コード例 #10
0
    def setUp(self):
        self.factory = RequestFactory()

        self.owner, self.owner_token = create_user_and_token("owner")

        self.repository = Repository.objects.create(
            owner=self.owner,
            name="Testing",
            slug="test",
            language=languages.LANGUAGE_EN,
        )
        self.example_intent_1 = RepositoryIntent.objects.create(
            text="bias",
            repository_version=self.repository.current_version().
            repository_version,
        )
        self.example = RepositoryExample.objects.create(
            repository_version_language=self.repository.current_version(),
            text="hi",
            intent=self.example_intent_1,
        )

        nlp_log = RepositoryNLPLog.objects.create(
            text="test",
            user_agent="python-requests/2.20.1",
            from_backend=True,
            repository_version_language=self.repository.current_version(),
            nlp_log=json.dumps({
                "intent": {
                    "name": "bias",
                    "confidence": 0.9994810819625854
                },
                "intent_ranking": [
                    {
                        "name": "bias",
                        "confidence": 0.9994810819625854
                    },
                    {
                        "name": "doubt",
                        "confidence": 0.039212167263031006
                    },
                    {
                        "name": "negative",
                        "confidence": 0.0
                    },
                    {
                        "name": "affirmative",
                        "confidence": 0.0
                    },
                ],
                "labels_list": [],
                "entities_list": [],
                "entities": {},
                "text":
                "test",
                "repository_version":
                int(self.repository.current_version().pk),
                "language":
                str(self.repository.language),
            }),
            user=self.owner,
        )

        RepositoryNLPLogIntent.objects.create(
            intent="bias",
            confidence=0.9994810819625854,
            is_default=True,
            repository_nlp_log=nlp_log,
        )

        RepositoryNLPLogIntent.objects.create(
            intent="doubt",
            confidence=0.039212167263031006,
            is_default=False,
            repository_nlp_log=nlp_log,
        )

        RepositoryNLPLogIntent.objects.create(
            intent="negative",
            confidence=0.0,
            is_default=False,
            repository_nlp_log=nlp_log,
        )

        RepositoryNLPLogIntent.objects.create(
            intent="affirmative",
            confidence=0.0,
            is_default=False,
            repository_nlp_log=nlp_log,
        )
コード例 #11
0
    def setUp(self):
        self.factory = RequestFactory()
        self.owner, self.owner_token = create_user_and_token("owneres")

        self.repository = Repository.objects.create(
            owner=self.owner,
            name="Testing Elasticsearch",
            slug="testinges",
            language=languages.LANGUAGE_EN,
        )
        self.example_intent_1 = RepositoryIntent.objects.create(
            text="biases",
            repository_version=self.repository.current_version().
            repository_version,
        )
        self.example = RepositoryExample.objects.create(
            repository_version_language=self.repository.current_version(),
            text="hies",
            intent=self.example_intent_1,
        )

        nlp_log = RepositoryNLPLog.objects.create(
            id=2,
            text="testes",
            user_agent="python-requests/2.20.1",
            from_backend=True,
            repository_version_language=self.repository.current_version(),
            nlp_log=json.dumps({
                "intent": {
                    "name": "biases",
                    "confidence": 0.9994810819625854
                },
                "intent_ranking": [
                    {
                        "name": "biases",
                        "confidence": 0.9994810819625854
                    },
                    {
                        "name": "doubtes",
                        "confidence": 0.039212167263031006
                    },
                    {
                        "name": "negativees",
                        "confidence": 0.0
                    },
                    {
                        "name": "affirmativees",
                        "confidence": 0.0
                    },
                ],
                "labels_list": [],
                "entities_list": [],
                "entities": {},
                "text":
                "testes",
                "repository_version":
                int(self.repository.current_version().pk),
                "language":
                str(self.repository.language),
            }),
            user=self.owner,
        )

        RepositoryNLPLogIntent.objects.create(
            intent="biases",
            confidence=0.9994810819625854,
            is_default=True,
            repository_nlp_log=nlp_log,
        )

        RepositoryNLPLogIntent.objects.create(
            intent="doubtes",
            confidence=0.039212167263031006,
            is_default=False,
            repository_nlp_log=nlp_log,
        )

        RepositoryNLPLogIntent.objects.create(
            intent="negativees",
            confidence=0.0,
            is_default=False,
            repository_nlp_log=nlp_log,
        )

        RepositoryNLPLogIntent.objects.create(
            intent="affirmativees",
            confidence=0.0,
            is_default=False,
            repository_nlp_log=nlp_log,
        )
        requests.delete(
            f"{settings.ELASTICSEARCH_DSL['default']['hosts']}/_data_stream/{REPOSITORYNLPLOG_INDEX_NAME}"
        )
        registry.update(nlp_log)