mojentic 0.8.0__py3-none-any.whl → 0.8.1__py3-none-any.whl

mojentic/llm/gateways/openai.py

@@ -91,6 +91,27 @@ class OpenAIGateway(LLMGateway):
                           num_tools=len(adapted_args['tools']))
             adapted_args['tools'] = None  # Set to None instead of removing the key
 
+        # Handle temperature restrictions for specific models
+        if 'temperature' in adapted_args:
+            temperature = adapted_args['temperature']
+
+            # Check if model supports temperature parameter at all
+            if capabilities.supported_temperatures == []:
+                # Model doesn't support temperature parameter at all - remove it
+                logger.warning("Model does not support temperature parameter, removing it",
+                              model=model,
+                              requested_temperature=temperature)
+                adapted_args.pop('temperature', None)
+            elif not capabilities.supports_temperature(temperature):
+                # Model supports temperature but not this specific value - use default
+                default_temp = 1.0
+                logger.warning("Model does not support requested temperature, using default",
+                              model=model,
+                              requested_temperature=temperature,
+                              default_temperature=default_temp,
+                              supported_temperatures=capabilities.supported_temperatures)
+                adapted_args['temperature'] = default_temp
+
         return adapted_args
 
     def _validate_model_parameters(self, model: str, args: dict) -> None:

mojentic/llm/gateways/openai_model_registry.py

@@ -34,6 +34,7 @@ class ModelCapabilities:
     supports_vision: bool = False
     max_context_tokens: Optional[int] = None
     max_output_tokens: Optional[int] = None
+    supported_temperatures: Optional[List[float]] = None  # None means all temperatures supported
 
     def get_token_limit_param(self) -> str:
         """Get the correct parameter name for token limits based on model type."""
@@ -41,6 +42,14 @@ class ModelCapabilities:
             return "max_completion_tokens"
         return "max_tokens"
 
+    def supports_temperature(self, temperature: float) -> bool:
+        """Check if the model supports a specific temperature value."""
+        if self.supported_temperatures is None:
+            return True  # All temperatures supported if not restricted
+        if self.supported_temperatures == []:
+            return False  # No temperature values supported (parameter not allowed)
+        return temperature in self.supported_temperatures
+
 
 class OpenAIModelRegistry:
     """
@@ -74,6 +83,9 @@ class OpenAIModelRegistry:
             # Deep research models and GPT-5 might have different capabilities
             is_deep_research = "deep-research" in model
             is_gpt5 = "gpt-5" in model
+            is_o1_series = model.startswith("o1")
+            is_o3_series = model.startswith("o3")
+            is_o4_series = model.startswith("o4")
             is_mini_or_nano = ("mini" in model or "nano" in model)
 
             # GPT-5 models may support more features than o1/o3/o4
@@ -91,13 +103,25 @@ class OpenAIModelRegistry:
                 context_tokens = 128000
                 output_tokens = 32768
 
+            # Temperature restrictions based on model series
+            if is_gpt5 or is_o1_series or is_o4_series:
+                # GPT-5, o1, and o4 series only support temperature=1.0
+                supported_temps = [1.0]
+            elif is_o3_series:
+                # o3 series doesn't support temperature parameter at all
+                supported_temps = []
+            else:
+                # Other reasoning models support all temperatures
+                supported_temps = None
+
             self._models[model] = ModelCapabilities(
                 model_type=ModelType.REASONING,
                 supports_tools=supports_tools,
                 supports_streaming=supports_streaming,
                 supports_vision=False,  # Vision support would need to be confirmed for GPT-5
                 max_context_tokens=context_tokens,
-                max_output_tokens=output_tokens
+                max_output_tokens=output_tokens,
+                supported_temperatures=supported_temps
             )
 
         # Chat Models (GPT-4 and GPT-4.1 series) - Updated 2025-09-28

mojentic/llm/gateways/openai_temperature_handling_spec.py

@@ -0,0 +1,245 @@
+import pytest
+from unittest.mock import MagicMock
+
+from mojentic.llm.gateways.openai import OpenAIGateway
+from mojentic.llm.gateways.openai_model_registry import get_model_registry
+from mojentic.llm.gateways.models import LLMMessage, MessageRole, LLMGatewayResponse
+
+
+@pytest.fixture
+def mock_openai_client(mocker):
+    """Mock the OpenAI client."""
+    mock_client = MagicMock()
+    mock_client.chat.completions.create.return_value = MagicMock()
+    mock_client.chat.completions.create.return_value.choices = [MagicMock()]
+    mock_client.chat.completions.create.return_value.choices[0].message.content = "Test response"
+    mock_client.chat.completions.create.return_value.choices[0].message.tool_calls = None
+    return mock_client
+
+
+@pytest.fixture
+def openai_gateway(mocker, mock_openai_client):
+    """Create an OpenAI gateway with mocked client."""
+    mocker.patch('mojentic.llm.gateways.openai.OpenAI', return_value=mock_openai_client)
+    return OpenAIGateway(api_key="test_key")
+
+
+class DescribeOpenAIGatewayTemperatureHandling:
+    """
+    Specification for OpenAI gateway temperature parameter handling.
+    """
+
+    class DescribeGPT5TemperatureRestrictions:
+        """
+        Specifications for GPT-5 model temperature restrictions.
+        """
+
+        def should_automatically_adjust_unsupported_temperature_for_gpt5(self, openai_gateway, mock_openai_client):
+            """
+            Given a GPT-5 model that only supports temperature=1.0
+            When calling complete with temperature=0.1 (unsupported)
+            Then it should automatically adjust to temperature=1.0
+            """
+            messages = [LLMMessage(role=MessageRole.User, content="Test message")]
+
+            openai_gateway.complete(
+                model="gpt-5",
+                messages=messages,
+                temperature=0.1
+            )
+
+            # Verify the API was called with temperature=1.0, not 0.1
+            call_args = mock_openai_client.chat.completions.create.call_args
+            assert call_args[1]['temperature'] == 1.0
+
+        def should_preserve_supported_temperature_for_gpt5(self, openai_gateway, mock_openai_client):
+            """
+            Given a GPT-5 model that supports temperature=1.0
+            When calling complete with temperature=1.0 (supported)
+            Then it should preserve the temperature value
+            """
+            messages = [LLMMessage(role=MessageRole.User, content="Test message")]
+
+            openai_gateway.complete(
+                model="gpt-5",
+                messages=messages,
+                temperature=1.0
+            )
+
+            # Verify the API was called with temperature=1.0
+            call_args = mock_openai_client.chat.completions.create.call_args
+            assert call_args[1]['temperature'] == 1.0
+
+        def should_preserve_any_temperature_for_gpt4o(self, openai_gateway, mock_openai_client):
+            """
+            Given a GPT-4o model that supports all temperatures
+            When calling complete with temperature=0.1
+            Then it should preserve the original temperature value
+            """
+            messages = [LLMMessage(role=MessageRole.User, content="Test message")]
+
+            openai_gateway.complete(
+                model="gpt-4o",
+                messages=messages,
+                temperature=0.1
+            )
+
+            # Verify the API was called with temperature=0.1
+            call_args = mock_openai_client.chat.completions.create.call_args
+            assert call_args[1]['temperature'] == 0.1
+
+        def should_automatically_adjust_unsupported_temperature_for_o1_mini(self, openai_gateway, mock_openai_client):
+            """
+            Given an o1-mini model that only supports temperature=1.0
+            When calling complete with temperature=0.1 (unsupported)
+            Then it should automatically adjust to temperature=1.0
+            """
+            messages = [LLMMessage(role=MessageRole.User, content="Test message")]
+
+            openai_gateway.complete(
+                model="o1-mini",
+                messages=messages,
+                temperature=0.1
+            )
+
+            # Verify the API was called with temperature=1.0, not 0.1
+            call_args = mock_openai_client.chat.completions.create.call_args
+            assert call_args[1]['temperature'] == 1.0
+
+        def should_automatically_adjust_unsupported_temperature_for_o4_mini(self, openai_gateway, mock_openai_client):
+            """
+            Given an o4-mini model that only supports temperature=1.0
+            When calling complete with temperature=0.1 (unsupported)
+            Then it should automatically adjust to temperature=1.0
+            """
+            messages = [LLMMessage(role=MessageRole.User, content="Test message")]
+
+            openai_gateway.complete(
+                model="o4-mini",
+                messages=messages,
+                temperature=0.1
+            )
+
+            # Verify the API was called with temperature=1.0, not 0.1
+            call_args = mock_openai_client.chat.completions.create.call_args
+            assert call_args[1]['temperature'] == 1.0
+
+        def should_remove_temperature_parameter_for_o3_mini(self, openai_gateway, mock_openai_client):
+            """
+            Given an o3-mini model that doesn't support temperature parameter at all
+            When calling complete with temperature=0.1
+            Then it should remove the temperature parameter entirely
+            """
+            messages = [LLMMessage(role=MessageRole.User, content="Test message")]
+
+            openai_gateway.complete(
+                model="o3-mini",
+                messages=messages,
+                temperature=0.1
+            )
+
+            # Verify the API was called without temperature parameter
+            call_args = mock_openai_client.chat.completions.create.call_args
+            assert 'temperature' not in call_args[1]
+
+
+class DescribeModelCapabilitiesTemperatureRestrictions:
+    """
+    Specification for model capabilities temperature restriction checks.
+    """
+
+    def should_identify_gpt5_temperature_restrictions(self):
+        """
+        Given the model registry
+        When checking GPT-5 model capabilities
+        Then it should indicate temperature=1.0 is supported and temperature=0.1 is not
+        """
+        registry = get_model_registry()
+        capabilities = registry.get_model_capabilities("gpt-5")
+
+        assert capabilities.supports_temperature(1.0) is True
+        assert capabilities.supports_temperature(0.1) is False
+        assert capabilities.supported_temperatures == [1.0]
+
+    def should_allow_all_temperatures_for_gpt4o(self):
+        """
+        Given the model registry
+        When checking GPT-4o model capabilities
+        Then it should indicate all temperature values are supported
+        """
+        registry = get_model_registry()
+        capabilities = registry.get_model_capabilities("gpt-4o")
+
+        assert capabilities.supports_temperature(1.0) is True
+        assert capabilities.supports_temperature(0.1) is True
+        assert capabilities.supports_temperature(0.7) is True
+        assert capabilities.supported_temperatures is None
+
+    def should_identify_all_gpt5_variants_temperature_restrictions(self):
+        """
+        Given the model registry
+        When checking all GPT-5 variant models
+        Then they should all have temperature restrictions to 1.0 only
+        """
+        registry = get_model_registry()
+        gpt5_models = [
+            "gpt-5",
+            "gpt-5-2025-08-07",
+            "gpt-5-chat-latest",
+            "gpt-5-codex",
+            "gpt-5-mini",
+            "gpt-5-mini-2025-08-07",
+            "gpt-5-nano",
+            "gpt-5-nano-2025-08-07"
+        ]
+
+        for model in gpt5_models:
+            capabilities = registry.get_model_capabilities(model)
+            assert capabilities.supports_temperature(1.0) is True
+            assert capabilities.supports_temperature(0.1) is False
+            assert capabilities.supported_temperatures == [1.0]
+
+    def should_identify_o1_series_temperature_restrictions(self):
+        """
+        Given the model registry
+        When checking o1 series models
+        Then they should have temperature restrictions to 1.0 only
+        """
+        registry = get_model_registry()
+        o1_models = ["o1", "o1-mini", "o1-pro", "o1-2024-12-17"]
+
+        for model in o1_models:
+            capabilities = registry.get_model_capabilities(model)
+            assert capabilities.supports_temperature(1.0) is True
+            assert capabilities.supports_temperature(0.1) is False
+            assert capabilities.supported_temperatures == [1.0]
+
+    def should_identify_o3_series_no_temperature_support(self):
+        """
+        Given the model registry
+        When checking o3 series models
+        Then they should not support temperature parameter at all
+        """
+        registry = get_model_registry()
+        o3_models = ["o3", "o3-mini", "o3-pro", "o3-deep-research"]
+
+        for model in o3_models:
+            capabilities = registry.get_model_capabilities(model)
+            assert capabilities.supports_temperature(1.0) is False
+            assert capabilities.supports_temperature(0.1) is False
+            assert capabilities.supported_temperatures == []
+
+    def should_identify_o4_series_temperature_restrictions(self):
+        """
+        Given the model registry
+        When checking o4 series models
+        Then they should have temperature restrictions to 1.0 only
+        """
+        registry = get_model_registry()
+        o4_models = ["o4-mini", "o4-mini-2025-04-16", "o4-mini-deep-research"]
+
+        for model in o4_models:
+            capabilities = registry.get_model_capabilities(model)
+            assert capabilities.supports_temperature(1.0) is True
+            assert capabilities.supports_temperature(0.1) is False
+            assert capabilities.supported_temperatures == [1.0]

{mojentic-0.8.0.dist-info → mojentic-0.8.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mojentic
-Version: 0.8.0
+Version: 0.8.1
 Summary: Mojentic is an agentic framework that aims to provide a simple and flexible way to assemble teams of agents to solve complex problems.
 Author-email: Stacey Vetzal <stacey@vetzal.com>
 Project-URL: Homepage, https://github.com/svetzal/mojentic
@@ -125,6 +125,17 @@ print(result)
 
 The framework automatically handles parameter differences between model types, so you can switch between any models without code changes.
 
+### Model-Specific Limitations
+
+Some models have specific parameter restrictions that are automatically handled:
+
+- **GPT-5 Series**: Only supports `temperature=1.0` (default). Other temperature values are automatically adjusted with a warning.
+- **o1 & o4 Series**: Only supports `temperature=1.0` (default). Other temperature values are automatically adjusted with a warning.
+- **o3 Series**: Does not support the `temperature` parameter at all. The parameter is automatically removed with a warning.
+- **All Reasoning Models** (o1, o3, o4, GPT-5): Use `max_completion_tokens` instead of `max_tokens`, and have limited tool support.
+
+The framework will automatically adapt parameters and log warnings when unsupported values are provided.
+
 ## 🏗️ Project Structure
 
 ```

{mojentic-0.8.0.dist-info → mojentic-0.8.1.dist-info}/RECORD

@@ -86,11 +86,12 @@ mojentic/llm/gateways/models.py,sha256=OyIaMHKrrx6dHo5FbC8qOFct7PRql9wqbe_BJlgDS
 mojentic/llm/gateways/ollama.py,sha256=OUUImBNzPte52Gsf-e7TBjDHRvYW5flU9ddxwG2zlzk,7909
 mojentic/llm/gateways/ollama_messages_adapter.py,sha256=kUN_p2FyN88_trXMcL-Xsn9xPBU7pGKlJwTUEUCf6G4,1404
 mojentic/llm/gateways/ollama_messages_adapter_spec.py,sha256=gVRbWDrHOa1EiZ0CkEWe0pGn-GKRqdGb-x56HBQeYSE,4981
-mojentic/llm/gateways/openai.py,sha256=42A-8etuDRBSy18q5Qp6S1yndyOJpu3p8Pu0Dd1orFU,12806
+mojentic/llm/gateways/openai.py,sha256=ru156JpPW8-Zs3_O7BCBuztu_PkP88uD0qbM1Y-RMU4,14032
 mojentic/llm/gateways/openai_message_adapter_spec.py,sha256=ITBSV5njldV_x0NPgjmg8Okf9KzevQJ8dTXM-t6ubcg,6612
 mojentic/llm/gateways/openai_messages_adapter.py,sha256=Scal68JKKdBHB35ok1c5DeWYdD6Wra5oXSsPxJyyXSQ,3947
-mojentic/llm/gateways/openai_model_registry.py,sha256=4BIWQOl-5yAbug3UHUtpbj3kpkadNoy4sMgThyPi-i8,12858
+mojentic/llm/gateways/openai_model_registry.py,sha256=CPfbwBhdZ94jzLjmaH9dRXGZFk4OD2pOUlW6RFWVAPM,14101
 mojentic/llm/gateways/openai_model_registry_spec.py,sha256=rCyXhiCOKMewkZjdZoawALoEk62yjENeYTpjYuMuXDM,6711
+mojentic/llm/gateways/openai_temperature_handling_spec.py,sha256=PxQpI57RGaWpt1Dj6z2uLeFcP-dRZTHkai-igZTZc9M,9947
 mojentic/llm/gateways/tokenizer_gateway.py,sha256=ztuqfunlJ6xmyUPPHcC_69-kegiNJD6jdSEde7hDh2w,485
 mojentic/llm/registry/__init__.py,sha256=P2MHlptrtRPMSWbWl9ojXPmjMwkW0rIn6jwzCkSgnhE,164
 mojentic/llm/registry/llm_registry.py,sha256=beyrgGrkXx5ZckUJzC1nQ461vra0fF6s_qRaEdi5bsg,2508
@@ -136,8 +137,8 @@ mojentic/tracer/tracer_system.py,sha256=7CPy_2tlsHtXQ4DcO5oo52N9a9WS0GH-mjeINzu6
 mojentic/tracer/tracer_system_spec.py,sha256=TNm0f9LV__coBx0JGEKyzzNN9mFjCSG_SSrRISO8Xeg,8632
 mojentic/utils/__init__.py,sha256=lqECkkoFvHFttDnafRE1vvh0Dmna_lwupMToP5VvX5k,115
 mojentic/utils/formatting.py,sha256=bPrwwdluXdQ8TsFxfWtHNOeMWKNvAfABSoUnnA1g7c8,947
-mojentic-0.8.0.dist-info/licenses/LICENSE.md,sha256=txSgV8n5zY1W3NiF5HHsCwlaW0e8We1cSC6TuJUqxXA,1060
-mojentic-0.8.0.dist-info/METADATA,sha256=EpjGKSzORxFrSG94rw5rUvm4vztoEMfGCTkXON-W49k,6154
-mojentic-0.8.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-mojentic-0.8.0.dist-info/top_level.txt,sha256=Q-BvPQ8Eu1jnEqK8Xkr6A9C8Xa1z38oPZRHuA5MCTqg,19
-mojentic-0.8.0.dist-info/RECORD,,
+mojentic-0.8.1.dist-info/licenses/LICENSE.md,sha256=txSgV8n5zY1W3NiF5HHsCwlaW0e8We1cSC6TuJUqxXA,1060
+mojentic-0.8.1.dist-info/METADATA,sha256=Lo7fKdzH3LXkdkL9d9ONoh00JxyggwF7snTIMvVSP7E,6896
+mojentic-0.8.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mojentic-0.8.1.dist-info/top_level.txt,sha256=Q-BvPQ8Eu1jnEqK8Xkr6A9C8Xa1z38oPZRHuA5MCTqg,19
+mojentic-0.8.1.dist-info/RECORD,,