hamtaa-texttools 1.1.2__py3-none-any.whl → 1.1.4__py3-none-any.whl

{hamtaa_texttools-1.1.2.dist-info → hamtaa_texttools-1.1.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hamtaa-texttools
-Version: 1.1.2
+Version: 1.1.4
 Summary: A high-level NLP toolkit built on top of modern LLMs.
 Author-email: Tohidi <the.mohammad.tohidi@gmail.com>, Montazer <montazerh82@gmail.com>, Givechi <mohamad.m.givechi@gmail.com>, MoosaviNejad <erfanmoosavi84@gmail.com>
 License: MIT License
@@ -40,14 +40,14 @@ Dynamic: license-file
 
 It provides both **sync (`TheTool`)** and **async (`AsyncTheTool`)** APIs for maximum flexibility.
 
-It provides ready-to-use utilities for **translation, question detection, keyword extraction, categorization, NER extractor, and more** — designed to help you integrate AI-powered text processing into your applications with minimal effort.
+It provides ready-to-use utilities for **translation, question detection, keyword extraction, categorization, NER extraction, and more** — designed to help you integrate AI-powered text processing into your applications with minimal effort.
 
 ---
 
 ## ✨ Features
 
 TextTools provides a rich collection of high-level NLP utilities built on top of LLMs.  
-Each tool is designed to work out-of-the-box with structured outputs (JSON / Pydantic).
+Each tool is designed to work with structured outputs (JSON / Pydantic).
 
 - **`categorize()`** - Classifies text into Islamic studies categories 
 - **`is_question()`** - Binary detection of whether input is a question
@@ -63,7 +63,7 @@ Each tool is designed to work out-of-the-box with structured outputs (JSON / Pyd
 
 ---
 
-## ⚙️ `with_analysis`, `logprobs`, `output_lang`, `user_prompt` and `temperature` parameters
+## ⚙️ `with_analysis`, `logprobs`, `output_lang`, `user_prompt`, `temperature` and `validator` parameters
 
 TextTools provides several optional flags to customize LLM behavior:
 
@@ -78,6 +78,8 @@ Note: This doubles token usage per call because it triggers an additional LLM re
 
 - **`temperature=0.0`** → Determines how creative the model should respond. Takes a float number from `0.0` to `1.0`.
 
+- **`validator=validation_function`** → Forces TheTool to validate the output result based on your custom validator. Validator should return bool (True if there were no problem, False if the validation failed.) If validator failed, TheTool will retry to get another output by modifying `temperature`.
+
 All these parameters can be used individually or together to tailor the behavior of any tool in **TextTools**.
 
 **Note:** There might be some tools that don't support some of the parameters above.

{hamtaa_texttools-1.1.2.dist-info → hamtaa_texttools-1.1.4.dist-info}/RECORD

@@ -1,4 +1,4 @@
-hamtaa_texttools-1.1.2.dist-info/licenses/LICENSE,sha256=Hb2YOBKy2MJQLnyLrX37B4ZVuac8eaIcE71SvVIMOLg,1082
+hamtaa_texttools-1.1.4.dist-info/licenses/LICENSE,sha256=Hb2YOBKy2MJQLnyLrX37B4ZVuac8eaIcE71SvVIMOLg,1082
 texttools/__init__.py,sha256=v3tQCH_Cjj47fCpuhK6sKSVAqEjNkc-cZbY4OJa4IZw,202
 texttools/batch/__init__.py,sha256=q50JsQsmQGp_8RW0KNasYeYWVV0R4FUNZ-ujXwEJemY,143
 texttools/batch/batch_manager.py,sha256=UCJaOq-tTy6kTZJvFuBDBSmzlDeVlLTeFlR83e6eXkQ,8808
@@ -16,15 +16,15 @@ texttools/prompts/summarize.yaml,sha256=o6rxGPfWtZd61Duvm8NVvCJqfq73b-wAuMSKR6UY
 texttools/prompts/text_to_question.yaml,sha256=UheKYpDn6iyKI8NxunHZtFpNyfCLZZe5cvkuXpurUJY,783
 texttools/prompts/translate.yaml,sha256=mGT2uBCei6uucWqVbs4silk-UV060v3G0jnt0P6sr50,634
 texttools/tools/__init__.py,sha256=hG1I28Q7BJ1Dbs95x6QMKXdsAlC5Eh_tqC-EbAibwiU,114
-texttools/tools/async_the_tool.py,sha256=ORn3xk6zJ3HdDNo_zj9cTlY05zsQ1un5H96xLIUUOg4,14446
-texttools/tools/the_tool.py,sha256=q-bvYv38CuBMstYOV_42zv2frHgsaLRzo_XAGNrdNWA,14212
-texttools/tools/internals/async_operator.py,sha256=vZvU5XPdNKz--ZYheEQOIRzVNi-Ni1x9GxgcDBlbYJM,6760
-texttools/tools/internals/base_operator.py,sha256=3GUSEN87Y_1CqkeK_QXkFG8O7GPliGSVUtiPp1taQRM,3051
+texttools/tools/async_the_tool.py,sha256=2ZY7Lo6Jj9xoTF8bfdh_g8VOXZ7ljMMesd1_QHXyf4s,15395
+texttools/tools/the_tool.py,sha256=XKgZuzriFnk8B-YihJfs6BKivxjGCgOFfe7hnCpEiXs,15161
+texttools/tools/internals/async_operator.py,sha256=pqxceWej3wJmL2RSYEWdowO77MVEN6E7kqppOnsoLOk,9069
+texttools/tools/internals/base_operator.py,sha256=7xRNKyPnh61XTvgPghyQeclBlhw7K7hZxk10ZMC9CQY,3492
 texttools/tools/internals/formatters.py,sha256=tACNLP6PeoqaRpNudVxBaHA25zyWqWYPZQuYysIu88g,941
-texttools/tools/internals/operator.py,sha256=M-0-6SmRmW7PX4CWhMAgoSwtk_2w1_YrItK3FhdObCU,6659
+texttools/tools/internals/operator.py,sha256=RqCau1F1lxoDQq4BqJvjOnwaf-86SQF_wpoP2V3LYKQ,8951
 texttools/tools/internals/output_models.py,sha256=gbVbzBWeyHUVNsCBuawdgz9ZEzsC7wfygGgZJsAaexY,1662
 texttools/tools/internals/prompt_loader.py,sha256=1khayXcRC5w0Vf2SufpNaN1IUIhbKzS5ATiKheoBcGE,2082
-hamtaa_texttools-1.1.2.dist-info/METADATA,sha256=J2YfhGNN2SpOq5P3iOsniaUzpuUaP20dTlPBiQX6Gss,7144
-hamtaa_texttools-1.1.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-hamtaa_texttools-1.1.2.dist-info/top_level.txt,sha256=5Mh0jIxxZ5rOXHGJ6Mp-JPKviywwN0MYuH0xk5bEWqE,10
-hamtaa_texttools-1.1.2.dist-info/RECORD,,
+hamtaa_texttools-1.1.4.dist-info/METADATA,sha256=-pHGh4p8ZW1agur7LwjY8EUZps9AU9TTWtxxkbleDAw,7449
+hamtaa_texttools-1.1.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+hamtaa_texttools-1.1.4.dist-info/top_level.txt,sha256=5Mh0jIxxZ5rOXHGJ6Mp-JPKviywwN0MYuH0xk5bEWqE,10
+hamtaa_texttools-1.1.4.dist-info/RECORD,,

texttools/tools/async_the_tool.py

@@ -1,4 +1,4 @@
-from typing import Literal, Any
+from typing import Literal, Any, Callable
 
 from openai import AsyncOpenAI
 
@@ -34,6 +34,7 @@ class AsyncTheTool:
         temperature: float | None = 0.0,
         logprobs: bool = False,
         top_logprobs: int | None = None,
+        validator: Callable[[Any], bool] | None = None,
     ) -> OutputModels.ToolOutput:
         """
         Categorize a text into a single Islamic studies domain category.
@@ -52,6 +53,7 @@ class AsyncTheTool:
             temperature=temperature,
             logprobs=logprobs,
             top_logprobs=top_logprobs,
+            validator=validator,
             # Internal parameters
             prompt_file="categorizer.yaml",
             output_model=OutputModels.CategorizerOutput,
@@ -69,6 +71,7 @@ class AsyncTheTool:
         temperature: float | None = 0.0,
         logprobs: bool = False,
         top_logprobs: int | None = None,
+        validator: Callable[[Any], bool] | None = None,
     ) -> OutputModels.ToolOutput:
         """
         Extract salient keywords from text.
@@ -88,6 +91,7 @@ class AsyncTheTool:
             temperature=temperature,
             logprobs=logprobs,
             top_logprobs=top_logprobs,
+            validator=validator,
             # Internal parameters
             prompt_file="extract_keywords.yaml",
             output_model=OutputModels.ListStrOutput,
@@ -104,6 +108,7 @@ class AsyncTheTool:
         temperature: float | None = 0.0,
         logprobs: bool = False,
         top_logprobs: int | None = None,
+        validator: Callable[[Any], bool] | None = None,
     ) -> OutputModels.ToolOutput:
         """
         Perform Named Entity Recognition (NER) over the input text.
@@ -123,6 +128,7 @@ class AsyncTheTool:
             temperature=temperature,
             logprobs=logprobs,
             top_logprobs=top_logprobs,
+            validator=validator,
             # Internal parameters
             prompt_file="extract_entities.yaml",
             output_model=OutputModels.ListDictStrStrOutput,
@@ -138,6 +144,7 @@ class AsyncTheTool:
         temperature: float | None = 0.0,
         logprobs: bool = False,
         top_logprobs: int | None = None,
+        validator: Callable[[Any], bool] | None = None,
     ) -> OutputModels.ToolOutput:
         """
         Detect if the input is phrased as a question.
@@ -156,6 +163,7 @@ class AsyncTheTool:
             temperature=temperature,
             logprobs=logprobs,
             top_logprobs=top_logprobs,
+            validator=validator,
             # Internal parameters
             prompt_file="is_question.yaml",
             output_model=OutputModels.BoolOutput,
@@ -173,6 +181,7 @@ class AsyncTheTool:
         temperature: float | None = 0.0,
         logprobs: bool = False,
         top_logprobs: int | None = None,
+        validator: Callable[[Any], bool] | None = None,
     ) -> OutputModels.ToolOutput:
         """
         Generate a single question from the given text.
@@ -192,6 +201,7 @@ class AsyncTheTool:
             temperature=temperature,
             logprobs=logprobs,
             top_logprobs=top_logprobs,
+            validator=validator,
             # Internal parameters
             prompt_file="text_to_question.yaml",
             output_model=OutputModels.StrOutput,
@@ -209,6 +219,7 @@ class AsyncTheTool:
         logprobs: bool = False,
         top_logprobs: int | None = None,
         mode: Literal["default", "reason"] = "default",
+        validator: Callable[[Any], bool] | None = None,
     ) -> OutputModels.ToolOutput:
         """
         Merge multiple questions into a single unified question.
@@ -229,6 +240,7 @@ class AsyncTheTool:
             temperature=temperature,
             logprobs=logprobs,
             top_logprobs=top_logprobs,
+            validator=validator,
             # Internal parameters
             prompt_file="merge_questions.yaml",
             output_model=OutputModels.StrOutput,
@@ -246,6 +258,7 @@ class AsyncTheTool:
         logprobs: bool = False,
         top_logprobs: int | None = None,
         mode: Literal["positive", "negative", "hard_negative"] = "positive",
+        validator: Callable[[Any], bool] | None = None,
     ) -> OutputModels.ToolOutput:
         """
         Rewrite a text with different modes.
@@ -265,6 +278,7 @@ class AsyncTheTool:
             temperature=temperature,
             logprobs=logprobs,
             top_logprobs=top_logprobs,
+            validator=validator,
             # Internal parameters
             prompt_file="rewrite.yaml",
             output_model=OutputModels.StrOutput,
@@ -282,6 +296,7 @@ class AsyncTheTool:
         temperature: float | None = 0.0,
         logprobs: bool = False,
         top_logprobs: int | None = None,
+        validator: Callable[[Any], bool] | None = None,
     ) -> OutputModels.ToolOutput:
         """
         Generate a list of questions about a subject.
@@ -302,6 +317,7 @@ class AsyncTheTool:
             temperature=temperature,
             logprobs=logprobs,
             top_logprobs=top_logprobs,
+            validator=validator,
             # Internal parameters
             prompt_file="subject_to_question.yaml",
             output_model=OutputModels.ReasonListStrOutput,
@@ -318,6 +334,7 @@ class AsyncTheTool:
         temperature: float | None = 0.0,
         logprobs: bool = False,
         top_logprobs: int | None = None,
+        validator: Callable[[Any], bool] | None = None,
     ) -> OutputModels.ToolOutput:
         """
         Summarize the given subject text.
@@ -337,6 +354,7 @@ class AsyncTheTool:
             temperature=temperature,
             logprobs=logprobs,
             top_logprobs=top_logprobs,
+            validator=validator,
             # Internal parameters
             prompt_file="summarize.yaml",
             output_model=OutputModels.StrOutput,
@@ -353,6 +371,7 @@ class AsyncTheTool:
         temperature: float | None = 0.0,
         logprobs: bool = False,
         top_logprobs: int | None = None,
+        validator: Callable[[Any], bool] | None = None,
     ) -> OutputModels.ToolOutput:
         """
         Translate text between languages.
@@ -372,6 +391,7 @@ class AsyncTheTool:
             temperature=temperature,
             logprobs=logprobs,
             top_logprobs=top_logprobs,
+            validator=validator,
             # Internal parameters
             prompt_file="translate.yaml",
             output_model=OutputModels.StrOutput,
@@ -411,4 +431,5 @@ class AsyncTheTool:
             user_prompt=None,
             with_analysis=False,
             mode=None,
+            validator=None,
         )

texttools/tools/internals/async_operator.py

@@ -1,4 +1,4 @@
-from typing import Any, TypeVar, Type, Literal
+from typing import Any, TypeVar, Type, Literal, Callable
 import logging
 
 from openai import AsyncOpenAI
@@ -115,6 +115,7 @@ class AsyncOperator(BaseOperator):
         temperature: float,
         logprobs: bool,
         top_logprobs: int | None,
+        validator: Callable[[Any], bool] | None,
         # Internal parameters
         prompt_file: str,
         output_model: Type[T],
@@ -179,6 +180,55 @@ class AsyncOperator(BaseOperator):
 
             output.result = parsed.result
 
+            # Retry logic if validation fails
+            if validator and not validator(output.result):
+                max_retries = 3
+                for attempt in range(max_retries):
+                    logger.warning(
+                        f"Validation failed, retrying for the {attempt + 1} time."
+                    )
+
+                    # Generate new temperature for retry
+                    retry_temperature = self._get_retry_temp(temperature)
+                    try:
+                        if resp_format == "vllm":
+                            parsed, completion = await self._vllm_completion(
+                                messages,
+                                output_model,
+                                retry_temperature,
+                                logprobs,
+                                top_logprobs,
+                            )
+                        elif resp_format == "parse":
+                            parsed, completion = await self._parse_completion(
+                                messages,
+                                output_model,
+                                retry_temperature,
+                                logprobs,
+                                top_logprobs,
+                            )
+
+                        output.result = parsed.result
+
+                        # Check if retry was successful
+                        if validator(output.result):
+                            logger.info(
+                                f"Validation passed on retry attempt {attempt + 1}"
+                            )
+                            break
+                        else:
+                            logger.warning(
+                                f"Validation still failing after retry attempt {attempt + 1}"
+                            )
+
+                    except Exception as e:
+                        logger.error(f"Retry attempt {attempt + 1} failed: {e}")
+                        # Continue to next retry attempt if this one fails
+
+            # Final check after all retries
+            if validator and not validator(output.result):
+                output.errors.append("Validation failed after all retry attempts")
+
             if logprobs:
                 output.logprobs = self._extract_logprobs(completion)
 

texttools/tools/internals/base_operator.py

@@ -3,6 +3,7 @@ import json
 import re
 import math
 import logging
+import random
 
 from pydantic import BaseModel
 from openai import OpenAI, AsyncOpenAI
@@ -86,3 +87,15 @@ class BaseOperator:
                 logprobs_data.append(token_entry)
 
         return logprobs_data
+
+    def _get_retry_temp(self, base_temp: float) -> float:
+        """
+        Calculate temperature for retry attempts.
+        """
+        delta_temp = random.choice([-1, 1]) * random.uniform(0.1, 0.9)
+        new_temp = base_temp + delta_temp
+        print(f"Base Temp: {base_temp}")
+        print(f"Delta Temp: {delta_temp}")
+        print(f"New Temp: {new_temp}")
+
+        return max(0.0, min(new_temp, 1.5))

texttools/tools/internals/operator.py

@@ -1,4 +1,4 @@
-from typing import Any, TypeVar, Type, Literal
+from typing import Any, TypeVar, Type, Literal, Callable
 import logging
 
 from openai import OpenAI
@@ -115,6 +115,7 @@ class Operator(BaseOperator):
         temperature: float,
         logprobs: bool,
         top_logprobs: int | None,
+        validator: Callable[[Any], bool] | None,
         # Internal parameters
         prompt_file: str,
         output_model: Type[T],
@@ -179,6 +180,55 @@ class Operator(BaseOperator):
 
             output.result = parsed.result
 
+            # Retry logic if validation fails
+            if validator and not validator(output.result):
+                max_retries = 3
+                for attempt in range(max_retries):
+                    logger.warning(
+                        f"Validation failed, retrying for the {attempt + 1} time."
+                    )
+
+                    # Generate new temperature for retry
+                    retry_temperature = self._get_retry_temp(temperature)
+                    try:
+                        if resp_format == "vllm":
+                            parsed, completion = self._vllm_completion(
+                                messages,
+                                output_model,
+                                retry_temperature,
+                                logprobs,
+                                top_logprobs,
+                            )
+                        elif resp_format == "parse":
+                            parsed, completion = self._parse_completion(
+                                messages,
+                                output_model,
+                                retry_temperature,
+                                logprobs,
+                                top_logprobs,
+                            )
+
+                        output.result = parsed.result
+
+                        # Check if retry was successful
+                        if validator(output.result):
+                            logger.info(
+                                f"Validation passed on retry attempt {attempt + 1}"
+                            )
+                            break
+                        else:
+                            logger.warning(
+                                f"Validation still failing after retry attempt {attempt + 1}"
+                            )
+
+                    except Exception as e:
+                        logger.error(f"Retry attempt {attempt + 1} failed: {e}")
+                        # Continue to next retry attempt if this one fails
+
+            # Final check after all retries
+            if validator and not validator(output.result):
+                output.errors.append("Validation failed after all retry attempts")
+
             if logprobs:
                 output.logprobs = self._extract_logprobs(completion)
 
@@ -188,5 +238,5 @@ class Operator(BaseOperator):
             return output
 
         except Exception as e:
-            logger.error(f"AsyncTheTool failed: {e}")
+            logger.error(f"TheTool failed: {e}")
             return output.errors.append(str(e))

texttools/tools/the_tool.py

@@ -1,4 +1,4 @@
-from typing import Literal, Any
+from typing import Literal, Any, Callable
 
 from openai import OpenAI
 
@@ -32,6 +32,7 @@ class TheTool:
         temperature: float | None = 0.0,
         logprobs: bool = False,
         top_logprobs: int | None = None,
+        validator: Callable[[Any], bool] | None = None,
     ) -> OutputModels.ToolOutput:
         """
         Categorize a text into a single Islamic studies domain category.
@@ -50,6 +51,7 @@ class TheTool:
             temperature=temperature,
             logprobs=logprobs,
             top_logprobs=top_logprobs,
+            validator=validator,
             # Internal parameters
             prompt_file="categorizer.yaml",
             output_model=OutputModels.CategorizerOutput,
@@ -67,6 +69,7 @@ class TheTool:
         temperature: float | None = 0.0,
         logprobs: bool = False,
         top_logprobs: int | None = None,
+        validator: Callable[[Any], bool] | None = None,
     ) -> OutputModels.ToolOutput:
         """
         Extract salient keywords from text.
@@ -86,6 +89,7 @@ class TheTool:
             temperature=temperature,
             logprobs=logprobs,
             top_logprobs=top_logprobs,
+            validator=validator,
             # Internal parameters
             prompt_file="extract_keywords.yaml",
             output_model=OutputModels.ListStrOutput,
@@ -102,6 +106,7 @@ class TheTool:
         temperature: float | None = 0.0,
         logprobs: bool = False,
         top_logprobs: int | None = None,
+        validator: Callable[[Any], bool] | None = None,
     ) -> OutputModels.ToolOutput:
         """
         Perform Named Entity Recognition (NER) over the input text.
@@ -121,6 +126,7 @@ class TheTool:
             temperature=temperature,
             logprobs=logprobs,
             top_logprobs=top_logprobs,
+            validator=validator,
             # Internal parameters
             prompt_file="extract_entities.yaml",
             output_model=OutputModels.ListDictStrStrOutput,
@@ -136,6 +142,7 @@ class TheTool:
         temperature: float | None = 0.0,
         logprobs: bool = False,
         top_logprobs: int | None = None,
+        validator: Callable[[Any], bool] | None = None,
     ) -> OutputModels.ToolOutput:
         """
         Detect if the input is phrased as a question.
@@ -154,6 +161,7 @@ class TheTool:
             temperature=temperature,
             logprobs=logprobs,
             top_logprobs=top_logprobs,
+            validator=validator,
             # Internal parameters
             prompt_file="is_question.yaml",
             output_model=OutputModels.BoolOutput,
@@ -171,6 +179,7 @@ class TheTool:
         temperature: float | None = 0.0,
         logprobs: bool = False,
         top_logprobs: int | None = None,
+        validator: Callable[[Any], bool] | None = None,
     ) -> OutputModels.ToolOutput:
         """
         Generate a single question from the given text.
@@ -190,6 +199,7 @@ class TheTool:
             temperature=temperature,
             logprobs=logprobs,
             top_logprobs=top_logprobs,
+            validator=validator,
             # Internal parameters
             prompt_file="text_to_question.yaml",
             output_model=OutputModels.StrOutput,
@@ -207,6 +217,7 @@ class TheTool:
         logprobs: bool = False,
         top_logprobs: int | None = None,
         mode: Literal["default", "reason"] = "default",
+        validator: Callable[[Any], bool] | None = None,
     ) -> OutputModels.ToolOutput:
         """
         Merge multiple questions into a single unified question.
@@ -227,6 +238,7 @@ class TheTool:
             temperature=temperature,
             logprobs=logprobs,
             top_logprobs=top_logprobs,
+            validator=validator,
             # Internal parameters
             prompt_file="merge_questions.yaml",
             output_model=OutputModels.StrOutput,
@@ -244,6 +256,7 @@ class TheTool:
         logprobs: bool = False,
         top_logprobs: int | None = None,
         mode: Literal["positive", "negative", "hard_negative"] = "positive",
+        validator: Callable[[Any], bool] | None = None,
     ) -> OutputModels.ToolOutput:
         """
         Rewrite a text with different modes.
@@ -263,6 +276,7 @@ class TheTool:
             temperature=temperature,
             logprobs=logprobs,
             top_logprobs=top_logprobs,
+            validator=validator,
             # Internal parameters
             prompt_file="rewrite.yaml",
             output_model=OutputModels.StrOutput,
@@ -280,6 +294,7 @@ class TheTool:
         temperature: float | None = 0.0,
         logprobs: bool = False,
         top_logprobs: int | None = None,
+        validator: Callable[[Any], bool] | None = None,
     ) -> OutputModels.ToolOutput:
         """
         Generate a list of questions about a subject.
@@ -300,6 +315,7 @@ class TheTool:
             temperature=temperature,
             logprobs=logprobs,
             top_logprobs=top_logprobs,
+            validator=validator,
             # Internal parameters
             prompt_file="subject_to_question.yaml",
             output_model=OutputModels.ReasonListStrOutput,
@@ -316,6 +332,7 @@ class TheTool:
         temperature: float | None = 0.0,
         logprobs: bool = False,
         top_logprobs: int | None = None,
+        validator: Callable[[Any], bool] | None = None,
     ) -> OutputModels.ToolOutput:
         """
         Summarize the given subject text.
@@ -335,6 +352,7 @@ class TheTool:
             temperature=temperature,
             logprobs=logprobs,
             top_logprobs=top_logprobs,
+            validator=validator,
             # Internal parameters
             prompt_file="summarize.yaml",
             output_model=OutputModels.StrOutput,
@@ -351,6 +369,7 @@ class TheTool:
         temperature: float | None = 0.0,
         logprobs: bool = False,
         top_logprobs: int | None = None,
+        validator: Callable[[Any], bool] | None = None,
     ) -> OutputModels.ToolOutput:
         """
         Translate text between languages.
@@ -370,6 +389,7 @@ class TheTool:
             temperature=temperature,
             logprobs=logprobs,
             top_logprobs=top_logprobs,
+            validator=validator,
             # Internal parameters
             prompt_file="translate.yaml",
             output_model=OutputModels.StrOutput,
@@ -409,4 +429,5 @@ class TheTool:
             user_prompt=None,
             with_analysis=False,
             mode=None,
+            validator=None,
         )