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

studyfetch_sdk/resources/v1/usage/analyst.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+from typing import List
 from typing_extensions import Literal
 
 import httpx
@@ -17,7 +18,12 @@ from ...._response import (
     async_to_streamed_response_wrapper,
 )
 from ...._base_client import make_request_options
-from ....types.v1.usage import analyst_list_events_params
+from ....types.v1.usage import (
+    analyst_list_events_params,
+    analyst_get_test_questions_params,
+    analyst_list_chat_messages_params,
+)
+from ....types.v1.usage.analyst_list_chat_messages_response import AnalystListChatMessagesResponse
 
 __all__ = ["AnalystResource", "AsyncAnalystResource"]
 
@@ -45,6 +51,8 @@ class AnalystResource(SyncAPIResource):
     def get_test_questions(
         self,
         *,
+        group_ids: List[str] | NotGiven = NOT_GIVEN,
+        user_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,
@@ -52,12 +60,37 @@ class AnalystResource(SyncAPIResource):
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> None:
-        """Get test results for users and groups in JWT"""
+        """
+        Get test results with full question data for user or group
+
+        Args:
+          group_ids: Array of group IDs to filter
+
+          user_id: User ID to get test results for
+
+          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 {})}
         return self._get(
             "/api/v1/usage-analyst/test-questions",
             options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=maybe_transform(
+                    {
+                        "group_ids": group_ids,
+                        "user_id": user_id,
+                    },
+                    analyst_get_test_questions_params.AnalystGetTestQuestionsParams,
+                ),
             ),
             cast_to=NoneType,
         )
@@ -65,21 +98,47 @@ class AnalystResource(SyncAPIResource):
     def list_chat_messages(
         self,
         *,
+        group_ids: List[str] | NotGiven = NOT_GIVEN,
+        user_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:
-        """Get chat messages for users and groups in JWT"""
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
+    ) -> AnalystListChatMessagesResponse:
+        """
+        Get all chat messages from sessions for user or group
+
+        Args:
+          group_ids: Array of group IDs to filter
+
+          user_id: User ID to get chat messages for
+
+          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._get(
             "/api/v1/usage-analyst/chat-messages",
             options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=maybe_transform(
+                    {
+                        "group_ids": group_ids,
+                        "user_id": user_id,
+                    },
+                    analyst_list_chat_messages_params.AnalystListChatMessagesParams,
+                ),
             ),
-            cast_to=NoneType,
+            cast_to=AnalystListChatMessagesResponse,
         )
 
     def list_events(
@@ -112,6 +171,8 @@ class AnalystResource(SyncAPIResource):
             "student_performance",
         ],
         start_date: str,
+        group_ids: List[str] | NotGiven = NOT_GIVEN,
+        user_ids: List[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,
@@ -129,6 +190,10 @@ class AnalystResource(SyncAPIResource):
 
           start_date: Start date for filtering (ISO 8601)
 
+          group_ids: Array of group IDs to filter
+
+          user_ids: Array of user IDs to filter
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -150,6 +215,8 @@ class AnalystResource(SyncAPIResource):
                         "end_date": end_date,
                         "event_type": event_type,
                         "start_date": start_date,
+                        "group_ids": group_ids,
+                        "user_ids": user_ids,
                     },
                     analyst_list_events_params.AnalystListEventsParams,
                 ),
@@ -181,6 +248,8 @@ class AsyncAnalystResource(AsyncAPIResource):
     async def get_test_questions(
         self,
         *,
+        group_ids: List[str] | NotGiven = NOT_GIVEN,
+        user_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,
@@ -188,12 +257,37 @@ class AsyncAnalystResource(AsyncAPIResource):
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> None:
-        """Get test results for users and groups in JWT"""
+        """
+        Get test results with full question data for user or group
+
+        Args:
+          group_ids: Array of group IDs to filter
+
+          user_id: User ID to get test results for
+
+          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 {})}
         return await self._get(
             "/api/v1/usage-analyst/test-questions",
             options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=await async_maybe_transform(
+                    {
+                        "group_ids": group_ids,
+                        "user_id": user_id,
+                    },
+                    analyst_get_test_questions_params.AnalystGetTestQuestionsParams,
+                ),
             ),
             cast_to=NoneType,
         )
@@ -201,21 +295,47 @@ class AsyncAnalystResource(AsyncAPIResource):
     async def list_chat_messages(
         self,
         *,
+        group_ids: List[str] | NotGiven = NOT_GIVEN,
+        user_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:
-        """Get chat messages for users and groups in JWT"""
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
+    ) -> AnalystListChatMessagesResponse:
+        """
+        Get all chat messages from sessions for user or group
+
+        Args:
+          group_ids: Array of group IDs to filter
+
+          user_id: User ID to get chat messages for
+
+          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._get(
             "/api/v1/usage-analyst/chat-messages",
             options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=await async_maybe_transform(
+                    {
+                        "group_ids": group_ids,
+                        "user_id": user_id,
+                    },
+                    analyst_list_chat_messages_params.AnalystListChatMessagesParams,
+                ),
             ),
-            cast_to=NoneType,
+            cast_to=AnalystListChatMessagesResponse,
         )
 
     async def list_events(
@@ -248,6 +368,8 @@ class AsyncAnalystResource(AsyncAPIResource):
             "student_performance",
         ],
         start_date: str,
+        group_ids: List[str] | NotGiven = NOT_GIVEN,
+        user_ids: List[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,
@@ -265,6 +387,10 @@ class AsyncAnalystResource(AsyncAPIResource):
 
           start_date: Start date for filtering (ISO 8601)
 
+          group_ids: Array of group IDs to filter
+
+          user_ids: Array of user IDs to filter
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -286,6 +412,8 @@ class AsyncAnalystResource(AsyncAPIResource):
                         "end_date": end_date,
                         "event_type": event_type,
                         "start_date": start_date,
+                        "group_ids": group_ids,
+                        "user_ids": user_ids,
                     },
                     analyst_list_events_params.AnalystListEventsParams,
                 ),

studyfetch_sdk/resources/v1/v1.py

@@ -2,6 +2,8 @@
 
 from __future__ import annotations
 
+import httpx
+
 from .folders import (
     FoldersResource,
     AsyncFoldersResource,
@@ -10,6 +12,7 @@ from .folders import (
     FoldersResourceWithStreamingResponse,
     AsyncFoldersResourceWithStreamingResponse,
 )
+from ..._types import NOT_GIVEN, Body, Query, Headers, NoneType, NotGiven
 from ..._compat import cached_property
 from .chat.chat import (
     ChatResource,
@@ -44,6 +47,12 @@ from .flashcards import (
     AsyncFlashcardsResourceWithStreamingResponse,
 )
 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 .embed.embed import (
     EmbedResource,
     AsyncEmbedResource,
@@ -76,6 +85,7 @@ from .upload.upload import (
     UploadResourceWithStreamingResponse,
     AsyncUploadResourceWithStreamingResponse,
 )
+from ..._base_client import make_request_options
 from .assignment_grader import (
     AssignmentGraderResource,
     AsyncAssignmentGraderResource,
@@ -108,6 +118,14 @@ from .audio_recaps.audio_recaps import (
     AudioRecapsResourceWithStreamingResponse,
     AsyncAudioRecapsResourceWithStreamingResponse,
 )
+from .data_analyst.data_analyst import (
+    DataAnalystResource,
+    AsyncDataAnalystResource,
+    DataAnalystResourceWithRawResponse,
+    AsyncDataAnalystResourceWithRawResponse,
+    DataAnalystResourceWithStreamingResponse,
+    AsyncDataAnalystResourceWithStreamingResponse,
+)
 
 __all__ = ["V1Resource", "AsyncV1Resource"]
 
@@ -165,6 +183,10 @@ class V1Resource(SyncAPIResource):
     def assignment_grader(self) -> AssignmentGraderResource:
         return AssignmentGraderResource(self._client)
 
+    @cached_property
+    def data_analyst(self) -> DataAnalystResource:
+        return DataAnalystResource(self._client)
+
     @cached_property
     def with_raw_response(self) -> V1ResourceWithRawResponse:
         """
@@ -184,6 +206,26 @@ class V1Resource(SyncAPIResource):
         """
         return V1ResourceWithStreamingResponse(self)
 
+    def test_mongodb(
+        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:
+        """Test MongoDB connection and get outbound IP"""
+        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
+        return self._get(
+            "/api/v1/test-mongodb",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=NoneType,
+        )
+
 
 class AsyncV1Resource(AsyncAPIResource):
     @cached_property
@@ -238,6 +280,10 @@ class AsyncV1Resource(AsyncAPIResource):
     def assignment_grader(self) -> AsyncAssignmentGraderResource:
         return AsyncAssignmentGraderResource(self._client)
 
+    @cached_property
+    def data_analyst(self) -> AsyncDataAnalystResource:
+        return AsyncDataAnalystResource(self._client)
+
     @cached_property
     def with_raw_response(self) -> AsyncV1ResourceWithRawResponse:
         """
@@ -257,11 +303,35 @@ class AsyncV1Resource(AsyncAPIResource):
         """
         return AsyncV1ResourceWithStreamingResponse(self)
 
+    async def test_mongodb(
+        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:
+        """Test MongoDB connection and get outbound IP"""
+        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
+        return await self._get(
+            "/api/v1/test-mongodb",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=NoneType,
+        )
+
 
 class V1ResourceWithRawResponse:
     def __init__(self, v1: V1Resource) -> None:
         self._v1 = v1
 
+        self.test_mongodb = to_raw_response_wrapper(
+            v1.test_mongodb,
+        )
+
     @cached_property
     def materials(self) -> MaterialsResourceWithRawResponse:
         return MaterialsResourceWithRawResponse(self._v1.materials)
@@ -314,11 +384,19 @@ class V1ResourceWithRawResponse:
     def assignment_grader(self) -> AssignmentGraderResourceWithRawResponse:
         return AssignmentGraderResourceWithRawResponse(self._v1.assignment_grader)
 
+    @cached_property
+    def data_analyst(self) -> DataAnalystResourceWithRawResponse:
+        return DataAnalystResourceWithRawResponse(self._v1.data_analyst)
+
 
 class AsyncV1ResourceWithRawResponse:
     def __init__(self, v1: AsyncV1Resource) -> None:
         self._v1 = v1
 
+        self.test_mongodb = async_to_raw_response_wrapper(
+            v1.test_mongodb,
+        )
+
     @cached_property
     def materials(self) -> AsyncMaterialsResourceWithRawResponse:
         return AsyncMaterialsResourceWithRawResponse(self._v1.materials)
@@ -371,11 +449,19 @@ class AsyncV1ResourceWithRawResponse:
     def assignment_grader(self) -> AsyncAssignmentGraderResourceWithRawResponse:
         return AsyncAssignmentGraderResourceWithRawResponse(self._v1.assignment_grader)
 
+    @cached_property
+    def data_analyst(self) -> AsyncDataAnalystResourceWithRawResponse:
+        return AsyncDataAnalystResourceWithRawResponse(self._v1.data_analyst)
+
 
 class V1ResourceWithStreamingResponse:
     def __init__(self, v1: V1Resource) -> None:
         self._v1 = v1
 
+        self.test_mongodb = to_streamed_response_wrapper(
+            v1.test_mongodb,
+        )
+
     @cached_property
     def materials(self) -> MaterialsResourceWithStreamingResponse:
         return MaterialsResourceWithStreamingResponse(self._v1.materials)
@@ -428,11 +514,19 @@ class V1ResourceWithStreamingResponse:
     def assignment_grader(self) -> AssignmentGraderResourceWithStreamingResponse:
         return AssignmentGraderResourceWithStreamingResponse(self._v1.assignment_grader)
 
+    @cached_property
+    def data_analyst(self) -> DataAnalystResourceWithStreamingResponse:
+        return DataAnalystResourceWithStreamingResponse(self._v1.data_analyst)
+
 
 class AsyncV1ResourceWithStreamingResponse:
     def __init__(self, v1: AsyncV1Resource) -> None:
         self._v1 = v1
 
+        self.test_mongodb = async_to_streamed_response_wrapper(
+            v1.test_mongodb,
+        )
+
     @cached_property
     def materials(self) -> AsyncMaterialsResourceWithStreamingResponse:
         return AsyncMaterialsResourceWithStreamingResponse(self._v1.materials)
@@ -484,3 +578,7 @@ class AsyncV1ResourceWithStreamingResponse:
     @cached_property
     def assignment_grader(self) -> AsyncAssignmentGraderResourceWithStreamingResponse:
         return AsyncAssignmentGraderResourceWithStreamingResponse(self._v1.assignment_grader)
+
+    @cached_property
+    def data_analyst(self) -> AsyncDataAnalystResourceWithStreamingResponse:
+        return AsyncDataAnalystResourceWithStreamingResponse(self._v1.data_analyst)

studyfetch_sdk/types/v1/__init__.py

@@ -4,6 +4,7 @@ from __future__ import annotations
 
 from .material import Material as Material
 from .component import Component as Component
+from .content_param import ContentParam as ContentParam
 from .chat_stream_params import ChatStreamParams as ChatStreamParams
 from .folder_list_params import FolderListParams as FolderListParams
 from .test_create_params import TestCreateParams as TestCreateParams
@@ -13,11 +14,13 @@ from .embed_verify_params import EmbedVerifyParams as EmbedVerifyParams
 from .folder_create_params import FolderCreateParams as FolderCreateParams
 from .folder_update_params import FolderUpdateParams as FolderUpdateParams
 from .material_list_params import MaterialListParams as MaterialListParams
+from .material_move_params import MaterialMoveParams as MaterialMoveParams
 from .component_list_params import ComponentListParams as ComponentListParams
 from .flashcard_rate_params import FlashcardRateParams as FlashcardRateParams
 from .embed_get_theme_params import EmbedGetThemeParams as EmbedGetThemeParams
 from .material_create_params import MaterialCreateParams as MaterialCreateParams
 from .material_list_response import MaterialListResponse as MaterialListResponse
+from .material_rename_params import MaterialRenameParams as MaterialRenameParams
 from .material_search_params import MaterialSearchParams as MaterialSearchParams
 from .scenario_create_params import ScenarioCreateParams as ScenarioCreateParams
 from .scenario_update_params import ScenarioUpdateParams as ScenarioUpdateParams
@@ -27,22 +30,31 @@ from .component_create_params import ComponentCreateParams as ComponentCreatePar
 from .component_list_response import ComponentListResponse as ComponentListResponse
 from .component_update_params import ComponentUpdateParams as ComponentUpdateParams
 from .explainer_create_params import ExplainerCreateParams as ExplainerCreateParams
+from .material_debug_response import MaterialDebugResponse as MaterialDebugResponse
 from .chat_send_message_params import ChatSendMessageParams as ChatSendMessageParams
 from .flashcard_get_all_params import FlashcardGetAllParams as FlashcardGetAllParams
 from .flashcard_get_due_params import FlashcardGetDueParams as FlashcardGetDueParams
+from .material_search_response import MaterialSearchResponse as MaterialSearchResponse
 from .usage_get_summary_params import UsageGetSummaryParams as UsageGetSummaryParams
 from .usage_list_events_params import UsageListEventsParams as UsageListEventsParams
 from .test_submit_answer_params import TestSubmitAnswerParams as TestSubmitAnswerParams
 from .assignment_grader_response import AssignmentGraderResponse as AssignmentGraderResponse
+from .data_analyst_stream_params import DataAnalystStreamParams as DataAnalystStreamParams
 from .flashcard_get_stats_params import FlashcardGetStatsParams as FlashcardGetStatsParams
 from .flashcard_get_types_response import FlashcardGetTypesResponse as FlashcardGetTypesResponse
+from .material_batch_create_params import MaterialBatchCreateParams as MaterialBatchCreateParams
 from .scenario_submit_answer_params import ScenarioSubmitAnswerParams as ScenarioSubmitAnswerParams
 from .flashcard_batch_process_params import FlashcardBatchProcessParams as FlashcardBatchProcessParams
+from .material_batch_create_response import MaterialBatchCreateResponse as MaterialBatchCreateResponse
 from .assignment_grader_create_params import AssignmentGraderCreateParams as AssignmentGraderCreateParams
 from .component_generate_embed_params import ComponentGenerateEmbedParams as ComponentGenerateEmbedParams
 from .explainer_handle_webhook_params import ExplainerHandleWebhookParams as ExplainerHandleWebhookParams
+from .data_analyst_send_message_params import DataAnalystSendMessageParams as DataAnalystSendMessageParams
 from .flashcard_batch_process_response import FlashcardBatchProcessResponse as FlashcardBatchProcessResponse
 from .flashcard_get_algorithm_response import FlashcardGetAlgorithmResponse as FlashcardGetAlgorithmResponse
 from .material_get_download_url_params import MaterialGetDownloadURLParams as MaterialGetDownloadURLParams
 from .component_generate_embed_response import ComponentGenerateEmbedResponse as ComponentGenerateEmbedResponse
 from .assignment_grader_get_all_response import AssignmentGraderGetAllResponse as AssignmentGraderGetAllResponse
+from .material_create_and_process_params import MaterialCreateAndProcessParams as MaterialCreateAndProcessParams
+from .material_get_download_url_response import MaterialGetDownloadURLResponse as MaterialGetDownloadURLResponse
+from .data_analyst_retrieve_session_params import DataAnalystRetrieveSessionParams as DataAnalystRetrieveSessionParams

studyfetch_sdk/types/v1/assignment_grader_response.py

@@ -1,5 +1,6 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
+from typing import Optional
 from datetime import datetime
 
 from pydantic import Field as FieldInfo
@@ -10,7 +11,10 @@ __all__ = ["AssignmentGraderResponse"]
 
 
 class AssignmentGraderResponse(BaseModel):
-    id: str
+    api_v: float = FieldInfo(alias="__v")
+    """Version key"""
+
+    api_id: str = FieldInfo(alias="_id")
     """Assignment grader ID"""
 
     created_at: datetime = FieldInfo(alias="createdAt")
@@ -19,8 +23,23 @@ class AssignmentGraderResponse(BaseModel):
     grade: float
     """Overall grade percentage"""
 
+    organization_id: str = FieldInfo(alias="organizationId")
+    """Organization ID"""
+
     rubric: object
     """Grading results"""
 
     title: str
     """Assignment title"""
+
+    updated_at: datetime = FieldInfo(alias="updatedAt")
+    """Update timestamp"""
+
+    material_id: Optional[str] = FieldInfo(alias="materialId", default=None)
+    """Material ID"""
+
+    text_to_grade: Optional[str] = FieldInfo(alias="textToGrade", default=None)
+    """Text that was graded"""
+
+    user_id: Optional[str] = FieldInfo(alias="userId", default=None)
+    """User ID"""

studyfetch_sdk/types/v1/component.py

@@ -34,7 +34,15 @@ class Component(BaseModel):
     """Component status"""
 
     type: Literal[
-        "chat", "flashcards", "scenarios", "practice_test", "audio_recap", "tutor_me", "explainers", "uploads"
+        "chat",
+        "data_analyst",
+        "flashcards",
+        "scenarios",
+        "practice_test",
+        "audio_recap",
+        "tutor_me",
+        "explainers",
+        "uploads",
     ]
     """Component type"""
 

studyfetch_sdk/types/v1/component_create_params.py

@@ -11,6 +11,7 @@ __all__ = [
     "ComponentCreateParams",
     "Config",
     "ConfigChatConfigDto",
+    "ConfigDataAnalystConfigDto",
     "ConfigFlashcardsConfigDto",
     "ConfigScenariosConfigDto",
     "ConfigScenariosConfigDtoCharacter",
@@ -32,7 +33,17 @@ class ComponentCreateParams(TypedDict, total=False):
     """Name of the component"""
 
     type: Required[
-        Literal["chat", "flashcards", "scenarios", "practice_test", "audio_recap", "tutor_me", "explainers", "uploads"]
+        Literal[
+            "chat",
+            "data_analyst",
+            "flashcards",
+            "scenarios",
+            "practice_test",
+            "audio_recap",
+            "tutor_me",
+            "explainers",
+            "uploads",
+        ]
     ]
     """Type of component to create"""
 
@@ -84,6 +95,50 @@ class ConfigChatConfigDto(TypedDict, total=False):
     """Temperature for response generation"""
 
 
+class ConfigDataAnalystConfigDto(TypedDict, total=False):
+    model: Required[str]
+    """AI model to use"""
+
+    enable_component_creation: Annotated[bool, PropertyInfo(alias="enableComponentCreation")]
+    """Enable component creation"""
+
+    enable_follow_ups: Annotated[bool, PropertyInfo(alias="enableFollowUps")]
+    """Enable follow-up questions"""
+
+    enable_history: Annotated[bool, PropertyInfo(alias="enableHistory")]
+    """Enable conversation history"""
+
+    enable_rag_search: Annotated[bool, PropertyInfo(alias="enableRAGSearch")]
+    """Enable RAG search"""
+
+    enable_voice: Annotated[bool, PropertyInfo(alias="enableVoice")]
+    """Enable voice interactions"""
+
+    enable_web_search: Annotated[bool, PropertyInfo(alias="enableWebSearch")]
+    """Enable web search"""
+
+    folders: List[str]
+    """Folder IDs"""
+
+    group_ids: Annotated[List[str], PropertyInfo(alias="groupIds")]
+    """Group IDs to filter data"""
+
+    materials: List[str]
+    """Material IDs"""
+
+    max_steps: Annotated[float, PropertyInfo(alias="maxSteps")]
+    """Maximum steps for multi-step tool calls"""
+
+    max_tokens: Annotated[float, PropertyInfo(alias="maxTokens")]
+    """Maximum tokens for response"""
+
+    system_prompt: Annotated[str, PropertyInfo(alias="systemPrompt")]
+    """System prompt for the data analyst"""
+
+    temperature: float
+    """Temperature for response generation"""
+
+
 class ConfigFlashcardsConfigDto(TypedDict, total=False):
     card_types: Annotated[List[str], PropertyInfo(alias="cardTypes")]
     """Types of flashcards"""
@@ -359,6 +414,7 @@ class ConfigTutorMeConfigDto(TypedDict, total=False):
 
 Config: TypeAlias = Union[
     ConfigChatConfigDto,
+    ConfigDataAnalystConfigDto,
     ConfigFlashcardsConfigDto,
     ConfigScenariosConfigDto,
     ConfigPracticeTestConfigDto,