prompture 0.0.37.dev3__tar.gz → 0.0.38.dev1__tar.gz

{prompture-0.0.37.dev3/prompture.egg-info → prompture-0.0.38.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prompture
-Version: 0.0.37.dev3
+Version: 0.0.38.dev1
 Summary: Ask LLMs to return structured JSON and run cross-model tests. API-first.
 Author-email: Juan Denis <juan@vene.co>
 License-Expression: MIT

{prompture-0.0.37.dev3 → prompture-0.0.38.dev1}/prompture/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.0.37.dev3'
-__version_tuple__ = version_tuple = (0, 0, 37, 'dev3')
+__version__ = version = '0.0.38.dev1'
+__version_tuple__ = version_tuple = (0, 0, 38, 'dev1')
 
 __commit_id__ = commit_id = None

{prompture-0.0.37.dev3 → prompture-0.0.38.dev1}/prompture/drivers/async_lmstudio_driver.py

@@ -15,6 +15,7 @@ logger = logging.getLogger(__name__)
 
 class AsyncLMStudioDriver(AsyncDriver):
     supports_json_mode = True
+    supports_json_schema = True
     supports_vision = True
 
     MODEL_PRICING = {"default": {"prompt": 0.0, "completion": 0.0}}
@@ -70,9 +71,21 @@ class AsyncLMStudioDriver(AsyncDriver):
             "temperature": merged_options.get("temperature", 0.7),
         }
 
-        # Native JSON mode support
+        # Native JSON mode support (LM Studio requires json_schema, not json_object)
         if merged_options.get("json_mode"):
-            payload["response_format"] = {"type": "json_object"}
+            json_schema = merged_options.get("json_schema")
+            if json_schema:
+                payload["response_format"] = {
+                    "type": "json_schema",
+                    "json_schema": {
+                        "name": "extraction",
+                        "schema": json_schema,
+                    },
+                }
+            else:
+                # No schema provided — omit response_format entirely;
+                # LM Studio rejects "json_object" type.
+                pass
 
         async with httpx.AsyncClient() as client:
             try:
@@ -128,7 +141,7 @@ class AsyncLMStudioDriver(AsyncDriver):
     async def unload_model(self, model: str) -> dict[str, Any]:
         """Unload a model from LM Studio via POST /api/v1/models/unload."""
         url = f"{self.base_url}/api/v1/models/unload"
-        payload = {"model": model}
+        payload = {"instance_id": model}
         async with httpx.AsyncClient() as client:
             r = await client.post(url, json=payload, headers=self._headers, timeout=30)
             r.raise_for_status()

{prompture-0.0.37.dev3 → prompture-0.0.38.dev1}/prompture/drivers/async_ollama_driver.py

@@ -15,6 +15,7 @@ logger = logging.getLogger(__name__)
 
 class AsyncOllamaDriver(AsyncDriver):
     supports_json_mode = True
+    supports_json_schema = True
     supports_vision = True
 
     MODEL_PRICING = {"default": {"prompt": 0.0, "completion": 0.0}}
@@ -42,9 +43,10 @@ class AsyncOllamaDriver(AsyncDriver):
             "stream": False,
         }
 
-        # Native JSON mode support
+        # Native JSON mode / structured output support
         if merged_options.get("json_mode"):
-            payload["format"] = "json"
+            json_schema = merged_options.get("json_schema")
+            payload["format"] = json_schema if json_schema else "json"
 
         if "temperature" in merged_options:
             payload["temperature"] = merged_options["temperature"]
@@ -95,7 +97,8 @@ class AsyncOllamaDriver(AsyncDriver):
         }
 
         if merged_options.get("json_mode"):
-            payload["format"] = "json"
+            json_schema = merged_options.get("json_schema")
+            payload["format"] = json_schema if json_schema else "json"
 
         if "temperature" in merged_options:
             payload["temperature"] = merged_options["temperature"]

{prompture-0.0.37.dev3 → prompture-0.0.38.dev1}/prompture/drivers/lmstudio_driver.py

@@ -12,6 +12,7 @@ logger = logging.getLogger(__name__)
 
 class LMStudioDriver(Driver):
     supports_json_mode = True
+    supports_json_schema = True
     supports_vision = True
 
     # LM Studio is local – costs are always zero.
@@ -82,9 +83,21 @@ class LMStudioDriver(Driver):
             "temperature": merged_options.get("temperature", 0.7),
         }
 
-        # Native JSON mode support
+        # Native JSON mode support (LM Studio requires json_schema, not json_object)
         if merged_options.get("json_mode"):
-            payload["response_format"] = {"type": "json_object"}
+            json_schema = merged_options.get("json_schema")
+            if json_schema:
+                payload["response_format"] = {
+                    "type": "json_schema",
+                    "json_schema": {
+                        "name": "extraction",
+                        "schema": json_schema,
+                    },
+                }
+            else:
+                # No schema provided — omit response_format entirely;
+                # LM Studio rejects "json_object" type.
+                pass
 
         try:
             logger.debug(f"Sending request to LM Studio endpoint: {self.endpoint}")
@@ -152,7 +165,7 @@ class LMStudioDriver(Driver):
     def unload_model(self, model: str) -> dict[str, Any]:
         """Unload a model from LM Studio via POST /api/v1/models/unload."""
         url = f"{self.base_url}/api/v1/models/unload"
-        payload = {"model": model}
+        payload = {"instance_id": model}
         r = requests.post(url, json=payload, headers=self._headers, timeout=30)
         r.raise_for_status()
         return r.json()

{prompture-0.0.37.dev3 → prompture-0.0.38.dev1}/prompture/drivers/ollama_driver.py

@@ -13,6 +13,7 @@ logger = logging.getLogger(__name__)
 
 class OllamaDriver(Driver):
     supports_json_mode = True
+    supports_json_schema = True
     supports_streaming = True
     supports_vision = True
 
@@ -64,9 +65,10 @@ class OllamaDriver(Driver):
             "stream": False,
         }
 
-        # Native JSON mode support
+        # Native JSON mode / structured output support
         if merged_options.get("json_mode"):
-            payload["format"] = "json"
+            json_schema = merged_options.get("json_schema")
+            payload["format"] = json_schema if json_schema else "json"
 
         # Add any Ollama-specific options from merged_options
         if "temperature" in merged_options:
@@ -152,7 +154,8 @@ class OllamaDriver(Driver):
         }
 
         if merged_options.get("json_mode"):
-            payload["format"] = "json"
+            json_schema = merged_options.get("json_schema")
+            payload["format"] = json_schema if json_schema else "json"
         if "temperature" in merged_options:
             payload["temperature"] = merged_options["temperature"]
         if "top_p" in merged_options:
@@ -210,9 +213,10 @@ class OllamaDriver(Driver):
             "stream": False,
         }
 
-        # Native JSON mode support
+        # Native JSON mode / structured output support
         if merged_options.get("json_mode"):
-            payload["format"] = "json"
+            json_schema = merged_options.get("json_schema")
+            payload["format"] = json_schema if json_schema else "json"
 
         if "temperature" in merged_options:
             payload["temperature"] = merged_options["temperature"]

{prompture-0.0.37.dev3 → prompture-0.0.38.dev1/prompture.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prompture
-Version: 0.0.37.dev3
+Version: 0.0.38.dev1
 Summary: Ask LLMs to return structured JSON and run cross-model tests. API-first.
 Author-email: Juan Denis <juan@vene.co>
 License-Expression: MIT