fix(api): use get_languages on /languages (#17)

Author: hspedro

babeltron/app/routers/translate.py

@@ -230,4 +230,4 @@ async def languages():
             detail="Translation model not loaded. Please check server logs.",
         )
 
-    return {"languages": model.supported_languages}
+    return {"languages": model.get_languages()}

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "babeltron"
-version = "0.7.1"
+version = "0.7.2"
 dynamic = ["version"]
 description = "A Python-based REST API that leverages single multilingual models like mBERT to provide efficient text translation services"
 authors = [

tests/unit/app/routers/test_translate.py

@@ -232,14 +232,14 @@ def test_languages_endpoint(mock_translation_model, mock_get_model, client):
     mock_model = MagicMock()
     mock_model.is_loaded = True
     mock_model._model_path = "/mocked/path"
-    mock_model.supported_languages = ["en", "fr", "es", "de"]
+    mock_model.get_languages.return_value = ["en", "fr", "es", "de"]
 
     # Make the factory return our mock model
     mock_get_model.return_value = mock_model
 
     # Configure the translation_model mock
     mock_translation_model.is_loaded = True
-    mock_translation_model.supported_languages = ["en", "fr", "es", "de"]
+    mock_translation_model.get_languages.return_value = ["en", "fr", "es", "de"]
 
     response = client.get("/api/v1/languages")
     assert response.status_code == status.HTTP_200_OK