def test_delete_document(self):
        # Setup Expected Response
        expected_response = {}
        expected_response = empty_pb2.Empty(**expected_response)
        operation = operations_pb2.Operation(
            name="operations/test_delete_document", done=True)
        operation.response.Pack(expected_response)

        # Mock the API response
        channel = ChannelStub(responses=[operation])
        patch = mock.patch("google.api_core.grpc_helpers.create_channel")
        with patch as create_channel:
            create_channel.return_value = channel
            client = dialogflow_v2beta1.DocumentsClient()

        # Setup Request
        name = client.document_path("[PROJECT]", "[KNOWLEDGE_BASE]",
                                    "[DOCUMENT]")

        response = client.delete_document(name)
        result = response.result()
        assert expected_response == result

        assert len(channel.requests) == 1
        expected_request = document_pb2.DeleteDocumentRequest(name=name)
        actual_request = channel.requests[0][1]
        assert expected_request == actual_request
    def test_get_document(self):
        # Setup Expected Response
        name_2 = "name2-1052831874"
        display_name = "displayName1615086568"
        mime_type = "mimeType-196041627"
        content_uri = "contentUri-388807514"
        enable_auto_reload = False
        expected_response = {
            "name": name_2,
            "display_name": display_name,
            "mime_type": mime_type,
            "content_uri": content_uri,
            "enable_auto_reload": enable_auto_reload,
        }
        expected_response = document_pb2.Document(**expected_response)

        # Mock the API response
        channel = ChannelStub(responses=[expected_response])
        patch = mock.patch("google.api_core.grpc_helpers.create_channel")
        with patch as create_channel:
            create_channel.return_value = channel
            client = dialogflow_v2beta1.DocumentsClient()

        # Setup Request
        name = client.document_path("[PROJECT]", "[KNOWLEDGE_BASE]",
                                    "[DOCUMENT]")

        response = client.get_document(name)
        assert expected_response == response

        assert len(channel.requests) == 1
        expected_request = document_pb2.GetDocumentRequest(name=name)
        actual_request = channel.requests[0][1]
        assert expected_request == actual_request
Exemplo n.º 3
0
    def test_list_documents(self):
        # Setup Expected Response
        next_page_token = ''
        documents_element = {}
        documents = [documents_element]
        expected_response = {
            'next_page_token': next_page_token,
            'documents': documents
        }
        expected_response = document_pb2.ListDocumentsResponse(
            **expected_response)

        # Mock the API response
        channel = ChannelStub(responses=[expected_response])
        client = dialogflow_v2beta1.DocumentsClient(channel=channel)

        # Setup Request
        parent = client.knowledge_base_path('[PROJECT]', '[KNOWLEDGE_BASE]')

        paged_list_response = client.list_documents(parent)
        resources = list(paged_list_response)
        assert len(resources) == 1

        assert expected_response.documents[0] == resources[0]

        assert len(channel.requests) == 1
        expected_request = document_pb2.ListDocumentsRequest(parent=parent)
        actual_request = channel.requests[0][1]
        assert expected_request == actual_request
    def test_list_documents(self):
        # Setup Expected Response
        next_page_token = ""
        documents_element = {}
        documents = [documents_element]
        expected_response = {
            "next_page_token": next_page_token,
            "documents": documents
        }
        expected_response = document_pb2.ListDocumentsResponse(
            **expected_response)

        # Mock the API response
        channel = ChannelStub(responses=[expected_response])
        patch = mock.patch("google.api_core.grpc_helpers.create_channel")
        with patch as create_channel:
            create_channel.return_value = channel
            client = dialogflow_v2beta1.DocumentsClient()

        # Setup Request
        parent = client.knowledge_base_path("[PROJECT]", "[KNOWLEDGE_BASE]")

        paged_list_response = client.list_documents(parent)
        resources = list(paged_list_response)
        assert len(resources) == 1

        assert expected_response.documents[0] == resources[0]

        assert len(channel.requests) == 1
        expected_request = document_pb2.ListDocumentsRequest(parent=parent)
        actual_request = channel.requests[0][1]
        assert expected_request == actual_request
Exemplo n.º 5
0
def setup_teardown():
    # Create a knowledge base to use in document management
    client = dialogflow.KnowledgeBasesClient()
    project_path = client.project_path(PROJECT_ID)
    knowledge_base = dialogflow.types.KnowledgeBase(
        display_name=KNOWLEDGE_BASE_NAME)
    response = client.create_knowledge_base(project_path, knowledge_base)
    pytest.KNOWLEDGE_BASE_ID = response.name.split(
        '/knowledgeBases/')[1].split('\n')[0]

    # Create a document to delete
    knowledge_base_path = client.knowledge_base_path(
        PROJECT_ID, pytest.KNOWLEDGE_BASE_ID)
    document = dialogflow.types.Document(
        display_name=DOCUMENT_DISPLAY_NAME, mime_type='text/html',
        content_uri='https://cloud.google.com/storage/docs/faq')
    document.knowledge_types.append(
        dialogflow.types.Document.KnowledgeType.Value('FAQ'))
    documents_client = dialogflow.DocumentsClient()
    response = documents_client.create_document(knowledge_base_path, document)
    document = response.result(timeout=90)
    pytest.DOCUMENT_ID = document.name.split('/documents/')[1].split('\n')[0]

    yield

    # Delete the created knowledge base
    client.delete_knowledge_base(knowledge_base_path, force=True)
Exemplo n.º 6
0
    def test_create_document(self):
        # Setup Expected Response
        name = 'name3373707'
        display_name = 'displayName1615086568'
        mime_type = 'mimeType-196041627'
        content_uri = 'contentUri-388807514'
        expected_response = {
            'name': name,
            'display_name': display_name,
            'mime_type': mime_type,
            'content_uri': content_uri
        }
        expected_response = document_pb2.Document(**expected_response)
        operation = operations_pb2.Operation(
            name='operations/test_create_document', done=True)
        operation.response.Pack(expected_response)

        # Mock the API response
        channel = ChannelStub(responses=[operation])
        client = dialogflow_v2beta1.DocumentsClient(channel=channel)

        # Setup Request
        parent = client.knowledge_base_path('[PROJECT]', '[KNOWLEDGE_BASE]')
        document = {}

        response = client.create_document(parent, document)
        result = response.result()
        assert expected_response == result

        assert len(channel.requests) == 1
        expected_request = document_pb2.CreateDocumentRequest(
            parent=parent, document=document)
        actual_request = channel.requests[0][1]
        assert expected_request == actual_request
    def test_get_document(self):
        # Setup Expected Response
        name_2 = 'name2-1052831874'
        display_name = 'displayName1615086568'
        mime_type = 'mimeType-196041627'
        content_uri = 'contentUri-388807514'
        expected_response = {
            'name': name_2,
            'display_name': display_name,
            'mime_type': mime_type,
            'content_uri': content_uri
        }
        expected_response = document_pb2.Document(**expected_response)

        # Mock the API response
        channel = ChannelStub(responses=[expected_response])
        patch = mock.patch('google.api_core.grpc_helpers.create_channel')
        with patch as create_channel:
            create_channel.return_value = channel
            client = dialogflow_v2beta1.DocumentsClient()

        # Setup Request
        name = client.document_path('[PROJECT]', '[KNOWLEDGE_BASE]',
                                    '[DOCUMENT]')

        response = client.get_document(name)
        assert expected_response == response

        assert len(channel.requests) == 1
        expected_request = document_pb2.GetDocumentRequest(name=name)
        actual_request = channel.requests[0][1]
        assert expected_request == actual_request
Exemplo n.º 8
0
    def test_list_documents_exception(self):
        channel = ChannelStub(responses=[CustomException()])
        client = dialogflow_v2beta1.DocumentsClient(channel=channel)

        # Setup request
        parent = client.knowledge_base_path('[PROJECT]', '[KNOWLEDGE_BASE]')

        paged_list_response = client.list_documents(parent)
        with pytest.raises(CustomException):
            list(paged_list_response)
    def test_reload_document_exception(self):
        # Mock the API response
        channel = ChannelStub(responses=[CustomException()])
        patch = mock.patch("google.api_core.grpc_helpers.create_channel")
        with patch as create_channel:
            create_channel.return_value = channel
            client = dialogflow_v2beta1.DocumentsClient()

        with pytest.raises(CustomException):
            client.reload_document()
Exemplo n.º 10
0
    def test_get_document_exception(self):
        # Mock the API response
        channel = ChannelStub(responses=[CustomException()])
        client = dialogflow_v2beta1.DocumentsClient(channel=channel)

        # Setup request
        name = client.document_path('[PROJECT]', '[KNOWLEDGE_BASE]',
                                    '[DOCUMENT]')

        with pytest.raises(CustomException):
            client.get_document(name)
Exemplo n.º 11
0
    def test_get_document_exception(self):
        # Mock the API response
        channel = ChannelStub(responses=[CustomException()])
        patch = mock.patch("google.api_core.grpc_helpers.create_channel")
        with patch as create_channel:
            create_channel.return_value = channel
            client = dialogflow_v2beta1.DocumentsClient()

        # Setup request
        name = client.document_path("[PROJECT]", "[KNOWLEDGE_BASE]", "[DOCUMENT]")

        with pytest.raises(CustomException):
            client.get_document(name)
    def test_list_documents_exception(self):
        channel = ChannelStub(responses=[CustomException()])
        patch = mock.patch("google.api_core.grpc_helpers.create_channel")
        with patch as create_channel:
            create_channel.return_value = channel
            client = dialogflow_v2beta1.DocumentsClient()

        # Setup request
        parent = client.knowledge_base_path("[PROJECT]", "[KNOWLEDGE_BASE]")

        paged_list_response = client.list_documents(parent)
        with pytest.raises(CustomException):
            list(paged_list_response)
Exemplo n.º 13
0
def create_document(project_id, knowledge_base_id, display_name, mime_type,
                    knowledge_type, content_uri):
    """Creates a Document.

    Args:
        project_id: The GCP project linked with the agent.
        knowledge_base_id: Id of the Knowledge base.
        display_name: The display name of the Document.
        mime_type: The mime_type of the Document. e.g. text/csv, text/html,
            text/plain, text/pdf etc.
        knowledge_type: The Knowledge type of the Document. e.g. FAQ,
            EXTRACTIVE_QA.
        content_uri: Uri of the document, e.g. gs://path/mydoc.csv,
            http://mypage.com/faq.html."""

    client = dialogflow.DocumentsClient()
    knowledge_base_path = client.knowledge_base_path(project_id,
                                                     knowledge_base_id)

    document = dialogflow.types.Document(display_name=display_name,
                                         mime_type=mime_type,
                                         content_uri=content_uri)

    document.knowledge_types.append(
        dialogflow.types.Document.KnowledgeType.Value(knowledge_type))

    response = client.create_document(knowledge_base_path, document)
    print('Waiting for results...')
    document = response.result(timeout=120)
    print('Created Document:')
    print(' - Display Name: {}'.format(document.display_name))
    print(' - Knowledge ID: {}'.format(document.name))
    print(' - MIME Type: {}'.format(document.mime_type))
    print(' - Knowledge Types:')
    # for knowledge_type in document.knowledge_types:
    #     print('    - {}'.format(KNOWLEDGE_TYPES[knowledge_type]))
    print(' - Source: {}\n'.format(document.content_uri))

    document = {}
    response = client.update_document(document)

    def callback(operation_future):
        # Handle result.
        result = operation_future.result()

    response.add_done_callback(callback)
    # Handle metadata.
    metadata = response.metadata()
Exemplo n.º 14
0
def delete_document(project_id, knowledge_base_id, document_id):
    """Deletes a Document.

    Args:
        project_id: The GCP project linked with the agent.
        knowledge_base_id: Id of the Knowledge base.
        document_id: Id of the Document."""
    import dialogflow_v2beta1 as dialogflow
    client = dialogflow.DocumentsClient()
    document_path = client.document_path(project_id, knowledge_base_id,
                                         document_id)

    response = client.delete_document(document_path)
    print('operation running:\n {}'.format(response.operation))
    print('Waiting for results...')
    print('Done.\n {}'.format(response.result()))
def delete_document(PROJECT_ID, KID, DID):
    """
    Deletes a Document.
        called by DELETE   /knowledge_base/{KID}/document/{DID}


    :parameter     PROJECT_ID: The GCP project linked with the agent.
    :parameter     KID: Id of the Knowledge base.
    :parameter     DID: Id of the Document.
    """
    client = dialogflow.DocumentsClient()
    document_path = client.document_path(PROJECT_ID, KID, DID)

    response = client.delete_document(document_path)
    print('operation running:\n {}'.format(response.operation))
    print('Waiting for results...')
    print('Done.\n {}'.format(response.result()))
def list_documents(PROJECT_ID, KID):
    """
    List all Documents in a Knowledge base by its KID.
       indirectly called by GET /knowledge_base/{KID}. directly called by get_knowledge_base Function
       format returned:
         "documents": [
           {
          "document-id": "searched from knowledge base management",
          "document-name": "stackoverflow",
          "MIME-type": "text/csv",
          "Knowledge-type": "FAQ",
          "content_uri": "gs://test/test01.csv"
          }
    :parameter PROJECT_ID: The GCP project linked with the agent.
    :parameter KID: ID to Knowledge base by.
    """
    client = dialogflow.DocumentsClient()
    knowledge_base_path = client.knowledge_base_path(PROJECT_ID, KID)
    #testing stout
    print('Documents for Knowledge Id: {}'.format(KID))
    for document in client.list_documents(knowledge_base_path):
        print(' - Display Name: {}'.format(document.display_name))
        print(' - document ID: {}'.format(document.name.split("/")[-1]))
        print(' - MIME Type: {}'.format(document.mime_type))
        print(' - Knowledge Types:')
        for knowledge_type in document.knowledge_types:
            print('    - {}'.format(KNOWLEDGE_TYPES[knowledge_type]))
        print(' - Source: {}\n'.format(document.content_uri))
        print(' - Detail Location: {}'.format(document.name))

    #content returned
    document_list = []
    for document in client.list_documents(knowledge_base_path):
        document_dict = dict()
        document_dict["document-name"] = str(document.display_name)
        document_dict["document-id"] = str(document.name.split("/")[-1])
        document_dict["MIME-type"] = str(document.mime_type)
        knowledge_type_list = [
            KNOWLEDGE_TYPES[i] for i in document.knowledge_types
        ]
        document_dict["Knowledge-type"] = ",".join(knowledge_type_list)
        document_dict["content_uri"] = document.content_uri

        document_list.append(document_dict)
    #returned only document list, need embeded into a dict of kb
    return document_dict
def get_document(PROJECT_ID, KID, DID):
    """
    Gets a Document in a specific knowledge base by kid
        called by GET   /knowledge_base/{KID}/document/{DID}
        format returned:
        {
      "knowledge-base-id": "searched from knowledge base management",
      "document-id": "searched from knowledge base management",
      "display-name": "stackoverflow",
      "MIME-type": "text/csv",
      "Knowledge-type": "FAQ",
      "content_uri": "gs://test/test01.csv"
    }
    :parameter PROJECT_ID: The GCP project linked with the agent.
    :parameter KID: Id of the Knowledge base.
    :parameter DID: Id of the Document.

    """
    client = dialogflow.DocumentsClient()
    document_path = client.document_path(PROJECT_ID, KID, DID)

    response = client.get_document(document_path)
    # testing stout
    print('Got Document:')
    print(' - Display Name: {}'.format(response.display_name))
    print(' - Knowledge ID: {}'.format(response.name.split("/")[-1]))
    print(' - MIME Type: {}'.format(response.mime_type))
    print(' - Knowledge Types:')
    for knowledge_type in response.knowledge_types:
        print('    - {}'.format(KNOWLEDGE_TYPES[knowledge_type]))
    print(' - Source: {}\n'.format(response.content_uri))
    print(' - Detail Location: {}'.format(response.name))

    # content returned
    document_dict = dict()
    document_dict["knowledge-base-id"] = str(KID)
    document_dict["document-name"] = str(response.display_name)
    document_dict["document-id"] = str(response.name.split("/")[-1])
    document_dict["MIME-type"] = str(response.mime_type)
    knowledge_type_list = [
        KNOWLEDGE_TYPES[i] for i in response.knowledge_types
    ]
    document_dict["Knowledge-type"] = ",".join(knowledge_type_list)
    document_dict["content_uri"] = response.content_uri
    # returned done, directly use as response
    return document_dict
Exemplo n.º 18
0
    def test_create_document_exception(self):
        # Setup Response
        error = status_pb2.Status()
        operation = operations_pb2.Operation(
            name='operations/test_create_document_exception', done=True)
        operation.error.CopyFrom(error)

        # Mock the API response
        channel = ChannelStub(responses=[operation])
        client = dialogflow_v2beta1.DocumentsClient(channel=channel)

        # Setup Request
        parent = client.knowledge_base_path('[PROJECT]', '[KNOWLEDGE_BASE]')
        document = {}

        response = client.create_document(parent, document)
        exception = response.exception()
        assert exception.errors[0] == error
Exemplo n.º 19
0
    def test_delete_document_exception(self):
        # Setup Response
        error = status_pb2.Status()
        operation = operations_pb2.Operation(
            name='operations/test_delete_document_exception', done=True)
        operation.error.CopyFrom(error)

        # Mock the API response
        channel = ChannelStub(responses=[operation])
        client = dialogflow_v2beta1.DocumentsClient(channel=channel)

        # Setup Request
        name = client.document_path('[PROJECT]', '[KNOWLEDGE_BASE]',
                                    '[DOCUMENT]')

        response = client.delete_document(name)
        exception = response.exception()
        assert exception.errors[0] == error
def counter(PROJECT_ID, KID):
    """
    count the number of documents in a specific knowledgement base
    :parameter PROJECT_ID: The GCP project linked with the agent.
    :parameter KID: ID to Knowledge base by.

    :return num: the number of documents
    """

    client = dialogflow.DocumentsClient()
    knowledge_base_path = client.knowledge_base_path(PROJECT_ID, KID)
    # testing stout
    print('Documents for Knowledge Id: {}'.format(KID))
    iterator = client.list_documents(knowledge_base_path)
    count = 0
    for i in iterator:
        count += 0
    return count
    def test_update_document_exception(self):
        # Setup Response
        error = status_pb2.Status()
        operation = operations_pb2.Operation(
            name="operations/test_update_document_exception", done=True)
        operation.error.CopyFrom(error)

        # Mock the API response
        channel = ChannelStub(responses=[operation])
        patch = mock.patch("google.api_core.grpc_helpers.create_channel")
        with patch as create_channel:
            create_channel.return_value = channel
            client = dialogflow_v2beta1.DocumentsClient()

        # Setup Request
        document = {}

        response = client.update_document(document)
        exception = response.exception()
        assert exception.errors[0] == error
Exemplo n.º 22
0
def list_documents(project_id, knowledge_base_id):
    """Lists the Documents belonging to a Knowledge base.

    Args:
        project_id: The GCP project linked with the agent.
        knowledge_base_id: Id of the Knowledge base."""
    import dialogflow_v2beta1 as dialogflow
    client = dialogflow.DocumentsClient()
    knowledge_base_path = client.knowledge_base_path(project_id,
                                                     knowledge_base_id)

    print('Documents for Knowledge Id: {}'.format(knowledge_base_id))
    for document in client.list_documents(knowledge_base_path):
        print(' - Display Name: {}'.format(document.display_name))
        print(' - Knowledge ID: {}'.format(document.name))
        print(' - MIME Type: {}'.format(document.mime_type))
        print(' - Knowledge Types:')
        for knowledge_type in document.knowledge_types:
            print('    - {}'.format(KNOWLEDGE_TYPES[knowledge_type]))
        print(' - Source: {}\n'.format(document.content_uri))
Exemplo n.º 23
0
def get_document(project_id, knowledge_base_id, document_id):
    """Gets a Document.

    Args:
        project_id: The GCP project linked with the agent.
        knowledge_base_id: Id of the Knowledge base.
        document_id: Id of the Document."""
    import dialogflow_v2beta1 as dialogflow
    client = dialogflow.DocumentsClient()
    document_path = client.document_path(project_id, knowledge_base_id,
                                         document_id)

    response = client.get_document(document_path)
    print('Got Document:')
    print(' - Display Name: {}'.format(response.display_name))
    print(' - Knowledge ID: {}'.format(response.name))
    print(' - MIME Type: {}'.format(response.mime_type))
    print(' - Knowledge Types:')
    for knowledge_type in response.knowledge_types:
        print('    - {}'.format(KNOWLEDGE_TYPES[knowledge_type]))
    print(' - Source: {}\n'.format(response.content_uri))
    def test_reload_document(self):
        # Setup Expected Response
        name = "name3373707"
        done = True
        expected_response = {"name": name, "done": done}
        expected_response = operations_pb2.Operation(**expected_response)

        # Mock the API response
        channel = ChannelStub(responses=[expected_response])
        patch = mock.patch("google.api_core.grpc_helpers.create_channel")
        with patch as create_channel:
            create_channel.return_value = channel
            client = dialogflow_v2beta1.DocumentsClient()

        response = client.reload_document()
        assert expected_response == response

        assert len(channel.requests) == 1
        expected_request = document_pb2.ReloadDocumentRequest()
        actual_request = channel.requests[0][1]
        assert expected_request == actual_request
    def test_reload_document_exception(self):
        # Setup Response
        error = status_pb2.Status()
        operation = operations_pb2.Operation(
            name="operations/test_reload_document_exception", done=True)
        operation.error.CopyFrom(error)

        # Mock the API response
        channel = ChannelStub(responses=[operation])
        patch = mock.patch("google.api_core.grpc_helpers.create_channel")
        with patch as create_channel:
            create_channel.return_value = channel
            client = dialogflow_v2beta1.DocumentsClient()

        # Setup Request
        name = client.document_path("[PROJECT]", "[KNOWLEDGE_BASE]",
                                    "[DOCUMENT]")

        response = client.reload_document(name)
        exception = response.exception()
        assert exception.errors[0] == error
    def test_create_document_exception(self):
        # Setup Response
        error = status_pb2.Status()
        operation = operations_pb2.Operation(
            name='operations/test_create_document_exception', done=True)
        operation.error.CopyFrom(error)

        # Mock the API response
        channel = ChannelStub(responses=[operation])
        patch = mock.patch('google.api_core.grpc_helpers.create_channel')
        with patch as create_channel:
            create_channel.return_value = channel
            client = dialogflow_v2beta1.DocumentsClient()

        # Setup Request
        parent = client.knowledge_base_path('[PROJECT]', '[KNOWLEDGE_BASE]')
        document = {}

        response = client.create_document(parent, document)
        exception = response.exception()
        assert exception.errors[0] == error
    def test_create_document(self):
        # Setup Expected Response
        name = "name3373707"
        display_name = "displayName1615086568"
        mime_type = "mimeType-196041627"
        content_uri = "contentUri-388807514"
        enable_auto_reload = False
        expected_response = {
            "name": name,
            "display_name": display_name,
            "mime_type": mime_type,
            "content_uri": content_uri,
            "enable_auto_reload": enable_auto_reload,
        }
        expected_response = document_pb2.Document(**expected_response)
        operation = operations_pb2.Operation(
            name="operations/test_create_document", done=True)
        operation.response.Pack(expected_response)

        # Mock the API response
        channel = ChannelStub(responses=[operation])
        patch = mock.patch("google.api_core.grpc_helpers.create_channel")
        with patch as create_channel:
            create_channel.return_value = channel
            client = dialogflow_v2beta1.DocumentsClient()

        # Setup Request
        parent = client.knowledge_base_path("[PROJECT]", "[KNOWLEDGE_BASE]")
        document = {}

        response = client.create_document(parent, document)
        result = response.result()
        assert expected_response == result

        assert len(channel.requests) == 1
        expected_request = document_pb2.CreateDocumentRequest(
            parent=parent, document=document)
        actual_request = channel.requests[0][1]
        assert expected_request == actual_request
Exemplo n.º 28
0
def create_document(project_id, knowledge_base_id, display_name, mime_type,
                    knowledge_type, content_uri):
    """Creates a Document.

    Args:
        project_id: The GCP project linked with the agent.
        knowledge_base_id: Id of the Knowledge base.
        display_name: The display name of the Document.
        mime_type: The mime_type of the Document. e.g. text/csv, text/html,
            text/plain, text/pdf etc.
        knowledge_type: The Knowledge type of the Document. e.g. FAQ,
            EXTRACTIVE_QA.
        content_uri: Uri of the document, e.g. gs://path/mydoc.csv,
            http://mypage.com/faq.html."""
    import dialogflow_v2beta1 as dialogflow
    client = dialogflow.DocumentsClient()
    knowledge_base_path = client.knowledge_base_path(project_id,
                                                     knowledge_base_id)

    document = dialogflow.types.Document(display_name=display_name,
                                         mime_type=mime_type,
                                         content_uri=content_uri)

    document.knowledge_types.append(
        dialogflow.types.Document.KnowledgeType.Value(knowledge_type))

    response = client.create_document(knowledge_base_path, document)
    print('Waiting for results...')
    document = response.result(timeout=90)
    print('Created Document:')
    print(' - Display Name: {}'.format(document.display_name))
    print(' - Knowledge ID: {}'.format(document.name))
    print(' - MIME Type: {}'.format(document.mime_type))
    print(' - Knowledge Types:')
    for knowledge_type in document.knowledge_types:
        print('    - {}'.format(KNOWLEDGE_TYPES[knowledge_type]))
    print(' - Source: {}\n'.format(document.content_uri))
    def test_update_document(self):
        # Setup Expected Response
        name = "name3373707"
        display_name = "displayName1615086568"
        mime_type = "mimeType-196041627"
        content_uri = "contentUri-388807514"
        expected_response = {
            "name": name,
            "display_name": display_name,
            "mime_type": mime_type,
            "content_uri": content_uri,
        }
        expected_response = document_pb2.Document(**expected_response)
        operation = operations_pb2.Operation(
            name="operations/test_update_document", done=True)
        operation.response.Pack(expected_response)

        # Mock the API response
        channel = ChannelStub(responses=[operation])
        patch = mock.patch("google.api_core.grpc_helpers.create_channel")
        with patch as create_channel:
            create_channel.return_value = channel
            client = dialogflow_v2beta1.DocumentsClient()

        # Setup Request
        document = {}

        response = client.update_document(document)
        result = response.result()
        assert expected_response == result

        assert len(channel.requests) == 1
        expected_request = document_pb2.UpdateDocumentRequest(
            document=document)
        actual_request = channel.requests[0][1]
        assert expected_request == actual_request
Exemplo n.º 30
0
    def test_delete_document(self):
        # Setup Expected Response
        expected_response = {}
        expected_response = empty_pb2.Empty(**expected_response)
        operation = operations_pb2.Operation(
            name='operations/test_delete_document', done=True)
        operation.response.Pack(expected_response)

        # Mock the API response
        channel = ChannelStub(responses=[operation])
        client = dialogflow_v2beta1.DocumentsClient(channel=channel)

        # Setup Request
        name = client.document_path('[PROJECT]', '[KNOWLEDGE_BASE]',
                                    '[DOCUMENT]')

        response = client.delete_document(name)
        result = response.result()
        assert expected_response == result

        assert len(channel.requests) == 1
        expected_request = document_pb2.DeleteDocumentRequest(name=name)
        actual_request = channel.requests[0][1]
        assert expected_request == actual_request