unique_toolkit 1.75.1__tar.gz → 1.76.0__tar.gz

{unique_toolkit-1.75.1 → unique_toolkit-1.76.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 1.75.1
+Version: 1.76.0
 Summary: 
 Author: Cedric Klinkert, Martin Fadler, Sadique Sheik, Fabian Schläpfer, Pascal Hauri, Ejiroghene Oki-Peter, Rami Azouz, Ricardo Visini, Sanzio Monti
 Author-email: Cedric Klinkert <cedric.klinkert@unique.ch>, Martin Fadler <martin.fadler@unique.ch>, Sadique Sheik <sadique@unique.ch>, Fabian Schläpfer <fabian@unique.ch>, Pascal Hauri <pascal@unique.ch>, Ejiroghene Oki-Peter <oki-peter.ejiroghene@unique.ai>, Rami Azouz <rami.ext@unique.ch>, Ricardo Visini <ricardo.visini@unique.ch>, Sanzio Monti <sanzio.monti@unique.ai>

{unique_toolkit-1.75.1 → unique_toolkit-1.76.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_toolkit"
-version = "1.75.1"
+version = "1.76.0"
 description = ""
 readme = "README.md"
 requires-python = ">=3.12"
@@ -127,6 +127,49 @@ extend-select = ["I"]
 
 [tool.basedpyright]
 pythonVersion = "3.12"
-typeCheckingMode = "standard"
+typeCheckingMode = "recommended"
 include = ["unique_toolkit"]
 exclude = ["**/tests/**", "**/test_*.py"]
+
+# --- Silenced rules ---
+# Any-cascade: all triggered by untyped third-party code; fixing requires
+# sprinkling `Any` everywhere which defeats the purpose.
+reportAny = "none"
+reportExplicitAny = "none"
+reportUnknownArgumentType = "none"
+reportUnknownMemberType = "none"
+reportUnknownVariableType = "none"
+reportUnknownParameterType = "none"
+reportUnknownLambdaType = "none"
+
+# basedpyright-only extension: requires every non-@final class to annotate
+# every attribute, which is impractical for mixin/base-class patterns.
+reportUnannotatedClassAttribute = "none"
+
+# SDK deprecations (e.g. Event → ChatEvent): tracked separately.
+reportDeprecated = "none"
+
+# Third-party libraries that ship no type stubs.
+reportMissingTypeStubs = "none"
+
+# Structural: circular imports require architectural refactoring.
+reportImportCycles = "none"
+
+# Style preferences.
+reportImplicitStringConcatenation = "none"
+reportCallInDefaultInitializer = "none"
+
+# Intentional patterns: callers often discard return values on purpose.
+reportUnusedCallResult = "none"
+reportUnusedParameter = "none"
+reportUnusedFunction = "none"
+
+# Toolkit legitimately accesses SDK-internal symbols.
+reportPrivateUsage = "none"
+reportPrivateLocalImportUsage = "none"
+
+# Requires adding @override to every overriding method — separate PR.
+reportImplicitOverride = "none"
+
+# Missing parameter type annotations on callbacks / dynamic patterns.
+reportMissingParameterType = "none"

{unique_toolkit-1.75.1 → unique_toolkit-1.76.0}/unique_toolkit/__init__.py

@@ -20,6 +20,7 @@ from unique_toolkit.services.knowledge_base import KnowledgeBaseService
 from unique_toolkit.short_term_memory import ShortTermMemoryService
 
 # Conditionally import config_checker (requires pydantic-settings)
+_CONFIG_CHECKER_AVAILABLE: bool = False
 try:
     from unique_toolkit._common.config_checker import (
         ConfigDiffer,  # noqa: F401, I001
@@ -33,17 +34,18 @@ try:
         register_config,  # noqa: F401, I001
     )
 
-    _CONFIG_CHECKER_AVAILABLE = True
+    _CONFIG_CHECKER_AVAILABLE = True  # pyright: ignore[reportConstantRedefinition]
 except ImportError:
-    _CONFIG_CHECKER_AVAILABLE = False
+    pass
 
 # Conditionally import langchain utilities if langchain is installed
+_LANGCHAIN_AVAILABLE: bool = False
 try:
     from unique_toolkit.framework_utilities.langchain.client import get_langchain_client  # noqa: F401, I001
 
-    _LANGCHAIN_AVAILABLE = True
+    _LANGCHAIN_AVAILABLE = True  # pyright: ignore[reportConstantRedefinition]
 except ImportError:
-    _LANGCHAIN_AVAILABLE = False
+    pass
 
 # You can add other classes you frequently use here as well
 

{unique_toolkit-1.75.1 → unique_toolkit-1.76.0}/unique_toolkit/_common/api_calling/human_verification_manager.py

@@ -143,7 +143,7 @@ class HumanVerificationManagerForApiCalling(
 
         class VerificationModel(BaseModel):
             confirmation: HumanConfirmation
-            modifiable_params: self._modifiable_payload_params_model  # type: ignore
+            modifiable_params: self._modifiable_payload_params_model  # pyright: ignore[reportInvalidTypeForm]
 
         self._verification_model = VerificationModel
 

{unique_toolkit-1.75.1 → unique_toolkit-1.76.0}/unique_toolkit/_common/base_model_type_attribute.py

@@ -72,8 +72,8 @@ class ParameterType(StrEnum):
                 return float
             case ParameterType.BOOLEAN:
                 return bool
-            case _:
-                raise ValueError(f"Invalid ParameterType: {self}")
+            case _:  # pyright: ignore[reportUnnecessaryComparison]
+                raise ValueError(f"Invalid ParameterType: {self}")  # pyright: ignore[reportUnreachable]
 
     @classmethod
     def from_python_type(cls, python_type: type) -> "ParameterType":
@@ -133,11 +133,11 @@ def convert_to_base_model_type(
     If the input is a list of Parameter as defined above, converts it to a BaseModel class
     If the input is a str (JSON schema), converts it to a BaseModel class using SchemaConverter from Jambo.
     """
-    if isinstance(value, type) and issubclass(value, BaseModel):
+    if isinstance(value, type) and issubclass(value, BaseModel):  # pyright: ignore[reportUnnecessaryIsInstance]
         return value
 
     if isinstance(value, list):
-        if all(isinstance(item, Parameter) for item in value):
+        if all(isinstance(item, Parameter) for item in value):  # pyright: ignore[reportUnnecessaryIsInstance]
             return create_pydantic_model_from_parameter_list("Parameters", value)
 
     converter = SchemaConverter()
@@ -202,7 +202,7 @@ def get_json_schema_extra_for_base_model_type(model: type[BaseModel]):
     """
     sample_params = base_model_to_parameter_list(model)
 
-    def _mutate(schema: dict) -> None:
+    def _mutate(schema: dict[str, Any]) -> None:
         json_default = json.dumps(model.model_json_schema())
         params_default = [p.model_dump() for p in sample_params]
 

{unique_toolkit-1.75.1 → unique_toolkit-1.76.0}/unique_toolkit/_common/chunk_relevancy_sorter/service.py

@@ -53,7 +53,7 @@ class ChunkRelevancySorter:
         "Use __init__ with company_id and user_id instead or use the classmethod `from_event`"
     )
     @overload
-    def __init__(self, event: ChatEvent | BaseEvent):
+    def __init__(self, event: ChatEvent | BaseEvent[Any]):
         """
         Initialize the ChunkRelevancySorter with an event (deprecated)
         """
@@ -66,7 +66,7 @@ class ChunkRelevancySorter:
 
     def __init__(
         self,
-        event: ChatEvent | BaseEvent | None = None,
+        event: ChatEvent | BaseEvent[Any] | None = None,
         company_id: str | None = None,
         user_id: str | None = None,
     ):
@@ -81,7 +81,7 @@ class ChunkRelevancySorter:
         self.logger = logging.getLogger(f"{module_name}.{__name__}")
 
     @classmethod
-    def from_event(cls, event: ChatEvent | BaseEvent):
+    def from_event(cls, event: ChatEvent | BaseEvent[Any]):
         return cls(company_id=event.company_id, user_id=event.user_id)
 
     async def run(
@@ -336,7 +336,9 @@ class ChunkRelevancySorter:
         sorted_chunk_relevancies = sorted(
             chunk_relevancies,
             key=lambda obj: (
-                relevancy_level_order[obj.relevancy.value.lower()],  # type: ignore
+                relevancy_level_order[obj.relevancy.value.lower()]
+                if obj.relevancy is not None
+                else len(relevancy_level_order),
                 chunk_order[obj.chunk.chunk_id],
             ),
         )
@@ -366,7 +368,8 @@ class ChunkRelevancySorter:
         return [
             chunk_relevancy
             for chunk_relevancy in chunk_relevancies
-            if chunk_relevancy.relevancy.value.lower() in levels_to_consider  # type: ignore
+            if chunk_relevancy.relevancy is not None
+            and chunk_relevancy.relevancy.value.lower() in levels_to_consider
         ]
 
     def _count_distinct_values(self, chunk_relevancies: list[ChunkRelevancy]):

{unique_toolkit-1.75.1 → unique_toolkit-1.76.0}/unique_toolkit/_common/docx_generator/schemas.py

@@ -66,7 +66,8 @@ class ContentField(BaseModel):
             #     content.download_image(self.download_path)
             #     content.add(sd) # type: ignore
             # else:
-            content.add(sd)  # type: ignore
+            # docxtpl subdoc is duck-type compatible with docx.Document but not a declared subtype
+            content.add(sd)  # pyright: ignore[reportArgumentType]
         return sd
 
     def __str__(self):

{unique_toolkit-1.75.1 → unique_toolkit-1.76.0}/unique_toolkit/_common/docx_generator/service.py

@@ -2,6 +2,7 @@ import io
 import logging
 import re
 from pathlib import Path
+from typing import Any
 
 from docxtpl import DocxTemplate
 from markdown_it import MarkdownIt
@@ -102,7 +103,7 @@ class DocxGeneratorService:
                             content,
                         )
                         bold = True
-                    current_section["text"] += content
+                    current_section["text"] += content  # pyright: ignore[reportOperatorIssue, reportArgumentType]
                 elif "runs" in current_section:
                     bold = False
                     italic = False
@@ -137,7 +138,7 @@ class DocxGeneratorService:
                             )
                         )
 
-                    current_section["runs"].extend(runs)
+                    current_section["runs"].extend(runs)  # pyright: ignore[reportAttributeAccessIssue]
 
         # Process remaining elements
         contents = []
@@ -167,7 +168,7 @@ class DocxGeneratorService:
     def generate_from_template(
         self,
         subdoc_content: list[HeadingField | ParagraphField | RunsField],
-        fields: dict | None = None,
+        fields: dict[str, Any] | None = None,
     ):
         """
         Generate a docx file from a template with the given content.

{unique_toolkit-1.75.1 → unique_toolkit-1.76.0}/unique_toolkit/_common/endpoint_builder.py

@@ -128,7 +128,7 @@ class ApiOperationProtocol(
 
     @staticmethod
     def create_path_from_model(
-        path_params: PathParamsType, *, model_dump_options: dict | None = None
+        path_params: PathParamsType, *, model_dump_options: dict[str, Any] | None = None
     ) -> str: ...
 
     @staticmethod
@@ -138,7 +138,7 @@ class ApiOperationProtocol(
 
     @staticmethod
     def create_payload_from_model(
-        payload: PayloadType, *, model_dump_options: dict | None = None
+        payload: PayloadType, *, model_dump_options: dict[str, Any] | None = None
     ) -> dict[str, Any]: ...
 
     @staticmethod
@@ -148,12 +148,16 @@ class ApiOperationProtocol(
 
     @staticmethod
     def create_query_params_from_model(
-        query_params: QueryParamsType, *, model_dump_options: dict | None = None
+        query_params: QueryParamsType,
+        *,
+        model_dump_options: dict[str, Any] | None = None,
     ) -> dict[str, Any]: ...
 
     @staticmethod
     def handle_response(
-        response: dict[str, Any], *, model_validate_options: dict | None = None
+        response: dict[str, Any],
+        *,
+        model_validate_options: dict[str, Any] | None = None,
     ) -> ResponseType: ...
 
     @staticmethod
@@ -175,13 +179,13 @@ def build_api_operation(
     path_template: Template,
     response_model_type: type[ResponseType],
     path_params_constructor: Callable[PathParamsSpec, PathParamsType] = EmptyModel,
-    payload_dump_options: dict | None = None,
+    payload_dump_options: dict[str, Any] | None = None,
     payload_constructor: Callable[PayloadParamSpec, PayloadType] = EmptyModel,
-    path_dump_options: dict | None = None,
+    path_dump_options: dict[str, Any] | None = None,
     query_params_constructor: Callable[QueryParamsSpec, QueryParamsType] = EmptyModel,
-    query_params_dump_options: dict | None = None,
-    response_validate_options: dict | None = None,
-    dump_options: dict | None = None,  # Deprecated
+    query_params_dump_options: dict[str, Any] | None = None,
+    response_validate_options: dict[str, Any] | None = None,
+    dump_options: dict[str, Any] | None = None,  # Deprecated
 ) -> type[
     ApiOperationProtocol[
         PathParamsSpec,
@@ -215,7 +219,8 @@ def build_api_operation(
             DeprecationWarning,
         )
 
-    class Operation(ApiOperationProtocol):
+    # TODO(UN-19502): tighten ApiOperationProtocol generic parameters away from Any
+    class Operation(ApiOperationProtocol[Any, Any, Any, Any, Any, Any, Any]):
         @staticmethod
         def path_dump_options() -> dict[str, Any]:
             return path_dump_options or {}
@@ -254,7 +259,9 @@ def build_api_operation(
 
         @staticmethod
         def create_path_from_model(
-            path_params: PathParamsType, *, model_dump_options: dict | None = None
+            path_params: PathParamsType,
+            *,
+            model_dump_options: dict[str, Any] | None = None,
         ) -> str:
             if model_dump_options is None:
                 if path_dump_options is None:
@@ -289,7 +296,7 @@ def build_api_operation(
 
         @staticmethod
         def create_payload_from_model(
-            payload: PayloadType, *, model_dump_options: dict | None = None
+            payload: PayloadType, *, model_dump_options: dict[str, Any] | None = None
         ) -> dict[str, Any]:
             if model_dump_options is None:
                 if payload_dump_options is None:
@@ -314,7 +321,9 @@ def build_api_operation(
 
         @staticmethod
         def create_query_params_from_model(
-            query_params: QueryParamsType, *, model_dump_options: dict | None = None
+            query_params: QueryParamsType,
+            *,
+            model_dump_options: dict[str, Any] | None = None,
         ) -> dict[str, Any]:
             if model_dump_options is None:
                 if query_params_dump_options is None:

{unique_toolkit-1.75.1 → unique_toolkit-1.76.0}/unique_toolkit/_common/endpoint_requestor.py

@@ -71,7 +71,8 @@ def build_fake_requestor(
     combined_model: Callable[CombinedParamsSpec, CombinedParamsType],
     return_value: dict[str, Any],
 ) -> type[EndpointRequestorProtocol[CombinedParamsSpec, ResponseType]]:
-    class FakeRequestor(EndpointRequestorProtocol):
+    # TODO(UN-19505): tighten EndpointRequestorProtocol generic parameters away from Any
+    class FakeRequestor(EndpointRequestorProtocol[Any, Any]):
         _operation = operation_type
 
         @classmethod
@@ -82,7 +83,7 @@ def build_fake_requestor(
             **kwargs: CombinedParamsSpec.kwargs,
         ) -> ResponseType:
             try:
-                path_params, payload_model, query_params = (
+                _path_params, _payload_model, _query_params = (
                     cls._operation.models_from_combined(combined=kwargs)
                 )
             except Exception as e:
@@ -126,7 +127,8 @@ def build_request_requestor(
 ) -> type[EndpointRequestorProtocol[CombinedParamsSpec, ResponseType]]:
     import requests
 
-    class RequestRequestor(EndpointRequestorProtocol):
+    # TODO(UN-19505): tighten EndpointRequestorProtocol generic parameters away from Any
+    class RequestRequestor(EndpointRequestorProtocol[Any, Any]):
         _operation = operation_type
 
         @classmethod
@@ -203,7 +205,8 @@ def build_httpx_requestor(
 ) -> type[EndpointRequestorProtocol[CombinedParamsSpec, ResponseType]]:
     import httpx
 
-    class HttpxRequestor(EndpointRequestorProtocol):
+    # TODO(UN-19505): tighten EndpointRequestorProtocol generic parameters away from Any
+    class HttpxRequestor(EndpointRequestorProtocol[Any, Any]):
         _operation = operation_type
 
         @classmethod
@@ -311,7 +314,8 @@ def build_aiohttp_requestor(
 ) -> type[EndpointRequestorProtocol[CombinedParamsSpec, ResponseType]]:
     import aiohttp
 
-    class AiohttpRequestor(EndpointRequestorProtocol):
+    # TODO(UN-19505): tighten EndpointRequestorProtocol generic parameters away from Any
+    class AiohttpRequestor(EndpointRequestorProtocol[Any, Any]):
         _operation = operation_type
 
         @classmethod

{unique_toolkit-1.75.1 → unique_toolkit-1.76.0}/unique_toolkit/_common/experimental/endpoint_builder.py

@@ -128,7 +128,7 @@ class ApiOperationProtocol(
 
     @staticmethod
     def create_path_from_model(
-        path_params: PathParamsType, *, model_dump_options: dict | None = None
+        path_params: PathParamsType, *, model_dump_options: dict[str, Any] | None = None
     ) -> str: ...
 
     @staticmethod
@@ -138,7 +138,7 @@ class ApiOperationProtocol(
 
     @staticmethod
     def create_payload_from_model(
-        payload: PayloadType, *, model_dump_options: dict | None = None
+        payload: PayloadType, *, model_dump_options: dict[str, Any] | None = None
     ) -> dict[str, Any]: ...
 
     @staticmethod
@@ -148,12 +148,16 @@ class ApiOperationProtocol(
 
     @staticmethod
     def create_query_params_from_model(
-        query_params: QueryParamsType, *, model_dump_options: dict | None = None
+        query_params: QueryParamsType,
+        *,
+        model_dump_options: dict[str, Any] | None = None,
     ) -> dict[str, Any]: ...
 
     @staticmethod
     def handle_response(
-        response: dict[str, Any], *, model_validate_options: dict | None = None
+        response: dict[str, Any],
+        *,
+        model_validate_options: dict[str, Any] | None = None,
     ) -> ResponseType: ...
 
     @staticmethod
@@ -175,13 +179,13 @@ def build_api_operation(
     path_template: Template,
     response_model_type: type[ResponseType],
     path_params_constructor: Callable[PathParamsSpec, PathParamsType] = EmptyModel,
-    payload_dump_options: dict | None = None,
+    payload_dump_options: dict[str, Any] | None = None,
     payload_constructor: Callable[PayloadParamSpec, PayloadType] = EmptyModel,
-    path_dump_options: dict | None = None,
+    path_dump_options: dict[str, Any] | None = None,
     query_params_constructor: Callable[QueryParamsSpec, QueryParamsType] = EmptyModel,
-    query_params_dump_options: dict | None = None,
-    response_validate_options: dict | None = None,
-    dump_options: dict | None = None,  # Deprecated
+    query_params_dump_options: dict[str, Any] | None = None,
+    response_validate_options: dict[str, Any] | None = None,
+    dump_options: dict[str, Any] | None = None,  # Deprecated
 ) -> type[
     ApiOperationProtocol[
         PathParamsSpec,
@@ -215,7 +219,8 @@ def build_api_operation(
             DeprecationWarning,
         )
 
-    class Operation(ApiOperationProtocol):
+    # TODO(UN-19502): tighten ApiOperationProtocol generic parameters away from Any
+    class Operation(ApiOperationProtocol[Any, Any, Any, Any, Any, Any, Any]):
         @staticmethod
         def path_dump_options() -> dict[str, Any]:
             return path_dump_options or {}
@@ -254,7 +259,9 @@ def build_api_operation(
 
         @staticmethod
         def create_path_from_model(
-            path_params: PathParamsType, *, model_dump_options: dict | None = None
+            path_params: PathParamsType,
+            *,
+            model_dump_options: dict[str, Any] | None = None,
         ) -> str:
             if model_dump_options is None:
                 if path_dump_options is None:
@@ -289,7 +296,7 @@ def build_api_operation(
 
         @staticmethod
         def create_payload_from_model(
-            payload: PayloadType, *, model_dump_options: dict | None = None
+            payload: PayloadType, *, model_dump_options: dict[str, Any] | None = None
         ) -> dict[str, Any]:
             if model_dump_options is None:
                 if payload_dump_options is None:
@@ -314,7 +321,9 @@ def build_api_operation(
 
         @staticmethod
         def create_query_params_from_model(
-            query_params: QueryParamsType, *, model_dump_options: dict | None = None
+            query_params: QueryParamsType,
+            *,
+            model_dump_options: dict[str, Any] | None = None,
         ) -> dict[str, Any]:
             if model_dump_options is None:
                 if query_params_dump_options is None:

{unique_toolkit-1.75.1 → unique_toolkit-1.76.0}/unique_toolkit/_common/experimental/endpoint_requestor.py

@@ -71,7 +71,8 @@ def build_fake_requestor(
     combined_model: Callable[CombinedParamsSpec, CombinedParamsType],
     return_value: dict[str, Any],
 ) -> type[EndpointRequestorProtocol[CombinedParamsSpec, ResponseType]]:
-    class FakeRequestor(EndpointRequestorProtocol):
+    # TODO(UN-19505): tighten EndpointRequestorProtocol generic parameters away from Any
+    class FakeRequestor(EndpointRequestorProtocol[Any, Any]):
         _operation = operation_type
 
         @classmethod
@@ -82,7 +83,7 @@ def build_fake_requestor(
             **kwargs: CombinedParamsSpec.kwargs,
         ) -> ResponseType:
             try:
-                path_params, payload_model, query_params = (
+                _path_params, _payload_model, _query_params = (
                     cls._operation.models_from_combined(combined=kwargs)
                 )
             except Exception as e:
@@ -126,7 +127,8 @@ def build_request_requestor(
 ) -> type[EndpointRequestorProtocol[CombinedParamsSpec, ResponseType]]:
     import requests
 
-    class RequestRequestor(EndpointRequestorProtocol):
+    # TODO(UN-19505): tighten EndpointRequestorProtocol generic parameters away from Any
+    class RequestRequestor(EndpointRequestorProtocol[Any, Any]):
         _operation = operation_type
 
         @classmethod
@@ -203,7 +205,8 @@ def build_httpx_requestor(
 ) -> type[EndpointRequestorProtocol[CombinedParamsSpec, ResponseType]]:
     import httpx
 
-    class HttpxRequestor(EndpointRequestorProtocol):
+    # TODO(UN-19505): tighten EndpointRequestorProtocol generic parameters away from Any
+    class HttpxRequestor(EndpointRequestorProtocol[Any, Any]):
         _operation = operation_type
 
         @classmethod
@@ -267,7 +270,11 @@ def build_httpx_requestor(
         ) -> ResponseType:
             headers = context.headers or {}
 
-            path_params, payload_model, query_model = (
+            # TODO(UN-19748): async path silently drops query params — build and
+            # pass query_params like the sync request() method does (see the
+            # sync branch above which calls create_query_params_from_model and
+            # passes params=query_params to client.request).
+            path_params, payload_model, _query_model = (
                 cls._operation.models_from_combined(combined=kwargs)
             )
 
@@ -323,7 +330,8 @@ def build_aiohttp_requestor(
 ) -> type[EndpointRequestorProtocol[CombinedParamsSpec, ResponseType]]:
     import aiohttp
 
-    class AiohttpRequestor(EndpointRequestorProtocol):
+    # TODO(UN-19505): tighten EndpointRequestorProtocol generic parameters away from Any
+    class AiohttpRequestor(EndpointRequestorProtocol[Any, Any]):
         _operation = operation_type
 
         @classmethod

{unique_toolkit-1.75.1 → unique_toolkit-1.76.0}/unique_toolkit/_common/experimental/write_up_agent/services/dataframe_handler/utils.py

@@ -1,6 +1,7 @@
 """Utility functions for DataFrame operations."""
 
 import re
+from typing import Any
 
 import pandas as pd
 
@@ -112,7 +113,7 @@ def limit_dataframe_rows(df: pd.DataFrame, max_rows: int) -> pd.DataFrame:
 
 def dataframe_to_dict_records(
     df: pd.DataFrame, columns: list[str] | None = None
-) -> list[dict]:
+) -> list[dict[str, Any]]:
     """
     Convert DataFrame to list of dict records.
 

{unique_toolkit-1.75.1 → unique_toolkit-1.76.0}/unique_toolkit/_common/experimental/write_up_agent/services/generation_handler/exceptions.py

@@ -1,5 +1,7 @@
 """Exceptions for generation handler operations."""
 
+from typing import Any
+
 
 class GenerationHandlerError(Exception):
     """Base exception for all generation handler errors."""
@@ -28,7 +30,7 @@ class PromptBuildError(GenerationHandlerError):
         self,
         message: str,
         prompt_type: str | None = None,
-        context: dict | None = None,
+        context: dict[str, Any] | None = None,
     ):
         super().__init__(message)
         self.prompt_type = prompt_type

{unique_toolkit-1.75.1 → unique_toolkit-1.76.0}/unique_toolkit/_common/pydantic/rjsf_tags.py

@@ -974,7 +974,7 @@ def ui_schema_for_model(
 
         # Dict -> additionalProperties (value side)
         elif origin is dict:
-            key_t, val_t = get_args(base) or (Any, Any)
+            _key_t, val_t = get_args(base) or (Any, Any)
             val_base, val_meta = _walk_annotated_chain(val_t)
             val_base = _unwrap_optional(val_base)
 

{unique_toolkit-1.75.1 → unique_toolkit-1.76.0}/unique_toolkit/_common/pydantic_helpers.py

@@ -35,15 +35,14 @@ NoneToDefault = BeforeValidator(_none_to_default)
 
 
 def get_configuration_dict(**kwargs: Unpack[ConfigDict]) -> ConfigDict:
-    config = {
+    base: dict[str, object] = {
         "alias_generator": to_camel,
         "field_title_generator": field_title_generator,
         "model_title_generator": model_title_generator,
         "populate_by_name": True,
-        # protected_namespaces=(),
     }
-    config.update(kwargs)
-    return ConfigDict(**config)
+    base.update(kwargs)
+    return ConfigDict(**base)  # pyright: ignore[reportArgumentType]
 
 
 ModelTypeA = TypeVar("ModelTypeA", bound=BaseModel)

{unique_toolkit-1.75.1 → unique_toolkit-1.76.0}/unique_toolkit/_common/string_utilities.py

@@ -25,8 +25,8 @@ def dict_to_markdown_table(data: dict[str, Any]) -> str:
     Returns:
         Markdown table string or JSON string
     """
-    if not isinstance(data, dict):
-        return json.dumps(data, indent=2)
+    if not isinstance(data, dict):  # pyright: ignore[reportUnnecessaryIsInstance]
+        return json.dumps(data, indent=2)  # pyright: ignore[reportUnreachable]
 
     if not _is_elementary_dict(data):
         return json.dumps(data, indent=2)

{unique_toolkit-1.75.1 → unique_toolkit-1.76.0}/unique_toolkit/_common/token/token_counting.py

@@ -96,7 +96,7 @@ def num_tokens_per_messages(
                     num_tokens += handle_message_with_images(value, encode)
                 else:
                     num_tokens += len(encode(value))
-            elif isinstance(value, str):
+            elif isinstance(value, str):  # pyright: ignore[reportUnnecessaryIsInstance]
                 num_tokens += len(encode(value))
 
             if key == "name":
@@ -167,7 +167,9 @@ def num_tokens_for_tools(
     return func_token_count
 
 
-def handle_message_with_images(message: list[dict], encode: Callable[[str], list[int]]):
+def handle_message_with_images(
+    message: list[dict[str, Any]], encode: Callable[[str], list[int]]
+):
     token_count = 0
     for item in message:
         if item.get("type") == "image_url":

{unique_toolkit-1.75.1 → unique_toolkit-1.76.0}/unique_toolkit/_common/utils/files.py

@@ -89,8 +89,8 @@ def get_common_name(extension: FileMimeType) -> str:
             return "markdown"
         case FileMimeType.HTML:
             return "html"
-        case _:
-            return "unknown"
+        case _:  # pyright: ignore[reportUnnecessaryComparison]
+            return "unknown"  # pyright: ignore[reportUnreachable]
 
 
 def get_file_extensions(mimes: list[FileMimeType]) -> list[str]:

{unique_toolkit-1.75.1 → unique_toolkit-1.76.0}/unique_toolkit/_common/utils/jinja/schema.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from datetime import date, datetime
 from typing import Annotated, Any
 
@@ -29,7 +31,7 @@ class ToolPromptParams(Jinja2PromptParams):
     tool_format_reminder_for_user_prompt: str = ""
 
     @classmethod
-    def from_tool(cls, tool: Tool) -> "ToolPromptParams":
+    def from_tool(cls, tool: Tool[Any]) -> ToolPromptParams:
         return cls(
             name=tool.name,
             tool_description_for_system_prompt=tool.tool_description_for_system_prompt(),