Trying to fix lint error

Author: ritaagarwala-sf

.pre-commit-config.yaml

@@ -53,7 +53,7 @@ repos:
     exclude: \.py$
 
 - repo: https://github.com/macisamuele/language-formatters-pre-commit-hooks
-  rev: v2.13.0
+  rev: v2.14.0
   hooks:
   - id: pretty-format-toml
     args: [--autofix]

pyproject.toml

@@ -72,7 +72,9 @@ use_parentheses = true
 
 [tool.mypy]
 check_untyped_defs = false
+explicit_package_bases = true
 ignore_missing_imports = true
+mypy_path = "src"
 no_implicit_optional = true
 plugins = [
   'pydantic.mypy'
@@ -82,8 +84,6 @@ strict_optional = true
 warn_redundant_casts = true
 warn_return_any = true
 warn_unused_configs = true
-explicit_package_bases = true
-mypy_path = "src"
 
 [tool.poetry]
 include = [

src/datacustomcode/llm_gateway/types/generate_text_request_builder.py

@@ -23,25 +23,25 @@
 
 
 class GenerateTextRequestBuilder:
-    def __init__(self):
+    def __init__(self) -> None:
         self._prompt = ""
         self._model_name = ""
         self._localization: Optional[Dict[str, Any]] = None
         self._tags: Optional[Dict[str, Any]] = None
 
-    def set_prompt(self, prompt: str):
+    def set_prompt(self, prompt: str) -> "GenerateTextRequestBuilder":
         self._prompt = prompt
         return self
 
-    def set_model(self, model_name: str):
+    def set_model(self, model_name: str) -> "GenerateTextRequestBuilder":
         self._model_name = model_name
         return self
 
     def set_localization(
         self,
         localization: Optional[Dict[str, Any]] = None,
         locale: Optional[str] = None,
-    ):
+    ) -> "GenerateTextRequestBuilder":
         """
         Set localization either from a dict or a simple locale string.
 
@@ -62,7 +62,7 @@ def set_localization(
 
         return self
 
-    def set_tags(self, tags: Dict[str, Any]):
+    def set_tags(self, tags: Dict[str, Any]) -> "GenerateTextRequestBuilder":
         self._tags = tags
         return self