diff --git a/docs/extras/code_samples/default_v2.txt b/docs/extras/code_samples/default_v2.txt
index 49c9b855..bb0a3cd5 100644
--- a/docs/extras/code_samples/default_v2.txt
+++ b/docs/extras/code_samples/default_v2.txt
@@ -1,4 +1,4 @@
-from mindee import ClientV2, InferenceParameters
+from mindee import ClientV2, InferenceParameters, PathInput
 
 input_path = "/path/to/the/file.ext"
 api_key = "MY_API_KEY"
@@ -16,7 +16,7 @@ params = InferenceParameters(
 )
 
 # Load a file from disk
-input_source = mindee_client.source_from_path(input_path)
+input_source = PathInput(input_path)
 
 # Send for processing
 response = mindee_client.enqueue_and_get_inference(
diff --git a/mindee/__init__.py b/mindee/__init__.py
index e4263fc4..b377c01a 100644
--- a/mindee/__init__.py
+++ b/mindee/__init__.py
@@ -5,6 +5,11 @@
 from mindee.input.local_response import LocalResponse
 from mindee.input.page_options import PageOptions
 from mindee.input.polling_options import PollingOptions
+from mindee.input.sources.base_64_input import Base64Input
+from mindee.input.sources.bytes_input import BytesInput
+from mindee.input.sources.file_input import FileInput
+from mindee.input.sources.path_input import PathInput
+from mindee.input.sources.url_input_source import UrlInputSource
 from mindee.parsing.common.api_response import ApiResponse
 from mindee.parsing.common.async_predict_response import AsyncPredictResponse
 from mindee.parsing.common.feedback_response import FeedbackResponse
@@ -18,6 +23,11 @@
     "Client",
     "ClientV2",
     "InferenceParameters",
+    "FileInput",
+    "PathInput",
+    "BytesInput",
+    "Base64Input",
+    "UrlInputSource",
     "LocalResponse",
     "PageOptions",
     "PollingOptions",