Updating path

Author: kumar-shivam-ranjan

ads/aqua/extension/model_handler.py

@@ -325,7 +325,7 @@ def get(self, model_id):
         if (
             len(path_list) == 4
             and is_valid_ocid(path_list[2])
-            and path_list[3] == "chat_template"
+            and path_list[3] == "tokenizer"
         ):
             return self.finish(AquaModelApp().get_hf_tokenizer_config(model_id))
         else:
@@ -335,6 +335,6 @@ def get(self, model_id):
 __handlers__ = [
     ("model/?([^/]*)", AquaModelHandler),
     ("model/?([^/]*)/license", AquaModelLicenseHandler),
-    ("model/?([^/]*)/chat_template", AquaModelTokenizerConfigHandler),
+    ("model/?([^/]*)/tokenizer", AquaModelTokenizerConfigHandler),
     ("model/hf/search/?([^/]*)", AquaHuggingFaceHandler),
 ]

tests/unitary/with_extras/aqua/test_model_handler.py

@@ -264,7 +264,7 @@ def setUp(self, ipython_init_mock) -> None:
     @patch.object(AquaModelApp, "get_hf_tokenizer_config")
     @patch("ads.aqua.extension.model_handler.urlparse")
     def test_get(self, mock_urlparse, mock_get_hf_tokenizer_config):
-        request_path = MagicMock(path="aqua/model/ocid1.xx./chat_template")
+        request_path = MagicMock(path="aqua/model/ocid1.xx./tokenizer")
         mock_urlparse.return_value = request_path
         self.model_tokenizer_config_handler.get(model_id="test_model_id")
         self.model_tokenizer_config_handler.finish.assert_called_with(