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

studyfetch_sdk/_models.py

@@ -208,14 +208,18 @@ class BaseModel(pydantic.BaseModel):
             else:
                 fields_values[name] = field_get_default(field)
 
+        extra_field_type = _get_extra_fields_type(__cls)
+
         _extra = {}
         for key, value in values.items():
             if key not in model_fields:
+                parsed = construct_type(value=value, type_=extra_field_type) if extra_field_type is not None else value
+
                 if PYDANTIC_V2:
-                    _extra[key] = value
+                    _extra[key] = parsed
                 else:
                     _fields_set.add(key)
-                    fields_values[key] = value
+                    fields_values[key] = parsed
 
         object.__setattr__(m, "__dict__", fields_values)
 
@@ -370,6 +374,23 @@ def _construct_field(value: object, field: FieldInfo, key: str) -> object:
     return construct_type(value=value, type_=type_, metadata=getattr(field, "metadata", None))
 
 
+def _get_extra_fields_type(cls: type[pydantic.BaseModel]) -> type | None:
+    if not PYDANTIC_V2:
+        # TODO
+        return None
+
+    schema = cls.__pydantic_core_schema__
+    if schema["type"] == "model":
+        fields = schema["schema"]
+        if fields["type"] == "model-fields":
+            extras = fields.get("extras_schema")
+            if extras and "cls" in extras:
+                # mypy can't narrow the type
+                return extras["cls"]  # type: ignore[no-any-return]
+
+    return None
+
+
 def is_basemodel(type_: type) -> bool:
     """Returns whether or not the given type is either a `BaseModel` or a union of `BaseModel`"""
     if is_union(type_):

studyfetch_sdk/_version.py

@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "studyfetch_sdk"
-__version__ = "0.1.0-alpha.19"  # x-release-please-version
+__version__ = "0.1.0-alpha.21"  # x-release-please-version

studyfetch_sdk/resources/v1/__init__.py

@@ -24,14 +24,6 @@ from .embed import (
     EmbedResourceWithStreamingResponse,
     AsyncEmbedResourceWithStreamingResponse,
 )
-from .tests import (
-    TestsResource,
-    AsyncTestsResource,
-    TestsResourceWithRawResponse,
-    AsyncTestsResourceWithRawResponse,
-    TestsResourceWithStreamingResponse,
-    AsyncTestsResourceWithStreamingResponse,
-)
 from .usage import (
     UsageResource,
     AsyncUsageResource,
@@ -40,14 +32,6 @@ from .usage import (
     UsageResourceWithStreamingResponse,
     AsyncUsageResourceWithStreamingResponse,
 )
-from .upload import (
-    UploadResource,
-    AsyncUploadResource,
-    UploadResourceWithRawResponse,
-    AsyncUploadResourceWithRawResponse,
-    UploadResourceWithStreamingResponse,
-    AsyncUploadResourceWithStreamingResponse,
-)
 from .folders import (
     FoldersResource,
     AsyncFoldersResource,
@@ -64,14 +48,6 @@ from .materials import (
     MaterialsResourceWithStreamingResponse,
     AsyncMaterialsResourceWithStreamingResponse,
 )
-from .scenarios import (
-    ScenariosResource,
-    AsyncScenariosResource,
-    ScenariosResourceWithRawResponse,
-    AsyncScenariosResourceWithRawResponse,
-    ScenariosResourceWithStreamingResponse,
-    AsyncScenariosResourceWithStreamingResponse,
-)
 from .components import (
     ComponentsResource,
     AsyncComponentsResource,
@@ -80,38 +56,6 @@ from .components import (
     ComponentsResourceWithStreamingResponse,
     AsyncComponentsResourceWithStreamingResponse,
 )
-from .explainers import (
-    ExplainersResource,
-    AsyncExplainersResource,
-    ExplainersResourceWithRawResponse,
-    AsyncExplainersResourceWithRawResponse,
-    ExplainersResourceWithStreamingResponse,
-    AsyncExplainersResourceWithStreamingResponse,
-)
-from .flashcards import (
-    FlashcardsResource,
-    AsyncFlashcardsResource,
-    FlashcardsResourceWithRawResponse,
-    AsyncFlashcardsResourceWithRawResponse,
-    FlashcardsResourceWithStreamingResponse,
-    AsyncFlashcardsResourceWithStreamingResponse,
-)
-from .audio_recaps import (
-    AudioRecapsResource,
-    AsyncAudioRecapsResource,
-    AudioRecapsResourceWithRawResponse,
-    AsyncAudioRecapsResourceWithRawResponse,
-    AudioRecapsResourceWithStreamingResponse,
-    AsyncAudioRecapsResourceWithStreamingResponse,
-)
-from .data_analyst import (
-    DataAnalystResource,
-    AsyncDataAnalystResource,
-    DataAnalystResourceWithRawResponse,
-    AsyncDataAnalystResourceWithRawResponse,
-    DataAnalystResourceWithStreamingResponse,
-    AsyncDataAnalystResourceWithStreamingResponse,
-)
 from .assignment_grader import (
     AssignmentGraderResource,
     AsyncAssignmentGraderResource,
@@ -152,60 +96,18 @@ __all__ = [
     "AsyncEmbedResourceWithRawResponse",
     "EmbedResourceWithStreamingResponse",
     "AsyncEmbedResourceWithStreamingResponse",
-    "ChatResource",
-    "AsyncChatResource",
-    "ChatResourceWithRawResponse",
-    "AsyncChatResourceWithRawResponse",
-    "ChatResourceWithStreamingResponse",
-    "AsyncChatResourceWithStreamingResponse",
-    "TestsResource",
-    "AsyncTestsResource",
-    "TestsResourceWithRawResponse",
-    "AsyncTestsResourceWithRawResponse",
-    "TestsResourceWithStreamingResponse",
-    "AsyncTestsResourceWithStreamingResponse",
-    "AudioRecapsResource",
-    "AsyncAudioRecapsResource",
-    "AudioRecapsResourceWithRawResponse",
-    "AsyncAudioRecapsResourceWithRawResponse",
-    "AudioRecapsResourceWithStreamingResponse",
-    "AsyncAudioRecapsResourceWithStreamingResponse",
-    "FlashcardsResource",
-    "AsyncFlashcardsResource",
-    "FlashcardsResourceWithRawResponse",
-    "AsyncFlashcardsResourceWithRawResponse",
-    "FlashcardsResourceWithStreamingResponse",
-    "AsyncFlashcardsResourceWithStreamingResponse",
-    "ScenariosResource",
-    "AsyncScenariosResource",
-    "ScenariosResourceWithRawResponse",
-    "AsyncScenariosResourceWithRawResponse",
-    "ScenariosResourceWithStreamingResponse",
-    "AsyncScenariosResourceWithStreamingResponse",
-    "ExplainersResource",
-    "AsyncExplainersResource",
-    "ExplainersResourceWithRawResponse",
-    "AsyncExplainersResourceWithRawResponse",
-    "ExplainersResourceWithStreamingResponse",
-    "AsyncExplainersResourceWithStreamingResponse",
-    "UploadResource",
-    "AsyncUploadResource",
-    "UploadResourceWithRawResponse",
-    "AsyncUploadResourceWithRawResponse",
-    "UploadResourceWithStreamingResponse",
-    "AsyncUploadResourceWithStreamingResponse",
     "AssignmentGraderResource",
     "AsyncAssignmentGraderResource",
     "AssignmentGraderResourceWithRawResponse",
     "AsyncAssignmentGraderResourceWithRawResponse",
     "AssignmentGraderResourceWithStreamingResponse",
     "AsyncAssignmentGraderResourceWithStreamingResponse",
-    "DataAnalystResource",
-    "AsyncDataAnalystResource",
-    "DataAnalystResourceWithRawResponse",
-    "AsyncDataAnalystResourceWithRawResponse",
-    "DataAnalystResourceWithStreamingResponse",
-    "AsyncDataAnalystResourceWithStreamingResponse",
+    "ChatResource",
+    "AsyncChatResource",
+    "ChatResourceWithRawResponse",
+    "AsyncChatResourceWithRawResponse",
+    "ChatResourceWithStreamingResponse",
+    "AsyncChatResourceWithStreamingResponse",
     "V1Resource",
     "AsyncV1Resource",
     "V1ResourceWithRawResponse",

studyfetch_sdk/resources/v1/chat.py

@@ -0,0 +1,246 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import List
+
+import httpx
+
+from ..._types import NOT_GIVEN, Body, Query, Headers, NoneType, NotGiven
+from ..._utils import maybe_transform, strip_not_given, async_maybe_transform
+from ..._compat import cached_property
+from ...types.v1 import chat_stream_params
+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__ = ["ChatResource", "AsyncChatResource"]
+
+
+class ChatResource(SyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> ChatResourceWithRawResponse:
+        """
+        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 ChatResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> ChatResourceWithStreamingResponse:
+        """
+        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 ChatResourceWithStreamingResponse(self)
+
+    def stream(
+        self,
+        *,
+        id: str | NotGiven = NOT_GIVEN,
+        component_id: str | NotGiven = NOT_GIVEN,
+        context: object | NotGiven = NOT_GIVEN,
+        group_ids: List[str] | NotGiven = NOT_GIVEN,
+        message: chat_stream_params.Message | NotGiven = NOT_GIVEN,
+        messages: List[str] | NotGiven = NOT_GIVEN,
+        session_id: str | NotGiven = NOT_GIVEN,
+        user_id: str | NotGiven = NOT_GIVEN,
+        x_component_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.
+        # 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:
+        """Streams a chat response in real-time using server-sent events (SSE).
+
+        Supports
+        both AI SDK format (with messages array) and custom format (with message
+        object).
+
+        Args:
+          id: Session ID (AI SDK uses "id")
+
+          component_id: Component ID
+
+          context: Additional context
+
+          group_ids: Group IDs for access control
+
+          message: Single message for custom format - contains text and optional images
+
+          messages: Messages array for AI SDK format - list of conversation messages with roles
+
+          session_id: Session ID
+
+          user_id: User ID
+
+          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
+        """
+        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
+        extra_headers = {**strip_not_given({"x-component-id": x_component_id}), **(extra_headers or {})}
+        return self._post(
+            "/api/v1/chat/stream",
+            body=maybe_transform(
+                {
+                    "id": id,
+                    "component_id": component_id,
+                    "context": context,
+                    "group_ids": group_ids,
+                    "message": message,
+                    "messages": messages,
+                    "session_id": session_id,
+                    "user_id": user_id,
+                },
+                chat_stream_params.ChatStreamParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=NoneType,
+        )
+
+
+class AsyncChatResource(AsyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> AsyncChatResourceWithRawResponse:
+        """
+        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 AsyncChatResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AsyncChatResourceWithStreamingResponse:
+        """
+        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 AsyncChatResourceWithStreamingResponse(self)
+
+    async def stream(
+        self,
+        *,
+        id: str | NotGiven = NOT_GIVEN,
+        component_id: str | NotGiven = NOT_GIVEN,
+        context: object | NotGiven = NOT_GIVEN,
+        group_ids: List[str] | NotGiven = NOT_GIVEN,
+        message: chat_stream_params.Message | NotGiven = NOT_GIVEN,
+        messages: List[str] | NotGiven = NOT_GIVEN,
+        session_id: str | NotGiven = NOT_GIVEN,
+        user_id: str | NotGiven = NOT_GIVEN,
+        x_component_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.
+        # 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:
+        """Streams a chat response in real-time using server-sent events (SSE).
+
+        Supports
+        both AI SDK format (with messages array) and custom format (with message
+        object).
+
+        Args:
+          id: Session ID (AI SDK uses "id")
+
+          component_id: Component ID
+
+          context: Additional context
+
+          group_ids: Group IDs for access control
+
+          message: Single message for custom format - contains text and optional images
+
+          messages: Messages array for AI SDK format - list of conversation messages with roles
+
+          session_id: Session ID
+
+          user_id: User ID
+
+          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
+        """
+        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
+        extra_headers = {**strip_not_given({"x-component-id": x_component_id}), **(extra_headers or {})}
+        return await self._post(
+            "/api/v1/chat/stream",
+            body=await async_maybe_transform(
+                {
+                    "id": id,
+                    "component_id": component_id,
+                    "context": context,
+                    "group_ids": group_ids,
+                    "message": message,
+                    "messages": messages,
+                    "session_id": session_id,
+                    "user_id": user_id,
+                },
+                chat_stream_params.ChatStreamParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=NoneType,
+        )
+
+
+class ChatResourceWithRawResponse:
+    def __init__(self, chat: ChatResource) -> None:
+        self._chat = chat
+
+        self.stream = to_raw_response_wrapper(
+            chat.stream,
+        )
+
+
+class AsyncChatResourceWithRawResponse:
+    def __init__(self, chat: AsyncChatResource) -> None:
+        self._chat = chat
+
+        self.stream = async_to_raw_response_wrapper(
+            chat.stream,
+        )
+
+
+class ChatResourceWithStreamingResponse:
+    def __init__(self, chat: ChatResource) -> None:
+        self._chat = chat
+
+        self.stream = to_streamed_response_wrapper(
+            chat.stream,
+        )
+
+
+class AsyncChatResourceWithStreamingResponse:
+    def __init__(self, chat: AsyncChatResource) -> None:
+        self._chat = chat
+
+        self.stream = async_to_streamed_response_wrapper(
+            chat.stream,
+        )