prompture 1.0.55.dev1__tar.gz → 1.0.55.dev2__tar.gz

{prompture-1.0.55.dev1 → prompture-1.0.55.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prompture
-Version: 1.0.55.dev1
+Version: 1.0.55.dev2
 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-1.0.55.dev1 → prompture-1.0.55.dev2}/prompture/_version.py

@@ -18,7 +18,7 @@ version_tuple: tuple[int | str, ...]
 commit_id: str | None
 __commit_id__: str | None
 
-__version__ = version = '1.0.55.dev1'
-__version_tuple__ = version_tuple = (1, 0, 55, 'dev1')
+__version__ = version = '1.0.55.dev2'
+__version_tuple__ = version_tuple = (1, 0, 55, 'dev2')
 
 __commit_id__ = commit_id = None

{prompture-1.0.55.dev1 → prompture-1.0.55.dev2}/prompture/drivers/async_claude_driver.py

@@ -67,6 +67,8 @@ class AsyncClaudeDriver(CostMixin, AsyncDriver):
             using_json_schema=bool(options.get("json_schema")),
         )
 
+        supports_temperature = self._get_model_config("claude", model)["supports_temperature"]
+
         client = anthropic.AsyncAnthropic(api_key=self.api_key)
 
         system_content, api_messages = _extract_anthropic_system_and_messages(messages)
@@ -74,9 +76,10 @@ class AsyncClaudeDriver(CostMixin, AsyncDriver):
         common_kwargs: dict[str, Any] = {
             "model": model,
             "messages": api_messages,
-            "temperature": opts["temperature"],
             "max_tokens": opts["max_tokens"],
         }
+        if supports_temperature:
+            common_kwargs["temperature"] = opts["temperature"]
         if system_content:
             common_kwargs["system"] = system_content
 
@@ -147,6 +150,8 @@ class AsyncClaudeDriver(CostMixin, AsyncDriver):
 
         self._validate_model_capabilities("claude", model, using_tool_use=True)
 
+        supports_temperature = self._get_model_config("claude", model)["supports_temperature"]
+
         client = anthropic.AsyncAnthropic(api_key=self.api_key)
 
         system_content, api_messages = _extract_anthropic_system_and_messages(messages)
@@ -155,10 +160,11 @@ class AsyncClaudeDriver(CostMixin, AsyncDriver):
         kwargs: dict[str, Any] = {
             "model": model,
             "messages": api_messages,
-            "temperature": opts["temperature"],
             "max_tokens": opts["max_tokens"],
             "tools": anthropic_tools,
         }
+        if supports_temperature:
+            kwargs["temperature"] = opts["temperature"]
         if system_content:
             kwargs["system"] = system_content
 
@@ -203,6 +209,7 @@ class AsyncClaudeDriver(CostMixin, AsyncDriver):
 
         opts = {**{"temperature": 0.0, "max_tokens": 512}, **options}
         model = options.get("model", self.model)
+        supports_temperature = self._get_model_config("claude", model)["supports_temperature"]
         client = anthropic.AsyncAnthropic(api_key=self.api_key)
 
         system_content, api_messages = _extract_anthropic_system_and_messages(messages)
@@ -210,9 +217,10 @@ class AsyncClaudeDriver(CostMixin, AsyncDriver):
         kwargs: dict[str, Any] = {
             "model": model,
             "messages": api_messages,
-            "temperature": opts["temperature"],
             "max_tokens": opts["max_tokens"],
         }
+        if supports_temperature:
+            kwargs["temperature"] = opts["temperature"]
         if system_content:
             kwargs["system"] = system_content
 

{prompture-1.0.55.dev1 → prompture-1.0.55.dev2}/prompture/drivers/async_google_driver.py

@@ -120,7 +120,10 @@ class AsyncGoogleDriver(CostMixin, AsyncDriver):
 
         config_dict: dict[str, Any] = {}
 
-        if "temperature" in merged_options:
+        model = merged_options.get("model", self.model)
+        supports_temperature = self._get_model_config("google", model)["supports_temperature"]
+
+        if supports_temperature and "temperature" in merged_options:
             config_dict.setdefault("temperature", merged_options["temperature"])
         if "max_tokens" in merged_options:
             config_dict.setdefault("max_output_tokens", merged_options["max_tokens"])

{prompture-1.0.55.dev1 → prompture-1.0.55.dev2}/prompture/drivers/claude_driver.py

@@ -278,6 +278,8 @@ class ClaudeDriver(CostMixin, Driver):
             using_json_schema=bool(options.get("json_schema")),
         )
 
+        supports_temperature = self._get_model_config("claude", model)["supports_temperature"]
+
         client = anthropic.Anthropic(api_key=self.api_key)
 
         # _do_generate uses the simple system/non-system split (no tool-message
@@ -294,9 +296,10 @@ class ClaudeDriver(CostMixin, Driver):
         common_kwargs: dict[str, Any] = {
             "model": model,
             "messages": api_messages,
-            "temperature": opts["temperature"],
             "max_tokens": opts["max_tokens"],
         }
+        if supports_temperature:
+            common_kwargs["temperature"] = opts["temperature"]
         if system_content:
             common_kwargs["system"] = system_content
 
@@ -378,6 +381,8 @@ class ClaudeDriver(CostMixin, Driver):
 
         self._validate_model_capabilities("claude", model, using_tool_use=True)
 
+        supports_temperature = self._get_model_config("claude", model)["supports_temperature"]
+
         client = anthropic.Anthropic(api_key=self.api_key)
 
         system_content, api_messages = _extract_anthropic_system_and_messages(messages)
@@ -386,10 +391,11 @@ class ClaudeDriver(CostMixin, Driver):
         kwargs: dict[str, Any] = {
             "model": model,
             "messages": api_messages,
-            "temperature": opts["temperature"],
             "max_tokens": opts["max_tokens"],
             "tools": anthropic_tools,
         }
+        if supports_temperature:
+            kwargs["temperature"] = opts["temperature"]
         if system_content:
             kwargs["system"] = system_content
 
@@ -434,6 +440,7 @@ class ClaudeDriver(CostMixin, Driver):
 
         opts = {**{"temperature": 0.0, "max_tokens": 512}, **options}
         model = options.get("model", self.model)
+        supports_temperature = self._get_model_config("claude", model)["supports_temperature"]
         client = anthropic.Anthropic(api_key=self.api_key)
 
         system_content, api_messages = _extract_anthropic_system_and_messages(messages)
@@ -441,9 +448,10 @@ class ClaudeDriver(CostMixin, Driver):
         kwargs: dict[str, Any] = {
             "model": model,
             "messages": api_messages,
-            "temperature": opts["temperature"],
             "max_tokens": opts["max_tokens"],
         }
+        if supports_temperature:
+            kwargs["temperature"] = opts["temperature"]
         if system_content:
             kwargs["system"] = system_content
 

{prompture-1.0.55.dev1 → prompture-1.0.55.dev2}/prompture/drivers/google_driver.py

@@ -166,7 +166,10 @@ class GoogleDriver(CostMixin, Driver):
 
         config_dict: dict[str, Any] = {}
 
-        if "temperature" in merged_options:
+        model = merged_options.get("model", self.model)
+        supports_temperature = self._get_model_config("google", model)["supports_temperature"]
+
+        if supports_temperature and "temperature" in merged_options:
             config_dict.setdefault("temperature", merged_options["temperature"])
         if "max_tokens" in merged_options:
             config_dict.setdefault("max_output_tokens", merged_options["max_tokens"])

{prompture-1.0.55.dev1 → prompture-1.0.55.dev2}/prompture.egg-info/PKG-INFO

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