chore(release): merge release v0.4.0

Author: github-actions[release-bot]

CHANGELOG.md

@@ -1,3 +1,9 @@
+## 0.4.0 (2025-01-18)
+
+### Fix
+
+- ensure all generation parameters are passed to the MBart model
+
 ## 0.3.0 (2025-01-16)
 
 ### Feat

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "translation-api"
-version = "0.3.0"
+version = "0.4.0"
 description = "A FastAPI project integrating translation services."
 authors = [{ name = "Grzegorz Gwóźdź" }]
 requires-python = ">=3.12,<3.13"

src/data/workers/mbart_translation_worker.py

@@ -90,12 +90,10 @@ def handle_command(
                 tokenizer.src_lang = source_language
                 inputs = tokenizer(text, return_tensors="pt").to(config.device)
 
-                if "forced_bos_token_id" in generation_parameters:
-                    kwargs = {"forced_bos_token_id": generation_parameters["forced_bos_token_id"]}
-                else:
-                    kwargs = {"forced_bos_token_id": tokenizer.lang_code_to_id[target_language]}
+                if "forced_bos_token_id" not in generation_parameters:
+                    generation_parameters["forced_bos_token_id"] = tokenizer.lang_code_to_id[target_language]
 
-                translation = model.generate(**inputs, **kwargs)
+                translation = model.generate(**inputs, **generation_parameters)
 
                 output = [tokenizer.decode(t, skip_special_tokens=True) for t in translation]