studyfetch-sdk 0.1.0a17__py3-none-any.whl → 0.1.0a19__py3-none-any.whl

studyfetch_sdk/resources/v1/materials/test.py

@@ -4,7 +4,7 @@ from __future__ import annotations
 
 import httpx
 
-from ...._types import NOT_GIVEN, Body, Query, Headers, NoneType, NotGiven
+from ...._types import NOT_GIVEN, Body, Query, Headers, NotGiven
 from ...._compat import cached_property
 from ...._resource import SyncAPIResource, AsyncAPIResource
 from ...._response import (
@@ -14,6 +14,10 @@ from ...._response import (
     async_to_streamed_response_wrapper,
 )
 from ...._base_client import make_request_options
+from ....types.v1.materials.test_perform_ocr_response import TestPerformOcrResponse
+from ....types.v1.materials.test_process_epub_response import TestProcessEpubResponse
+from ....types.v1.materials.test_process_image_response import TestProcessImageResponse
+from ....types.v1.materials.test_process_video_response import TestProcessVideoResponse
 
 __all__ = ["TestResource", "AsyncTestResource"]
 
@@ -49,14 +53,14 @@ class TestResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> None:
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
+    ) -> TestPerformOcrResponse:
+        """Test OCR functionality with a sample PDF"""
         return self._post(
             "/api/v1/materials/test/ocr",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=NoneType,
+            cast_to=TestPerformOcrResponse,
         )
 
     def process_epub(
@@ -68,14 +72,14 @@ class TestResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> None:
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
+    ) -> TestProcessEpubResponse:
+        """Test EPUB processing functionality"""
         return self._post(
             "/api/v1/materials/test/epub-processing",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=NoneType,
+            cast_to=TestProcessEpubResponse,
         )
 
     def process_image(
@@ -87,14 +91,14 @@ class TestResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> None:
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
+    ) -> TestProcessImageResponse:
+        """Test image processing with OCR and AI vision"""
         return self._post(
             "/api/v1/materials/test/image-processing",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=NoneType,
+            cast_to=TestProcessImageResponse,
         )
 
     def process_video(
@@ -106,14 +110,14 @@ class TestResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> None:
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
+    ) -> TestProcessVideoResponse:
+        """Test video processing setup and dependencies"""
         return self._post(
             "/api/v1/materials/test/video-processing",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=NoneType,
+            cast_to=TestProcessVideoResponse,
         )
 
 
@@ -146,14 +150,14 @@ class AsyncTestResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> None:
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
+    ) -> TestPerformOcrResponse:
+        """Test OCR functionality with a sample PDF"""
         return await self._post(
             "/api/v1/materials/test/ocr",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=NoneType,
+            cast_to=TestPerformOcrResponse,
         )
 
     async def process_epub(
@@ -165,14 +169,14 @@ class AsyncTestResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> None:
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
+    ) -> TestProcessEpubResponse:
+        """Test EPUB processing functionality"""
         return await self._post(
             "/api/v1/materials/test/epub-processing",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=NoneType,
+            cast_to=TestProcessEpubResponse,
         )
 
     async def process_image(
@@ -184,14 +188,14 @@ class AsyncTestResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> None:
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
+    ) -> TestProcessImageResponse:
+        """Test image processing with OCR and AI vision"""
         return await self._post(
             "/api/v1/materials/test/image-processing",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=NoneType,
+            cast_to=TestProcessImageResponse,
         )
 
     async def process_video(
@@ -203,14 +207,14 @@ class AsyncTestResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> None:
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
+    ) -> TestProcessVideoResponse:
+        """Test video processing setup and dependencies"""
         return await self._post(
             "/api/v1/materials/test/video-processing",
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=NoneType,
+            cast_to=TestProcessVideoResponse,
         )