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

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.7.4.dist-info → mojentic-0.8.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mojentic
-Version: 0.7.4
+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
@@ -39,7 +39,7 @@ Dynamic: license-file
 
 # Mojentic
 
-Mojentic is a framework that provides a simple and flexible way to interact with Large Language Models (LLMs). It offers integration with various LLM providers and includes tools for structured output generation, task automation, and more. The future direction is to facilitate a team of agents, but the current focus is on robust LLM interaction capabilities.
+Mojentic is a framework that provides a simple and flexible way to interact with Large Language Models (LLMs). It offers integration with various LLM providers and includes tools for structured output generation, task automation, and more. With comprehensive support for all OpenAI models including GPT-5 and automatic parameter adaptation, Mojentic handles the complexities of different model types seamlessly. The future direction is to facilitate a team of agents, but the current focus is on robust LLM interaction capabilities.
 
 [![GitHub](https://img.shields.io/github/license/svetzal/mojentic)](LICENSE.md)
 [![Python Version](https://img.shields.io/badge/python-3.11%2B-blue)](https://www.python.org/downloads/)
@@ -48,6 +48,8 @@ Mojentic is a framework that provides a simple and flexible way to interact with
 ## 🚀 Features
 
 - **LLM Integration**: Support for multiple LLM providers (OpenAI, Ollama)
+- **Latest OpenAI Models**: Full support for GPT-5, GPT-4.1, and all reasoning models (o1, o3, o4 series)
+- **Automatic Model Adaptation**: Seamless parameter handling across different OpenAI model types
 - **Structured Output**: Generate structured data from LLM responses using Pydantic models
 - **Tools Integration**: Utilities for date resolution, image analysis, and more
 - **Multi-modal Capabilities**: Process and analyze images alongside text
@@ -84,8 +86,9 @@ from mojentic.llm.gateways.models import LLMMessage
 from mojentic.llm.tools.date_resolver import ResolveDateTool
 from pydantic import BaseModel, Field
 
-# Initialize with OpenAI
-openai_llm = LLMBroker(model="gpt-4o", gateway=OpenAIGateway(api_key="your_api_key"))
+# Initialize with OpenAI (supports all models including GPT-5, GPT-4.1, reasoning models)
+openai_llm = LLMBroker(model="gpt-5", gateway=OpenAIGateway(api_key="your_api_key"))
+# Or use other models: "gpt-4o", "gpt-4.1", "o1-mini", "o3-mini", etc.
 
 # Or use Ollama for local LLMs
 ollama_llm = LLMBroker(model="llama3")
@@ -99,7 +102,7 @@ class Sentiment(BaseModel):
     label: str = Field(..., description="Label for the sentiment")
 
 sentiment = openai_llm.generate_object(
-    messages=[LLMMessage(content="Hello, how are you?")], 
+    messages=[LLMMessage(content="Hello, how are you?")],
     object_model=Sentiment
 )
 print(sentiment.label)
@@ -118,6 +121,21 @@ result = openai_llm.generate(messages=[
 print(result)
 ```
 
+## 🤖 OpenAI Model Support
+
+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.7.4.dist-info → mojentic-0.8.1.dist-info}/RECORD

@@ -2,7 +2,7 @@ _examples/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 _examples/async_dispatcher_example.py,sha256=rlaShGpXUsTN7RwVxOo8sbxQjYYJ13gKRB9PWQ3sVTE,8454
 _examples/async_llm_example.py,sha256=IgYBMawHhaKWbA6zkurH-raqBez9nODfjHEAHI7PRek,8708
 _examples/broker_as_tool.py,sha256=QJHryYqIXSUYkK62LhMspbNdK3yL0DoenWHxUK5c_fk,2971
-_examples/broker_examples.py,sha256=7tES2KyKc3uigBuxkwOTf5UGhp7nVeVqLhd8fEHSWEc,1901
+_examples/broker_examples.py,sha256=-fNqOLl4lwja3_sQwVJyT-2Qwy-B0ol9UWs6zSP2QKk,2397
 _examples/broker_image_examples.py,sha256=8dpZ2RFmRYhJqoeWklzlPxmvj9m4M393452SO_D2bJY,1133
 _examples/characterize_ollama.py,sha256=_TzPEAoTuB-saTiaypEQAsUpiBeM37l9I5wCMU3GM4E,2521
 _examples/characterize_openai.py,sha256=JQQNjsHIEegjLAS3uzDmq_Ci_ZqXTqjcoVaC-rS_R_8,564
@@ -14,6 +14,7 @@ _examples/design_analysis.py,sha256=66JxvnA065QV__WXpxJX_eTBDan34nGcKvao75UPLlw,
 _examples/embeddings.py,sha256=94DAMsMB35BU98hOfOeOsbqntcequFSdtaStox2hTvk,267
 _examples/ensures_files_exist.py,sha256=LAwX9YtUZGI6NT254PY7oC5yg6q-b9Q_Dez9gGrabQM,2009
 _examples/ephemeral_task_manager_example.py,sha256=Q1YpLPwDkB-ytIGgCh2eWOr5EJ0FE-osxJ0B0wDlQkY,1491
+_examples/fetch_openai_models.py,sha256=06FrBZfV4FJ7MM9MczbrzFD8wUmXKZOjzPFFxwSVlj4,3545
 _examples/file_deduplication.py,sha256=jYayx4BCrk1uJwVBkjIXzguLQOWLREi4PBPzoK9LuOU,1665
 _examples/file_tool.py,sha256=fUzaIF8qqIy3-E6TJ89jOSD2dwjuO_gBi_1vfCzjuC4,2335
 _examples/image_analysis.py,sha256=Kj49vLQD1DIpvv5P7rir4BqzsVCTgq-tfppqXcYVJkA,503
@@ -22,6 +23,7 @@ _examples/image_broker_splat.py,sha256=O7rzTFUka32if4G4VuXvhu1O-2lRMWfi0r8gjIE8-
 _examples/iterative_solver.py,sha256=ANGdC74ymHosVt6xUBjplkJl_W3ALTGxOkDpPLEDcm8,1331
 _examples/list_models.py,sha256=8noMpGeXOdX5Pf0NXCt_CRurOKEg_5luhWveGntBhe8,578
 _examples/model_characterization.py,sha256=XwLiUP1ZIrNs4ZLmjLDW-nJQsB66H-BV0bWgBgT3N7k,2571
+_examples/openai_gateway_enhanced_demo.py,sha256=vNgJSeQS78PSUibFR9I3WSrBEoSW5qr0dkf2UNlZRdI,5995
 _examples/oversized_embeddings.py,sha256=_z2JoqZn0g7VtRsFVWIkngVqzjhQQvCEUYWVxs1I7MM,284
 _examples/raw.py,sha256=Y2wvgynFuoUs28agE4ijsLYec8VRjiReklqlCH2lERs,442
 _examples/react.py,sha256=VQ-5MmjUXoHzBFPTV_JrocuOkDzZ8oyUUSYLlEToJ_0,939
@@ -84,9 +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=Wxx_WfG2czOv9Ng8q4JCLFIHGqNs3vaMtE5ggSLBjHk,7787
+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=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
@@ -132,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.7.4.dist-info/licenses/LICENSE.md,sha256=txSgV8n5zY1W3NiF5HHsCwlaW0e8We1cSC6TuJUqxXA,1060
-mojentic-0.7.4.dist-info/METADATA,sha256=shFHb00Ezpkzo6QJOp1vAIhXGONL2jFH7pzIJimPBao,5475
-mojentic-0.7.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-mojentic-0.7.4.dist-info/top_level.txt,sha256=Q-BvPQ8Eu1jnEqK8Xkr6A9C8Xa1z38oPZRHuA5MCTqg,19
-mojentic-0.7.4.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,,