hamtaa-texttools 1.1.9__py3-none-any.whl → 1.1.10__py3-none-any.whl

texttools/tools/sync_tools.py

@@ -3,7 +3,7 @@ from typing import Literal, Any, Callable
 from openai import OpenAI
 
 from texttools.tools.internals.operator import Operator
-import texttools.tools.internals.output_models as OutputModels
+import texttools.tools.internals.output_models as OM
 
 
 class TheTool:
@@ -22,7 +22,7 @@ class TheTool:
         client: OpenAI,
         model: str,
     ):
-        self.operator = Operator(client=client, model=model)
+        self._operator = Operator(client=client, model=model)
 
     def categorize(
         self,
@@ -33,7 +33,7 @@ class TheTool:
         logprobs: bool = False,
         top_logprobs: int | None = None,
         validator: Callable[[Any], bool] | None = None,
-    ) -> OutputModels.ToolOutput:
+    ) -> OM.ToolOutput:
         """
         Categorize a text into a single Islamic studies domain category.
 
@@ -43,7 +43,7 @@ class TheTool:
                 - logprobs (list | None): Probability data if logprobs enabled
                 - analysis (str | None): Detailed reasoning if with_analysis enabled
         """
-        return self.operator.run(
+        return self._operator.run(
             # User parameters
             text=text,
             with_analysis=with_analysis,
@@ -54,8 +54,7 @@ class TheTool:
             validator=validator,
             # Internal parameters
             prompt_file="categorizer.yaml",
-            output_model=OutputModels.CategorizerOutput,
-            resp_format="parse",
+            output_model=OM.CategorizerOutput,
             mode=None,
             output_lang=None,
         )
@@ -70,7 +69,7 @@ class TheTool:
         logprobs: bool = False,
         top_logprobs: int | None = None,
         validator: Callable[[Any], bool] | None = None,
-    ) -> OutputModels.ToolOutput:
+    ) -> OM.ToolOutput:
         """
         Extract salient keywords from text.
 
@@ -80,7 +79,7 @@ class TheTool:
                 - logprobs (list | None): Probability data if logprobs enabled
                 - analysis (str | None): Detailed reasoning if with_analysis enabled
         """
-        return self.operator.run(
+        return self._operator.run(
             # User parameters
             text=text,
             with_analysis=with_analysis,
@@ -92,8 +91,7 @@ class TheTool:
             validator=validator,
             # Internal parameters
             prompt_file="extract_keywords.yaml",
-            output_model=OutputModels.ListStrOutput,
-            resp_format="parse",
+            output_model=OM.ListStrOutput,
             mode=None,
         )
 
@@ -107,7 +105,7 @@ class TheTool:
         logprobs: bool = False,
         top_logprobs: int | None = None,
         validator: Callable[[Any], bool] | None = None,
-    ) -> OutputModels.ToolOutput:
+    ) -> OM.ToolOutput:
         """
         Perform Named Entity Recognition (NER) over the input text.
 
@@ -117,7 +115,7 @@ class TheTool:
                 - logprobs (list | None): Probability data if logprobs enabled
                 - analysis (str | None): Detailed reasoning if with_analysis enabled
         """
-        return self.operator.run(
+        return self._operator.run(
             # User parameters
             text=text,
             with_analysis=with_analysis,
@@ -129,8 +127,7 @@ class TheTool:
             validator=validator,
             # Internal parameters
             prompt_file="extract_entities.yaml",
-            output_model=OutputModels.ListDictStrStrOutput,
-            resp_format="parse",
+            output_model=OM.ListDictStrStrOutput,
             mode=None,
         )
 
@@ -143,7 +140,7 @@ class TheTool:
         logprobs: bool = False,
         top_logprobs: int | None = None,
         validator: Callable[[Any], bool] | None = None,
-    ) -> OutputModels.ToolOutput:
+    ) -> OM.ToolOutput:
         """
         Detect if the input is phrased as a question.
 
@@ -153,7 +150,7 @@ class TheTool:
                 - logprobs (list | None): Probability data if logprobs enabled
                 - analysis (str | None): Detailed reasoning if with_analysis enabled
         """
-        return self.operator.run(
+        return self._operator.run(
             # User parameters
             text=text,
             with_analysis=with_analysis,
@@ -164,8 +161,7 @@ class TheTool:
             validator=validator,
             # Internal parameters
             prompt_file="is_question.yaml",
-            output_model=OutputModels.BoolOutput,
-            resp_format="parse",
+            output_model=OM.BoolOutput,
             mode=None,
             output_lang=None,
         )
@@ -180,7 +176,7 @@ class TheTool:
         logprobs: bool = False,
         top_logprobs: int | None = None,
         validator: Callable[[Any], bool] | None = None,
-    ) -> OutputModels.ToolOutput:
+    ) -> OM.ToolOutput:
         """
         Generate a single question from the given text.
 
@@ -190,7 +186,7 @@ class TheTool:
                 - logprobs (list | None): Probability data if logprobs enabled
                 - analysis (str | None): Detailed reasoning if with_analysis enabled
         """
-        return self.operator.run(
+        return self._operator.run(
             # User parameters
             text=text,
             with_analysis=with_analysis,
@@ -202,8 +198,7 @@ class TheTool:
             validator=validator,
             # Internal parameters
             prompt_file="text_to_question.yaml",
-            output_model=OutputModels.StrOutput,
-            resp_format="parse",
+            output_model=OM.StrOutput,
             mode=None,
         )
 
@@ -218,7 +213,7 @@ class TheTool:
         top_logprobs: int | None = None,
         mode: Literal["default", "reason"] = "default",
         validator: Callable[[Any], bool] | None = None,
-    ) -> OutputModels.ToolOutput:
+    ) -> OM.ToolOutput:
         """
         Merge multiple questions into a single unified question.
 
@@ -229,7 +224,7 @@ class TheTool:
                 - analysis (str | None): Detailed reasoning if with_analysis enabled
         """
         text = ", ".join(text)
-        return self.operator.run(
+        return self._operator.run(
             # User parameters
             text=text,
             with_analysis=with_analysis,
@@ -241,8 +236,7 @@ class TheTool:
             validator=validator,
             # Internal parameters
             prompt_file="merge_questions.yaml",
-            output_model=OutputModels.StrOutput,
-            resp_format="parse",
+            output_model=OM.StrOutput,
             mode=mode,
         )
 
@@ -257,7 +251,7 @@ class TheTool:
         top_logprobs: int | None = None,
         mode: Literal["positive", "negative", "hard_negative"] = "positive",
         validator: Callable[[Any], bool] | None = None,
-    ) -> OutputModels.ToolOutput:
+    ) -> OM.ToolOutput:
         """
         Rewrite a text with different modes.
 
@@ -267,7 +261,7 @@ class TheTool:
                 - logprobs (list | None): Probability data if logprobs enabled
                 - analysis (str | None): Detailed reasoning if with_analysis enabled
         """
-        return self.operator.run(
+        return self._operator.run(
             # User parameters
             text=text,
             with_analysis=with_analysis,
@@ -279,8 +273,7 @@ class TheTool:
             validator=validator,
             # Internal parameters
             prompt_file="rewrite.yaml",
-            output_model=OutputModels.StrOutput,
-            resp_format="parse",
+            output_model=OM.StrOutput,
             mode=mode,
         )
 
@@ -295,7 +288,7 @@ class TheTool:
         logprobs: bool = False,
         top_logprobs: int | None = None,
         validator: Callable[[Any], bool] | None = None,
-    ) -> OutputModels.ToolOutput:
+    ) -> OM.ToolOutput:
         """
         Generate a list of questions about a subject.
 
@@ -305,7 +298,7 @@ class TheTool:
                 - logprobs (list | None): Probability data if logprobs enabled
                 - analysis (str | None): Detailed reasoning if with_analysis enabled
         """
-        return self.operator.run(
+        return self._operator.run(
             # User parameters
             text=text,
             number_of_questions=number_of_questions,
@@ -318,8 +311,7 @@ class TheTool:
             validator=validator,
             # Internal parameters
             prompt_file="subject_to_question.yaml",
-            output_model=OutputModels.ReasonListStrOutput,
-            resp_format="parse",
+            output_model=OM.ReasonListStrOutput,
             mode=None,
         )
 
@@ -333,7 +325,7 @@ class TheTool:
         logprobs: bool = False,
         top_logprobs: int | None = None,
         validator: Callable[[Any], bool] | None = None,
-    ) -> OutputModels.ToolOutput:
+    ) -> OM.ToolOutput:
         """
         Summarize the given subject text.
 
@@ -343,7 +335,7 @@ class TheTool:
                 - logprobs (list | None): Probability data if logprobs enabled
                 - analysis (str | None): Detailed reasoning if with_analysis enabled
         """
-        return self.operator.run(
+        return self._operator.run(
             # User parameters
             text=text,
             with_analysis=with_analysis,
@@ -355,8 +347,7 @@ class TheTool:
             validator=validator,
             # Internal parameters
             prompt_file="summarize.yaml",
-            output_model=OutputModels.StrOutput,
-            resp_format="parse",
+            output_model=OM.StrOutput,
             mode=None,
         )
 
@@ -370,7 +361,7 @@ class TheTool:
         logprobs: bool = False,
         top_logprobs: int | None = None,
         validator: Callable[[Any], bool] | None = None,
-    ) -> OutputModels.ToolOutput:
+    ) -> OM.ToolOutput:
         """
         Translate text between languages.
 
@@ -380,7 +371,7 @@ class TheTool:
                 - logprobs (list | None): Probability data if logprobs enabled
                 - analysis (str | None): Detailed reasoning if with_analysis enabled
         """
-        return self.operator.run(
+        return self._operator.run(
             # User parameters
             text=text,
             target_language=target_language,
@@ -392,8 +383,7 @@ class TheTool:
             validator=validator,
             # Internal parameters
             prompt_file="translate.yaml",
-            output_model=OutputModels.StrOutput,
-            resp_format="parse",
+            output_model=OM.StrOutput,
             mode=None,
             output_lang=None,
         )
@@ -406,7 +396,7 @@ class TheTool:
         temperature: float | None = None,
         logprobs: bool | None = None,
         top_logprobs: int | None = None,
-    ) -> OutputModels.ToolOutput:
+    ) -> OM.ToolOutput:
         """
         Custom tool that can do almost anything!
 
@@ -414,7 +404,7 @@ class TheTool:
             ToolOutput: Object with fields:
                 - result (str): The output result
         """
-        return self.operator.run(
+        return self._operator.run(
             # User paramaeters
             text=prompt,
             output_model=output_model,
@@ -425,7 +415,6 @@ class TheTool:
             top_logprobs=top_logprobs,
             # Internal parameters
             prompt_file="run_custom.yaml",
-            resp_format="parse",
             user_prompt=None,
             with_analysis=False,
             mode=None,

hamtaa_texttools-1.1.9.dist-info/RECORD

@@ -1,30 +0,0 @@
-hamtaa_texttools-1.1.9.dist-info/licenses/LICENSE,sha256=Hb2YOBKy2MJQLnyLrX37B4ZVuac8eaIcE71SvVIMOLg,1082
-texttools/__init__.py,sha256=lFYe1jdssHC1h8qcPpV3whANxiDi8aiiFdY-7L0Ck10,164
-texttools/batch/__init__.py,sha256=DJGJTfR6F3Yv4_alsj9g1tesGzdcSV27Zw74DonhW_s,102
-texttools/batch/batch_manager.py,sha256=ZgLiO9maCHnx2cJbUjsYXFnlUsMLI2TP3Vc9uKU0BLg,8706
-texttools/batch/batch_runner.py,sha256=X0YQmaowO_jUSAFWBHdxOLoRrX_gvmrJDgp9qPlOSEw,10254
-texttools/prompts/README.md,sha256=-5YO93CN93QLifqZpUeUnCOCBbDiOTV-cFQeJ7Gg0I4,1377
-texttools/prompts/categorizer.yaml,sha256=GMqIIzQFhgnlpkgU1qi3FAD3mD4A2jiWD5TilQ2XnnE,1204
-texttools/prompts/extract_entities.yaml,sha256=KiKjeDpHaeh3JVtZ6q1pa3k4DYucUIU9WnEcRTCA-SE,651
-texttools/prompts/extract_keywords.yaml,sha256=0O7ypL_OsEOxtvlQ2CZjnsv9637DJwAKprZsf9Vo2_s,769
-texttools/prompts/is_question.yaml,sha256=d0-vKRbXWkxvO64ikvxRjEmpAXGpCYIPGhgexvPPjws,471
-texttools/prompts/merge_questions.yaml,sha256=0J85GvTirZB4ELwH3sk8ub_WcqqpYf6PrMKr3djlZeo,1792
-texttools/prompts/rewrite.yaml,sha256=LO7He_IA3MZKz8a-LxH9DHJpOjpYwaYN1pbjp1Y0tFo,5392
-texttools/prompts/run_custom.yaml,sha256=38OkCoVITbuuS9c08UZSP1jZW4WjSmRIi8fR0RAiPu4,108
-texttools/prompts/subject_to_question.yaml,sha256=C7x7rNNm6U_ZG9HOn6zuzYOtvJUZ2skuWbL1-aYdd3E,1147
-texttools/prompts/summarize.yaml,sha256=o6rxGPfWtZd61Duvm8NVvCJqfq73b-wAuMSKR6UYUqY,459
-texttools/prompts/text_to_question.yaml,sha256=UheKYpDn6iyKI8NxunHZtFpNyfCLZZe5cvkuXpurUJY,783
-texttools/prompts/translate.yaml,sha256=mGT2uBCei6uucWqVbs4silk-UV060v3G0jnt0P6sr50,634
-texttools/tools/__init__.py,sha256=3fPoeB-E5wGxWgv7axztHkeolR7ZDUJudd0xmpPFjao,113
-texttools/tools/async_tools.py,sha256=2ZY7Lo6Jj9xoTF8bfdh_g8VOXZ7ljMMesd1_QHXyf4s,15395
-texttools/tools/sync_tools.py,sha256=XKgZuzriFnk8B-YihJfs6BKivxjGCgOFfe7hnCpEiXs,15161
-texttools/tools/internals/async_operator.py,sha256=egBsrcpGBmkDY5YzUvGHh1TjPmsH9IOVXDGmYMWjzMs,8960
-texttools/tools/internals/base_operator.py,sha256=qV9LlVo_DzSCzQnjYTFi-6mlHN4gE0edPE2y_9WwQFw,3292
-texttools/tools/internals/formatters.py,sha256=tACNLP6PeoqaRpNudVxBaHA25zyWqWYPZQuYysIu88g,941
-texttools/tools/internals/operator.py,sha256=xgbt1Mm67SEC-KD9jwXjXGTCcaCsaVLhG6iCYOqLDcc,8709
-texttools/tools/internals/output_models.py,sha256=ekpbyocmXj_dee7ieOT1zOkMo9cPHT7xcUFCZoUaXA0,1886
-texttools/tools/internals/prompt_loader.py,sha256=8uD7JUatKXSLXhGwWs46iQpcjWdhF9p32SFDLMndy1o,1940
-hamtaa_texttools-1.1.9.dist-info/METADATA,sha256=nQFuGr_7aVHlO7nRsTbubEtO0QVUofcdUKwMATzHhUU,9129
-hamtaa_texttools-1.1.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-hamtaa_texttools-1.1.9.dist-info/top_level.txt,sha256=5Mh0jIxxZ5rOXHGJ6Mp-JPKviywwN0MYuH0xk5bEWqE,10
-hamtaa_texttools-1.1.9.dist-info/RECORD,,

texttools/batch/__init__.py

@@ -1,3 +0,0 @@
-from .batch_runner import BatchJobRunner, BatchConfig
-
-__all__ = ["BatchJobRunner", "BatchConfig"]

texttools/tools/__init__.py

@@ -1,4 +0,0 @@
-from .sync_tools import TheTool
-from .async_tools import AsyncTheTool
-
-__all__ = ["TheTool", "AsyncTheTool"]