studyfetch-sdk 0.1.0a2__py3-none-any.whl → 0.1.0a3__py3-none-any.whl

studyfetch_sdk/resources/v1/usage.py

@@ -9,13 +9,7 @@ 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 ...types.v1 import (
-    usage_get_stats_params,
-    usage_track_chat_params,
-    usage_get_summary_params,
-    usage_list_events_params,
-    usage_track_event_params,
-)
+from ...types.v1 import usage_get_stats_params, usage_get_summary_params, usage_list_events_params
 from ..._resource import SyncAPIResource, AsyncAPIResource
 from ..._response import (
     to_raw_response_wrapper,
@@ -253,281 +247,6 @@ class UsageResource(SyncAPIResource):
             cast_to=NoneType,
         )
 
-    def track_chat(
-        self,
-        *,
-        input_tokens: float,
-        model: Literal[
-            "openai:gpt-4.1",
-            "openai:gpt-4o",
-            "openai:gpt-4o-mini",
-            "openai:gpt-4",
-            "openai:gpt-3.5-turbo",
-            "openai:o1",
-            "openai:o1-mini",
-            "openai:o3-mini",
-            "anthropic:claude-3-opus",
-            "anthropic:claude-3-sonnet",
-            "anthropic:claude-3-haiku",
-            "google:gemini-pro",
-            "meta:llama-3",
-        ],
-        output_tokens: float,
-        session_id: str,
-        component_id: str | NotGiven = NOT_GIVEN,
-        group_id: str | NotGiven = NOT_GIVEN,
-        metadata: object | NotGiven = NOT_GIVEN,
-        reasoning_tokens: float | 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:
-        """
-        Args:
-          input_tokens: Number of input tokens
-
-          model: AI model used
-
-          output_tokens: Number of output tokens
-
-          session_id: Chat session ID
-
-          component_id: Component ID
-
-          group_id: Group ID
-
-          metadata: Additional metadata
-
-          reasoning_tokens: Number of reasoning tokens
-
-          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 {})}
-        return self._post(
-            "/api/v1/usage/chat",
-            body=maybe_transform(
-                {
-                    "input_tokens": input_tokens,
-                    "model": model,
-                    "output_tokens": output_tokens,
-                    "session_id": session_id,
-                    "component_id": component_id,
-                    "group_id": group_id,
-                    "metadata": metadata,
-                    "reasoning_tokens": reasoning_tokens,
-                    "user_id": user_id,
-                },
-                usage_track_chat_params.UsageTrackChatParams,
-            ),
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=NoneType,
-        )
-
-    def track_event(
-        self,
-        *,
-        event_type: Literal[
-            "material_created",
-            "material_uploaded",
-            "material_processed",
-            "material_deleted",
-            "component_created",
-            "component_accessed",
-            "component_deleted",
-            "component_usage",
-            "chat_message_sent",
-            "chat_session_started",
-            "chat_session_ended",
-            "test_created",
-            "test_started",
-            "test_completed",
-            "test_question_answered",
-            "test_retaken",
-            "audio_recap_create",
-            "api_call",
-            "cache_hit",
-            "sso_login",
-            "sso_logout",
-            "student_performance",
-        ],
-        cache_hit: bool | NotGiven = NOT_GIVEN,
-        component_name: str | NotGiven = NOT_GIVEN,
-        component_type: str | NotGiven = NOT_GIVEN,
-        content_size: float | NotGiven = NOT_GIVEN,
-        endpoint: str | NotGiven = NOT_GIVEN,
-        file_size: float | NotGiven = NOT_GIVEN,
-        file_type: str | NotGiven = NOT_GIVEN,
-        group_id: str | NotGiven = NOT_GIVEN,
-        input_tokens: float | NotGiven = NOT_GIVEN,
-        ip_address: str | NotGiven = NOT_GIVEN,
-        metadata: object | NotGiven = NOT_GIVEN,
-        method: str | NotGiven = NOT_GIVEN,
-        model: Literal[
-            "openai:gpt-4.1",
-            "openai:gpt-4o",
-            "openai:gpt-4o-mini",
-            "openai:gpt-4",
-            "openai:gpt-3.5-turbo",
-            "openai:o1",
-            "openai:o1-mini",
-            "openai:o3-mini",
-            "anthropic:claude-3-opus",
-            "anthropic:claude-3-sonnet",
-            "anthropic:claude-3-haiku",
-            "google:gemini-pro",
-            "meta:llama-3",
-        ]
-        | NotGiven = NOT_GIVEN,
-        output_tokens: float | NotGiven = NOT_GIVEN,
-        performance_data: object | NotGiven = NOT_GIVEN,
-        reasoning_tokens: float | NotGiven = NOT_GIVEN,
-        resource_id: str | NotGiven = NOT_GIVEN,
-        resource_type: str | NotGiven = NOT_GIVEN,
-        response_time: float | NotGiven = NOT_GIVEN,
-        session_id: str | NotGiven = NOT_GIVEN,
-        status_code: float | NotGiven = NOT_GIVEN,
-        total_tokens: float | NotGiven = NOT_GIVEN,
-        user_agent: 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:
-        """
-        Args:
-          event_type: Type of usage event
-
-          cache_hit: Whether response was served from cache
-
-          component_name: Component name
-
-          component_type: Component type
-
-          content_size: Size of content in bytes
-
-          endpoint: API endpoint accessed
-
-          file_size: File size in bytes
-
-          file_type: File MIME type
-
-          group_id: Group ID for collaborative sessions
-
-          input_tokens: Number of input tokens
-
-          ip_address: Client IP address
-
-          metadata: Additional metadata
-
-          method: HTTP method used
-
-          model: AI model used
-
-          output_tokens: Number of output tokens
-
-          performance_data: Performance metrics data
-
-          reasoning_tokens: Number of reasoning tokens
-
-          resource_id: Resource ID being accessed
-
-          resource_type: Type of resource
-
-          response_time: Response time in milliseconds
-
-          session_id: Session ID
-
-          status_code: HTTP status code
-
-          total_tokens: Total number of tokens
-
-          user_agent: Client user agent
-
-          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 {})}
-        return self._post(
-            "/api/v1/usage/track",
-            body=maybe_transform(
-                {
-                    "event_type": event_type,
-                    "cache_hit": cache_hit,
-                    "component_name": component_name,
-                    "component_type": component_type,
-                    "content_size": content_size,
-                    "endpoint": endpoint,
-                    "file_size": file_size,
-                    "file_type": file_type,
-                    "group_id": group_id,
-                    "input_tokens": input_tokens,
-                    "ip_address": ip_address,
-                    "metadata": metadata,
-                    "method": method,
-                    "model": model,
-                    "output_tokens": output_tokens,
-                    "performance_data": performance_data,
-                    "reasoning_tokens": reasoning_tokens,
-                    "resource_id": resource_id,
-                    "resource_type": resource_type,
-                    "response_time": response_time,
-                    "session_id": session_id,
-                    "status_code": status_code,
-                    "total_tokens": total_tokens,
-                    "user_agent": user_agent,
-                    "user_id": user_id,
-                },
-                usage_track_event_params.UsageTrackEventParams,
-            ),
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=NoneType,
-        )
-
-    def track_performance(
-        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/usage/performance",
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=NoneType,
-        )
-
 
 class AsyncUsageResource(AsyncAPIResource):
     @cached_property
@@ -754,281 +473,6 @@ class AsyncUsageResource(AsyncAPIResource):
             cast_to=NoneType,
         )
 
-    async def track_chat(
-        self,
-        *,
-        input_tokens: float,
-        model: Literal[
-            "openai:gpt-4.1",
-            "openai:gpt-4o",
-            "openai:gpt-4o-mini",
-            "openai:gpt-4",
-            "openai:gpt-3.5-turbo",
-            "openai:o1",
-            "openai:o1-mini",
-            "openai:o3-mini",
-            "anthropic:claude-3-opus",
-            "anthropic:claude-3-sonnet",
-            "anthropic:claude-3-haiku",
-            "google:gemini-pro",
-            "meta:llama-3",
-        ],
-        output_tokens: float,
-        session_id: str,
-        component_id: str | NotGiven = NOT_GIVEN,
-        group_id: str | NotGiven = NOT_GIVEN,
-        metadata: object | NotGiven = NOT_GIVEN,
-        reasoning_tokens: float | 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:
-        """
-        Args:
-          input_tokens: Number of input tokens
-
-          model: AI model used
-
-          output_tokens: Number of output tokens
-
-          session_id: Chat session ID
-
-          component_id: Component ID
-
-          group_id: Group ID
-
-          metadata: Additional metadata
-
-          reasoning_tokens: Number of reasoning tokens
-
-          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 {})}
-        return await self._post(
-            "/api/v1/usage/chat",
-            body=await async_maybe_transform(
-                {
-                    "input_tokens": input_tokens,
-                    "model": model,
-                    "output_tokens": output_tokens,
-                    "session_id": session_id,
-                    "component_id": component_id,
-                    "group_id": group_id,
-                    "metadata": metadata,
-                    "reasoning_tokens": reasoning_tokens,
-                    "user_id": user_id,
-                },
-                usage_track_chat_params.UsageTrackChatParams,
-            ),
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=NoneType,
-        )
-
-    async def track_event(
-        self,
-        *,
-        event_type: Literal[
-            "material_created",
-            "material_uploaded",
-            "material_processed",
-            "material_deleted",
-            "component_created",
-            "component_accessed",
-            "component_deleted",
-            "component_usage",
-            "chat_message_sent",
-            "chat_session_started",
-            "chat_session_ended",
-            "test_created",
-            "test_started",
-            "test_completed",
-            "test_question_answered",
-            "test_retaken",
-            "audio_recap_create",
-            "api_call",
-            "cache_hit",
-            "sso_login",
-            "sso_logout",
-            "student_performance",
-        ],
-        cache_hit: bool | NotGiven = NOT_GIVEN,
-        component_name: str | NotGiven = NOT_GIVEN,
-        component_type: str | NotGiven = NOT_GIVEN,
-        content_size: float | NotGiven = NOT_GIVEN,
-        endpoint: str | NotGiven = NOT_GIVEN,
-        file_size: float | NotGiven = NOT_GIVEN,
-        file_type: str | NotGiven = NOT_GIVEN,
-        group_id: str | NotGiven = NOT_GIVEN,
-        input_tokens: float | NotGiven = NOT_GIVEN,
-        ip_address: str | NotGiven = NOT_GIVEN,
-        metadata: object | NotGiven = NOT_GIVEN,
-        method: str | NotGiven = NOT_GIVEN,
-        model: Literal[
-            "openai:gpt-4.1",
-            "openai:gpt-4o",
-            "openai:gpt-4o-mini",
-            "openai:gpt-4",
-            "openai:gpt-3.5-turbo",
-            "openai:o1",
-            "openai:o1-mini",
-            "openai:o3-mini",
-            "anthropic:claude-3-opus",
-            "anthropic:claude-3-sonnet",
-            "anthropic:claude-3-haiku",
-            "google:gemini-pro",
-            "meta:llama-3",
-        ]
-        | NotGiven = NOT_GIVEN,
-        output_tokens: float | NotGiven = NOT_GIVEN,
-        performance_data: object | NotGiven = NOT_GIVEN,
-        reasoning_tokens: float | NotGiven = NOT_GIVEN,
-        resource_id: str | NotGiven = NOT_GIVEN,
-        resource_type: str | NotGiven = NOT_GIVEN,
-        response_time: float | NotGiven = NOT_GIVEN,
-        session_id: str | NotGiven = NOT_GIVEN,
-        status_code: float | NotGiven = NOT_GIVEN,
-        total_tokens: float | NotGiven = NOT_GIVEN,
-        user_agent: 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:
-        """
-        Args:
-          event_type: Type of usage event
-
-          cache_hit: Whether response was served from cache
-
-          component_name: Component name
-
-          component_type: Component type
-
-          content_size: Size of content in bytes
-
-          endpoint: API endpoint accessed
-
-          file_size: File size in bytes
-
-          file_type: File MIME type
-
-          group_id: Group ID for collaborative sessions
-
-          input_tokens: Number of input tokens
-
-          ip_address: Client IP address
-
-          metadata: Additional metadata
-
-          method: HTTP method used
-
-          model: AI model used
-
-          output_tokens: Number of output tokens
-
-          performance_data: Performance metrics data
-
-          reasoning_tokens: Number of reasoning tokens
-
-          resource_id: Resource ID being accessed
-
-          resource_type: Type of resource
-
-          response_time: Response time in milliseconds
-
-          session_id: Session ID
-
-          status_code: HTTP status code
-
-          total_tokens: Total number of tokens
-
-          user_agent: Client user agent
-
-          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 {})}
-        return await self._post(
-            "/api/v1/usage/track",
-            body=await async_maybe_transform(
-                {
-                    "event_type": event_type,
-                    "cache_hit": cache_hit,
-                    "component_name": component_name,
-                    "component_type": component_type,
-                    "content_size": content_size,
-                    "endpoint": endpoint,
-                    "file_size": file_size,
-                    "file_type": file_type,
-                    "group_id": group_id,
-                    "input_tokens": input_tokens,
-                    "ip_address": ip_address,
-                    "metadata": metadata,
-                    "method": method,
-                    "model": model,
-                    "output_tokens": output_tokens,
-                    "performance_data": performance_data,
-                    "reasoning_tokens": reasoning_tokens,
-                    "resource_id": resource_id,
-                    "resource_type": resource_type,
-                    "response_time": response_time,
-                    "session_id": session_id,
-                    "status_code": status_code,
-                    "total_tokens": total_tokens,
-                    "user_agent": user_agent,
-                    "user_id": user_id,
-                },
-                usage_track_event_params.UsageTrackEventParams,
-            ),
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=NoneType,
-        )
-
-    async def track_performance(
-        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/usage/performance",
-            options=make_request_options(
-                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
-            ),
-            cast_to=NoneType,
-        )
-
 
 class UsageResourceWithRawResponse:
     def __init__(self, usage: UsageResource) -> None:
@@ -1043,15 +487,6 @@ class UsageResourceWithRawResponse:
         self.list_events = to_raw_response_wrapper(
             usage.list_events,
         )
-        self.track_chat = to_raw_response_wrapper(
-            usage.track_chat,
-        )
-        self.track_event = to_raw_response_wrapper(
-            usage.track_event,
-        )
-        self.track_performance = to_raw_response_wrapper(
-            usage.track_performance,
-        )
 
 
 class AsyncUsageResourceWithRawResponse:
@@ -1067,15 +502,6 @@ class AsyncUsageResourceWithRawResponse:
         self.list_events = async_to_raw_response_wrapper(
             usage.list_events,
         )
-        self.track_chat = async_to_raw_response_wrapper(
-            usage.track_chat,
-        )
-        self.track_event = async_to_raw_response_wrapper(
-            usage.track_event,
-        )
-        self.track_performance = async_to_raw_response_wrapper(
-            usage.track_performance,
-        )
 
 
 class UsageResourceWithStreamingResponse:
@@ -1091,15 +517,6 @@ class UsageResourceWithStreamingResponse:
         self.list_events = to_streamed_response_wrapper(
             usage.list_events,
         )
-        self.track_chat = to_streamed_response_wrapper(
-            usage.track_chat,
-        )
-        self.track_event = to_streamed_response_wrapper(
-            usage.track_event,
-        )
-        self.track_performance = to_streamed_response_wrapper(
-            usage.track_performance,
-        )
 
 
 class AsyncUsageResourceWithStreamingResponse:
@@ -1115,12 +532,3 @@ class AsyncUsageResourceWithStreamingResponse:
         self.list_events = async_to_streamed_response_wrapper(
             usage.list_events,
         )
-        self.track_chat = async_to_streamed_response_wrapper(
-            usage.track_chat,
-        )
-        self.track_event = async_to_streamed_response_wrapper(
-            usage.track_event,
-        )
-        self.track_performance = async_to_streamed_response_wrapper(
-            usage.track_performance,
-        )