openaivec 0.14.7__py3-none-any.whl → 0.14.8__py3-none-any.whl

openaivec/_responses.py

@@ -1,7 +1,7 @@
 import warnings
 from dataclasses import dataclass, field
 from logging import Logger, getLogger
-from typing import Any, Generic, List, Type, cast
+from typing import Any, Generic, cast
 
 from openai import AsyncOpenAI, BadRequestError, InternalServerError, OpenAI, RateLimitError
 from openai.types.responses import ParsedResponse
@@ -120,11 +120,11 @@ class Message(BaseModel, Generic[ResponseFormat]):
 
 
 class Request(BaseModel):
-    user_messages: List[Message[str]]
+    user_messages: list[Message[str]]
 
 
 class Response(BaseModel, Generic[ResponseFormat]):
-    assistant_messages: List[Message[ResponseFormat]]
+    assistant_messages: list[Message[ResponseFormat]]
 
 
 @dataclass(frozen=True)
@@ -150,7 +150,7 @@ class BatchResponses(Generic[ResponseFormat]):
         system_message (str): System prompt prepended to every request.
         temperature (float): Sampling temperature.
         top_p (float): Nucleus‑sampling parameter.
-        response_format (Type[ResponseFormat]): Expected Pydantic model class or ``str`` for each assistant message.
+        response_format (type[ResponseFormat]): Expected Pydantic model class or ``str`` for each assistant message.
         cache (BatchingMapProxy[str, ResponseFormat]): Order‑preserving batching proxy with de‑duplication and caching.
 
     Notes:
@@ -165,7 +165,7 @@ class BatchResponses(Generic[ResponseFormat]):
     system_message: str
     temperature: float | None = None
     top_p: float = 1.0
-    response_format: Type[ResponseFormat] = str  # type: ignore[assignment]
+    response_format: type[ResponseFormat] = str  # type: ignore[assignment]
     cache: BatchingMapProxy[str, ResponseFormat] = field(default_factory=lambda: BatchingMapProxy(batch_size=None))
     _vectorized_system_message: str = field(init=False)
     _model_json_schema: dict = field(init=False)
@@ -178,7 +178,7 @@ class BatchResponses(Generic[ResponseFormat]):
         system_message: str,
         temperature: float | None = 0.0,
         top_p: float = 1.0,
-        response_format: Type[ResponseFormat] = str,
+        response_format: type[ResponseFormat] = str,
         batch_size: int | None = None,
     ) -> "BatchResponses":
         """Factory constructor.
@@ -189,7 +189,7 @@ class BatchResponses(Generic[ResponseFormat]):
             system_message (str): System prompt for the model.
             temperature (float, optional): Sampling temperature. Defaults to 0.0.
             top_p (float, optional): Nucleus sampling parameter. Defaults to 1.0.
-            response_format (Type[ResponseFormat], optional): Expected output type. Defaults to ``str``.
+            response_format (type[ResponseFormat], optional): Expected output type. Defaults to ``str``.
             batch_size (int | None, optional): Max unique prompts per API call. Defaults to None
                 (automatic batch size optimization). Set to a positive integer for fixed batch size.
 
@@ -242,12 +242,12 @@ class BatchResponses(Generic[ResponseFormat]):
     @observe(_LOGGER)
     @backoff(exceptions=[RateLimitError, InternalServerError], scale=1, max_retries=12)
     def _request_llm(
-        self, user_messages: List[Message[str]], **extra_api_params: Any
+        self, user_messages: list[Message[str]], **extra_api_params: Any
     ) -> ParsedResponse[Response[ResponseFormat]]:
         """Make a single call to the OpenAI JSON‑mode endpoint.
 
         Args:
-            user_messages (List[Message[str]]): Sequence of ``Message[str]`` representing the
+            user_messages (list[Message[str]]): Sequence of ``Message[str]`` representing the
                 prompts for this minibatch.  Each message carries a unique `id`
                 so we can restore ordering later.
 
@@ -265,7 +265,7 @@ class BatchResponses(Generic[ResponseFormat]):
             body: response_format  # type: ignore
 
         class ResponseT(BaseModel):
-            assistant_messages: List[MessageT]
+            assistant_messages: list[MessageT]
 
         # Build base API parameters (cannot be overridden by caller)
         api_params: dict[str, Any] = {
@@ -300,7 +300,7 @@ class BatchResponses(Generic[ResponseFormat]):
         return cast(ParsedResponse[Response[ResponseFormat]], completion)
 
     @observe(_LOGGER)
-    def _predict_chunk(self, user_messages: List[str], **api_kwargs: Any) -> List[ResponseFormat | None]:
+    def _predict_chunk(self, user_messages: list[str], **api_kwargs: Any) -> list[ResponseFormat | None]:
         """Helper executed for every unique minibatch.
 
             This method:
@@ -316,11 +316,11 @@ class BatchResponses(Generic[ResponseFormat]):
         if not responses.output_parsed:
             return [None] * len(messages)
         response_dict = {message.id: message.body for message in responses.output_parsed.assistant_messages}
-        sorted_responses: List[ResponseFormat | None] = [response_dict.get(m.id, None) for m in messages]
+        sorted_responses: list[ResponseFormat | None] = [response_dict.get(m.id, None) for m in messages]
         return sorted_responses
 
     @observe(_LOGGER)
-    def parse(self, inputs: List[str], **api_kwargs: Any) -> List[ResponseFormat | None]:
+    def parse(self, inputs: list[str], **api_kwargs: Any) -> list[ResponseFormat | None]:
         """Batched predict.
 
         Accepts arbitrary keyword arguments that are forwarded to the underlying
@@ -329,16 +329,16 @@ class BatchResponses(Generic[ResponseFormat]):
         configured values but can be overridden explicitly.
 
         Args:
-            inputs (List[str]): Prompts that require responses. Duplicates are de‑duplicated.
+            inputs (list[str]): Prompts that require responses. Duplicates are de‑duplicated.
             **api_kwargs: Extra keyword args forwarded to the OpenAI Responses API.
 
         Returns:
-            List[ResponseFormat | None]: Assistant responses aligned to ``inputs``.
+            list[ResponseFormat | None]: Assistant responses aligned to ``inputs``.
         """
         if not api_kwargs:
             return self.cache.map(inputs, self._predict_chunk)  # type: ignore[return-value]
 
-        def _predict_with(xs: List[str]) -> List[ResponseFormat | None]:
+        def _predict_with(xs: list[str]) -> list[ResponseFormat | None]:
             return self._predict_chunk(xs, **api_kwargs)
 
         return self.cache.map(inputs, _predict_with)  # type: ignore[return-value]
@@ -385,7 +385,7 @@ class AsyncBatchResponses(Generic[ResponseFormat]):
         system_message (str): System prompt prepended to every request.
         temperature (float): Sampling temperature.
         top_p (float): Nucleus‑sampling parameter.
-        response_format (Type[ResponseFormat]): Expected Pydantic model class or ``str`` for each assistant message.
+        response_format (type[ResponseFormat]): Expected Pydantic model class or ``str`` for each assistant message.
         cache (AsyncBatchingMapProxy[str, ResponseFormat]): Async batching proxy with de‑duplication
             and concurrency control.
     """
@@ -395,7 +395,7 @@ class AsyncBatchResponses(Generic[ResponseFormat]):
     system_message: str
     temperature: float | None = 0.0
     top_p: float = 1.0
-    response_format: Type[ResponseFormat] = str  # type: ignore[assignment]
+    response_format: type[ResponseFormat] = str  # type: ignore[assignment]
     cache: AsyncBatchingMapProxy[str, ResponseFormat] = field(
         default_factory=lambda: AsyncBatchingMapProxy(batch_size=None, max_concurrency=8)
     )
@@ -410,7 +410,7 @@ class AsyncBatchResponses(Generic[ResponseFormat]):
         system_message: str,
         temperature: float | None = None,
         top_p: float = 1.0,
-        response_format: Type[ResponseFormat] = str,
+        response_format: type[ResponseFormat] = str,
         batch_size: int | None = None,
         max_concurrency: int = 8,
     ) -> "AsyncBatchResponses":
@@ -422,7 +422,7 @@ class AsyncBatchResponses(Generic[ResponseFormat]):
             system_message (str): System prompt.
             temperature (float, optional): Sampling temperature. Defaults to 0.0.
             top_p (float, optional): Nucleus sampling parameter. Defaults to 1.0.
-            response_format (Type[ResponseFormat], optional): Expected output type. Defaults to ``str``.
+            response_format (type[ResponseFormat], optional): Expected output type. Defaults to ``str``.
             batch_size (int | None, optional): Max unique prompts per API call. Defaults to None
                 (automatic batch size optimization). Set to a positive integer for fixed batch size.
             max_concurrency (int, optional): Max concurrent API calls. Defaults to 8.
@@ -482,12 +482,12 @@ class AsyncBatchResponses(Generic[ResponseFormat]):
     @backoff_async(exceptions=[RateLimitError, InternalServerError], scale=1, max_retries=12)
     @observe(_LOGGER)
     async def _request_llm(
-        self, user_messages: List[Message[str]], **extra_api_params: Any
+        self, user_messages: list[Message[str]], **extra_api_params: Any
     ) -> ParsedResponse[Response[ResponseFormat]]:
         """Make a single async call to the OpenAI JSON‑mode endpoint.
 
         Args:
-            user_messages (List[Message[str]]): Sequence of ``Message[str]`` representing the minibatch prompts.
+            user_messages (list[Message[str]]): Sequence of ``Message[str]`` representing the minibatch prompts.
 
         Returns:
             ParsedResponse[Response[ResponseFormat]]: Parsed response with assistant messages (arbitrary order).
@@ -502,7 +502,7 @@ class AsyncBatchResponses(Generic[ResponseFormat]):
             body: response_format  # type: ignore
 
         class ResponseT(BaseModel):
-            assistant_messages: List[MessageT]
+            assistant_messages: list[MessageT]
 
         # Build base API parameters (cannot be overridden by caller)
         api_params: dict[str, Any] = {
@@ -537,7 +537,7 @@ class AsyncBatchResponses(Generic[ResponseFormat]):
         return cast(ParsedResponse[Response[ResponseFormat]], completion)
 
     @observe(_LOGGER)
-    async def _predict_chunk(self, user_messages: List[str], **api_kwargs: Any) -> List[ResponseFormat | None]:
+    async def _predict_chunk(self, user_messages: list[str], **api_kwargs: Any) -> list[ResponseFormat | None]:
         """Async helper executed for every unique minibatch.
 
             This method:
@@ -553,11 +553,11 @@ class AsyncBatchResponses(Generic[ResponseFormat]):
             return [None] * len(messages)
         response_dict = {message.id: message.body for message in responses.output_parsed.assistant_messages}
         # Ensure proper handling for missing IDs - this shouldn't happen in normal operation
-        sorted_responses: List[ResponseFormat | None] = [response_dict.get(m.id, None) for m in messages]
+        sorted_responses: list[ResponseFormat | None] = [response_dict.get(m.id, None) for m in messages]
         return sorted_responses
 
     @observe(_LOGGER)
-    async def parse(self, inputs: List[str], **api_kwargs: Any) -> List[ResponseFormat | None]:
+    async def parse(self, inputs: list[str], **api_kwargs: Any) -> list[ResponseFormat | None]:
         """Batched predict (async).
 
         Accepts arbitrary keyword arguments forwarded to ``AsyncOpenAI.responses.parse``.
@@ -566,16 +566,16 @@ class AsyncBatchResponses(Generic[ResponseFormat]):
         changing the public surface again.
 
         Args:
-            inputs (List[str]): Prompts that require responses. Duplicates are de‑duplicated.
+            inputs (list[str]): Prompts that require responses. Duplicates are de‑duplicated.
             **api_kwargs: Extra keyword args for the OpenAI Responses API.
 
         Returns:
-            List[ResponseFormat | None]: Assistant responses aligned to ``inputs``.
+            list[ResponseFormat | None]: Assistant responses aligned to ``inputs``.
         """
         if not api_kwargs:
             return await self.cache.map(inputs, self._predict_chunk)  # type: ignore[return-value]
 
-        async def _predict_with(xs: List[str]) -> List[ResponseFormat | None]:
+        async def _predict_with(xs: list[str]) -> list[ResponseFormat | None]:
             return await self._predict_chunk(xs, **api_kwargs)
 
         return await self.cache.map(inputs, _predict_with)  # type: ignore[return-value]

openaivec/_schema.py

@@ -25,8 +25,11 @@ This module is intentionally **internal** (``__all__ = []``). Public users
 should interact through higher‑level batch APIs once a schema has been inferred.
 
 Design constraints:
-* Flat schema only (no nesting / arrays) to simplify Spark & pandas alignment.
-* Primitive types limited to {string, integer, float, boolean}.
+* Flat schema only (no nested objects). Top-level arrays permitted ONLY as homogeneous arrays of primitives
+    (e.g. array of strings) – represented via specialized primitive array type names
+    (string_array, integer_array, float_array, boolean_array).
+* Primitive scalar types limited to {string, integer, float, boolean}; optional array variants
+    {string_array, integer_array, float_array, boolean_array}.
 * Optional enumerations for *closed*, *observed* categorical sets only.
 * Validation retries ensure a structurally coherent suggestion before returning.
 
@@ -49,7 +52,7 @@ authoritative contract is the ordered list of ``FieldSpec`` instances.
 
 from dataclasses import dataclass
 from enum import Enum
-from typing import List, Literal, Optional, Type
+from typing import Literal
 
 from openai import OpenAI
 from openai.types.responses import ParsedResponse
@@ -95,12 +98,24 @@ class FieldSpec(BaseModel):
             "negated forms (is_not_active)."
         )
     )
-    type: Literal["string", "integer", "float", "boolean"] = Field(
+    type: Literal[
+        "string",
+        "integer",
+        "float",
+        "boolean",
+        "string_array",
+        "integer_array",
+        "float_array",
+        "boolean_array",
+    ] = Field(
         description=(
             "Primitive type. Use 'integer' only if all observed numeric values are whole numbers. "
             "Use 'float' if any value can contain a decimal or represents a ratio/score. Use 'boolean' only for "
             "explicit binary states (yes/no, true/false, present/absent) consistently encoded. Use 'string' otherwise. "
-            "Never output arrays, objects, or composite encodings; flatten to the most specific scalar value."
+            "Array variants (string_array, integer_array, float_array, boolean_array) are ONLY allowed when the value "
+            "is a repeatable homogeneous collection whose individual elements would otherwise stand as valid scalar "
+            "extractions (e.g. keywords, error_codes, tag_ids). Do not encode objects or mixed-type arrays; flatten or "
+            "choose the most informative level."
         )
     )
     description: str = Field(
@@ -112,7 +127,7 @@ class FieldSpec(BaseModel):
             "state the transformation (e.g. sentiment of comment_text, normalized date, language code)."
         )
     )
-    enum_values: Optional[List[str]] = Field(
+    enum_values: list[str] | None = Field(
         default=None,
         description=(
             "Optional finite categorical label set (classification) for a string field. Provide ONLY when a closed, "
@@ -168,7 +183,7 @@ class InferredSchema(BaseModel):
             "reduce hallucinated fields. Internal diagnostic / quality aid; not required for downstream extraction."
         )
     )
-    fields: List[FieldSpec] = Field(
+    fields: list[FieldSpec] = Field(
         description=(
             "Ordered list of proposed fields derived strictly from observable, repeatable signals in the "
             "examples and aligned with the purpose."
@@ -199,13 +214,13 @@ class InferredSchema(BaseModel):
             return cls.model_validate_json(f.read())
 
     @property
-    def model(self) -> Type[BaseModel]:
+    def model(self) -> type[BaseModel]:
         """Dynamically materialized Pydantic model for the inferred schema.
 
         Equivalent to calling :meth:`build_model` each access (not cached).
 
         Returns:
-            Type[BaseModel]: Fresh model type reflecting ``fields`` ordering.
+            type[BaseModel]: Fresh model type reflecting ``fields`` ordering.
         """
         return self.build_model()
 
@@ -220,7 +235,7 @@ class InferredSchema(BaseModel):
             instructions=self.inference_prompt, response_format=self.model, top_p=None, temperature=None
         )
 
-    def build_model(self) -> Type[BaseModel]:
+    def build_model(self) -> type[BaseModel]:
         """Create a new dynamic ``BaseModel`` class adhering to this schema.
 
         Implementation details:
@@ -231,9 +246,14 @@ class InferredSchema(BaseModel):
               introduced later by modifying this logic if needed.
 
         Returns:
-            Type[BaseModel]: New (not cached) model type; order matches ``fields``.
+            type[BaseModel]: New (not cached) model type; order matches ``fields``.
         """
-        type_map: dict[str, type] = {"string": str, "integer": int, "float": float, "boolean": bool}
+        type_map: dict[str, type] = {
+            "string": str,
+            "integer": int,
+            "float": float,
+            "boolean": bool,
+        }
         fields: dict[str, tuple[type, object]] = {}
 
         for spec in self.fields:
@@ -254,7 +274,11 @@ class InferredSchema(BaseModel):
                 enum_cls = Enum(enum_class_name, members)  # type: ignore[arg-type]
                 py_type = enum_cls
             else:
-                py_type = type_map[spec.type]
+                if spec.type.endswith("_array"):
+                    base = spec.type.rsplit("_", 1)[0]
+                    py_type = list[type_map[base]]  # type: ignore[index]
+                else:
+                    py_type = type_map[spec.type]
             fields[spec.name] = (py_type, Field(description=spec.description))
 
         model = create_model("InferredSchema", **fields)  # type: ignore[call-arg]
@@ -282,7 +306,7 @@ class SchemaInferenceInput(BaseModel):
             relevance & exclusion of outcome labels.
     """
 
-    examples: List[str] = Field(
+    examples: list[str] = Field(
         description=(
             "Representative sample texts (strings). Provide only data the schema should generalize over; "
             "exclude outliers not in scope."
@@ -306,7 +330,8 @@ Task:
    to concrete recurring evidence in the examples (or flags gaps). Use concise bullet‑style
    sentences (still a plain string) such as: "purpose facet -> supporting pattern / gap".
    This MUST NOT introduce new domain facts beyond the examples & purpose.
-4. Propose a minimal flat set of scalar fields (no nesting / arrays) that are reliably extractable.
+4. Propose a minimal flat set of scalar fields (and ONLY when justified,
+   homogeneous primitive arrays) that are reliably extractable.
 5. Skip fields likely missing in a large share (>~20%) of realistic inputs.
 6. Provide enum_values ONLY when a small stable closed categorical set (2–24 lowercase tokens)
     is clearly evidenced; never invent.
@@ -324,7 +349,11 @@ Rules:
     (e.g. *_count, *_seconds, *_ms, *_usd, *_ratio, *_score). Avoid ambiguous bare numeric names.
 - Boolean field names MUST start with 'is_' followed by a positive predicate (e.g. is_active,
   is_delayed). Avoid negated forms.
-- No arrays, objects, composite encodings, or merged multi-concept fields.
+- No nested objects or mixed-type arrays. Homogeneous primitive arrays are allowed ONLY if each element is an atomic
+    scalar signal (use *_array types: string_array, integer_array, float_array, boolean_array). The array is expected to
+    contain 0..N such elements per record.
+- Array field names MUST end with '_array' (e.g. keywords_array, tag_ids_array). Do not use plural-only forms
+    (e.g. keywords) for arrays; the suffix makes container semantics explicit.
 - Descriptions: concise, objective extraction rules (no marketing/emotion/speculation).
 - enum_values only for string fields with stable closed vocab; omit otherwise.
 - Exclude direct outcome labels (e.g. attrition_probability, will_buy, purchase_likelihood)
@@ -465,12 +494,21 @@ def _basic_field_list_validation(parsed: InferredSchema) -> None:
         raise ValueError("no fields suggested")
     if len(names) != len(set(names)):
         raise ValueError("duplicate field names detected")
-    allowed = {"string", "integer", "float", "boolean"}
+    allowed = {
+        "string",
+        "integer",
+        "float",
+        "boolean",
+        "string_array",
+        "integer_array",
+        "float_array",
+        "boolean_array",
+    }
     for f in parsed.fields:
         if f.type not in allowed:
             raise ValueError(f"unsupported field type: {f.type}")
         if f.enum_values is not None:
             if f.type != "string":
-                raise ValueError(f"enum_values only allowed for string field: {f.name}")
+                raise ValueError(f"enum_values only allowed for plain string field: {f.name}")
             if not (2 <= len(f.enum_values) <= 24):
                 raise ValueError(f"enum_values length out of bounds for field {f.name}")

openaivec/_serialize.py

@@ -4,19 +4,19 @@ This module provides utilities for converting Pydantic BaseModel classes
 to and from JSON schema representations with simplified, maintainable code.
 """
 
-from typing import Any, Dict, List, Literal, Tuple, Type, Union
+from typing import Any, Literal
 
 from pydantic import BaseModel, Field, create_model
 
 __all__ = []
 
 
-def serialize_base_model(obj: Type[BaseModel]) -> Dict[str, Any]:
+def serialize_base_model(obj: type[BaseModel]) -> dict[str, Any]:
     """Serialize a Pydantic BaseModel to JSON schema."""
     return obj.model_json_schema()
 
 
-def dereference_json_schema(json_schema: Dict[str, Any]) -> Dict[str, Any]:
+def dereference_json_schema(json_schema: dict[str, Any]) -> dict[str, Any]:
     """Dereference JSON schema by resolving $ref pointers with circular reference protection."""
     model_map = json_schema.get("$defs", {})
 
@@ -61,7 +61,7 @@ def dereference_json_schema(json_schema: Dict[str, Any]) -> Dict[str, Any]:
 # ============================================================================
 
 
-def _resolve_union_type(union_options: List[Dict[str, Any]]) -> Type:
+def _resolve_union_type(union_options: list[dict[str, Any]]) -> type:
     """Resolve anyOf/oneOf to Union type."""
     union_types = []
     for option in union_options:
@@ -75,12 +75,14 @@ def _resolve_union_type(union_options: List[Dict[str, Any]]) -> Type:
     elif len(union_types) == 2 and type(None) in union_types:
         # Optional type: T | None
         non_none_type = next(t for t in union_types if t is not type(None))
-        return Union[non_none_type, type(None)]  # type: ignore[return-value]
+        return non_none_type | None  # type: ignore[return-value]
     else:
+        from typing import Union
+
         return Union[tuple(union_types)]  # type: ignore[return-value]
 
 
-def _resolve_basic_type(type_name: str, field_def: Dict[str, Any]) -> Type:
+def _resolve_basic_type(type_name: str, field_def: dict[str, Any]) -> type:
     """Resolve basic JSON schema types to Python types."""
     type_mapping = {
         "string": str,
@@ -101,14 +103,14 @@ def _resolve_basic_type(type_name: str, field_def: Dict[str, Any]) -> Type:
     elif type_name == "array":
         if "items" in field_def:
             inner_type = parse_field(field_def["items"])
-            return List[inner_type]
+            return list[inner_type]
         else:
-            return List[Any]
+            return list[Any]
     else:
         raise ValueError(f"Unsupported type: {type_name}")
 
 
-def parse_field(field_def: Dict[str, Any]) -> Type:
+def parse_field(field_def: dict[str, Any]) -> type:
     """Parse a JSON schema field definition to a Python type.
 
     Simplified version with clear separation of concerns.
@@ -141,17 +143,19 @@ def _create_field_info(description: str | None, default_value: Any, is_required:
         return Field(default=default_value, description=description) if description else Field(default=default_value)
 
 
-def _make_optional_if_needed(field_type: Type, is_required: bool, has_default: bool) -> Type:
+def _make_optional_if_needed(field_type: type, is_required: bool, has_default: bool) -> type:
     """Make field type optional if needed."""
     if is_required or has_default:
         return field_type
 
     # Check if already nullable
+    from typing import Union
+
     if hasattr(field_type, "__origin__") and field_type.__origin__ is Union and type(None) in field_type.__args__:
         return field_type
 
     # Make optional
-    return Union[field_type, type(None)]  # type: ignore[return-value]
+    return field_type | None  # type: ignore[return-value]
 
 
 # ============================================================================
@@ -159,7 +163,7 @@ def _make_optional_if_needed(field_type: Type, is_required: bool, has_default: b
 # ============================================================================
 
 
-def _process_enum_field(field_name: str, field_def: Dict[str, Any], is_required: bool) -> Tuple[Type, Field]:  # type: ignore[type-arg]
+def _process_enum_field(field_name: str, field_def: dict[str, Any], is_required: bool) -> tuple[type, Field]:  # type: ignore[type-arg]
     """Process enum field with Literal type."""
     enum_values = field_def["enum"]
 
@@ -175,14 +179,14 @@ def _process_enum_field(field_name: str, field_def: Dict[str, Any], is_required:
     has_default = default_value is not None
 
     if not is_required and not has_default:
-        literal_type = Union[literal_type, type(None)]  # type: ignore[assignment]
+        literal_type = literal_type | None  # type: ignore[assignment]
         default_value = None
 
     field_info = _create_field_info(description, default_value, is_required)
     return literal_type, field_info  # type: ignore[return-value]
 
 
-def _process_regular_field(field_name: str, field_def: Dict[str, Any], is_required: bool) -> Tuple[Type, Field]:  # type: ignore[type-arg]
+def _process_regular_field(field_name: str, field_def: dict[str, Any], is_required: bool) -> tuple[type, Field]:  # type: ignore[type-arg]
     """Process regular (non-enum) field."""
     field_type = parse_field(field_def)
     description = field_def.get("description")
@@ -204,7 +208,7 @@ def _process_regular_field(field_name: str, field_def: Dict[str, Any], is_requir
 # ============================================================================
 
 
-def deserialize_base_model(json_schema: Dict[str, Any]) -> Type[BaseModel]:
+def deserialize_base_model(json_schema: dict[str, Any]) -> type[BaseModel]:
     """Deserialize a JSON schema to a Pydantic BaseModel class.
 
     Refactored version with clear separation of concerns and simplified logic.

openaivec/_util.py

@@ -2,8 +2,9 @@ import asyncio
 import functools
 import re
 import time
+from collections.abc import Awaitable, Callable
 from dataclasses import dataclass
-from typing import Awaitable, Callable, List, Type, TypeVar
+from typing import TypeVar
 
 import numpy as np
 import tiktoken
@@ -36,14 +37,14 @@ def get_exponential_with_cutoff(scale: float) -> float:
 
 
 def backoff(
-    exceptions: List[Type[Exception]],
+    exceptions: list[type[Exception]],
     scale: int | None = None,
     max_retries: int | None = None,
 ) -> Callable[..., V]:
     """Decorator implementing exponential back‑off retry logic.
 
     Args:
-        exceptions (List[Type[Exception]]): List of exception types that trigger a retry.
+        exceptions (list[type[Exception]]): List of exception types that trigger a retry.
         scale (int | None): Initial scale parameter for the exponential jitter.
             This scale is used as the mean for the first delay's exponential
             distribution and doubles with each subsequent retry. If ``None``,
@@ -88,14 +89,14 @@ def backoff(
 
 
 def backoff_async(
-    exceptions: List[Type[Exception]],
+    exceptions: list[type[Exception]],
     scale: int | None = None,
     max_retries: int | None = None,
 ) -> Callable[..., Awaitable[V]]:
     """Asynchronous version of the backoff decorator.
 
     Args:
-        exceptions (List[Type[Exception]]): List of exception types that trigger a retry.
+        exceptions (list[type[Exception]]): List of exception types that trigger a retry.
         scale (int | None): Initial scale parameter for the exponential jitter.
             This scale is used as the mean for the first delay's exponential
             distribution and doubles with each subsequent retry. If ``None``,
@@ -145,7 +146,7 @@ class TextChunker:
 
     enc: tiktoken.Encoding
 
-    def split(self, original: str, max_tokens: int, sep: List[str]) -> List[str]:
+    def split(self, original: str, max_tokens: int, sep: list[str]) -> list[str]:
         """Token‑aware sentence segmentation.
 
         The text is first split by the given separators, then greedily packed
@@ -154,11 +155,11 @@ class TextChunker:
         Args:
             original (str): Original text to split.
             max_tokens (int): Maximum number of tokens allowed per chunk.
-            sep (List[str]): List of separator patterns used by
+            sep (list[str]): List of separator patterns used by
                 :pyfunc:`re.split`.
 
         Returns:
-            List[str]: List of text chunks respecting the ``max_tokens`` limit.
+            list[str]: List of text chunks respecting the ``max_tokens`` limit.
         """
         sentences = re.split(f"({'|'.join(sep)})", original)
         sentences = [s.strip() for s in sentences if s.strip()]