File tree Expand file tree Collapse file tree 6 files changed +11
-10
lines changed
pydantic_ai_slim/pydantic_ai/models Expand file tree Collapse file tree 6 files changed +11
-10
lines changed Original file line number Diff line number Diff line change @@ -258,11 +258,11 @@ def _estimate_cost(messages: Iterable[Message]) -> result.Cost:
258
258
request_tokens += _string_cost (message .model_response_str ())
259
259
elif isinstance (message , RetryPrompt ):
260
260
request_tokens += _string_cost (message .model_response ())
261
- elif isinstance (message , ModelResponse ): # pyright: ignore[reportUnnecessaryIsInstance]
261
+ elif isinstance (message , ModelResponse ):
262
262
for item in message .parts :
263
263
if isinstance (item , TextPart ):
264
264
response_tokens += _string_cost (item .content )
265
- elif isinstance (item , ToolCallPart ): # pyright: ignore[reportUnnecessaryIsInstance]
265
+ elif isinstance (item , ToolCallPart ):
266
266
call = item
267
267
if isinstance (call .args , ArgsJson ):
268
268
args_str = call .args .args_json
Original file line number Diff line number Diff line change @@ -289,7 +289,7 @@ def _message_to_gemini_content(m: Message) -> _GeminiContent | None:
289
289
return _content_tool_return (m )
290
290
elif isinstance (m , RetryPrompt ):
291
291
return _content_retry_prompt (m )
292
- elif isinstance (m , ModelResponse ): # pyright: ignore[reportUnnecessaryIsInstance]
292
+ elif isinstance (m , ModelResponse ):
293
293
return _content_model_response (m )
294
294
else :
295
295
assert_never (m )
@@ -445,7 +445,7 @@ def _content_model_response(m: ModelResponse) -> _GeminiContent:
445
445
for item in m .parts :
446
446
if isinstance (item , ToolCallPart ):
447
447
parts .append (_function_call_part_from_call (item ))
448
- elif isinstance (item , TextPart ): # pyright: ignore[reportUnnecessaryIsInstance]
448
+ elif isinstance (item , TextPart ):
449
449
parts .append (_GeminiTextPart (text = item .content ))
450
450
else :
451
451
assert_never (item )
Original file line number Diff line number Diff line change @@ -271,13 +271,13 @@ def _map_message(message: Message) -> chat.ChatCompletionMessageParam:
271
271
tool_call_id = _guard_tool_call_id (t = message , model_source = 'Groq' ),
272
272
content = message .model_response (),
273
273
)
274
- elif isinstance (message , ModelResponse ): # pyright: ignore[reportUnnecessaryIsInstance]
274
+ elif isinstance (message , ModelResponse ):
275
275
texts : list [str ] = []
276
276
tool_calls : list [chat .ChatCompletionMessageToolCallParam ] = []
277
277
for item in message .parts :
278
278
if isinstance (item , TextPart ):
279
279
texts .append (item .content )
280
- elif isinstance (item , ToolCallPart ): # pyright: ignore[reportUnnecessaryIsInstance]
280
+ elif isinstance (item , ToolCallPart ):
281
281
tool_calls .append (_map_tool_call (item ))
282
282
else :
283
283
assert_never (item )
Original file line number Diff line number Diff line change @@ -259,13 +259,13 @@ def _map_message(message: Message) -> chat.ChatCompletionMessageParam:
259
259
tool_call_id = _guard_tool_call_id (t = message , model_source = 'OpenAI' ),
260
260
content = message .model_response (),
261
261
)
262
- elif isinstance (message , ModelResponse ): # pyright: ignore[reportUnnecessaryIsInstance]
262
+ elif isinstance (message , ModelResponse ):
263
263
texts : list [str ] = []
264
264
tool_calls : list [chat .ChatCompletionMessageToolCallParam ] = []
265
265
for item in message .parts :
266
266
if isinstance (item , TextPart ):
267
267
texts .append (item .content )
268
- elif isinstance (item , ToolCallPart ): # pyright: ignore[reportUnnecessaryIsInstance]
268
+ elif isinstance (item , ToolCallPart ):
269
269
tool_calls .append (_map_tool_call (item ))
270
270
else :
271
271
assert_never (item )
Original file line number Diff line number Diff line change @@ -146,7 +146,7 @@ async def request_stream(
146
146
for item in msg .parts :
147
147
if isinstance (item , TextPart ):
148
148
texts .append (item .content )
149
- elif isinstance (item , ToolCallPart ): # pyright: ignore[reportUnnecessaryIsInstance]
149
+ elif isinstance (item , ToolCallPart ):
150
150
tool_calls .append (item )
151
151
else :
152
152
assert_never (item )
Original file line number Diff line number Diff line change @@ -121,8 +121,9 @@ quote-style = "single"
121
121
122
122
[tool .pyright ]
123
123
typeCheckingMode = " strict"
124
- reportUnnecessaryTypeIgnoreComment = true
125
124
reportMissingTypeStubs = false
125
+ reportUnnecessaryIsInstance = false
126
+ reportUnnecessaryTypeIgnoreComment = true
126
127
include = [" pydantic_ai_slim" , " tests" , " pydantic_ai_examples" ]
127
128
venvPath = " .venv"
128
129
# see https://github.com/microsoft/pyright/issues/7771 - we don't want to error on decorated functions in tests
You can’t perform that action at this time.
0 commit comments