refactor: move off python 3.10

.github/workflows/quality.yml

@@ -17,7 +17,7 @@ jobs:
     timeout-minutes: 90 # TODO: need to reduce this after we figure out our testing strategy.
     strategy:
       matrix:
-        python-version: ["3.10", "3.11", "3.12"] # Need to add 3.13 once we resolve outlines issues.
+        python-version: ["3.11", "3.12", "3.13"]
     steps:
       - uses: actions/checkout@v4
       - name: Free disk space

cli/decompose/decompose.py

@@ -1,7 +1,7 @@
 import json
 import keyword
 import re
-from enum import Enum
+from enum import StrEnum
 from graphlib import TopologicalSorter
 from pathlib import Path
 from typing import Annotated
@@ -13,7 +13,7 @@
 
 # Must maintain declaration order
 # Newer versions must be declared on the bottom
-class DecompVersion(str, Enum):
+class DecompVersion(StrEnum):
     latest = "latest"
     v1 = "v1"
     # v2 = "v2"

cli/decompose/pipeline.py

@@ -1,8 +1,6 @@
 import re
-from enum import Enum
-from typing import Literal, TypedDict
-
-from typing_extensions import NotRequired
+from enum import StrEnum
+from typing import Literal, NotRequired, TypedDict
 
 from mellea import MelleaSession
 from mellea.backends import ModelOption
@@ -46,7 +44,7 @@ class DecompPipelineResult(TypedDict):
     final_response: NotRequired[str]
 
 
-class DecompBackend(str, Enum):
+class DecompBackend(StrEnum):
     ollama = "ollama"
     openai = "openai"
     rits = "rits"

cli/decompose/prompt_modules/subtask_constraint_assign/_subtask_constraint_assign.py

@@ -1,8 +1,6 @@
 import re
 from collections.abc import Callable, Sequence
-from typing import TypedDict, TypeVar, cast, final
-
-from typing_extensions import Unpack
+from typing import TypedDict, TypeVar, Unpack, cast, final
 
 from mellea import MelleaSession
 from mellea.backends import ModelOption

cli/decompose/prompt_modules/subtask_prompt_generator/_subtask_prompt_generator.py

@@ -1,8 +1,6 @@
 import re
 from collections.abc import Callable, Sequence
-from typing import TypedDict, TypeVar, cast, final
-
-from typing_extensions import Unpack
+from typing import TypedDict, TypeVar, Unpack, cast, final
 
 from mellea import MelleaSession
 from mellea.backends import ModelOption

docs/examples/agents/react/react_from_scratch/react.py

@@ -85,7 +85,6 @@ def call_tool(self, tool: ReactTool, kwargs_json: str):
 
     def tool_name_schema(self):
         names = self.tool_names()
-        # Python 3.10 compatible: use Enum instead of Literal[*names] (requires 3.11+)
         ToolEnum = Enum("ToolEnum", {name: name for name in names})
         return pydantic.create_model("ToolSelectionSchema", tool=(ToolEnum, ...))
 

docs/examples/agents/react/react_from_scratch/react_instruct.py

@@ -82,7 +82,6 @@ def call_tool(self, tool: ReactTool, kwargs_json: str):
 
     def tool_name_schema(self):
         names = self.tool_names()
-        # Python 3.10 compatible: use Enum instead of Literal[*names] (requires 3.11+)
         ToolEnum = Enum("ToolEnum", {name: name for name in names})
         return pydantic.create_model("ToolSelectionSchema", tool=(ToolEnum, ...))
 

docs/examples/bedrock/bedrock_litellm_example.py

@@ -2,7 +2,7 @@
 # SKIP REASON: Requires an AWS bearer token for Bedrock.
 #!/usr/bin/env -S uv run --script
 # /// script
-# requires-python = ">=3.10"
+# requires-python = ">=3.11"
 # dependencies = [
 #   "mellea[litellm]",
 #   "boto3" # including so that this example works before the next release.

docs/examples/bedrock/bedrock_openai_example.py

@@ -2,7 +2,7 @@
 # SKIP REASON: Requires an AWS bearer token for Bedrock.
 #!/usr/bin/env -S uv run --script
 # /// script
-# requires-python = ">=3.10"
+# requires-python = ">=3.11"
 # dependencies = [
 #   "mellea[openai]"
 # ]

mellea/backends/tools.py

@@ -435,7 +435,6 @@ def validate_tool_arguments(
             if len(unique_types) == 1:
                 param_type = unique_types[0]
             else:
-                # Use modern union syntax (Python 3.10+)
                 from functools import reduce
                 from operator import or_
 

mellea/formatters/granite/base/types.py

@@ -3,13 +3,12 @@
 """Common shared types."""
 
 # Standard
-from typing import Annotated, Literal, TypeAlias
+from typing import Annotated, Any, Literal, TypeAlias
 
 import pydantic
 
 # Third Party
 from pydantic import Field, StringConstraints
-from typing_extensions import Any
 
 
 class NoDefaultsMixin:

pyproject.toml

@@ -10,7 +10,7 @@ authors = [
 ]
 description = "mellea is a library for writing generative programs"
 readme = "README.md"
-requires-python = ">=3.10"
+requires-python = ">=3.11"
 classifiers = [
     "Programming Language :: Python :: 3",
     "License :: OSI Approved :: Apache Software License",
@@ -166,7 +166,7 @@ docs = [
 # -----------------------------
 
 [tool.ruff]
-target-version = "py310"
+target-version = "py311"
 respect-gitignore = true
 
 [tool.ruff.format]
@@ -199,7 +199,6 @@ ignore = [
   "F811",    # redefinition of the same function
   "PL",      # Pylint
   "RUF012",  # Mutable Class Attributes
-  "PD901",   # Generic variable name 'df' for DataFrames (deprecated rule, but needed while PD is enabled)
   "C901",    # Complexity warnings
 ]
 
@@ -232,7 +231,7 @@ split-on-trailing-comma = false
 install_types = true
 non_interactive = true
 disable_error_code = ["empty-body", "import-untyped"]
-python_version = "3.10"
+python_version = "3.11"
 
 [[tool.mypy.overrides]]
 # Keep import-not-found suppressed for optional dependencies