selectools 0.25.1__tar.gz → 0.26.0__tar.gz

{selectools-0.25.1/src/selectools.egg-info → selectools-0.26.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: selectools
-Version: 0.25.1
+Version: 0.26.0
 Summary: Production-ready Python framework for AI agents with multi-agent graphs, hybrid RAG, guardrails, audit logging, 50 evaluators, and a visual builder. Supports OpenAI, Anthropic, Gemini, Ollama. By NichevLabs.
 Author-email: John Nichev <johnnichev@gmail.com>
 Maintainer-email: NichevLabs <support@nichevlabs.com>
@@ -108,6 +108,19 @@ result = AgentGraph.chain(planner, writer, reviewer).run("Write a blog post")
 # selectools serve agent.yaml
 ```
 
+## What's New in v0.26
+
+### v0.26.0 — Safety Patch & Verified Registry
+
+A mid-bake safety patch plus a fully source-verified model registry.
+
+- **Confirm parser safety fix** — `RegexConfirmParser` treated replies like "se você não pode apagar, tudo bem" / "tú no puedes borrar" as confirmations: the restated-verb branch matched the action verb without checking for negation elsewhere in the reply. A negation token anywhere now vetoes that branch; bare confirms are unaffected. **All `selectools.pending` consumers should upgrade immediately.**
+- **Model registry refresh** — 152 → 115 models, every entry source-verified: claude-fable-5 / opus-4-8 / opus-4-7, the gpt-5.5 family, and the gemini-3.5 line added; retired and shut-down models removed (**BREAKING** if you referenced their registry constants — e.g. `AnthropicModels.CLAUDE_3_5_SONNET` — but those IDs 404 at their providers regardless); `claude-opus-4-1` mispricing corrected ($5/$25 → $15/$75); gpt-5 family context specs fixed (400k/128k); `GeminiEmbeddingProvider` default moved off the retired `text-embedding-004` to `gemini-embedding-001`.
+- **Cache-aware cost calculation** — `calculate_cost(..., cache_read_input_tokens=..., cache_creation_input_tokens=...)` prices prompt-cache traffic at Anthropic's published rates (reads 0.1×, 5-min writes 1.25×); `AnthropicProvider`'s `UsageStats.cost_usd` is now cache-accurate.
+- **Fixes** — A2A returns JSON-RPC -32602 (not HTTP 500) for non-dict `message.parts` elements; Gemini embedding dimension constant corrected to the actual 3072 for `gemini-embedding-001`/`-2` (vectors themselves unchanged); bake-hunt test module no longer requires optional extras in CI.
+
+See `CHANGELOG.md` for the full entry (7,420 tests, 111 examples, 115 models).
+
 ## What's New in v0.25
 
 ### v0.25.0 — Hardening & v1.0 Prep
@@ -659,7 +672,7 @@ report.to_html("report.html")
 - **Advanced Chunking**: Semantic + contextual chunking for better retrieval
 - **Dynamic Tool Loading**: Plugin system with hot-reload support
 - **Response Caching**: InMemoryCache and RedisCache with stats tracking
-- **152 Model Registry**: Type-safe constants with pricing and metadata
+- **115 Model Registry**: Type-safe constants with pricing and metadata
 - **Pre-built Toolbox**: 48 tools for files, data, text, datetime, web, code, search, GitHub, DB, calculator, email, PDF, Slack, Notion, Linear
 - **Persistent Sessions**: 4 backends (JSON file, SQLite, Redis, Supabase) with TTL and cross-session search
 - **Entity Memory**: LLM-based named entity extraction and tracking
@@ -1035,7 +1048,7 @@ See [docs/modules/STREAMING.md](https://github.com/johnnichev/selectools/blob/ma
 ```python
 from selectools.models import OpenAI, Anthropic, Gemini, Ollama
 
-# IDE autocomplete for all 152 models with pricing metadata
+# IDE autocomplete for all 115 models with pricing metadata
 model = OpenAI.GPT_4O_MINI
 print(f"Cost: ${model.prompt_cost}/${model.completion_cost} per 1M tokens")
 print(f"Context: {model.context_window:,} tokens")

{selectools-0.25.1 → selectools-0.26.0}/README.md

@@ -30,6 +30,19 @@ result = AgentGraph.chain(planner, writer, reviewer).run("Write a blog post")
 # selectools serve agent.yaml
 ```
 
+## What's New in v0.26
+
+### v0.26.0 — Safety Patch & Verified Registry
+
+A mid-bake safety patch plus a fully source-verified model registry.
+
+- **Confirm parser safety fix** — `RegexConfirmParser` treated replies like "se você não pode apagar, tudo bem" / "tú no puedes borrar" as confirmations: the restated-verb branch matched the action verb without checking for negation elsewhere in the reply. A negation token anywhere now vetoes that branch; bare confirms are unaffected. **All `selectools.pending` consumers should upgrade immediately.**
+- **Model registry refresh** — 152 → 115 models, every entry source-verified: claude-fable-5 / opus-4-8 / opus-4-7, the gpt-5.5 family, and the gemini-3.5 line added; retired and shut-down models removed (**BREAKING** if you referenced their registry constants — e.g. `AnthropicModels.CLAUDE_3_5_SONNET` — but those IDs 404 at their providers regardless); `claude-opus-4-1` mispricing corrected ($5/$25 → $15/$75); gpt-5 family context specs fixed (400k/128k); `GeminiEmbeddingProvider` default moved off the retired `text-embedding-004` to `gemini-embedding-001`.
+- **Cache-aware cost calculation** — `calculate_cost(..., cache_read_input_tokens=..., cache_creation_input_tokens=...)` prices prompt-cache traffic at Anthropic's published rates (reads 0.1×, 5-min writes 1.25×); `AnthropicProvider`'s `UsageStats.cost_usd` is now cache-accurate.
+- **Fixes** — A2A returns JSON-RPC -32602 (not HTTP 500) for non-dict `message.parts` elements; Gemini embedding dimension constant corrected to the actual 3072 for `gemini-embedding-001`/`-2` (vectors themselves unchanged); bake-hunt test module no longer requires optional extras in CI.
+
+See `CHANGELOG.md` for the full entry (7,420 tests, 111 examples, 115 models).
+
 ## What's New in v0.25
 
 ### v0.25.0 — Hardening & v1.0 Prep
@@ -581,7 +594,7 @@ report.to_html("report.html")
 - **Advanced Chunking**: Semantic + contextual chunking for better retrieval
 - **Dynamic Tool Loading**: Plugin system with hot-reload support
 - **Response Caching**: InMemoryCache and RedisCache with stats tracking
-- **152 Model Registry**: Type-safe constants with pricing and metadata
+- **115 Model Registry**: Type-safe constants with pricing and metadata
 - **Pre-built Toolbox**: 48 tools for files, data, text, datetime, web, code, search, GitHub, DB, calculator, email, PDF, Slack, Notion, Linear
 - **Persistent Sessions**: 4 backends (JSON file, SQLite, Redis, Supabase) with TTL and cross-session search
 - **Entity Memory**: LLM-based named entity extraction and tracking
@@ -957,7 +970,7 @@ See [docs/modules/STREAMING.md](https://github.com/johnnichev/selectools/blob/ma
 ```python
 from selectools.models import OpenAI, Anthropic, Gemini, Ollama
 
-# IDE autocomplete for all 152 models with pricing metadata
+# IDE autocomplete for all 115 models with pricing metadata
 model = OpenAI.GPT_4O_MINI
 print(f"Cost: ${model.prompt_cost}/${model.completion_cost} per 1M tokens")
 print(f"Context: {model.context_window:,} tokens")

{selectools-0.25.1 → selectools-0.26.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "selectools"
-version = "0.25.1"
+version = "0.26.0"
 description = "Production-ready Python framework for AI agents with multi-agent graphs, hybrid RAG, guardrails, audit logging, 50 evaluators, and a visual builder. Supports OpenAI, Anthropic, Gemini, Ollama. By NichevLabs."
 readme = "README.md"
 requires-python = ">=3.9"

{selectools-0.25.1 → selectools-0.26.0}/src/selectools/__init__.py

@@ -1,6 +1,6 @@
 """Public exports for the selectools package."""
 
-__version__ = "0.25.1"
+__version__ = "0.26.0"
 
 # Import submodules (lazy loading for optional dependencies)
 from . import embeddings, evals, guardrails, models, observe, patterns, rag, toolbox

{selectools-0.25.1 → selectools-0.26.0}/src/selectools/a2a/server.py

@@ -94,6 +94,10 @@ def _extract_prompt(parts: List[Dict[str, Any]]) -> str:
     """
     chunks: List[str] = []
     for part in parts:
+        if not isinstance(part, dict):
+            # Defensive: validation rejects non-dict parts up front, but a
+            # stray list element must never crash prompt assembly.
+            continue
         kind = part.get("kind")
         if kind == "text" and isinstance(part.get("text"), str):
             chunks.append(part["text"])
@@ -300,6 +304,8 @@ class A2AServer:
         parts = message.get("parts")
         if not isinstance(parts, list) or not parts:
             return None, "'message.parts' must be a non-empty array"
+        if not all(isinstance(p, dict) for p in parts):
+            return None, "every entry in 'message.parts' must be an object"
         if not any(
             isinstance(p, dict) and p.get("kind") == "text" and isinstance(p.get("text"), str)
             for p in parts

{selectools-0.25.1 → selectools-0.26.0}/src/selectools/embeddings/gemini.py

@@ -16,14 +16,19 @@ class GeminiEmbeddingProvider(EmbeddingProvider):
     """
     Google Gemini embedding provider.
 
-    Supports models: text-embedding-001, text-embedding-004
+    Supports models: gemini-embedding-001, gemini-embedding-2
+
+    Requests do not set ``output_dimensionality``, so vectors come back at
+    each model's default dimensionality (3072 for gemini-embedding-001 and
+    gemini-embedding-2, which support MRL truncation but are not truncated
+    here). See ``_get_model_dimension`` for the full dimension/MRL story.
 
     Example:
         >>> from selectools.embeddings import GeminiEmbeddingProvider
         >>> from selectools.models import Gemini
         >>>
         >>> embedder = GeminiEmbeddingProvider(
-        ...     model=Gemini.Embeddings.EMBEDDING_004.id
+        ...     model=Gemini.Embeddings.EMBEDDING_001.id
         ... )
         >>> embedding = embedder.embed_text("Hello, world!")
         >>> print(f"Dimension: {len(embedding)}")
@@ -31,7 +36,9 @@ class GeminiEmbeddingProvider(EmbeddingProvider):
 
     def __init__(
         self,
-        model: str = "text-embedding-004",
+        # Default changed from text-embedding-004 (retired by Google) to its
+        # documented replacement gemini-embedding-001.
+        model: str = "gemini-embedding-001",
         api_key: Optional[str] = None,
         task_type: str = "retrieval_document",
     ):
@@ -39,7 +46,7 @@ class GeminiEmbeddingProvider(EmbeddingProvider):
         Initialize Gemini embedding provider.
 
         Args:
-            model: Model name (default: text-embedding-004)
+            model: Model name (default: gemini-embedding-001)
             api_key: Google API key (defaults to GEMINI_API_KEY or GOOGLE_API_KEY env var)
             task_type: Task type for embeddings (retrieval_document, retrieval_query,
                       semantic_similarity, classification, clustering)
@@ -64,12 +71,31 @@ class GeminiEmbeddingProvider(EmbeddingProvider):
         self._dimension = self._get_model_dimension()
 
     def _get_model_dimension(self) -> int:
-        """Get the embedding dimension for the model."""
-        if "embedding-004" in self.model or "embedding-001" in self.model:
-            return 768
-        else:
-            logger.warning(f"Unknown Gemini embedding model '{self.model}', assuming dimension 768")
+        """Get the embedding dimension for the model.
+
+        The declared dimension must match what the API actually returns. This
+        provider never passes ``output_dimensionality`` in its requests (see
+        ``embed_text``/``embed_texts``/``embed_query``), so the API returns
+        each model's default, full-size vector:
+
+        - ``gemini-embedding-001`` and ``gemini-embedding-2``: 3072 by
+          default. Both are trained with Matryoshka Representation Learning
+          (MRL), so vectors *can* be truncated server-side (Google recommends
+          768, 1536, or 3072) by passing ``output_dimensionality`` — but this
+          provider does not request truncation, and existing vector store
+          indexes built with it hold 3072-dim vectors. Note that
+          ``gemini-embedding-001`` only normalizes the full 3072-dim output;
+          truncated outputs would need manual re-normalization.
+        - ``text-embedding-004`` (retired by Google): natively 768.
+
+        Source: https://ai.google.dev/gemini-api/docs/embeddings
+        """
+        if "embedding-004" in self.model:
             return 768
+        if "embedding-001" in self.model or "embedding-2" in self.model:
+            return 3072
+        logger.warning(f"Unknown Gemini embedding model '{self.model}', assuming dimension 3072")
+        return 3072
 
     def embed_text(self, text: str) -> List[float]:
         """