Adding grok-4-1-fast-non-reasoning as default

Author: brightsparc

docs/models/grok.md

@@ -10,7 +10,7 @@ pip/uv-add "pydantic-ai-slim[grok]"
 
 ## Configuration
 
-To use Grok from [xAI](https://x.ai/api) through their API, go to [console.x.ai]https://console.x.ai) and follow your nose until you find the place to create an API key.
+To use Grok from [xAI](https://x.ai/api) through their API, go your [console.x.ai](https://console.x.ai/team/default/api-keys) and follow your nose until you find the place to create an API key.
 
 ## Environment variable
 
@@ -25,7 +25,7 @@ You can then use `GrokModel` by name:
 ```python
 from pydantic_ai import Agent
 
-agent = Agent('grok:grok-4-fast-non-reasoning')
+agent = Agent('grok:grok-4-1-fast-non-reasoning')
 ...
 ```
 
@@ -35,7 +35,7 @@ Or initialise the model directly with just the model name:
 from pydantic_ai import Agent
 from pydantic_ai.models.grok import GrokModel
 
-model = GrokModel('grok-4-fast-non-reasoning')
+model = GrokModel('grok-4-1-fast-non-reasoning')
 agent = Agent(model)
 ...
 ```
@@ -46,7 +46,7 @@ You can provide your own `api_key` inline like so:
 from pydantic_ai import Agent
 from pydantic_ai.models.grok import GrokModel
 
-model = GrokModel('grok-4-fast-non-reasoning', api_key='your-api-key')
+model = GrokModel('grok-4-1-fast-non-reasoning', api_key='your-api-key')
 agent = Agent(model)
 ...
 ```
@@ -60,7 +60,7 @@ async_client = AsyncClient(api_key='your-api-key')
 from pydantic_ai import Agent
 from pydantic_ai.models.grok import GrokModel
 
-model = GrokModel('grok-4-fast-non-reasoning', client=async_client)
+model = GrokModel('grok-4-1-fast-non-reasoning', client=async_client)
 agent = Agent(model)
 ...
 ```

pydantic_ai_slim/pydantic_ai/models/__init__.py

@@ -175,6 +175,8 @@
         'grok:grok-3-mini-fast',
         'grok:grok-4',
         'grok:grok-4-0709',
+        'grok:grok-4-1-fast-non-reasoning',
+        'grok:grok-4-1-fast-reasoning',
         'grok:grok-4-fast-non-reasoning',
         'grok:grok-4-fast-reasoning',
         'grok:grok-code-fast-1',

pydantic_ai_slim/pydantic_ai/models/grok.py

@@ -63,7 +63,7 @@ def __init__(
         """Initialize the Grok model.
 
         Args:
-            model_name: The name of the Grok model to use (e.g., "grok-3", "grok-4-fast-non-reasoning")
+            model_name: The name of the Grok model to use (e.g., "grok-4-1-fast-non-reasoning")
             api_key: The xAI API key. If not provided, uses XAI_API_KEY environment variable.
             client: Optional AsyncClient instance for testing. If provided, api_key is ignored.
             settings: Optional model settings.

pydantic_ai_slim/pydantic_ai/providers/grok.py

@@ -25,6 +25,8 @@
 GrokModelName = Literal[
     'grok-4',
     'grok-4-0709',
+    'grok-4-1-fast-reasoning',
+    'grok-4-1-fast-non-reasoning',
     'grok-4-fast-reasoning',
     'grok-4-fast-non-reasoning',
     'grok-code-fast-1',

tests/models/test_grok.py

@@ -64,16 +64,16 @@
 
 
 def test_grok_init():
-    m = GrokModel('grok-4-fast-non-reasoning', api_key='foobar')
+    m = GrokModel('grok-4-1-fast-non-reasoning', api_key='foobar')
     # Check model properties without accessing private attributes
-    assert m.model_name == 'grok-4-fast-non-reasoning'
+    assert m.model_name == 'grok-4-1-fast-non-reasoning'
     assert m.system == 'xai'
 
 
 async def test_grok_request_simple_success(allow_model_requests: None):
     response = create_response(content='world')
     mock_client = MockGrok.create_mock(response)
-    m = GrokModel('grok-4-fast-non-reasoning', client=mock_client)
+    m = GrokModel('grok-4-1-fast-non-reasoning', client=mock_client)
     agent = Agent(m)
 
     result = await agent.run('hello')
@@ -94,7 +94,7 @@ async def test_grok_request_simple_success(allow_model_requests: None):
             ),
             ModelResponse(
                 parts=[TextPart(content='world')],
-                model_name='grok-4-fast-non-reasoning',
+                model_name='grok-4-1-fast-non-reasoning',
                 timestamp=IsDatetime(),
                 provider_name='xai',
                 provider_response_id='grok-123',
@@ -107,7 +107,7 @@ async def test_grok_request_simple_success(allow_model_requests: None):
             ),
             ModelResponse(
                 parts=[TextPart(content='world')],
-                model_name='grok-4-fast-non-reasoning',
+                model_name='grok-4-1-fast-non-reasoning',
                 timestamp=IsDatetime(),
                 provider_name='xai',
                 provider_response_id='grok-123',
@@ -124,7 +124,7 @@ async def test_grok_request_simple_usage(allow_model_requests: None):
         usage=SimpleNamespace(prompt_tokens=2, completion_tokens=1),
     )
     mock_client = MockGrok.create_mock(response)
-    m = GrokModel('grok-4-fast-non-reasoning', client=mock_client)
+    m = GrokModel('grok-4-1-fast-non-reasoning', client=mock_client)
     agent = Agent(m)
 
     result = await agent.run('Hello')
@@ -142,7 +142,7 @@ async def test_grok_image_input(allow_model_requests: None):
     """Test that Grok model handles image inputs (text is extracted from content)."""
     response = create_response(content='done')
     mock_client = MockGrok.create_mock(response)
-    model = GrokModel('grok-4-fast-non-reasoning', client=mock_client)
+    model = GrokModel('grok-4-1-fast-non-reasoning', client=mock_client)
     agent = Agent(model)
 
     image_url = ImageUrl('https://example.com/image.png')
@@ -160,7 +160,7 @@ async def test_grok_request_structured_response(allow_model_requests: None):
     )
     response = create_response(tool_calls=[tool_call])
     mock_client = MockGrok.create_mock(response)
-    m = GrokModel('grok-4-fast-non-reasoning', client=mock_client)
+    m = GrokModel('grok-4-1-fast-non-reasoning', client=mock_client)
     agent = Agent(m, output_type=list[int])
 
     result = await agent.run('Hello')
@@ -179,7 +179,7 @@ async def test_grok_request_structured_response(allow_model_requests: None):
                         tool_call_id='123',
                     )
                 ],
-                model_name='grok-4-fast-non-reasoning',
+                model_name='grok-4-1-fast-non-reasoning',
                 timestamp=IsDatetime(),
                 provider_name='xai',
                 provider_response_id='grok-123',
@@ -214,7 +214,7 @@ async def test_grok_request_tool_call(allow_model_requests: None):
         create_response(content='final response'),
     ]
     mock_client = MockGrok.create_mock(responses)
-    m = GrokModel('grok-4-fast-non-reasoning', client=mock_client)
+    m = GrokModel('grok-4-1-fast-non-reasoning', client=mock_client)
     agent = Agent(m, system_prompt='this is the system prompt')
 
     @agent.tool_plain
@@ -247,7 +247,7 @@ async def get_location(loc_name: str) -> str:
                     input_tokens=2,
                     output_tokens=1,
                 ),
-                model_name='grok-4-fast-non-reasoning',
+                model_name='grok-4-1-fast-non-reasoning',
                 timestamp=IsDatetime(),
                 provider_name='xai',
                 provider_response_id='grok-123',
@@ -277,7 +277,7 @@ async def get_location(loc_name: str) -> str:
                     input_tokens=3,
                     output_tokens=2,
                 ),
-                model_name='grok-4-fast-non-reasoning',
+                model_name='grok-4-1-fast-non-reasoning',
                 timestamp=IsDatetime(),
                 provider_name='xai',
                 provider_response_id='grok-123',
@@ -297,7 +297,7 @@ async def get_location(loc_name: str) -> str:
             ),
             ModelResponse(
                 parts=[TextPart(content='final response')],
-                model_name='grok-4-fast-non-reasoning',
+                model_name='grok-4-1-fast-non-reasoning',
                 timestamp=IsDatetime(),
                 provider_name='xai',
                 provider_response_id='grok-123',
@@ -369,7 +369,7 @@ def grok_reasoning_text_chunk(
 async def test_grok_stream_text(allow_model_requests: None):
     stream = [grok_text_chunk('hello '), grok_text_chunk('world')]
     mock_client = MockGrok.create_mock_stream(stream)
-    m = GrokModel('grok-4-fast-non-reasoning', client=mock_client)
+    m = GrokModel('grok-4-1-fast-non-reasoning', client=mock_client)
     agent = Agent(m)
 
     async with agent.run_stream('') as result:
@@ -387,7 +387,7 @@ async def test_grok_stream_text_finish_reason(allow_model_requests: None):
         grok_text_chunk('.', 'stop'),
     ]
     mock_client = MockGrok.create_mock_stream(stream)
-    m = GrokModel('grok-4-fast-non-reasoning', client=mock_client)
+    m = GrokModel('grok-4-1-fast-non-reasoning', client=mock_client)
     agent = Agent(m)
 
     async with agent.run_stream('') as result:
@@ -402,7 +402,7 @@ async def test_grok_stream_text_finish_reason(allow_model_requests: None):
                     ModelResponse(
                         parts=[TextPart(content='hello world.')],
                         usage=RequestUsage(input_tokens=2, output_tokens=1),
-                        model_name='grok-4-fast-non-reasoning',
+                        model_name='grok-4-1-fast-non-reasoning',
                         timestamp=IsDatetime(),
                         provider_name='xai',
                         provider_response_id='grok-123',
@@ -479,7 +479,7 @@ async def test_grok_stream_structured(allow_model_requests: None):
         grok_tool_chunk(None, '}', finish_reason='stop', accumulated_args='{"first": "One", "second": "Two"}'),
     ]
     mock_client = MockGrok.create_mock_stream(stream)
-    m = GrokModel('grok-4-fast-non-reasoning', client=mock_client)
+    m = GrokModel('grok-4-1-fast-non-reasoning', client=mock_client)
     agent = Agent(m, output_type=MyTypedDict)
 
     async with agent.run_stream('') as result:
@@ -500,7 +500,7 @@ async def test_grok_stream_structured_finish_reason(allow_model_requests: None):
         grok_tool_chunk(None, None, finish_reason='stop', accumulated_args='{"first": "One", "second": "Two"}'),
     ]
     mock_client = MockGrok.create_mock_stream(stream)
-    m = GrokModel('grok-4-fast-non-reasoning', client=mock_client)
+    m = GrokModel('grok-4-1-fast-non-reasoning', client=mock_client)
     agent = Agent(m, output_type=MyTypedDict)
 
     async with agent.run_stream('') as result:
@@ -518,7 +518,7 @@ async def test_grok_stream_native_output(allow_model_requests: None):
         grok_text_chunk('}'),
     ]
     mock_client = MockGrok.create_mock_stream(stream)
-    m = GrokModel('grok-4-fast-non-reasoning', client=mock_client)
+    m = GrokModel('grok-4-1-fast-non-reasoning', client=mock_client)
     agent = Agent(m, output_type=NativeOutput(MyTypedDict))
 
     async with agent.run_stream('') as result:
@@ -537,7 +537,7 @@ async def test_grok_stream_tool_call_with_empty_text(allow_model_requests: None)
         grok_tool_chunk(None, '}', finish_reason='stop', accumulated_args='{"first": "One", "second": "Two"}'),
     ]
     mock_client = MockGrok.create_mock_stream(stream)
-    m = GrokModel('grok-4-fast-non-reasoning', client=mock_client)
+    m = GrokModel('grok-4-1-fast-non-reasoning', client=mock_client)
     agent = Agent(m, output_type=[str, MyTypedDict])
 
     async with agent.run_stream('') as result:
@@ -554,7 +554,7 @@ async def test_grok_no_delta(allow_model_requests: None):
         grok_text_chunk('world'),
     ]
     mock_client = MockGrok.create_mock_stream(stream)
-    m = GrokModel('grok-4-fast-non-reasoning', client=mock_client)
+    m = GrokModel('grok-4-1-fast-non-reasoning', client=mock_client)
     agent = Agent(m)
 
     async with agent.run_stream('') as result:
@@ -571,7 +571,7 @@ async def test_grok_none_delta(allow_model_requests: None):
         grok_text_chunk('world'),
     ]
     mock_client = MockGrok.create_mock_stream(stream)
-    m = GrokModel('grok-4-fast-non-reasoning', client=mock_client)
+    m = GrokModel('grok-4-1-fast-non-reasoning', client=mock_client)
     agent = Agent(m)
 
     async with agent.run_stream('') as result:
@@ -597,7 +597,7 @@ async def test_grok_parallel_tool_calls(allow_model_requests: None, parallel_too
     )
     response = create_response(content='', tool_calls=[tool_call], finish_reason='tool_calls')
     mock_client = MockGrok.create_mock(response)
-    m = GrokModel('grok-4-fast-non-reasoning', client=mock_client)
+    m = GrokModel('grok-4-1-fast-non-reasoning', client=mock_client)
     agent = Agent(m, output_type=list[int], model_settings=ModelSettings(parallel_tool_calls=parallel_tool_calls))
 
     await agent.run('Hello')
@@ -607,7 +607,7 @@ async def test_grok_parallel_tool_calls(allow_model_requests: None, parallel_too
 async def test_grok_penalty_parameters(allow_model_requests: None) -> None:
     response = create_response(content='test response')
     mock_client = MockGrok.create_mock(response)
-    m = GrokModel('grok-4-fast-non-reasoning', client=mock_client)
+    m = GrokModel('grok-4-1-fast-non-reasoning', client=mock_client)
 
     settings = ModelSettings(
         temperature=0.7,
@@ -631,7 +631,7 @@ async def test_grok_penalty_parameters(allow_model_requests: None) -> None:
 async def test_grok_image_url_input(allow_model_requests: None):
     response = create_response(content='world')
     mock_client = MockGrok.create_mock(response)
-    m = GrokModel('grok-4-fast-non-reasoning', client=mock_client)
+    m = GrokModel('grok-4-1-fast-non-reasoning', client=mock_client)
     agent = Agent(m)
 
     result = await agent.run(
@@ -647,7 +647,7 @@ async def test_grok_image_url_input(allow_model_requests: None):
 
 @pytest.mark.skipif(os.getenv('XAI_API_KEY') is None, reason='Requires XAI_API_KEY (gRPC, no cassettes)')
 async def test_grok_image_url_tool_response(allow_model_requests: None, xai_api_key: str):
-    m = GrokModel('grok-4-fast-non-reasoning', api_key=xai_api_key)
+    m = GrokModel('grok-4-1-fast-non-reasoning', api_key=xai_api_key)
     agent = Agent(m)
 
     @agent.tool_plain
@@ -684,7 +684,7 @@ async def get_image() -> ImageUrl:
 async def test_grok_image_as_binary_content_tool_response(
     allow_model_requests: None, image_content: BinaryContent, xai_api_key: str
 ):
-    m = GrokModel('grok-4-fast-non-reasoning', api_key=xai_api_key)
+    m = GrokModel('grok-4-1-fast-non-reasoning', api_key=xai_api_key)
     agent = Agent(m)
 
     @agent.tool_plain
@@ -724,7 +724,7 @@ async def test_grok_image_as_binary_content_input(
     allow_model_requests: None, image_content: BinaryContent, xai_api_key: str
 ):
     """Test passing binary image content directly as input (not from a tool)."""
-    m = GrokModel('grok-4-fast-non-reasoning', api_key=xai_api_key)
+    m = GrokModel('grok-4-1-fast-non-reasoning', api_key=xai_api_key)
     agent = Agent(m)
 
     result = await agent.run(['What fruit is in the image?', image_content])
@@ -923,7 +923,7 @@ async def test_grok_model_retries(allow_model_requests: None):
     success_response = create_response(content='Success after retry')
 
     mock_client = MockGrok.create_mock(success_response)
-    m = GrokModel('grok-4-fast-non-reasoning', client=mock_client)
+    m = GrokModel('grok-4-1-fast-non-reasoning', client=mock_client)
     agent = Agent(m)
     result = await agent.run('hello')
     assert result.output == 'Success after retry'
@@ -933,7 +933,7 @@ async def test_grok_model_settings(allow_model_requests: None):
     """Test Grok model with various settings."""
     response = create_response(content='response with settings')
     mock_client = MockGrok.create_mock(response)
-    m = GrokModel('grok-4-fast-non-reasoning', client=mock_client)
+    m = GrokModel('grok-4-1-fast-non-reasoning', client=mock_client)
     agent = Agent(
         m,
         model_settings=ModelSettings(
@@ -965,7 +965,7 @@ async def test_grok_model_multiple_tool_calls(allow_model_requests: None):
     ]
 
     mock_client = MockGrok.create_mock(responses)
-    m = GrokModel('grok-4-fast-non-reasoning', client=mock_client)
+    m = GrokModel('grok-4-1-fast-non-reasoning', client=mock_client)
     agent = Agent(m)
 
     @agent.tool_plain
@@ -995,7 +995,7 @@ async def test_grok_stream_with_tool_calls(allow_model_requests: None):
     ]
 
     mock_client = MockGrok.create_mock_stream([stream1, stream2])
-    m = GrokModel('grok-4-fast-non-reasoning', client=mock_client)
+    m = GrokModel('grok-4-1-fast-non-reasoning', client=mock_client)
     agent = Agent(m)
 
     @agent.tool_plain
@@ -1017,14 +1017,14 @@ async def get_info(query: str) -> str:
 async def test_grok_model_invalid_api_key():
     """Test Grok model with invalid API key."""
     with pytest.raises(ValueError, match='XAI API key is required'):
-        GrokModel('grok-4-fast-non-reasoning', api_key='')
+        GrokModel('grok-4-1-fast-non-reasoning', api_key='')
 
 
 async def test_grok_model_properties():
     """Test Grok model properties."""
-    m = GrokModel('grok-4-fast-non-reasoning', api_key='test-key')
+    m = GrokModel('grok-4-1-fast-non-reasoning', api_key='test-key')
 
-    assert m.model_name == 'grok-4-fast-non-reasoning'
+    assert m.model_name == 'grok-4-1-fast-non-reasoning'
     assert m.system == 'xai'