Cortex Agentのリソース予算¶

通知を送信¶

支出がしきい値に達すると、通知を送信できます。詳細については、 予算に関する通知 をご参照ください。

1. 通知を送信するメールアドレスを設定します。

   CopyExpand

   CALL my_budget!SET_EMAIL_NOTIFICATIONS(
     'budgets_notification_integration',
      'costadmin@example.com, budgetadmin@example.com'
   );
   

   Show lessSee more

   Scroll to top

2. 通知のしきい値を設定します。

   CopyExpand

   CALL my_budget!SET_NOTIFICATION_THRESHOLD(80);
   

   Show lessSee more

   Scroll to top

アクセス権を取り消す¶

1. Cortex Agentへのアクセスを取り消すストアドプロシージャを作成します。ストアドプロシージャでは、特定のロールへのアクセスを制限して、そのロールのUSAGEを取り消すことができます。

   CopyExpand

   -- Create a stored procedure that revokes access to the Cortex Agent object
   CREATE OR REPLACE PROCEDURE budgets_db.budgets_schema.sp_revoke_agent_access(
      agent_name STRING, role_name STRING
   )
   RETURNS STRING
   LANGUAGE SQL
   AS
   BEGIN
      EXECUTE IMMEDIATE 'REVOKE ROLE agent_' || agent_name || '_role FROM ROLE ' || role_name;
      RETURN 'Access revoked for ' || agent_name;
   END;
   

   Show lessSee more

   Scroll to top

   重要

   :code:`role_name`とユーザーが、他のロールを介してCortex Agentにアクセスできないことを確認します。ロールと権限を正しく構成する方法について詳しくは、:ref:`label-snowflake-intelligence-user-privileges`を参照してください。

2. 予算の100%が使用されると、アクセスをブロックするカスタムアクションを設定します。通知にカスタムアクションを使用することもできます。

   CopyExpand

   -- Provide access to the stored procedures
   GRANT USAGE ON DATABASE budgets_db TO APPLICATION SNOWFLAKE;
   GRANT USAGE ON SCHEMA budgets_db.budgets_schema TO APPLICATION SNOWFLAKE;
   GRANT USAGE ON PROCEDURE budgets_db.budgets_schema.sp_revoke_agent_access(STRING, STRING)
      TO APPLICATION SNOWFLAKE;
   
   -- Block access at 100% of the budget
   CALL budgets_db.budgets_schema.my_budget!ADD_CUSTOM_ACTION(
      SYSTEM$REFERENCE('PROCEDURE',
         'budgets_db.budgets_schema.sp_revoke_agent_access(string, string)'),
      ARRAY_CONSTRUCT('AGENT_NAME', 'ROLE_NAME'),
      'ACTUAL',
      100);
   

   Show lessSee more

   Scroll to top

注釈

また、カスタムアクションを使用して、支出が予算制限を超えると予測された場合にアクションを起こすこともできます。詳細については、 予算のカスタムアクション をご参照ください。

支出制限の例外処理¶

場合によっては、収益時期やその他のピーク時など、予算制限に達した後に再度アクセスが必要になることがあります。これらの例外シナリオを処理するために、100%を超えて最大500%までのしきい値を設定できます。

ワークフローでは、支出が予算のしきい値に達すると、構成されたストアドプロシージャを使用してアクセスが取り消されることを前提としています。管理者は一部のユーザーを復元し、アクセス権を再度付与します。支出が200%に達すると、失効プロシージャが強制停止として再実行されます。

1. ロールへのアクセスを復元するストアドプロシージャを作成します。

   CopyExpand

   -- Create a stored procedure that reinstates access to the Cortex Agent object
   CREATE OR REPLACE PROCEDURE budgets_db.budgets_schema.sp_reinstate_agent_access(
      agent_name STRING, role_name STRING
   )
   RETURNS STRING
   LANGUAGE SQL
   AS
   BEGIN
      EXECUTE IMMEDIATE 'GRANT ROLE agent_' || agent_name || '_role TO ROLE ' || role_name;
      RETURN 'Access reinstated for ' || agent_name;
   END;
   

   Show lessSee more

   Scroll to top

2. アクセスを復元するストアドプロシージャで、100%を超えるしきい値を構成します。これにより、例外期間の有効な予算を引き上げることができます。支出が予算の200%に達すると、アクセスは再び取り消されます。

   CopyExpand

   -- Add grants for this procedure
   GRANT USAGE ON DATABASE budgets_db TO APPLICATION SNOWFLAKE;
   GRANT USAGE ON SCHEMA budgets_db.budgets_schema TO APPLICATION SNOWFLAKE;
   GRANT USAGE ON PROCEDURE budgets_db.budgets_schema.sp_revoke_agent_access(STRING, STRING)
      TO APPLICATION SNOWFLAKE;
   
   -- Issue a reinstatement for a subset of users
   CALL budgets_db.budgets_schema.sp_reinstate_agent_access('my_agent', 'power_user_role');
   
   -- Set another threshold at 200% as a hard stop
   CALL budgets_db.budgets_schema.my_budget!ADD_CUSTOM_ACTION(
      SYSTEM$REFERENCE(
         'PROCEDURE',
         'budgets_db.budgets_schema.sp_revoke_agent_access(string, string)'
      ),
      ARRAY_CONSTRUCT('my_agent', 'power_user_role'),
      'ACTUAL',
      200
   );
   

   Show lessSee more

   Scroll to top

アクセスの復元¶

次の予算期間の開始時にユーザーがCortex Agentに再度アクセスできるようにするには、予算サイクルの再開時に以下のストアドプロシージャが呼び出されるように設定します。

1. ロールへのアクセスを復元するストアドプロシージャを作成します。

   CopyExpand

   -- Create a stored procedure that reinstates access to the Cortex Agent object
   CREATE OR REPLACE PROCEDURE budgets_db.budgets_schema.sp_reinstate_agent_access(
      agent_name STRING, role_name STRING
   )
   RETURNS STRING
   LANGUAGE SQL
   AS
   BEGIN
      EXECUTE IMMEDIATE 'GRANT ROLE agent_' || agent_name || '_role TO ROLE ' || role_name;
      RETURN 'Access reinstated for ' || agent_name;
   END;
   

   Show lessSee more

   Scroll to top

2. 予算のサイクル開始アクションを設定します。

   CopyExpand

   GRANT USAGE ON DATABASE budgets_db TO APPLICATION SNOWFLAKE;
   GRANT USAGE ON SCHEMA budgets_db.budgets_schema TO APPLICATION SNOWFLAKE;
   GRANT USAGE ON PROCEDURE budgets_db.budgets_schema.sp_reinstate_agent_access(STRING, STRING)
      TO APPLICATION SNOWFLAKE;
   
   CALL budgets_db.budgets_schema.my_budget!SET_CYCLE_START_ACTION(
      SYSTEM$REFERENCE('PROCEDURE', 'budgets_db.budgets_schema.sp_reinstate_agent_access(string, string)'),
      ARRAY_CONSTRUCT('my_agent', 'power_user_role')
   );
   

   Show lessSee more

   Scroll to top

支出予測に基づくアラートの設定¶

アラートを受信したり、実際の支出ではなく予測支出に基づいてアクションを実行したりするには、トリガータイプを``PROJECTED``に設定します。たとえば、予測消費量が予算制限の75%に達したときに``alert_team``という名前のストアドプロシージャを呼び出すには、以下のコマンドを実行します。

CALL budget_db.sch1.my_budget!ADD_CUSTOM_ACTION(
   SYSTEM$REFERENCE('PROCEDURE', 'code_db.sch1.alert_team(string, string, string)'),
   ARRAY_CONSTRUCT('admin@example.com', 'Budget Alert', 'Spending at 75% of budget limit'),
   'PROJECTED',
   75);