fix: correct test assertions and add missing pyright ignore

Author: zeke

src/replicate/_client.py

@@ -350,7 +350,7 @@ def stream(
         """
         from .lib._stream import stream as _stream
 
-        return _stream(type(self), ref, input=input)  # type: ignore[return-value, arg-type]  # pyright: ignore[reportDeprecated]
+        return _stream(type(self), ref, input=input)  # type: ignore[return-value, arg-type]  # pyright: ignore[reportDeprecated, reportGeneralTypeIssues]
 
     def copy(
         self,
@@ -757,7 +757,7 @@ async def stream(
         """
         from .lib._stream import stream as _stream
 
-        return _stream(type(self), ref, input=input)  # type: ignore[return-value, arg-type]  # pyright: ignore[reportDeprecated]
+        return _stream(type(self), ref, input=input)  # type: ignore[return-value, arg-type]  # pyright: ignore[reportDeprecated, reportGeneralTypeIssues]
 
     def copy(
         self,

tests/lib/test_stream.py

@@ -67,7 +67,7 @@ def test_stream_calls_use_with_streaming_true():
         mock_use.assert_called_once()
         call_args = mock_use.call_args
         assert call_args.kwargs["streaming"] is True
-        assert call_args.args[0] == "anthropic/claude-4.5-sonnet"
+        assert call_args.args[1] == "anthropic/claude-4.5-sonnet"
 
         # Verify the mock function was called with the input
         mock_function.assert_called_once_with(prompt="Hello")
@@ -101,12 +101,14 @@ def test_stream_returns_iterator():
 
 def test_stream_works_same_as_use_with_streaming():
     """Test that stream() produces the same output as use() with streaming=True."""
-    with patch("replicate.lib._stream.use") as mock_use:
+    with patch("replicate.lib._stream.use") as mock_stream_use, \
+         patch("replicate.lib._predictions_use.use") as mock_predictions_use:
         # Create a mock function that returns an iterator
         mock_function = Mock()
         expected_output = ["Test", " ", "output"]
         mock_function.return_value = iter(expected_output.copy())
-        mock_use.return_value = mock_function
+        mock_stream_use.return_value = mock_function
+        mock_predictions_use.return_value = mock_function
 
         client = Replicate(bearer_token="test-token")