From c7c7d4c8a2134425cd76d2e1592ea4e2ea65adb0 Mon Sep 17 00:00:00 2001
From: Varun Villait <vvillait88@yahoo.com>
Date: Tue, 17 Dec 2024 09:53:21 -0700
Subject: [PATCH] Add Updated Title Roles to Autocomplete (#24)

* add support for utr in autocomplete

* add test

* update test

* fix
---
 README.md                                    | 2 +-
 lib/peopledatalabs/api_resource.rb           | 2 +-
 lib/peopledatalabs/resources/autocomplete.rb | 3 ++-
 lib/peopledatalabs/version.rb                | 2 +-
 spec/peopledatalabs_spec.rb                  | 6 ++++++
 5 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index 931fc91..81b233c 100644
--- a/README.md
+++ b/README.md
@@ -1,5 +1,5 @@
 <p align="center">
-<img src="https://i.imgur.com/S7DkZtr.png" width="250" alt="People Data Labs Logo">
+<img src="https://www.peopledatalabs.com/images/company-logo.png" style="background-color: white; padding: 5px 10px;" width="250" alt="People Data Labs Logo">
 </p>
 <h1 align="center">People Data Labs Ruby Library</h1>
 <p align="center">Official Ruby client for the People Data Labs API.</p>
diff --git a/lib/peopledatalabs/api_resource.rb b/lib/peopledatalabs/api_resource.rb
index 294fe99..3901f55 100644
--- a/lib/peopledatalabs/api_resource.rb
+++ b/lib/peopledatalabs/api_resource.rb
@@ -3,7 +3,7 @@ class APIResource
 
     protected
 
-    VALID_AUTOCOMPLETE_FIELDS = ['company', 'country', 'industry', 'location', 'major', 'region', 'role', 'school', 'sub_role', 'skill', 'title'].freeze
+    VALID_AUTOCOMPLETE_FIELDS = ['class', 'company', 'country', 'industry', 'location', 'major', 'region', 'role', 'school', 'sub_role', 'skill', 'title'].freeze
 
     def self.get(path:, headers: {}, params: {})
       request = check(params: params, path: path)
diff --git a/lib/peopledatalabs/resources/autocomplete.rb b/lib/peopledatalabs/resources/autocomplete.rb
index 9f5c9bc..e2c11a3 100644
--- a/lib/peopledatalabs/resources/autocomplete.rb
+++ b/lib/peopledatalabs/resources/autocomplete.rb
@@ -2,7 +2,7 @@
 
 module Peopledatalabs
   class Autocomplete < APIResource
-    def self.retrieve(field:, text: '', size: 10, pretty: false, titlecase: false)
+    def self.retrieve(field:, text: '', size: 10, pretty: false, titlecase: false, updated_title_roles: false)
 
       params = {
         'field' => field,
@@ -10,6 +10,7 @@ def self.retrieve(field:, text: '', size: 10, pretty: false, titlecase: false)
         'size' => size,
         'pretty' => pretty,
         'titlecase' => titlecase,
+        'updated_title_roles' => updated_title_roles,
       };
 
       headers = {
diff --git a/lib/peopledatalabs/version.rb b/lib/peopledatalabs/version.rb
index c867e85..519ea0d 100644
--- a/lib/peopledatalabs/version.rb
+++ b/lib/peopledatalabs/version.rb
@@ -1,3 +1,3 @@
 module Peopledatalabs
-  VERSION = "3.1.0"
+  VERSION = "3.2.0"
 end
diff --git a/spec/peopledatalabs_spec.rb b/spec/peopledatalabs_spec.rb
index cc9e8c4..714aa1b 100644
--- a/spec/peopledatalabs_spec.rb
+++ b/spec/peopledatalabs_spec.rb
@@ -219,6 +219,12 @@
       expect(result).to be_an_instance_of(Hash)
     end
 
+    it "should return autocomplete record for class" do
+      result = Peopledatalabs::Autocomplete.retrieve(field: 'class', text: 'sales', size: 10, updated_title_roles: true)
+      expect(result['status']).to eq(200)
+      expect(result).to be_an_instance_of(Hash)
+    end
+
     it "should error" do
       result = Peopledatalabs::Autocomplete.retrieve(field: nil)
       expect(result['status']).to eq(400)