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

pycarlo/lib/schema.json

@@ -29761,6 +29761,95 @@
                 "ofType": null
               }
             },
+            {
+              "args": [
+                {
+                  "defaultValue": null,
+                  "description": "Monitor uuid",
+                  "name": "monitorUuid",
+                  "type": {
+                    "kind": "NON_NULL",
+                    "name": null,
+                    "ofType": {
+                      "kind": "SCALAR",
+                      "name": "String",
+                      "ofType": null
+                    }
+                  }
+                }
+              ],
+              "deprecationReason": null,
+              "description": "(experimental) Collibra monitor note. Has extra debug information about table relationships if they could not be found",
+              "isDeprecated": false,
+              "name": "getCollibraMonitorNote",
+              "type": {
+                "kind": "OBJECT",
+                "name": "CollibraMonitorNoteResponse",
+                "ofType": null
+              }
+            },
+            {
+              "args": [
+                {
+                  "defaultValue": null,
+                  "description": "Monitor uuid",
+                  "name": "monitorUuid",
+                  "type": {
+                    "kind": "NON_NULL",
+                    "name": null,
+                    "ofType": {
+                      "kind": "SCALAR",
+                      "name": "String",
+                      "ofType": null
+                    }
+                  }
+                }
+              ],
+              "deprecationReason": null,
+              "description": "(experimental) Get table search names for a monitor in Collibra format",
+              "isDeprecated": false,
+              "name": "getCollibraMonitorTableSearchNames",
+              "type": {
+                "kind": "LIST",
+                "name": null,
+                "ofType": {
+                  "kind": "SCALAR",
+                  "name": "String",
+                  "ofType": null
+                }
+              }
+            },
+            {
+              "args": [
+                {
+                  "defaultValue": null,
+                  "description": "Table search name in Collibra format (e.g., '>project>dataset>table')",
+                  "name": "tableSearchName",
+                  "type": {
+                    "kind": "NON_NULL",
+                    "name": null,
+                    "ofType": {
+                      "kind": "SCALAR",
+                      "name": "String",
+                      "ofType": null
+                    }
+                  }
+                }
+              ],
+              "deprecationReason": null,
+              "description": "(experimental) Get Collibra domain information for a table search name",
+              "isDeprecated": false,
+              "name": "getCollibraDomainsForTableSearchName",
+              "type": {
+                "kind": "LIST",
+                "name": null,
+                "ofType": {
+                  "kind": "OBJECT",
+                  "name": "CollibraDomainSearchResponse",
+                  "ofType": null
+                }
+              }
+            },
             {
               "args": [
                 {
@@ -30367,6 +30456,33 @@
                 }
               }
             },
+            {
+              "args": [
+                {
+                  "defaultValue": null,
+                  "description": "Type of AI agent to retrieve configuration for",
+                  "name": "agentType",
+                  "type": {
+                    "kind": "NON_NULL",
+                    "name": null,
+                    "ofType": {
+                      "kind": "ENUM",
+                      "name": "AiAgentType",
+                      "ofType": null
+                    }
+                  }
+                }
+              ],
+              "deprecationReason": null,
+              "description": "(experimental) Get AI agent configuration for the account",
+              "isDeprecated": false,
+              "name": "getAiAgentConfig",
+              "type": {
+                "kind": "OBJECT",
+                "name": "AiAgent",
+                "ofType": null
+              }
+            },
             {
               "args": [
                 {
@@ -146986,6 +147102,124 @@
           "name": "CollibraPingResponse",
           "possibleTypes": null
         },
+        {
+          "description": "Collibra monitor note response. ",
+          "enumValues": null,
+          "fields": [
+            {
+              "args": [],
+              "deprecationReason": null,
+              "description": "Monitor uuid",
+              "isDeprecated": false,
+              "name": "monitorUuid",
+              "type": {
+                "kind": "NON_NULL",
+                "name": null,
+                "ofType": {
+                  "kind": "SCALAR",
+                  "name": "String",
+                  "ofType": null
+                }
+              }
+            },
+            {
+              "args": [],
+              "deprecationReason": null,
+              "description": "Collibra note",
+              "isDeprecated": false,
+              "name": "collibraNote",
+              "type": {
+                "kind": "SCALAR",
+                "name": "String",
+                "ofType": null
+              }
+            }
+          ],
+          "inputFields": null,
+          "interfaces": [],
+          "kind": "OBJECT",
+          "name": "CollibraMonitorNoteResponse",
+          "possibleTypes": null
+        },
+        {
+          "description": "Collibra domain search response for table search name",
+          "enumValues": null,
+          "fields": [
+            {
+              "args": [],
+              "deprecationReason": null,
+              "description": "Collibra table asset ID",
+              "isDeprecated": false,
+              "name": "collibraTableAssetId",
+              "type": {
+                "kind": "NON_NULL",
+                "name": null,
+                "ofType": {
+                  "kind": "SCALAR",
+                  "name": "String",
+                  "ofType": null
+                }
+              }
+            },
+            {
+              "args": [],
+              "deprecationReason": null,
+              "description": "Collibra table asset domain ID",
+              "isDeprecated": false,
+              "name": "collibraTableAssetDomainId",
+              "type": {
+                "kind": "NON_NULL",
+                "name": null,
+                "ofType": {
+                  "kind": "SCALAR",
+                  "name": "String",
+                  "ofType": null
+                }
+              }
+            },
+            {
+              "args": [],
+              "deprecationReason": null,
+              "description": "Collibra schema asset ID",
+              "isDeprecated": false,
+              "name": "collibraSchemaAssetId",
+              "type": {
+                "kind": "SCALAR",
+                "name": "String",
+                "ofType": null
+              }
+            },
+            {
+              "args": [],
+              "deprecationReason": null,
+              "description": "Collibra database asset ID",
+              "isDeprecated": false,
+              "name": "collibraDatabaseAssetId",
+              "type": {
+                "kind": "SCALAR",
+                "name": "String",
+                "ofType": null
+              }
+            },
+            {
+              "args": [],
+              "deprecationReason": null,
+              "description": "Collibra database asset domain ID",
+              "isDeprecated": false,
+              "name": "collibraDatabaseAssetDomainId",
+              "type": {
+                "kind": "SCALAR",
+                "name": "String",
+                "ofType": null
+              }
+            }
+          ],
+          "inputFields": null,
+          "interfaces": [],
+          "kind": "OBJECT",
+          "name": "CollibraDomainSearchResponse",
+          "possibleTypes": null
+        },
         {
           "description": null,
           "enumValues": null,
@@ -147520,6 +147754,94 @@
           "name": "AirflowCapabilitiesResponse",
           "possibleTypes": null
         },
+        {
+          "description": "AI Agent configuration for an account.",
+          "enumValues": null,
+          "fields": [
+            {
+              "args": [],
+              "deprecationReason": null,
+              "description": "",
+              "isDeprecated": false,
+              "name": "createdTime",
+              "type": {
+                "kind": "NON_NULL",
+                "name": null,
+                "ofType": {
+                  "kind": "SCALAR",
+                  "name": "DateTime",
+                  "ofType": null
+                }
+              }
+            },
+            {
+              "args": [],
+              "deprecationReason": null,
+              "description": "",
+              "isDeprecated": false,
+              "name": "updatedTime",
+              "type": {
+                "kind": "NON_NULL",
+                "name": null,
+                "ofType": {
+                  "kind": "SCALAR",
+                  "name": "DateTime",
+                  "ofType": null
+                }
+              }
+            },
+            {
+              "args": [],
+              "deprecationReason": null,
+              "description": null,
+              "isDeprecated": false,
+              "name": "agentType",
+              "type": {
+                "kind": "NON_NULL",
+                "name": null,
+                "ofType": {
+                  "kind": "ENUM",
+                  "name": "AiAgentType",
+                  "ofType": null
+                }
+              }
+            },
+            {
+              "args": [],
+              "deprecationReason": null,
+              "description": "Custom context prompt for the AI agent",
+              "isDeprecated": false,
+              "name": "contextPrompt",
+              "type": {
+                "kind": "SCALAR",
+                "name": "String",
+                "ofType": null
+              }
+            }
+          ],
+          "inputFields": null,
+          "interfaces": [],
+          "kind": "OBJECT",
+          "name": "AiAgent",
+          "possibleTypes": null
+        },
+        {
+          "description": "Types of AI agents available in the system.",
+          "enumValues": [
+            {
+              "deprecationReason": null,
+              "description": null,
+              "isDeprecated": false,
+              "name": "TROUBLESHOOTING"
+            }
+          ],
+          "fields": null,
+          "inputFields": null,
+          "interfaces": null,
+          "kind": "ENUM",
+          "name": "AiAgentType",
+          "possibleTypes": null
+        },
         {
           "description": null,
           "enumValues": null,
@@ -173968,6 +174290,43 @@
                 "ofType": null
               }
             },
+            {
+              "args": [
+                {
+                  "defaultValue": null,
+                  "description": "Type of AI agent to configure",
+                  "name": "agentType",
+                  "type": {
+                    "kind": "NON_NULL",
+                    "name": null,
+                    "ofType": {
+                      "kind": "ENUM",
+                      "name": "AiAgentType",
+                      "ofType": null
+                    }
+                  }
+                },
+                {
+                  "defaultValue": null,
+                  "description": "Custom context prompt for the agent. Pass null to clear.",
+                  "name": "contextPrompt",
+                  "type": {
+                    "kind": "SCALAR",
+                    "name": "String",
+                    "ofType": null
+                  }
+                }
+              ],
+              "deprecationReason": null,
+              "description": "(experimental) Configure the context prompt for an AI agent",
+              "isDeprecated": false,
+              "name": "configureAiAgentPrompt",
+              "type": {
+                "kind": "OBJECT",
+                "name": "ConfigureAiAgentPrompt",
+                "ofType": null
+              }
+            },
             {
               "args": [
                 {
@@ -194775,6 +195134,45 @@
           "name": "BulkSetEtlJobGeneratesAlerts",
           "possibleTypes": null
         },
+        {
+          "description": "Configure or update the context prompt for an AI agent type",
+          "enumValues": null,
+          "fields": [
+            {
+              "args": [],
+              "deprecationReason": null,
+              "description": "The configured AI agent",
+              "isDeprecated": false,
+              "name": "aiAgent",
+              "type": {
+                "kind": "OBJECT",
+                "name": "AiAgent",
+                "ofType": null
+              }
+            },
+            {
+              "args": [],
+              "deprecationReason": null,
+              "description": "Whether the operation succeeded",
+              "isDeprecated": false,
+              "name": "success",
+              "type": {
+                "kind": "NON_NULL",
+                "name": null,
+                "ofType": {
+                  "kind": "SCALAR",
+                  "name": "Boolean",
+                  "ofType": null
+                }
+              }
+            }
+          ],
+          "inputFields": null,
+          "interfaces": [],
+          "kind": "OBJECT",
+          "name": "ConfigureAiAgentPrompt",
+          "possibleTypes": null
+        },
         {
           "description": "Create an account secret",
           "enumValues": null,

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"""
 
@@ -25580,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",
@@ -42148,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",
@@ -44225,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",
@@ -60259,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",
@@ -60509,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",

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

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

{pycarlo-0.10.169.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=AUFKs5akQLoT7EQ3rUuoP1J-VihNuL8bF8VjZ-ssxDQ,6256212
-pycarlo/lib/schema.py,sha256=sBtWcFjF9Wa4xspsCyhZvv4zVrqOoQcfFgdOyKqeBb8,2718603
-pycarlo-0.10.169.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-pycarlo-0.10.169.dist-info/METADATA,sha256=axQaPxDX1bNsWqlJa3pu87MKoqfRu8GW50lY0OtxJr0,8742
-pycarlo-0.10.169.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-pycarlo-0.10.169.dist-info/top_level.txt,sha256=TIE04H4pgzGaFxAB-gvkmVAUOAoHxxFfhnEcpuQ5bF4,8
-pycarlo-0.10.169.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,,