hamtaa-texttools 1.1.19__tar.gz → 1.1.21__tar.gz

{hamtaa_texttools-1.1.19/hamtaa_texttools.egg-info → hamtaa_texttools-1.1.21}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hamtaa-texttools
-Version: 1.1.19
+Version: 1.1.21
 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>, Zareshahi <a.zareshahi1377@gmail.com>
 License: MIT License
@@ -50,7 +50,7 @@ It provides ready-to-use utilities for **translation, question detection, keywor
 TextTools provides a rich collection of high-level NLP utilities,
 Each tool is designed to work with structured outputs (JSON / Pydantic).
 
-- **`categorize()`** - Classifies text into given categories (You have to create a category tree)
+- **`categorize()`** - Classifies text into given categories
 - **`extract_keywords()`** - Extracts keywords from text
 - **`extract_entities()`** - Named Entity Recognition (NER) system
 - **`is_question()`** - Binary detection of whether input is a question
@@ -61,7 +61,7 @@ Each tool is designed to work with structured outputs (JSON / Pydantic).
 - **`summarize()`** - Text summarization
 - **`translate()`** - Text translation between languages
 - **`propositionize()`** - Convert text to atomic independence meaningful sentences 
-- **`check_fact()`** - Check a statement is relevant to source text or not 
+- **`check_fact()`** - Check whether a statement is relevant to the source text
 - **`run_custom()`** - Allows users to define a custom tool with an arbitrary BaseModel
 
 ---
@@ -99,21 +99,21 @@ Each tool is designed to work with structured outputs (JSON / Pydantic).
 
 TextTools provides several optional flags to customize LLM behavior:
 
-- **`with_analysis (bool)`** → Adds a reasoning step before generating the final output.
+- **`with_analysis: bool`** → Adds a reasoning step before generating the final output.
 **Note:** This doubles token usage per call because it triggers an additional LLM request.
 
-- **`logprobs (bool)`** → Returns token-level probabilities for the generated output. You can also specify `top_logprobs=<N>` to get the top N alternative tokens and their probabilities.  
+- **`logprobs: bool`** → Returns token-level probabilities for the generated output. You can also specify `top_logprobs=<N>` to get the top N alternative tokens and their probabilities.  
 **Note:** This feature works if it's supported by the model.
 
-- **`output_lang (str)`** → Forces the model to respond in a specific language. The model will ignore other instructions about language and respond strictly in the requested language.
+- **`output_lang: str`** → Forces the model to respond in a specific language. The model will ignore other instructions about language and respond strictly in the requested language.
 
-- **`user_prompt (str)`** → Allows you to inject a custom instruction or prompt into the model alongside the main template. This gives you fine-grained control over how the model interprets or modifies the input text.
+- **`user_prompt: str`** → Allows you to inject a custom instruction or prompt into the model alongside the main template. This gives you fine-grained control over how the model interprets or modifies the input text.
 
-- **`temperature (float)`** → Determines how creative the model should respond. Takes a float number from `0.0` to `2.0`.
+- **`temperature: float`** → Determines how creative the model should respond. Takes a float number from `0.0` to `2.0`.
 
-- **`validator (Callable)`** → Forces TheTool to validate the output result based on your custom validator. Validator should return a bool (True if there were no problem, False if the validation fails.) If the validator fails, TheTool will retry to get another output by modifying `temperature`. You can specify `max_validation_retries=<N>` to change the number of retries.
+- **`validator: Callable (Experimental)`** → Forces TheTool to validate the output result based on your custom validator. Validator should return a bool (True if there were no problem, False if the validation fails.) If the validator fails, TheTool will retry to get another output by modifying `temperature`. You can specify `max_validation_retries=<N>` to change the number of retries.
 
-- **`priority (int)`** → Task execution priority level. Higher values = higher priority. Affects processing order in queues.
+- **`priority: int (Experimental)`** → Task execution priority level. Higher values = higher priority. Affects processing order in queues.
 **Note:** This feature works if it's supported by the model and vLLM.
 
 **Note:** There might be some tools that don't support some of the parameters above.
@@ -125,11 +125,12 @@ TextTools provides several optional flags to customize LLM behavior:
 Every tool of `TextTools` returns a `ToolOutput` object which is a BaseModel with attributes:
 - **`result: Any`** → The output of LLM
 - **`analysis: str`** → The reasoning step before generating the final output
-- **`logprobs: list`** → Token-level probabilities for the generated output 
-- **`process: str`** → The tool name which processed the input
-- **`processed_at: datetime`** → The process time
-- **`execution_time: float`** → The execution time (seconds)
+- **`logprobs: list`** → Token-level probabilities for the generated output
 - **`errors: list[str]`** → Any error that have occured during calling LLM
+- **`ToolOutputMetadata`** →
+    - **`tool_name: str`** → The tool name which processed the input
+    - **`processed_at: datetime`** → The process time
+    - **`execution_time: float`** → The execution time (seconds)
 
 **Note:** You can use `repr(ToolOutput)` to see details of your ToolOutput.
 
@@ -224,26 +225,6 @@ Use **TextTools** when you need to:
 
 ---
 
-## 🔍 Logging
-
-TextTools uses Python's standard `logging` module. The library's default logger level is `WARNING`, so if you want to modify it, follow instructions:
-
-
-```python
-import logging
-
-# Default: warnings and errors only
-logging.basicConfig(level=logging.WARNING)
-
-# Debug everything (verbose)
-logging.basicConfig(level=logging.DEBUG)
-
-# Complete silence
-logging.basicConfig(level=logging.CRITICAL)
-```
-
----
-
 ## 📚 Batch Processing
 
 Process large datasets efficiently using OpenAI's batch API.

{hamtaa_texttools-1.1.19 → hamtaa_texttools-1.1.21}/README.md

@@ -15,7 +15,7 @@ It provides ready-to-use utilities for **translation, question detection, keywor
 TextTools provides a rich collection of high-level NLP utilities,
 Each tool is designed to work with structured outputs (JSON / Pydantic).
 
-- **`categorize()`** - Classifies text into given categories (You have to create a category tree)
+- **`categorize()`** - Classifies text into given categories
 - **`extract_keywords()`** - Extracts keywords from text
 - **`extract_entities()`** - Named Entity Recognition (NER) system
 - **`is_question()`** - Binary detection of whether input is a question
@@ -26,7 +26,7 @@ Each tool is designed to work with structured outputs (JSON / Pydantic).
 - **`summarize()`** - Text summarization
 - **`translate()`** - Text translation between languages
 - **`propositionize()`** - Convert text to atomic independence meaningful sentences 
-- **`check_fact()`** - Check a statement is relevant to source text or not 
+- **`check_fact()`** - Check whether a statement is relevant to the source text
 - **`run_custom()`** - Allows users to define a custom tool with an arbitrary BaseModel
 
 ---
@@ -64,21 +64,21 @@ Each tool is designed to work with structured outputs (JSON / Pydantic).
 
 TextTools provides several optional flags to customize LLM behavior:
 
-- **`with_analysis (bool)`** → Adds a reasoning step before generating the final output.
+- **`with_analysis: bool`** → Adds a reasoning step before generating the final output.
 **Note:** This doubles token usage per call because it triggers an additional LLM request.
 
-- **`logprobs (bool)`** → Returns token-level probabilities for the generated output. You can also specify `top_logprobs=<N>` to get the top N alternative tokens and their probabilities.  
+- **`logprobs: bool`** → Returns token-level probabilities for the generated output. You can also specify `top_logprobs=<N>` to get the top N alternative tokens and their probabilities.  
 **Note:** This feature works if it's supported by the model.
 
-- **`output_lang (str)`** → Forces the model to respond in a specific language. The model will ignore other instructions about language and respond strictly in the requested language.
+- **`output_lang: str`** → Forces the model to respond in a specific language. The model will ignore other instructions about language and respond strictly in the requested language.
 
-- **`user_prompt (str)`** → Allows you to inject a custom instruction or prompt into the model alongside the main template. This gives you fine-grained control over how the model interprets or modifies the input text.
+- **`user_prompt: str`** → Allows you to inject a custom instruction or prompt into the model alongside the main template. This gives you fine-grained control over how the model interprets or modifies the input text.
 
-- **`temperature (float)`** → Determines how creative the model should respond. Takes a float number from `0.0` to `2.0`.
+- **`temperature: float`** → Determines how creative the model should respond. Takes a float number from `0.0` to `2.0`.
 
-- **`validator (Callable)`** → Forces TheTool to validate the output result based on your custom validator. Validator should return a bool (True if there were no problem, False if the validation fails.) If the validator fails, TheTool will retry to get another output by modifying `temperature`. You can specify `max_validation_retries=<N>` to change the number of retries.
+- **`validator: Callable (Experimental)`** → Forces TheTool to validate the output result based on your custom validator. Validator should return a bool (True if there were no problem, False if the validation fails.) If the validator fails, TheTool will retry to get another output by modifying `temperature`. You can specify `max_validation_retries=<N>` to change the number of retries.
 
-- **`priority (int)`** → Task execution priority level. Higher values = higher priority. Affects processing order in queues.
+- **`priority: int (Experimental)`** → Task execution priority level. Higher values = higher priority. Affects processing order in queues.
 **Note:** This feature works if it's supported by the model and vLLM.
 
 **Note:** There might be some tools that don't support some of the parameters above.
@@ -90,11 +90,12 @@ TextTools provides several optional flags to customize LLM behavior:
 Every tool of `TextTools` returns a `ToolOutput` object which is a BaseModel with attributes:
 - **`result: Any`** → The output of LLM
 - **`analysis: str`** → The reasoning step before generating the final output
-- **`logprobs: list`** → Token-level probabilities for the generated output 
-- **`process: str`** → The tool name which processed the input
-- **`processed_at: datetime`** → The process time
-- **`execution_time: float`** → The execution time (seconds)
+- **`logprobs: list`** → Token-level probabilities for the generated output
 - **`errors: list[str]`** → Any error that have occured during calling LLM
+- **`ToolOutputMetadata`** →
+    - **`tool_name: str`** → The tool name which processed the input
+    - **`processed_at: datetime`** → The process time
+    - **`execution_time: float`** → The execution time (seconds)
 
 **Note:** You can use `repr(ToolOutput)` to see details of your ToolOutput.
 
@@ -189,26 +190,6 @@ Use **TextTools** when you need to:
 
 ---
 
-## 🔍 Logging
-
-TextTools uses Python's standard `logging` module. The library's default logger level is `WARNING`, so if you want to modify it, follow instructions:
-
-
-```python
-import logging
-
-# Default: warnings and errors only
-logging.basicConfig(level=logging.WARNING)
-
-# Debug everything (verbose)
-logging.basicConfig(level=logging.DEBUG)
-
-# Complete silence
-logging.basicConfig(level=logging.CRITICAL)
-```
-
----
-
 ## 📚 Batch Processing
 
 Process large datasets efficiently using OpenAI's batch API.

{hamtaa_texttools-1.1.19 → hamtaa_texttools-1.1.21/hamtaa_texttools.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hamtaa-texttools
-Version: 1.1.19
+Version: 1.1.21
 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>, Zareshahi <a.zareshahi1377@gmail.com>
 License: MIT License
@@ -50,7 +50,7 @@ It provides ready-to-use utilities for **translation, question detection, keywor
 TextTools provides a rich collection of high-level NLP utilities,
 Each tool is designed to work with structured outputs (JSON / Pydantic).
 
-- **`categorize()`** - Classifies text into given categories (You have to create a category tree)
+- **`categorize()`** - Classifies text into given categories
 - **`extract_keywords()`** - Extracts keywords from text
 - **`extract_entities()`** - Named Entity Recognition (NER) system
 - **`is_question()`** - Binary detection of whether input is a question
@@ -61,7 +61,7 @@ Each tool is designed to work with structured outputs (JSON / Pydantic).
 - **`summarize()`** - Text summarization
 - **`translate()`** - Text translation between languages
 - **`propositionize()`** - Convert text to atomic independence meaningful sentences 
-- **`check_fact()`** - Check a statement is relevant to source text or not 
+- **`check_fact()`** - Check whether a statement is relevant to the source text
 - **`run_custom()`** - Allows users to define a custom tool with an arbitrary BaseModel
 
 ---
@@ -99,21 +99,21 @@ Each tool is designed to work with structured outputs (JSON / Pydantic).
 
 TextTools provides several optional flags to customize LLM behavior:
 
-- **`with_analysis (bool)`** → Adds a reasoning step before generating the final output.
+- **`with_analysis: bool`** → Adds a reasoning step before generating the final output.
 **Note:** This doubles token usage per call because it triggers an additional LLM request.
 
-- **`logprobs (bool)`** → Returns token-level probabilities for the generated output. You can also specify `top_logprobs=<N>` to get the top N alternative tokens and their probabilities.  
+- **`logprobs: bool`** → Returns token-level probabilities for the generated output. You can also specify `top_logprobs=<N>` to get the top N alternative tokens and their probabilities.  
 **Note:** This feature works if it's supported by the model.
 
-- **`output_lang (str)`** → Forces the model to respond in a specific language. The model will ignore other instructions about language and respond strictly in the requested language.
+- **`output_lang: str`** → Forces the model to respond in a specific language. The model will ignore other instructions about language and respond strictly in the requested language.
 
-- **`user_prompt (str)`** → Allows you to inject a custom instruction or prompt into the model alongside the main template. This gives you fine-grained control over how the model interprets or modifies the input text.
+- **`user_prompt: str`** → Allows you to inject a custom instruction or prompt into the model alongside the main template. This gives you fine-grained control over how the model interprets or modifies the input text.
 
-- **`temperature (float)`** → Determines how creative the model should respond. Takes a float number from `0.0` to `2.0`.
+- **`temperature: float`** → Determines how creative the model should respond. Takes a float number from `0.0` to `2.0`.
 
-- **`validator (Callable)`** → Forces TheTool to validate the output result based on your custom validator. Validator should return a bool (True if there were no problem, False if the validation fails.) If the validator fails, TheTool will retry to get another output by modifying `temperature`. You can specify `max_validation_retries=<N>` to change the number of retries.
+- **`validator: Callable (Experimental)`** → Forces TheTool to validate the output result based on your custom validator. Validator should return a bool (True if there were no problem, False if the validation fails.) If the validator fails, TheTool will retry to get another output by modifying `temperature`. You can specify `max_validation_retries=<N>` to change the number of retries.
 
-- **`priority (int)`** → Task execution priority level. Higher values = higher priority. Affects processing order in queues.
+- **`priority: int (Experimental)`** → Task execution priority level. Higher values = higher priority. Affects processing order in queues.
 **Note:** This feature works if it's supported by the model and vLLM.
 
 **Note:** There might be some tools that don't support some of the parameters above.
@@ -125,11 +125,12 @@ TextTools provides several optional flags to customize LLM behavior:
 Every tool of `TextTools` returns a `ToolOutput` object which is a BaseModel with attributes:
 - **`result: Any`** → The output of LLM
 - **`analysis: str`** → The reasoning step before generating the final output
-- **`logprobs: list`** → Token-level probabilities for the generated output 
-- **`process: str`** → The tool name which processed the input
-- **`processed_at: datetime`** → The process time
-- **`execution_time: float`** → The execution time (seconds)
+- **`logprobs: list`** → Token-level probabilities for the generated output
 - **`errors: list[str]`** → Any error that have occured during calling LLM
+- **`ToolOutputMetadata`** →
+    - **`tool_name: str`** → The tool name which processed the input
+    - **`processed_at: datetime`** → The process time
+    - **`execution_time: float`** → The execution time (seconds)
 
 **Note:** You can use `repr(ToolOutput)` to see details of your ToolOutput.
 
@@ -224,26 +225,6 @@ Use **TextTools** when you need to:
 
 ---
 
-## 🔍 Logging
-
-TextTools uses Python's standard `logging` module. The library's default logger level is `WARNING`, so if you want to modify it, follow instructions:
-
-
-```python
-import logging
-
-# Default: warnings and errors only
-logging.basicConfig(level=logging.WARNING)
-
-# Debug everything (verbose)
-logging.basicConfig(level=logging.DEBUG)
-
-# Complete silence
-logging.basicConfig(level=logging.CRITICAL)
-```
-
----
-
 ## 📚 Batch Processing
 
 Process large datasets efficiently using OpenAI's batch API.

{hamtaa_texttools-1.1.19 → hamtaa_texttools-1.1.21}/hamtaa_texttools.egg-info/SOURCES.txt

@@ -12,16 +12,15 @@ tests/test_all_tools.py
 tests/test_output_validation.py
 texttools/__init__.py
 texttools/batch/batch_config.py
+texttools/batch/batch_manager.py
 texttools/batch/batch_runner.py
-texttools/batch/internals/batch_manager.py
-texttools/batch/internals/utils.py
 texttools/internals/async_operator.py
 texttools/internals/exceptions.py
-texttools/internals/formatters.py
 texttools/internals/models.py
 texttools/internals/operator_utils.py
 texttools/internals/prompt_loader.py
 texttools/internals/sync_operator.py
+texttools/internals/text_to_chunks.py
 texttools/prompts/README.md
 texttools/prompts/categorize.yaml
 texttools/prompts/check_fact.yaml

{hamtaa_texttools-1.1.19 → hamtaa_texttools-1.1.21}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "hamtaa-texttools"
-version = "1.1.19"
+version = "1.1.21"
 authors = [
   { name = "Tohidi", email = "the.mohammad.tohidi@gmail.com" },
   { name = "Montazer", email = "montazerh82@gmail.com" },

{hamtaa_texttools-1.1.19 → hamtaa_texttools-1.1.21}/tests/test_all_async_tools.py

@@ -24,12 +24,11 @@ async def main():
         "سلام حالت چطوره؟",
         categories=["هیچکدام", "دینی", "فلسفه"],
         logprobs=True,
-        top_logprobs=-1,
     )
     keywords_task = t.extract_keywords("Tomorrow, we will be dead by the car crash")
     entities_task = t.extract_entities("We will be dead by the car crash")
     detection_task = t.is_question("We will be dead by the car crash")
-    question_task = t.text_to_question("We will be dead by the car crash")
+    question_task = t.text_to_question("We will be dead by the car crash", 2)
     merged_task = t.merge_questions(
         ["چرا ما موجوداتی اجتماعی هستیم؟", "چرا باید در کنار هم زندگی کنیم؟"],
         mode="default",
@@ -47,7 +46,10 @@ async def main():
         "جنگ جهانی دوم در سال ۱۹۳۹ آغاز شد و آلمان به لهستان حمله کرد.",
         output_lang="Persian",
     )
-    check_fact_task = t.check_fact(text="امام نهم در ایران به خاک سپرده شد", source_text="حرم مطهر امام رضا علیه السلام در مشهد مقدس هست")
+    check_fact_task = t.check_fact(
+        text="امام نهم در ایران به خاک سپرده شد",
+        source_text="حرم مطهر امام رضا علیه السلام در مشهد مقدس هست",
+    )
     (
         category,
         keywords,

{hamtaa_texttools-1.1.19 → hamtaa_texttools-1.1.21}/tests/test_all_tools.py

@@ -28,19 +28,20 @@ category = t.categorize(
 print(repr(category))
 
 # Categorizer: tree mode
-tree = CategoryTree("category_test_tree")
-tree.add_node("اخلاق")
-tree.add_node("معرفت شناسی")
-tree.add_node("متافیزیک", description="اراده قدرت در حیطه متافیزیک است")
-tree.add_node("فلسفه ذهن", description="فلسفه ذهن به چگونگی درک ما از جهان می پردازد")
-tree.add_node("آگاهی", "فلسفه ذهن", description="آگاهی خیلی مهم است")
+tree = CategoryTree()
+tree.add_node("اخلاق", "root")
+tree.add_node("معرفت شناسی", "root")
+tree.add_node("متافیزیک", "root", description="اراده قدرت در حیطه متافیزیک است")
+tree.add_node(
+    "فلسفه ذهن", "root", description="فلسفه ذهن به چگونگی درک ما از جهان می پردازد"
+)
+tree.add_node("آگاهی", "فلسفه ذهن")
 tree.add_node("ذهن و بدن", "فلسفه ذهن")
 tree.add_node("امکان و ضرورت", "متافیزیک")
 
 categories = t.categorize(
     "اراده قدرت مفهومی مهم در مابعد الطبیعه است که توسط نیچه مطرح شده",
     tree,
-    mode="category_tree",
 )
 print(repr(categories))
 
@@ -51,7 +52,12 @@ keywords = t.extract_keywords(
 print(repr(keywords))
 
 # NER Extractor
-entities = t.extract_entities("We will be dead by the car crash", with_analysis=True)
+entities = t.extract_entities(
+    "Ali will be dead by the car crash",
+    entities=["EVENT"],
+    with_analysis=True,
+    logprobs=True,
+)
 print(repr(entities))
 
 
@@ -60,7 +66,7 @@ detection = t.is_question("We will be dead by the car crash")
 print(repr(detection))
 
 # Question from Text Generator
-question = t.text_to_question("We will be dead by the car crash")
+question = t.text_to_question("We will be dead by the car crash", 2)
 print(repr(question))
 
 # Question Merger
@@ -77,7 +83,7 @@ rewritten = t.rewrite(
 )
 print(repr(rewritten))
 
-# Question Generator from Subject
+# Question from Subject Generator
 questions = t.subject_to_question("Friendship", 3)
 print(repr(questions))
 
@@ -89,14 +95,14 @@ print(repr(summary))
 translation = t.translate("سلام حالت چطوره؟", target_language="English")
 print(repr(translation))
 
-# propositionize
+# Propositionizer
 propositionize = t.propositionize(
     "جنگ جهانی دوم در سال ۱۹۳۹ آغاز شد و آلمان به لهستان حمله کرد.",
     output_lang="Persian",
 )
 print(repr(propositionize))
 
-# check_fact
+# Check Fact
 check_fact = t.check_fact(
     text="امام نهم در ایران به خاک سپرده شد",
     source_text="حرم مطهر امام رضا علیه السلام در مشهد مقدس هست",
@@ -104,7 +110,7 @@ check_fact = t.check_fact(
 print(repr(check_fact))
 
 
-# Custom tool
+# Run Custom
 class Student(BaseModel):
     result: list[dict[str, str]]
 

{hamtaa_texttools-1.1.19 → hamtaa_texttools-1.1.21}/texttools/batch/batch_config.py

@@ -1,7 +1,20 @@
 from dataclasses import dataclass
 from collections.abc import Callable
 
-from texttools.batch.internals.utils import import_data, export_data
+
+def export_data(data) -> list[dict[str, str]]:
+    """
+    Produces a structure of the following form from an initial data structure:
+    [{"id": str, "text": str},...]
+    """
+    return data
+
+
+def import_data(data) -> object:
+    """
+    Takes the output and adds and aggregates it to the original structure.
+    """
+    return data
 
 
 @dataclass

{hamtaa_texttools-1.1.19/texttools/batch/internals → hamtaa_texttools-1.1.21/texttools/batch}/batch_manager.py

@@ -1,7 +1,7 @@
 import json
 import uuid
 from pathlib import Path
-from typing import Any, Type, TypeVar
+from typing import Type, TypeVar
 import logging
 
 from pydantic import BaseModel
@@ -31,7 +31,7 @@ class BatchManager:
         prompt_template: str,
         state_dir: Path = Path(".batch_jobs"),
         custom_json_schema_obj_str: dict | None = None,
-        **client_kwargs: Any,
+        **client_kwargs: object,
     ):
         self._client = client
         self._model = model
@@ -51,7 +51,7 @@ class BatchManager:
     def _state_file(self, job_name: str) -> Path:
         return self._state_dir / f"{job_name}.json"
 
-    def _load_state(self, job_name: str) -> list[dict[str, Any]]:
+    def _load_state(self, job_name: str) -> list[dict[str, object]]:
         """
         Loads the state (job information) from the state file for the given job name.
         Returns an empty list if the state file does not exist.
@@ -62,7 +62,7 @@ class BatchManager:
                 return json.load(f)
         return []
 
-    def _save_state(self, job_name: str, jobs: list[dict[str, Any]]) -> None:
+    def _save_state(self, job_name: str, jobs: list[dict[str, object]]) -> None:
         """
         Saves the job state to the state file for the given job name.
         """
@@ -77,11 +77,11 @@ class BatchManager:
         if path.exists():
             path.unlink()
 
-    def _build_task(self, text: str, idx: str) -> dict[str, Any]:
+    def _build_task(self, text: str, idx: str) -> dict[str, object]:
         """
         Builds a single task dictionary for the batch job, including the prompt, model, and response format configuration.
         """
-        response_format_config: dict[str, Any]
+        response_format_config: dict[str, object]
 
         if self._custom_json_schema_obj_str:
             response_format_config = {

{hamtaa_texttools-1.1.19 → hamtaa_texttools-1.1.21}/texttools/batch/batch_runner.py

@@ -2,16 +2,16 @@ import json
 import os
 import time
 from pathlib import Path
-from typing import Any, Type, TypeVar
+from typing import Type, TypeVar
 import logging
 
 from dotenv import load_dotenv
 from openai import OpenAI
 from pydantic import BaseModel
 
-from texttools.batch.internals.batch_manager import BatchManager
+from texttools.batch.batch_manager import BatchManager
 from texttools.batch.batch_config import BatchConfig
-from texttools.internals.models import StrOutput
+from texttools.internals.models import Str
 from texttools.internals.exceptions import TextToolsError, ConfigurationError
 
 # Base Model type for output models
@@ -26,7 +26,7 @@ class BatchJobRunner:
     """
 
     def __init__(
-        self, config: BatchConfig = BatchConfig(), output_model: Type[T] = StrOutput
+        self, config: BatchConfig = BatchConfig(), output_model: Type[T] = Str
     ):
         try:
             self._config = config
@@ -38,7 +38,7 @@ class BatchJobRunner:
             self._output_model = output_model
             self._manager = self._init_manager()
             self._data = self._load_data()
-            self._parts: list[list[dict[str, Any]]] = []
+            self._parts: list[list[dict[str, object]]] = []
             # Map part index to job name
             self._part_idx_to_job_name: dict[int, str] = {}
             # Track retry attempts per part
@@ -130,8 +130,8 @@ class BatchJobRunner:
 
     def _save_results(
         self,
-        output_data: list[dict[str, Any]] | dict[str, Any],
-        log: list[Any],
+        output_data: list[dict[str, object]] | dict[str, object],
+        log: list[object],
         part_idx: int,
     ):
         part_suffix = f"_part_{part_idx + 1}" if len(self._parts) > 1 else ""