"""
@generated by mypy-protobuf.  Do not edit manually!
isort:skip_file
"""

import builtins
import collections.abc
import dbtlabs.proto.public.v1.common.vortex_telemetry_contexts_pb2
import dbtlabs.proto.public.v1.events.vortex_pb2
import google.protobuf.descriptor
import google.protobuf.internal.containers
import google.protobuf.message
import sys
import typing

if sys.version_info >= (3, 10):
    import typing as typing_extensions
else:
    import typing_extensions

DESCRIPTOR: google.protobuf.descriptor.FileDescriptor

@typing.final
class ToolCalled(google.protobuf.message.Message):
    DESCRIPTOR: google.protobuf.descriptor.Descriptor

    @typing.final
    class ArgumentsEntry(google.protobuf.message.Message):
        DESCRIPTOR: google.protobuf.descriptor.Descriptor

        KEY_FIELD_NUMBER: builtins.int
        VALUE_FIELD_NUMBER: builtins.int
        key: builtins.str
        value: builtins.str
        def __init__(
            self,
            *,
            key: builtins.str = ...,
            value: builtins.str = ...,
        ) -> None: ...
        def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ...

    ENRICHMENT_FIELD_NUMBER: builtins.int
    EVENT_ID_FIELD_NUMBER: builtins.int
    START_TIME_MS_FIELD_NUMBER: builtins.int
    END_TIME_MS_FIELD_NUMBER: builtins.int
    TOOL_NAME_FIELD_NUMBER: builtins.int
    ARGUMENTS_FIELD_NUMBER: builtins.int
    ERROR_MESSAGE_FIELD_NUMBER: builtins.int
    DBT_CLOUD_ENVIRONMENT_ID_DEV_FIELD_NUMBER: builtins.int
    DBT_CLOUD_ENVIRONMENT_ID_PROD_FIELD_NUMBER: builtins.int
    DBT_CLOUD_USER_ID_FIELD_NUMBER: builtins.int
    LOCAL_USER_ID_FIELD_NUMBER: builtins.int
    HOST_FIELD_NUMBER: builtins.int
    MULTICELL_ACCOUNT_PREFIX_FIELD_NUMBER: builtins.int
    CTX_FIELD_NUMBER: builtins.int
    DBT_MCP_VERSION_FIELD_NUMBER: builtins.int
    AUTHENTICATION_METHOD_FIELD_NUMBER: builtins.int
    TRACE_ID_FIELD_NUMBER: builtins.int
    DISABLED_TOOLSETS_FIELD_NUMBER: builtins.int
    DISABLED_TOOLS_FIELD_NUMBER: builtins.int
    USER_AGENT_FIELD_NUMBER: builtins.int
    ATTRIBUTION_ID_FIELD_NUMBER: builtins.int
    event_id: builtins.str
    """event_id is the unique identifier for this event. It is a generated UUID."""
    start_time_ms: builtins.int
    """The start of the tool call in milliseconds since the Unix epoch"""
    end_time_ms: builtins.int
    """The end of the tool call in milliseconds since the Unix epoch"""
    tool_name: builtins.str
    """The name of the tool, e.g. get_dimentions, dbt build, get_mart_models"""
    error_message: builtins.str
    """An error message if the call fails"""
    dbt_cloud_environment_id_dev: builtins.str
    """The user can optionally set their dev environment id to use tools like the execute_sql"""
    dbt_cloud_environment_id_prod: builtins.str
    """The user can optionally set their prod environment id to use tools like list_metrics"""
    dbt_cloud_user_id: builtins.str
    """The user can optionally set their user id to use tools like execute_sql"""
    local_user_id: builtins.str
    """The user id found at ~/.dbt/.user.yml if present"""
    host: builtins.str
    """The dbt Cloud host that the user configured"""
    multicell_account_prefix: builtins.str
    """The multicell account prefix that the user configured: https://docs.getdbt.com/docs/cloud/about-cloud/access-regions-ip-addresses"""
    dbt_mcp_version: builtins.str
    """The version of dbt_mcp being used (e.g. 0.8.2)"""
    authentication_method: builtins.str
    """How the instance of dbt-mcp was authenticated. Only applicable to local MCP (e.g. env-vars, oauth, null)"""
    trace_id: builtins.str
    """The trace that this tool call is associated with, i.e., the tools it is called with for a single user message. Only applicable to tool calls within an agent run in the dbt Platform."""
    user_agent: builtins.str
    """Indicates where this request is coming from with the user_agent header. This is only relevant for remote MCP. For local MCP, the user_agent can be attained from the enrichment field."""
    attribution_id: builtins.str
    """Attribution identifier for tracking. Uses the format "type__value" (e.g. "thread_id__<x-dbt-copilot-thread-id>"). Only applicable to tool calls within an agent run in the dbt Platform."""
    @property
    def enrichment(self) -> dbtlabs.proto.public.v1.events.vortex_pb2.VortexMessageEnrichment: ...
    @property
    def arguments(self) -> google.protobuf.internal.containers.ScalarMap[builtins.str, builtins.str]:
        """The arguments supplied to the tool, e.g. a list of metrics or a dbt CLI selector"""

    @property
    def ctx(self) -> dbtlabs.proto.public.v1.common.vortex_telemetry_contexts_pb2.VortexTelemetryDbtCloudContext:
        """Common dbt Platform context"""

    @property
    def disabled_toolsets(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
        """The list of disabled toolsets from the DISABLE_* environment variables or x-dbt-disable-toolsets header."""

    @property
    def disabled_tools(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
        """This list of disabled tools from the DISABLE_TOOLS environment variable or x-dbt-disable-tools header."""

    def __init__(
        self,
        *,
        enrichment: dbtlabs.proto.public.v1.events.vortex_pb2.VortexMessageEnrichment | None = ...,
        event_id: builtins.str = ...,
        start_time_ms: builtins.int = ...,
        end_time_ms: builtins.int = ...,
        tool_name: builtins.str = ...,
        arguments: collections.abc.Mapping[builtins.str, builtins.str] | None = ...,
        error_message: builtins.str = ...,
        dbt_cloud_environment_id_dev: builtins.str = ...,
        dbt_cloud_environment_id_prod: builtins.str = ...,
        dbt_cloud_user_id: builtins.str = ...,
        local_user_id: builtins.str = ...,
        host: builtins.str = ...,
        multicell_account_prefix: builtins.str = ...,
        ctx: dbtlabs.proto.public.v1.common.vortex_telemetry_contexts_pb2.VortexTelemetryDbtCloudContext | None = ...,
        dbt_mcp_version: builtins.str = ...,
        authentication_method: builtins.str = ...,
        trace_id: builtins.str = ...,
        disabled_toolsets: collections.abc.Iterable[builtins.str] | None = ...,
        disabled_tools: collections.abc.Iterable[builtins.str] | None = ...,
        user_agent: builtins.str = ...,
        attribution_id: builtins.str = ...,
    ) -> None: ...
    def HasField(self, field_name: typing.Literal["ctx", b"ctx", "enrichment", b"enrichment"]) -> builtins.bool: ...
    def ClearField(self, field_name: typing.Literal["arguments", b"arguments", "attribution_id", b"attribution_id", "authentication_method", b"authentication_method", "ctx", b"ctx", "dbt_cloud_environment_id_dev", b"dbt_cloud_environment_id_dev", "dbt_cloud_environment_id_prod", b"dbt_cloud_environment_id_prod", "dbt_cloud_user_id", b"dbt_cloud_user_id", "dbt_mcp_version", b"dbt_mcp_version", "disabled_tools", b"disabled_tools", "disabled_toolsets", b"disabled_toolsets", "end_time_ms", b"end_time_ms", "enrichment", b"enrichment", "error_message", b"error_message", "event_id", b"event_id", "host", b"host", "local_user_id", b"local_user_id", "multicell_account_prefix", b"multicell_account_prefix", "start_time_ms", b"start_time_ms", "tool_name", b"tool_name", "trace_id", b"trace_id", "user_agent", b"user_agent"]) -> None: ...

Global___ToolCalled: typing_extensions.TypeAlias = ToolCalled