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

studyfetch_sdk/resources/v1/data_analyst/sessions.py

@@ -0,0 +1,222 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+import httpx
+
+from ...._types import NOT_GIVEN, Body, Query, Headers, NoneType, NotGiven
+from ...._utils import maybe_transform, async_maybe_transform
+from ...._compat import cached_property
+from ...._resource import SyncAPIResource, AsyncAPIResource
+from ...._response import (
+    to_raw_response_wrapper,
+    to_streamed_response_wrapper,
+    async_to_raw_response_wrapper,
+    async_to_streamed_response_wrapper,
+)
+from ...._base_client import make_request_options
+from ....types.v1.data_analyst import session_retrieve_params
+
+__all__ = ["SessionsResource", "AsyncSessionsResource"]
+
+
+class SessionsResource(SyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> SessionsResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/GoStudyFetchGo/studyfetch-sdk-python#accessing-raw-response-data-eg-headers
+        """
+        return SessionsResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> SessionsResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/GoStudyFetchGo/studyfetch-sdk-python#with_streaming_response
+        """
+        return SessionsResourceWithStreamingResponse(self)
+
+    def create(
+        self,
+        *,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        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 {})}
+        return self._post(
+            "/api/v1/data-analyst/sessions/create",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=NoneType,
+        )
+
+    def retrieve(
+        self,
+        user_id: str,
+        *,
+        limit: str,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> None:
+        """
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not user_id:
+            raise ValueError(f"Expected a non-empty value for `user_id` but received {user_id!r}")
+        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
+        return self._get(
+            f"/api/v1/data-analyst/sessions/{user_id}",
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=maybe_transform({"limit": limit}, session_retrieve_params.SessionRetrieveParams),
+            ),
+            cast_to=NoneType,
+        )
+
+
+class AsyncSessionsResource(AsyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> AsyncSessionsResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/GoStudyFetchGo/studyfetch-sdk-python#accessing-raw-response-data-eg-headers
+        """
+        return AsyncSessionsResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AsyncSessionsResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/GoStudyFetchGo/studyfetch-sdk-python#with_streaming_response
+        """
+        return AsyncSessionsResourceWithStreamingResponse(self)
+
+    async def create(
+        self,
+        *,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        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 {})}
+        return await self._post(
+            "/api/v1/data-analyst/sessions/create",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=NoneType,
+        )
+
+    async def retrieve(
+        self,
+        user_id: str,
+        *,
+        limit: str,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> None:
+        """
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not user_id:
+            raise ValueError(f"Expected a non-empty value for `user_id` but received {user_id!r}")
+        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
+        return await self._get(
+            f"/api/v1/data-analyst/sessions/{user_id}",
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=await async_maybe_transform({"limit": limit}, session_retrieve_params.SessionRetrieveParams),
+            ),
+            cast_to=NoneType,
+        )
+
+
+class SessionsResourceWithRawResponse:
+    def __init__(self, sessions: SessionsResource) -> None:
+        self._sessions = sessions
+
+        self.create = to_raw_response_wrapper(
+            sessions.create,
+        )
+        self.retrieve = to_raw_response_wrapper(
+            sessions.retrieve,
+        )
+
+
+class AsyncSessionsResourceWithRawResponse:
+    def __init__(self, sessions: AsyncSessionsResource) -> None:
+        self._sessions = sessions
+
+        self.create = async_to_raw_response_wrapper(
+            sessions.create,
+        )
+        self.retrieve = async_to_raw_response_wrapper(
+            sessions.retrieve,
+        )
+
+
+class SessionsResourceWithStreamingResponse:
+    def __init__(self, sessions: SessionsResource) -> None:
+        self._sessions = sessions
+
+        self.create = to_streamed_response_wrapper(
+            sessions.create,
+        )
+        self.retrieve = to_streamed_response_wrapper(
+            sessions.retrieve,
+        )
+
+
+class AsyncSessionsResourceWithStreamingResponse:
+    def __init__(self, sessions: AsyncSessionsResource) -> None:
+        self._sessions = sessions
+
+        self.create = async_to_streamed_response_wrapper(
+            sessions.create,
+        )
+        self.retrieve = async_to_streamed_response_wrapper(
+            sessions.retrieve,
+        )

studyfetch_sdk/resources/v1/data_analyst/test.py

@@ -0,0 +1,190 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+import httpx
+
+from ...._types import NOT_GIVEN, Body, Query, Headers, NoneType, NotGiven
+from ...._compat import cached_property
+from ...._resource import SyncAPIResource, AsyncAPIResource
+from ...._response import (
+    to_raw_response_wrapper,
+    to_streamed_response_wrapper,
+    async_to_raw_response_wrapper,
+    async_to_streamed_response_wrapper,
+)
+from ...._base_client import make_request_options
+
+__all__ = ["TestResource", "AsyncTestResource"]
+
+
+class TestResource(SyncAPIResource):
+    __test__ = False
+
+    @cached_property
+    def with_raw_response(self) -> TestResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/GoStudyFetchGo/studyfetch-sdk-python#accessing-raw-response-data-eg-headers
+        """
+        return TestResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> TestResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/GoStudyFetchGo/studyfetch-sdk-python#with_streaming_response
+        """
+        return TestResourceWithStreamingResponse(self)
+
+    def upload_image(
+        self,
+        *,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        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 {})}
+        return self._post(
+            "/api/v1/data-analyst/test/image",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=NoneType,
+        )
+
+    def upload_image_citation(
+        self,
+        *,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        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 {})}
+        return self._post(
+            "/api/v1/data-analyst/test/image-citation",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=NoneType,
+        )
+
+
+class AsyncTestResource(AsyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> AsyncTestResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/GoStudyFetchGo/studyfetch-sdk-python#accessing-raw-response-data-eg-headers
+        """
+        return AsyncTestResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AsyncTestResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/GoStudyFetchGo/studyfetch-sdk-python#with_streaming_response
+        """
+        return AsyncTestResourceWithStreamingResponse(self)
+
+    async def upload_image(
+        self,
+        *,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        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 {})}
+        return await self._post(
+            "/api/v1/data-analyst/test/image",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=NoneType,
+        )
+
+    async def upload_image_citation(
+        self,
+        *,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        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 {})}
+        return await self._post(
+            "/api/v1/data-analyst/test/image-citation",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=NoneType,
+        )
+
+
+class TestResourceWithRawResponse:
+    __test__ = False
+
+    def __init__(self, test: TestResource) -> None:
+        self._test = test
+
+        self.upload_image = to_raw_response_wrapper(
+            test.upload_image,
+        )
+        self.upload_image_citation = to_raw_response_wrapper(
+            test.upload_image_citation,
+        )
+
+
+class AsyncTestResourceWithRawResponse:
+    def __init__(self, test: AsyncTestResource) -> None:
+        self._test = test
+
+        self.upload_image = async_to_raw_response_wrapper(
+            test.upload_image,
+        )
+        self.upload_image_citation = async_to_raw_response_wrapper(
+            test.upload_image_citation,
+        )
+
+
+class TestResourceWithStreamingResponse:
+    __test__ = False
+
+    def __init__(self, test: TestResource) -> None:
+        self._test = test
+
+        self.upload_image = to_streamed_response_wrapper(
+            test.upload_image,
+        )
+        self.upload_image_citation = to_streamed_response_wrapper(
+            test.upload_image_citation,
+        )
+
+
+class AsyncTestResourceWithStreamingResponse:
+    def __init__(self, test: AsyncTestResource) -> None:
+        self._test = test
+
+        self.upload_image = async_to_streamed_response_wrapper(
+            test.upload_image,
+        )
+        self.upload_image_citation = async_to_streamed_response_wrapper(
+            test.upload_image_citation,
+        )

studyfetch_sdk/resources/v1/materials/materials.py

@@ -41,6 +41,7 @@ from ....types.v1 import (
     material_search_params,
     material_batch_create_params,
     material_get_download_url_params,
+    material_create_and_process_params,
 )
 from ...._resource import SyncAPIResource, AsyncAPIResource
 from ...._response import (
@@ -51,6 +52,7 @@ from ...._response import (
 )
 from ...._base_client import make_request_options
 from ....types.v1.material import Material
+from ....types.v1.content_param import ContentParam
 from ....types.v1.material_list_response import MaterialListResponse
 from ....types.v1.material_debug_response import MaterialDebugResponse
 from ....types.v1.material_search_response import MaterialSearchResponse
@@ -95,7 +97,7 @@ class MaterialsResource(SyncAPIResource):
     def create(
         self,
         *,
-        content: material_create_params.Content,
+        content: ContentParam,
         name: str,
         folder_id: str | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -277,6 +279,63 @@ class MaterialsResource(SyncAPIResource):
             cast_to=MaterialBatchCreateResponse,
         )
 
+    def create_and_process(
+        self,
+        *,
+        content: ContentParam,
+        name: str,
+        folder_id: str | NotGiven = NOT_GIVEN,
+        poll_interval_ms: float | NotGiven = NOT_GIVEN,
+        timeout_ms: float | NotGiven = NOT_GIVEN,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> Material:
+        """Creates a material and waits for processing to finish before returning.
+
+        Useful
+        for synchronous API usage.
+
+        Args:
+          content: Content details
+
+          name: Name of the material
+
+          folder_id: Folder ID to place the material in
+
+          poll_interval_ms: Polling interval in milliseconds (default: 2 seconds)
+
+          timeout_ms: Maximum time to wait for processing in milliseconds (default: 5 minutes)
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return self._post(
+            "/api/v1/materials/upload-and-process",
+            body=maybe_transform(
+                {
+                    "content": content,
+                    "name": name,
+                    "folder_id": folder_id,
+                    "poll_interval_ms": poll_interval_ms,
+                    "timeout_ms": timeout_ms,
+                },
+                material_create_and_process_params.MaterialCreateAndProcessParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=Material,
+        )
+
     def debug(
         self,
         id: str,
@@ -546,7 +605,7 @@ class AsyncMaterialsResource(AsyncAPIResource):
     async def create(
         self,
         *,
-        content: material_create_params.Content,
+        content: ContentParam,
         name: str,
         folder_id: str | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -730,6 +789,63 @@ class AsyncMaterialsResource(AsyncAPIResource):
             cast_to=MaterialBatchCreateResponse,
         )
 
+    async def create_and_process(
+        self,
+        *,
+        content: ContentParam,
+        name: str,
+        folder_id: str | NotGiven = NOT_GIVEN,
+        poll_interval_ms: float | NotGiven = NOT_GIVEN,
+        timeout_ms: float | NotGiven = NOT_GIVEN,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> Material:
+        """Creates a material and waits for processing to finish before returning.
+
+        Useful
+        for synchronous API usage.
+
+        Args:
+          content: Content details
+
+          name: Name of the material
+
+          folder_id: Folder ID to place the material in
+
+          poll_interval_ms: Polling interval in milliseconds (default: 2 seconds)
+
+          timeout_ms: Maximum time to wait for processing in milliseconds (default: 5 minutes)
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return await self._post(
+            "/api/v1/materials/upload-and-process",
+            body=await async_maybe_transform(
+                {
+                    "content": content,
+                    "name": name,
+                    "folder_id": folder_id,
+                    "poll_interval_ms": poll_interval_ms,
+                    "timeout_ms": timeout_ms,
+                },
+                material_create_and_process_params.MaterialCreateAndProcessParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=Material,
+        )
+
     async def debug(
         self,
         id: str,
@@ -983,6 +1099,9 @@ class MaterialsResourceWithRawResponse:
         self.batch_create = to_raw_response_wrapper(
             materials.batch_create,
         )
+        self.create_and_process = to_raw_response_wrapper(
+            materials.create_and_process,
+        )
         self.debug = to_raw_response_wrapper(
             materials.debug,
         )
@@ -1034,6 +1153,9 @@ class AsyncMaterialsResourceWithRawResponse:
         self.batch_create = async_to_raw_response_wrapper(
             materials.batch_create,
         )
+        self.create_and_process = async_to_raw_response_wrapper(
+            materials.create_and_process,
+        )
         self.debug = async_to_raw_response_wrapper(
             materials.debug,
         )
@@ -1085,6 +1207,9 @@ class MaterialsResourceWithStreamingResponse:
         self.batch_create = to_streamed_response_wrapper(
             materials.batch_create,
         )
+        self.create_and_process = to_streamed_response_wrapper(
+            materials.create_and_process,
+        )
         self.debug = to_streamed_response_wrapper(
             materials.debug,
         )
@@ -1136,6 +1261,9 @@ class AsyncMaterialsResourceWithStreamingResponse:
         self.batch_create = async_to_streamed_response_wrapper(
             materials.batch_create,
         )
+        self.create_and_process = async_to_streamed_response_wrapper(
+            materials.create_and_process,
+        )
         self.debug = async_to_streamed_response_wrapper(
             materials.debug,
         )