Skip to content

Commit 76ea1b3

Browse files
authored
Remove hardcoded n parameter from OpenAIModel requests (#1807)
1 parent 0810eed commit 76ea1b3

File tree

2 files changed

+0
-5
lines changed

2 files changed

+0
-5
lines changed

pydantic_ai_slim/pydantic_ai/models/openai.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -277,7 +277,6 @@ async def _completions_create(
277277
return await self.client.chat.completions.create(
278278
model=self._model_name,
279279
messages=openai_messages,
280-
n=1,
281280
parallel_tool_calls=model_settings.get('parallel_tool_calls', NOT_GIVEN),
282281
tools=tools or NOT_GIVEN,
283282
tool_choice=tool_choice or NOT_GIVEN,

tests/models/test_openai.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -188,7 +188,6 @@ async def test_request_simple_success(allow_model_requests: None):
188188
{
189189
'messages': [{'content': 'hello', 'role': 'user'}],
190190
'model': 'gpt-4o',
191-
'n': 1,
192191
'extra_headers': {'User-Agent': IsStr(regex=r'pydantic-ai\/.*')},
193192
'extra_body': None,
194193
},
@@ -199,7 +198,6 @@ async def test_request_simple_success(allow_model_requests: None):
199198
{'content': 'hello', 'role': 'user'},
200199
],
201200
'model': 'gpt-4o',
202-
'n': 1,
203201
'extra_headers': {'User-Agent': IsStr(regex=r'pydantic-ai\/.*')},
204202
'extra_body': None,
205203
},
@@ -586,7 +584,6 @@ async def test_system_prompt_role(
586584
{'content': 'hello', 'role': 'user'},
587585
],
588586
'model': 'gpt-4o',
589-
'n': 1,
590587
'extra_headers': {'User-Agent': IsStr(regex=r'pydantic-ai\/.*')},
591588
'extra_body': None,
592589
}
@@ -663,7 +660,6 @@ async def test_image_url_input(allow_model_requests: None):
663660
],
664661
}
665662
],
666-
'n': 1,
667663
'extra_headers': {'User-Agent': IsStr(regex=r'pydantic-ai\/.*')},
668664
'extra_body': None,
669665
}

0 commit comments

Comments
 (0)