agentsinsightsv2response.py

"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""

from __future__ import annotations
from .agentstimesavedinsight import (
    AgentsTimeSavedInsight,
    AgentsTimeSavedInsightTypedDict,
)
from .agentsusagebydepartmentinsight import (
    AgentsUsageByDepartmentInsight,
    AgentsUsageByDepartmentInsightTypedDict,
)
from .agentusersinsight import AgentUsersInsight, AgentUsersInsightTypedDict
from .labeledcountinfo import LabeledCountInfo, LabeledCountInfoTypedDict
from .peragentinsight import PerAgentInsight, PerAgentInsightTypedDict
from glean.api_client.types import BaseModel, UNSET_SENTINEL
import pydantic
from pydantic import model_serializer
from typing import List, Optional
from typing_extensions import Annotated, NotRequired, TypedDict


class AgentsInsightsV2ResponseTypedDict(TypedDict):
    monthly_active_users: NotRequired[int]
    r"""Number of current Monthly Active Users, in the specified departments."""
    weekly_active_users: NotRequired[int]
    r"""Number of current Weekly Active Users, in the specified departments."""
    monthly_active_user_timeseries: NotRequired[LabeledCountInfoTypedDict]
    weekly_active_user_timeseries: NotRequired[LabeledCountInfoTypedDict]
    daily_active_user_timeseries: NotRequired[LabeledCountInfoTypedDict]
    shared_agents_count: NotRequired[int]
    r"""Total number of shared agents."""
    top_agents_insights: NotRequired[List[PerAgentInsightTypedDict]]
    agents_usage_by_department_insights: NotRequired[
        List[AgentsUsageByDepartmentInsightTypedDict]
    ]
    agent_users_insights: NotRequired[List[AgentUsersInsightTypedDict]]
    agents_time_saved_insights: NotRequired[List[AgentsTimeSavedInsightTypedDict]]
    r"""Insights for agents time saved over the specified time period."""
    daily_agent_runs_timeseries: NotRequired[LabeledCountInfoTypedDict]
    successful_runs_timeseries: NotRequired[LabeledCountInfoTypedDict]
    failed_runs_timeseries: NotRequired[LabeledCountInfoTypedDict]
    paused_runs_timeseries: NotRequired[LabeledCountInfoTypedDict]
    upvotes_timeseries: NotRequired[LabeledCountInfoTypedDict]
    downvotes_timeseries: NotRequired[LabeledCountInfoTypedDict]


class AgentsInsightsV2Response(BaseModel):
    monthly_active_users: Annotated[
        Optional[int], pydantic.Field(alias="monthlyActiveUsers")
    ] = None
    r"""Number of current Monthly Active Users, in the specified departments."""

    weekly_active_users: Annotated[
        Optional[int], pydantic.Field(alias="weeklyActiveUsers")
    ] = None
    r"""Number of current Weekly Active Users, in the specified departments."""

    monthly_active_user_timeseries: Annotated[
        Optional[LabeledCountInfo], pydantic.Field(alias="monthlyActiveUserTimeseries")
    ] = None

    weekly_active_user_timeseries: Annotated[
        Optional[LabeledCountInfo], pydantic.Field(alias="weeklyActiveUserTimeseries")
    ] = None

    daily_active_user_timeseries: Annotated[
        Optional[LabeledCountInfo], pydantic.Field(alias="dailyActiveUserTimeseries")
    ] = None

    shared_agents_count: Annotated[
        Optional[int], pydantic.Field(alias="sharedAgentsCount")
    ] = None
    r"""Total number of shared agents."""

    top_agents_insights: Annotated[
        Optional[List[PerAgentInsight]], pydantic.Field(alias="topAgentsInsights")
    ] = None

    agents_usage_by_department_insights: Annotated[
        Optional[List[AgentsUsageByDepartmentInsight]],
        pydantic.Field(alias="agentsUsageByDepartmentInsights"),
    ] = None

    agent_users_insights: Annotated[
        Optional[List[AgentUsersInsight]], pydantic.Field(alias="agentUsersInsights")
    ] = None

    agents_time_saved_insights: Annotated[
        Optional[List[AgentsTimeSavedInsight]],
        pydantic.Field(alias="agentsTimeSavedInsights"),
    ] = None
    r"""Insights for agents time saved over the specified time period."""

    daily_agent_runs_timeseries: Annotated[
        Optional[LabeledCountInfo], pydantic.Field(alias="dailyAgentRunsTimeseries")
    ] = None

    successful_runs_timeseries: Annotated[
        Optional[LabeledCountInfo], pydantic.Field(alias="successfulRunsTimeseries")
    ] = None

    failed_runs_timeseries: Annotated[
        Optional[LabeledCountInfo], pydantic.Field(alias="failedRunsTimeseries")
    ] = None

    paused_runs_timeseries: Annotated[
        Optional[LabeledCountInfo], pydantic.Field(alias="pausedRunsTimeseries")
    ] = None

    upvotes_timeseries: Annotated[
        Optional[LabeledCountInfo], pydantic.Field(alias="upvotesTimeseries")
    ] = None

    downvotes_timeseries: Annotated[
        Optional[LabeledCountInfo], pydantic.Field(alias="downvotesTimeseries")
    ] = None

    @model_serializer(mode="wrap")
    def serialize_model(self, handler):
        optional_fields = set(
            [
                "monthlyActiveUsers",
                "weeklyActiveUsers",
                "monthlyActiveUserTimeseries",
                "weeklyActiveUserTimeseries",
                "dailyActiveUserTimeseries",
                "sharedAgentsCount",
                "topAgentsInsights",
                "agentsUsageByDepartmentInsights",
                "agentUsersInsights",
                "agentsTimeSavedInsights",
                "dailyAgentRunsTimeseries",
                "successfulRunsTimeseries",
                "failedRunsTimeseries",
                "pausedRunsTimeseries",
                "upvotesTimeseries",
                "downvotesTimeseries",
            ]
        )
        serialized = handler(self)
        m = {}

        for n, f in type(self).model_fields.items():
            k = f.alias or n
            val = serialized.get(k, serialized.get(n))

            if val != UNSET_SENTINEL:
                if val is not None or k not in optional_fields:
                    m[k] = val

        return m


try:
    AgentsInsightsV2Response.model_rebuild()
except NameError:
    pass