verifiers 0.2.1.dev11__tar.gz → 0.2.1.dev12__tar.gz

{verifiers-0.2.1.dev11 → verifiers-0.2.1.dev12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: verifiers
-Version: 0.2.1.dev11
+Version: 0.2.1.dev12
 Summary: Verifiers: Environments for LLM Reinforcement Learning
 Project-URL: Homepage, https://github.com/primeintellect-ai/verifiers
 Project-URL: Documentation, https://github.com/primeintellect-ai/verifiers

{verifiers-0.2.1.dev11 → verifiers-0.2.1.dev12}/verifiers/v1/dialects/chat.py

@@ -30,6 +30,16 @@ from verifiers.v1.types import (
     content_to_parts,
 )
 
+
+class ModdedChatCompletion(ChatCompletion):
+    """The OpenAI SDK closes `service_tier` to a fixed `Literal`, but providers return tiers
+    outside it (e.g. Prime's `provisioned`), which makes `model_validate` reject an otherwise
+    valid completion. Widen the field to a plain string — we don't consume it — so parsing stays
+    lenient about the label instead of dropping it."""
+
+    service_tier: str | None = None
+
+
 FINISH_REASONS = frozenset({"stop", "length", "tool_calls"})
 
 # Providers name the model's reasoning differently; read them in the v0 client's precedence.
@@ -262,7 +272,7 @@ class ChatStreamParser(StreamParser):
             self.message["reasoning_details"] = self.reasoning_details
         head = self.head or {}
         return response_from_wire(
-            ChatCompletion.model_validate(
+            ModdedChatCompletion.model_validate(
                 {
                     "id": head.get("id", "vf-intercept"),
                     "object": "chat.completion",
@@ -284,7 +294,7 @@ class ChatStreamParser(StreamParser):
 class ChatDialect(Dialect[dict, ChatCompletion]):
     routes = ("/v1/chat/completions",)
     upstream_path = "/chat/completions"
-    response_type = ChatCompletion
+    response_type = ModdedChatCompletion
 
     def parse_request(self, body: dict) -> tuple[Messages, list[Tool] | None]:
         messages: Messages = []