From a887fc65e858e5c691236c5f0613e9c8a96c38c0 Mon Sep 17 00:00:00 2001
From: Jack Forgash <58153492+forgxyz@users.noreply.github.com>
Date: Mon, 15 Sep 2025 09:38:31 -0600
Subject: [PATCH] define api integrations for v2

---
 macros/create_udfs.sql                 |  2 +-
 macros/streamline/api_integrations.sql | 19 +++++++++++++++++++
 macros/streamline/streamline_udfs.sql  |  2 +-
 3 files changed, 21 insertions(+), 2 deletions(-)

diff --git a/macros/create_udfs.sql b/macros/create_udfs.sql
index 2644b0f..5480601 100644
--- a/macros/create_udfs.sql
+++ b/macros/create_udfs.sql
@@ -7,7 +7,7 @@
             ) }}
             {{ create_udf_get_chainhead() }}
             {{ create_udf_get_chainhead_testnet() }}
-            {{ create_udf_bulk_grpc() }}
+            {{ create_udf_bulk_grpc_v2() }}
             
             {{ run_create_udf_array_disjunctive_union() }}
             {{ run_create_address_array_adj() }}
diff --git a/macros/streamline/api_integrations.sql b/macros/streamline/api_integrations.sql
index 376eb7e..a70b3e6 100644
--- a/macros/streamline/api_integrations.sql
+++ b/macros/streamline/api_integrations.sql
@@ -2,9 +2,20 @@
 {% macro create_aws_flow_api() %}
     {% if target.name == "prod" %}
         {% set sql %}
+        -- Likely deprecated endpoint
         CREATE api integration IF NOT EXISTS aws_flow_api_prod api_provider = aws_api_gateway api_aws_role_arn = 'arn:aws:iam::490041342817:role/flow-api-prod-rolesnowflakeudfsAF733095-FNY67ODG1RFG' api_allowed_prefixes = (
             'https://quxfxtl934.execute-api.us-east-1.amazonaws.com/prod/'
         ) enabled = TRUE;
+
+        {% endset %}
+        {% do run_query(sql) %}
+
+        {% set sql %}
+        -- New v2 deployment for prod
+        CREATE api integration IF NOT EXISTS aws_flow_api_prod_v2 api_provider = aws_api_gateway api_aws_role_arn = 'arn:aws:iam::924682671219:role/flow-api-prod-rolesnowflakeudfsAF733095-RmrgKIWbzoFL' api_allowed_prefixes = (
+            'https://rajpkbgko9.execute-api.us-east-1.amazonaws.com/prod/'
+        ) enabled = TRUE;
+
         {% endset %}
         {% do run_query(sql) %}
 
@@ -26,6 +37,14 @@
 
     {% elif target.name == "dev" %}
         {{ log("Generating api integration for target:" ~ target.name, info=True) }}
+        -- New v2 deployment for dev
+        {% set sql %}
+        CREATE api integration IF NOT EXISTS aws_flow_api_stg_v2 api_provider = aws_api_gateway api_aws_role_arn = 'arn:aws:iam::704693948482:role/flow-api-stg-rolesnowflakeudfsAF733095-ybejBONVMTd4' api_allowed_prefixes = (
+            'https://2hcu4hei27.execute-api.us-east-1.amazonaws.com/stg/'
+        ) enabled = TRUE;    
+        {% endset %}
+        {% do run_query(sql) %}
+
         {% set sql %}
         CREATE api integration IF NOT EXISTS aws_flow_api_dev_2 api_provider = aws_api_gateway api_aws_role_arn = 'arn:aws:iam::490041342817:role/flow-api-dev-rolesnowflakeudfsAF733095-i1JsMNTpSzX0' api_allowed_prefixes = (
             'https://sicl8dvvv9.execute-api.us-east-1.amazonaws.com/dev/'
diff --git a/macros/streamline/streamline_udfs.sql b/macros/streamline/streamline_udfs.sql
index cb9dce7..d5a3041 100644
--- a/macros/streamline/streamline_udfs.sql
+++ b/macros/streamline/streamline_udfs.sql
@@ -49,7 +49,7 @@
     CREATE
     OR REPLACE EXTERNAL FUNCTION streamline.udf_bulk_grpc_v2(json variant) returns variant api_integration = 
     {% if target.name == "prod" %} 
-        aws_flow_api_stg_v2 AS 'https://rajpkbgko9.execute-api.us-east-1.amazonaws.com/prod/udf_bulk_grpc'
+        aws_flow_api_prod_v2 AS 'https://rajpkbgko9.execute-api.us-east-1.amazonaws.com/prod/udf_bulk_grpc'
     {% elif target.name == "dev" %}
         aws_flow_api_stg_v2 AS 'https://2hcu4hei27.execute-api.us-east-1.amazonaws.com/stg/udf_bulk_grpc'
     {% elif  target.name == "sbx" %}