def complete_query(project_id, tenant_id, query): """Complete job title given partial text (autocomplete)""" client = talent_v4beta1.CompletionClient() # project_id = 'Your Google Cloud Project ID' # tenant_id = 'Your Tenant ID (using tenancy is optional)' # query = '[partially typed job title]' if isinstance(project_id, six.binary_type): project_id = project_id.decode("utf-8") if isinstance(tenant_id, six.binary_type): tenant_id = tenant_id.decode("utf-8") if isinstance(query, six.binary_type): query = query.decode("utf-8") parent = f"projects/{project_id}/tenants/{tenant_id}" request = talent_v4beta1.CompleteQueryRequest( parent=parent, query=query, page_size=5, # limit for number of results language_codes=["en-US"], # language code ) response = client.complete_query(request=request) for result in response.completion_results: print(f"Suggested title: {result.suggestion}") # Suggestion type is JOB_TITLE or COMPANY_TITLE print( f"Suggestion type: {talent_v4beta1.CompleteQueryRequest.CompletionType(result.type_).name}" )
def test_complete_query(self): # Setup Expected Response expected_response = {} expected_response = completion_service_pb2.CompleteQueryResponse( **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 = talent_v4beta1.CompletionClient() # Setup Request parent = client.tenant_path("[PROJECT]", "[TENANT]") query = "query107944136" page_size = 883849137 response = client.complete_query(parent, query, page_size) assert expected_response == response assert len(channel.requests) == 1 expected_request = completion_service_pb2.CompleteQueryRequest( parent=parent, query=query, page_size=page_size) actual_request = channel.requests[0][1] assert expected_request == actual_request
def sample_complete_query(project_id, tenant_id, query, num_results, language_code): """Complete job title given partial text (autocomplete)""" # [START job_search_autocomplete_job_title_core] client = talent_v4beta1.CompletionClient() # project_id = 'Your Google Cloud Project ID' # tenant_id = 'Your Tenant ID (using tenancy is optional)' # query = '[partially typed job title]' # num_results = 5 # language_code = 'en-US' if isinstance(project_id, six.binary_type): project_id = project_id.decode('utf-8') if isinstance(tenant_id, six.binary_type): tenant_id = tenant_id.decode('utf-8') if isinstance(query, six.binary_type): query = query.decode('utf-8') if isinstance(language_code, six.binary_type): language_code = language_code.decode('utf-8') parent = client.tenant_path(project_id, tenant_id) language_codes = [language_code] response = client.complete_query(parent, query, num_results, language_codes=language_codes) for result in response.completion_results: print('Suggested title: {}'.format(result.suggestion)) # Suggestion type is JOB_TITLE or COMPANY_TITLE print('Suggestion type: {}'.format( enums.CompleteQueryRequest.CompletionType(result.type).name))
def test_complete_query_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 = talent_v4beta1.CompletionClient() # Setup request parent = client.tenant_path("[PROJECT]", "[TENANT]") query = "query107944136" page_size = 883849137 with pytest.raises(CustomException): client.complete_query(parent, query, page_size)
def test_complete_query_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 = talent_v4beta1.CompletionClient() # Setup request name = client.project_path('[PROJECT]') query = 'query107944136' page_size = 883849137 with pytest.raises(CustomException): client.complete_query(name, query, page_size)