pycarlo 0.10.168__py3-none-any.whl → 0.10.170__py3-none-any.whl

pycarlo/lib/schema.py

@@ -302,6 +302,18 @@ class AggregationFunction(sgqlc.types.Enum):
     __choices__ = ("AVG", "MAX", "MEDIAN", "MIN")
 
 
+class AiAgentType(sgqlc.types.Enum):
+    """Types of AI agents available in the system.
+
+    Enumeration Choices:
+
+    * `TROUBLESHOOTING`None
+    """
+
+    __schema__ = schema
+    __choices__ = ("TROUBLESHOOTING",)
+
+
 class AirflowRunState(sgqlc.types.Enum):
     """Enumeration Choices:
 
@@ -12730,6 +12742,21 @@ class AggregatedQueryResults(sgqlc.types.Type):
     """Data necessary to paginate aggregated queries"""
 
 
+class AiAgent(sgqlc.types.Type):
+    """AI Agent configuration for an account."""
+
+    __schema__ = schema
+    __field_names__ = ("created_time", "updated_time", "agent_type", "context_prompt")
+    created_time = sgqlc.types.Field(sgqlc.types.non_null(DateTime), graphql_name="createdTime")
+
+    updated_time = sgqlc.types.Field(sgqlc.types.non_null(DateTime), graphql_name="updatedTime")
+
+    agent_type = sgqlc.types.Field(sgqlc.types.non_null(AiAgentType), graphql_name="agentType")
+
+    context_prompt = sgqlc.types.Field(String, graphql_name="contextPrompt")
+    """Custom context prompt for the AI agent"""
+
+
 class AirflowCapabilitiesResponse(sgqlc.types.Type):
     """Airflow capabilities information"""
 
@@ -15244,6 +15271,39 @@ class CollectionStorage(sgqlc.types.Type):
     """Storage resources"""
 
 
+class CollibraDomainSearchResponse(sgqlc.types.Type):
+    """Collibra domain search response for table search name"""
+
+    __schema__ = schema
+    __field_names__ = (
+        "collibra_table_asset_id",
+        "collibra_table_asset_domain_id",
+        "collibra_schema_asset_id",
+        "collibra_database_asset_id",
+        "collibra_database_asset_domain_id",
+    )
+    collibra_table_asset_id = sgqlc.types.Field(
+        sgqlc.types.non_null(String), graphql_name="collibraTableAssetId"
+    )
+    """Collibra table asset ID"""
+
+    collibra_table_asset_domain_id = sgqlc.types.Field(
+        sgqlc.types.non_null(String), graphql_name="collibraTableAssetDomainId"
+    )
+    """Collibra table asset domain ID"""
+
+    collibra_schema_asset_id = sgqlc.types.Field(String, graphql_name="collibraSchemaAssetId")
+    """Collibra schema asset ID"""
+
+    collibra_database_asset_id = sgqlc.types.Field(String, graphql_name="collibraDatabaseAssetId")
+    """Collibra database asset ID"""
+
+    collibra_database_asset_domain_id = sgqlc.types.Field(
+        String, graphql_name="collibraDatabaseAssetDomainId"
+    )
+    """Collibra database asset domain ID"""
+
+
 class CollibraIntegrationConnection(sgqlc.types.relay.Connection):
     __schema__ = schema
     __field_names__ = ("page_info", "edges")
@@ -15268,6 +15328,18 @@ class CollibraIntegrationEdge(sgqlc.types.Type):
     """A cursor for use in pagination"""
 
 
+class CollibraMonitorNoteResponse(sgqlc.types.Type):
+    """Collibra monitor note response."""
+
+    __schema__ = schema
+    __field_names__ = ("monitor_uuid", "collibra_note")
+    monitor_uuid = sgqlc.types.Field(sgqlc.types.non_null(String), graphql_name="monitorUuid")
+    """Monitor uuid"""
+
+    collibra_note = sgqlc.types.Field(String, graphql_name="collibraNote")
+    """Collibra note"""
+
+
 class CollibraPingResponse(sgqlc.types.Type):
     """Collibra ping information"""
 
@@ -15382,6 +15454,18 @@ class ComparisonMonitorResponseEdge(sgqlc.types.Type):
     """A cursor for use in pagination"""
 
 
+class ConfigureAiAgentPrompt(sgqlc.types.Type):
+    """Configure or update the context prompt for an AI agent type"""
+
+    __schema__ = schema
+    __field_names__ = ("ai_agent", "success")
+    ai_agent = sgqlc.types.Field(AiAgent, graphql_name="aiAgent")
+    """The configured AI agent"""
+
+    success = sgqlc.types.Field(sgqlc.types.non_null(Boolean), graphql_name="success")
+    """Whether the operation succeeded"""
+
+
 class ConfigureMetadataEvents(sgqlc.types.Type):
     """Configure collection of metadata via S3 events"""
 
@@ -24155,6 +24239,55 @@ class MigrateConnectionToSnowflakeAgent(sgqlc.types.Type):
     """True if the connection was migrated successfully."""
 
 
+class MonitorAtRisk(sgqlc.types.Type):
+    """Monitor that could be affected if this table monitor is deleted"""
+
+    __schema__ = schema
+    __field_names__ = ("uuid", "description", "creator", "created_time", "monitor_type")
+    uuid = sgqlc.types.Field(sgqlc.types.non_null(UUID), graphql_name="uuid")
+    """Monitor UUID"""
+
+    description = sgqlc.types.Field(sgqlc.types.non_null(String), graphql_name="description")
+    """Monitor description"""
+
+    creator = sgqlc.types.Field(String, graphql_name="creator")
+    """ID of user who created the monitor"""
+
+    created_time = sgqlc.types.Field(sgqlc.types.non_null(DateTime), graphql_name="createdTime")
+    """When the monitor was created"""
+
+    monitor_type = sgqlc.types.Field(
+        sgqlc.types.non_null(UserDefinedMonitors), graphql_name="monitorType"
+    )
+    """Type of monitor"""
+
+
+class MonitorAtRiskConnection(sgqlc.types.relay.Connection):
+    """Relay connection for monitors at risk"""
+
+    __schema__ = schema
+    __field_names__ = ("page_info", "edges")
+    page_info = sgqlc.types.Field(sgqlc.types.non_null("PageInfo"), graphql_name="pageInfo")
+    """Pagination data for this connection."""
+
+    edges = sgqlc.types.Field(
+        sgqlc.types.non_null(sgqlc.types.list_of("MonitorAtRiskEdge")), graphql_name="edges"
+    )
+    """Contains the nodes in this connection."""
+
+
+class MonitorAtRiskEdge(sgqlc.types.Type):
+    """A Relay edge containing a `MonitorAtRisk` and its cursor."""
+
+    __schema__ = schema
+    __field_names__ = ("node", "cursor")
+    node = sgqlc.types.Field(MonitorAtRisk, graphql_name="node")
+    """The item at the end of the edge"""
+
+    cursor = sgqlc.types.Field(sgqlc.types.non_null(String), graphql_name="cursor")
+    """A cursor for use in pagination"""
+
+
 class MonitorBootstrap(sgqlc.types.Type):
     __schema__ = schema
     __field_names__ = (
@@ -25531,6 +25664,7 @@ class Mutation(sgqlc.types.Type):
         "set_etl_job_generates_incidents",
         "set_etl_job_generates_alerts",
         "bulk_set_etl_job_generates_alerts",
+        "configure_ai_agent_prompt",
         "create_account_secret",
         "update_account_secret",
         "delete_account_secret",
@@ -42099,6 +42233,33 @@ class Mutation(sgqlc.types.Type):
     * `mcons` (`[String]!`): MCONs of jobs to set generate alerts for
     """
 
+    configure_ai_agent_prompt = sgqlc.types.Field(
+        ConfigureAiAgentPrompt,
+        graphql_name="configureAiAgentPrompt",
+        args=sgqlc.types.ArgDict(
+            (
+                (
+                    "agent_type",
+                    sgqlc.types.Arg(
+                        sgqlc.types.non_null(AiAgentType), graphql_name="agentType", default=None
+                    ),
+                ),
+                (
+                    "context_prompt",
+                    sgqlc.types.Arg(String, graphql_name="contextPrompt", default=None),
+                ),
+            )
+        ),
+    )
+    """(experimental) Configure the context prompt for an AI agent
+
+    Arguments:
+
+    * `agent_type` (`AiAgentType!`): Type of AI agent to configure
+    * `context_prompt` (`String`): Custom context prompt for the
+      agent. Pass null to clear.
+    """
+
     create_account_secret = sgqlc.types.Field(
         CreateAccountSecret,
         graphql_name="createAccountSecret",
@@ -43940,6 +44101,7 @@ class Query(sgqlc.types.Type):
         "get_job_execution_history_logs",
         "get_job_executions",
         "get_table_monitor",
+        "get_table_monitor_coverage",
         "get_monitors",
         "get_monitors_count",
         "get_monitor_namespaces",
@@ -44175,12 +44337,16 @@ class Query(sgqlc.types.Type):
         "get_alerts_count_by_date",
         "get_alert_access_request",
         "get_collibra_ping",
+        "get_collibra_monitor_note",
+        "get_collibra_monitor_table_search_names",
+        "get_collibra_domains_for_table_search_name",
         "get_alation_table_flags",
         "get_airflow_task_results",
         "get_airflow_task_stats",
         "get_airflow_tasks_for_source_and_destination_tables",
         "get_airflow_dag_runs",
         "get_airflow_capabilities",
+        "get_ai_agent_config",
         "fix_sql_query",
         "create_sql_query",
         "get_agent_operation_logs",
@@ -50395,6 +50561,36 @@ class Query(sgqlc.types.Type):
     * `monitor_uuid` (`UUID!`)None
     """
 
+    get_table_monitor_coverage = sgqlc.types.Field(
+        "TableMonitorCoverage",
+        graphql_name="getTableMonitorCoverage",
+        args=sgqlc.types.ArgDict(
+            (
+                (
+                    "monitor_uuid",
+                    sgqlc.types.Arg(
+                        sgqlc.types.non_null(UUID), graphql_name="monitorUuid", default=None
+                    ),
+                ),
+                ("search", sgqlc.types.Arg(String, graphql_name="search", default=None)),
+                (
+                    "monitors_search",
+                    sgqlc.types.Arg(String, graphql_name="monitorsSearch", default=None),
+                ),
+            )
+        ),
+    )
+    """(experimental) Get table monitor coverage information
+
+    Arguments:
+
+    * `monitor_uuid` (`UUID!`): UUID of the table monitor
+    * `search` (`String`): Search for tables by name, database, or
+      schema (case-insensitive substring match)
+    * `monitors_search` (`String`): Search for monitors by description
+      (case-insensitive substring match)
+    """
+
     get_monitors = sgqlc.types.Field(
         sgqlc.types.list_of("Monitor"),
         graphql_name="getMonitors",
@@ -60179,6 +60375,73 @@ class Query(sgqlc.types.Type):
       existing integration
     """
 
+    get_collibra_monitor_note = sgqlc.types.Field(
+        CollibraMonitorNoteResponse,
+        graphql_name="getCollibraMonitorNote",
+        args=sgqlc.types.ArgDict(
+            (
+                (
+                    "monitor_uuid",
+                    sgqlc.types.Arg(
+                        sgqlc.types.non_null(String), graphql_name="monitorUuid", default=None
+                    ),
+                ),
+            )
+        ),
+    )
+    """(experimental) Collibra monitor note. Has extra debug information
+    about table relationships if they could not be found
+
+    Arguments:
+
+    * `monitor_uuid` (`String!`): Monitor uuid
+    """
+
+    get_collibra_monitor_table_search_names = sgqlc.types.Field(
+        sgqlc.types.list_of(String),
+        graphql_name="getCollibraMonitorTableSearchNames",
+        args=sgqlc.types.ArgDict(
+            (
+                (
+                    "monitor_uuid",
+                    sgqlc.types.Arg(
+                        sgqlc.types.non_null(String), graphql_name="monitorUuid", default=None
+                    ),
+                ),
+            )
+        ),
+    )
+    """(experimental) Get table search names for a monitor in Collibra
+    format
+
+    Arguments:
+
+    * `monitor_uuid` (`String!`): Monitor uuid
+    """
+
+    get_collibra_domains_for_table_search_name = sgqlc.types.Field(
+        sgqlc.types.list_of(CollibraDomainSearchResponse),
+        graphql_name="getCollibraDomainsForTableSearchName",
+        args=sgqlc.types.ArgDict(
+            (
+                (
+                    "table_search_name",
+                    sgqlc.types.Arg(
+                        sgqlc.types.non_null(String), graphql_name="tableSearchName", default=None
+                    ),
+                ),
+            )
+        ),
+    )
+    """(experimental) Get Collibra domain information for a table search
+    name
+
+    Arguments:
+
+    * `table_search_name` (`String!`): Table search name in Collibra
+      format (e.g., '>project>dataset>table')
+    """
+
     get_alation_table_flags = sgqlc.types.Field(
         sgqlc.types.list_of(GetAlationTableFlags),
         graphql_name="getAlationTableFlags",
@@ -60429,6 +60692,28 @@ class Query(sgqlc.types.Type):
     * `etl_container_uuid` (`UUID`): UUID of ETL Container
     """
 
+    get_ai_agent_config = sgqlc.types.Field(
+        AiAgent,
+        graphql_name="getAiAgentConfig",
+        args=sgqlc.types.ArgDict(
+            (
+                (
+                    "agent_type",
+                    sgqlc.types.Arg(
+                        sgqlc.types.non_null(AiAgentType), graphql_name="agentType", default=None
+                    ),
+                ),
+            )
+        ),
+    )
+    """(experimental) Get AI agent configuration for the account
+
+    Arguments:
+
+    * `agent_type` (`AiAgentType!`): Type of AI agent to retrieve
+      configuration for
+    """
+
     fix_sql_query = sgqlc.types.Field(
         AIMessageOutput,
         graphql_name="fixSqlQuery",
@@ -64402,6 +64687,80 @@ class TableMonitorConnection(sgqlc.types.relay.Connection):
     """Contains the nodes in this connection."""
 
 
+class TableMonitorCoverage(sgqlc.types.Type):
+    """Coverage information for a table monitor"""
+
+    __schema__ = schema
+    __field_names__ = (
+        "tables_at_risk",
+        "tables_at_risk_count",
+        "non_table_monitors_at_risk",
+        "non_table_monitors_at_risk_count",
+    )
+    tables_at_risk = sgqlc.types.Field(
+        "WarehouseTableConnection",
+        graphql_name="tablesAtRisk",
+        args=sgqlc.types.ArgDict(
+            (
+                ("before", sgqlc.types.Arg(String, graphql_name="before", default=None)),
+                ("after", sgqlc.types.Arg(String, graphql_name="after", default=None)),
+                ("first", sgqlc.types.Arg(Int, graphql_name="first", default=None)),
+                ("last", sgqlc.types.Arg(Int, graphql_name="last", default=None)),
+            )
+        ),
+    )
+    """Paginated list of tables that are only covered by this monitor and
+    would become unmonitored if this monitor is deleted
+
+    Arguments:
+
+    * `before` (`String`)None
+    * `after` (`String`)None
+    * `first` (`Int`)None
+    * `last` (`Int`)None
+    """
+
+    tables_at_risk_count = sgqlc.types.Field(
+        sgqlc.types.non_null(Int), graphql_name="tablesAtRiskCount"
+    )
+    """Number of tables that will become unmonitored if this monitor is
+    deleted (tables only covered by this monitor)
+    """
+
+    non_table_monitors_at_risk = sgqlc.types.Field(
+        MonitorAtRiskConnection,
+        graphql_name="nonTableMonitorsAtRisk",
+        args=sgqlc.types.ArgDict(
+            (
+                ("before", sgqlc.types.Arg(String, graphql_name="before", default=None)),
+                ("after", sgqlc.types.Arg(String, graphql_name="after", default=None)),
+                ("first", sgqlc.types.Arg(Int, graphql_name="first", default=None)),
+                ("last", sgqlc.types.Arg(Int, graphql_name="last", default=None)),
+            )
+        ),
+    )
+    """Paginated list of monitors that cover tables which are only
+    covered by this monitor. These monitors are at risk of covering
+    unmonitored tables if this monitor is deleted. Sorted by creation
+    date descending.
+
+    Arguments:
+
+    * `before` (`String`)None
+    * `after` (`String`)None
+    * `first` (`Int`)None
+    * `last` (`Int`)None
+    """
+
+    non_table_monitors_at_risk_count = sgqlc.types.Field(
+        sgqlc.types.non_null(Int), graphql_name="nonTableMonitorsAtRiskCount"
+    )
+    """Number of non-table monitors that cover tables which are only
+    covered by this monitor (monitors at risk of covering unmonitored
+    tables if this monitor is deleted)
+    """
+
+
 class TableMonitorEdge(sgqlc.types.Type):
     """A Relay edge containing a `TableMonitor` and its cursor."""
 
@@ -77060,6 +77419,7 @@ class WarehouseTable(sgqlc.types.Type, Node):
         "total_downstream_nodes",
         "total_upstream_nodes",
         "monitored_metrics",
+        "table_monitor_count",
     )
     table_id = sgqlc.types.Field(String, graphql_name="tableId")
 
@@ -77560,6 +77920,11 @@ class WarehouseTable(sgqlc.types.Type, Node):
         graphql_name="monitoredMetrics",
     )
 
+    table_monitor_count = sgqlc.types.Field(Int, graphql_name="tableMonitorCount")
+    """Number of table monitors covering this table (eventually
+    consistent)
+    """
+
 
 class WarehouseTableHealth(sgqlc.types.Type, Node):
     __schema__ = schema

{pycarlo-0.10.168.dist-info → pycarlo-0.10.170.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pycarlo
-Version: 0.10.168
+Version: 0.10.170
 Summary: Monte Carlo's Python SDK
 Home-page: https://www.montecarlodata.com/
 Author: Monte Carlo Data, Inc

{pycarlo-0.10.168.dist-info → pycarlo-0.10.170.dist-info}/RECORD

@@ -35,10 +35,10 @@ pycarlo/features/user/queries.py,sha256=m97RvM0oiBlrU5xmOwe_JJ5N0G0NG5hIOeyQqN2O
 pycarlo/features/user/service.py,sha256=DHkhuonySaHro07NTd0YNe3cNkDk62CiRTY77dhVaMs,2890
 pycarlo/lib/README.md,sha256=QGNeUefPzLKGyZqn5aITpcFgkC9WQTNS292BGisRFHk,139
 pycarlo/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pycarlo/lib/schema.json,sha256=2fgHIEleLsCd7TqIXqhiEaDmryOC3gR-oAIrBzaesAo,6243800
-pycarlo/lib/schema.py,sha256=xgaRFYMF4XURQs8ODoOxNiy27x1Hf0_h3GyVyxp9T-Y,2712937
-pycarlo-0.10.168.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-pycarlo-0.10.168.dist-info/METADATA,sha256=_Tu0lG4o0TMPgotpuyN1FtZa557SexksarjuW5hbBqE,8742
-pycarlo-0.10.168.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-pycarlo-0.10.168.dist-info/top_level.txt,sha256=TIE04H4pgzGaFxAB-gvkmVAUOAoHxxFfhnEcpuQ5bF4,8
-pycarlo-0.10.168.dist-info/RECORD,,
+pycarlo/lib/schema.json,sha256=VR8Il5do9rfdah8vUmHBdyY6QwyZTMxT1V9qUbhbqnU,6268885
+pycarlo/lib/schema.py,sha256=mBmvkv_hRhiaM2O6LNEi9sKjv94ugYleqYyOx2hBxzM,2725066
+pycarlo-0.10.170.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+pycarlo-0.10.170.dist-info/METADATA,sha256=0B83ShVrNKP4svMHGM5X1BZkpjoR7-QGLo3VN99boUs,8742
+pycarlo-0.10.170.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+pycarlo-0.10.170.dist-info/top_level.txt,sha256=TIE04H4pgzGaFxAB-gvkmVAUOAoHxxFfhnEcpuQ5bF4,8
+pycarlo-0.10.170.dist-info/RECORD,,