Skip to content

Commit 9f22b36

Browse files
waleedlatif1waleedlatif1
committed
feat(athena): add AWS Athena integration
1 parent 98be968 commit 9f22b36

File tree

31 files changed

+2862
-117
lines changed

31 files changed

+2862
-117
lines changed

‎apps/docs/components/icons.tsx‎

Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4687,6 +4687,33 @@ export function CloudFormationIcon(props: SVGProps<SVGSVGElement>) {
46874687
)
46884688
}
46894689

4690+
export function AthenaIcon(props: SVGProps<SVGSVGElement>) {
4691+
return (
4692+
<svg
4693+
{...props}
4694+
viewBox='0 0 80 80'
4695+
version='1.1'
4696+
xmlns='http://www.w3.org/2000/svg'
4697+
xmlnsXlink='http://www.w3.org/1999/xlink'
4698+
>
4699+
<g
4700+
id='Icon-Architecture/64/Arch_Amazon-Athena_64'
4701+
stroke='none'
4702+
strokeWidth='1'
4703+
fill='none'
4704+
fillRule='evenodd'
4705+
transform='translate(40, 40) scale(1.25) translate(-40, -40)'
4706+
>
4707+
<path
4708+
d='M38.29505,27.2267312 C42.787319,27.2267312 45.2478437,28.2331825 45.6964751,28.7379193 C45.2478437,29.2426562 42.787319,30.2491074 38.29505,30.2491074 C33.8027811,30.2491074 31.3422564,29.2426562 30.893625,28.7379193 C31.3422564,28.2331825 33.8027811,27.2267312 38.29505,27.2267312 L38.29505,27.2267312 Z M37.7838882,35.2823712 C37.6191254,35.1977447 37.5029973,35.0294991 37.5029973,34.8300223 C37.5029973,34.5499487 37.7292981,34.3212556 38.0062188,34.3212556 C38.0866151,34.3212556 38.1600636,34.3444272 38.2285494,34.3796882 L37.7838882,35.2823712 Z M43.5674612,43.5908834 C43.4930201,43.6513309 43.322302,43.7681961 42.9709403,43.9092403 C42.6582879,44.0341652 42.2880677,44.1470006 41.8682202,44.2457316 C40.7525971,44.5076708 39.3808968,44.6517374 38.0052262,44.6517374 C34.9968155,44.6517374 32.9005556,44.0019265 32.4489466,43.5989431 L31.1159556,31.150783 C33.1596104,31.9869737 36.1700063,32.2640249 38.29505,32.2640249 C40.3843621,32.2640249 43.3292498,31.9950334 45.3719121,31.1910813 L44.5748967,36.6656121 C43.0731726,36.0994203 41.1992434,35.2773339 39.4235763,34.4129344 C39.2429327,33.786295 38.6801584,33.3248789 38.0062188,33.3248789 C37.1883598,33.3248789 36.5233532,34.0008837 36.5233532,34.8300223 C36.5233532,35.6611757 37.1883598,36.3361731 38.0062188,36.3361731 C38.1997655,36.3361731 38.3843793,36.2958747 38.5531123,36.2273675 C41.0344805,37.4524373 42.8835961,38.2382552 44.2751474,38.7228428 L43.5674612,43.5908834 Z M28.8718062,28.8467249 L30.4787403,43.8498003 C30.5918907,46.6344162 37.6995217,46.6666549 38.0052262,46.6666549 C39.5268012,46.6666549 41.0573091,46.5034466 42.3148665,46.2092686 C42.8299985,46.0883736 43.2964958,45.9453144 43.7004625,45.7831136 C44.8736534,45.3116229 45.4890327,44.6688642 45.5317122,43.8739793 L46.2006891,39.2759376 C46.6562683,39.3696313 47.0284735,39.4109371 47.3252452,39.4109371 C48.2592321,39.4109371 48.5053839,39.0281028 48.6751094,38.7641486 C48.853768,38.48609 48.9053804,38.1445615 48.8220064,37.8010181 C48.6314374,37.0111704 47.5168068,35.971473 46.7723963,35.3539008 L47.7133311,28.8850083 L47.7043982,28.8840008 C47.7083684,28.8346354 47.7242492,28.7882923 47.7242492,28.7379193 C47.7242492,25.9543109 41.7967568,25.2118138 38.29505,25.2118138 C34.7933433,25.2118138 28.8658509,25.9543109 28.8658509,28.7379193 C28.8658509,28.7751953 28.8787541,28.8084414 28.8807391,28.8457174 L28.8718062,28.8467249 Z M37.8355007,20.0596698 C46.4865427,20.0596698 53.5246954,27.2035597 53.5246954,35.98457 C53.5246954,44.7655803 46.4865427,51.9094701 37.8355007,51.9094701 C29.1834661,51.9094701 22.1453133,44.7655803 22.1453133,35.98457 C22.1453133,27.2035597 29.1834661,20.0596698 37.8355007,20.0596698 L37.8355007,20.0596698 Z M12.9850945,41.8348828 L12.9850945,43.8498003 L21.91802,43.8498003 L21.91802,43.7309201 C24.7735785,49.7494786 30.8261318,53.9243876 37.8355007,53.9243876 C47.5803298,53.9243876 55.50979,45.8768072 55.50979,35.98457 C55.50979,26.0923327 47.5803298,18.0447524 37.8355007,18.0447524 C30.253432,18.0447524 23.7909567,22.9248825 21.2857674,29.7453781 L12.9850945,29.7453781 L12.9850945,31.7602955 L20.6763434,31.7602955 C20.3666686,33.0568949 20.1850325,34.4018523 20.1701443,35.7901304 L11,35.7901304 L11,37.8050479 L20.2515331,37.8050479 C20.3914823,39.2044081 20.7061198,40.548358 21.1448257,41.8348828 L12.9850945,41.8348828 Z M67.0799136,66.035049 C65.8789314,67.2560889 63.7965672,67.2631412 62.5965775,66.046131 L51.9326496,55.220987 C53.6487638,53.9223727 55.1802643,52.3900279 56.4934043,50.6763406 L67.0918241,61.4853653 C67.688345,62.0918555 68.0168782,62.8998374 68.014902,63.7591997 C68.0139005,64.6205769 67.6823898,65.4275513 67.0799136,66.035049 L67.0799136,66.035049 Z M68.4972711,60.0628336 L57.6616325,49.0100039 C60.0635969,45.2562127 61.4650736,40.7851108 61.4650736,35.98457 C61.4650736,22.7586518 50.8646687,12 37.8355007,12 C28.4728022,12 19.9825528,17.6196048 16.2039254,26.316996 L18.0202869,27.1290077 C21.4812992,19.1630316 29.2588997,14.0149175 37.8355007,14.0149175 C49.7708816,14.0149175 59.4799791,23.8698788 59.4799791,35.98457 C59.4799791,48.0982537 49.7708816,57.9542225 37.8355007,57.9542225 C29.8623684,57.9542225 22.5572205,53.5244265 18.7686675,46.3936336 L17.0217843,47.3507194 C21.1557437,55.1343455 29.1318536,59.9691399 37.8355007,59.9691399 C42.3912926,59.9691399 46.6483279,58.6503765 50.2602074,56.3735197 L61.1941082,67.4716851 C62.1648195,68.4569797 63.4561235,69 64.8278238,69 C66.2074645,69 67.5067089,68.4529499 68.4813903,67.462618 C69.4580568,66.4773233 69.9980025,65.1635972 70,63.7622221 C70.0029653,62.3628619 69.4679823,61.0491357 68.4972711,60.0628336 L68.4972711,60.0628336 Z'
4709+
id='Amazon-Athena_Icon_64_Squid'
4710+
fill='currentColor'
4711+
/>
4712+
</g>
4713+
</svg>
4714+
)
4715+
}
4716+
46904717
export function CloudWatchIcon(props: SVGProps<SVGSVGElement>) {
46914718
return (
46924719
<svg

‎apps/docs/components/ui/icon-mapping.ts‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ import {
1616
ArxivIcon,
1717
AsanaIcon,
1818
AshbyIcon,
19+
AthenaIcon,
1920
AttioIcon,
2021
AzureIcon,
2122
BoxCompanyIcon,
@@ -205,6 +206,7 @@ export const blockTypeToIconMap: Record<string, IconComponent> = {
205206
arxiv: ArxivIcon,
206207
asana: AsanaIcon,
207208
ashby: AshbyIcon,
209+
athena: AthenaIcon,
208210
attio: AttioIcon,
209211
box: BoxCompanyIcon,
210212
brandfetch: BrandfetchIcon,

‎apps/docs/content/docs/en/tools/athena.mdx‎

Lines changed: 238 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,238 @@
1+
---
2+
title: Athena
3+
description: Run SQL queries on data in Amazon S3 using AWS Athena
4+
---
5+
6+
import { BlockInfoCard } from "@/components/ui/block-info-card"
7+
8+
<BlockInfoCard
9+
type="athena"
10+
color="linear-gradient(45deg, #4D27A8 0%, #A166FF 100%)"
11+
/>
12+
13+
{/* MANUAL-CONTENT-START:intro */}
14+
[Amazon Athena](https://aws.amazon.com/athena/) is an interactive query service from AWS that makes it easy to analyze data directly in Amazon S3 using standard SQL. Athena is serverless, so there is no infrastructure to manage, and you pay only for the queries you run.
15+
16+
With Athena, you can:
17+
18+
- **Query data in S3**: Run SQL queries directly against data stored in Amazon S3 without loading it into a database
19+
- **Support multiple formats**: Query CSV, JSON, Parquet, ORC, Avro, and other common data formats
20+
- **Integrate with AWS Glue**: Use the AWS Glue Data Catalog to manage table metadata and schemas
21+
- **Scale automatically**: Handle queries of any size without provisioning servers or clusters
22+
- **Save and reuse queries**: Create named queries for frequently used SQL statements
23+
24+
In Sim, the Athena integration enables your agents to run SQL queries against data in S3, check query execution status, retrieve results, and manage saved queries — all within your agent workflows. Supported operations include:
25+
26+
- **Start Query**: Execute SQL queries against your S3 data
27+
- **Get Query Execution**: Check the status and details of a running or completed query
28+
- **Get Query Results**: Retrieve the results of a completed query
29+
- **Stop Query**: Cancel a running query execution
30+
- **List Query Executions**: View recent query execution IDs
31+
- **Create Named Query**: Save a query for reuse
32+
- **Get Named Query**: Retrieve details of a saved query
33+
- **List Named Queries**: View all saved query IDs
34+
35+
This integration empowers Sim agents to automate data analysis tasks using AWS Athena, enabling workflows that query, monitor, and manage large-scale data in S3 without manual effort or infrastructure management.
36+
{/* MANUAL-CONTENT-END */}
37+
38+
39+
## Usage Instructions
40+
41+
Integrate AWS Athena into workflows. Execute SQL queries against data in S3, check query status, retrieve results, manage named queries, and list executions. Requires AWS access key and secret access key.
42+
43+
44+
45+
## Tools
46+
47+
### `athena_start_query`
48+
49+
Start an SQL query execution in AWS Athena
50+
51+
#### Input
52+
53+
| Parameter | Type | Required | Description |
54+
| --------- | ---- | -------- | ----------- |
55+
| `awsRegion` | string | Yes | AWS region \(e.g., us-east-1\) |
56+
| `awsAccessKeyId` | string | Yes | AWS access key ID |
57+
| `awsSecretAccessKey` | string | Yes | AWS secret access key |
58+
| `queryString` | string | Yes | SQL query string to execute |
59+
| `database` | string | No | Database name within the catalog |
60+
| `catalog` | string | No | Data catalog name \(default: AwsDataCatalog\) |
61+
| `outputLocation` | string | No | S3 output location for query results \(e.g., s3://bucket/path/\) |
62+
| `workGroup` | string | No | Workgroup to execute the query in \(default: primary\) |
63+
64+
#### Output
65+
66+
| Parameter | Type | Description |
67+
| --------- | ---- | ----------- |
68+
| `queryExecutionId` | string | Unique ID of the started query execution |
69+
70+
### `athena_get_query_execution`
71+
72+
Get the status and details of an Athena query execution
73+
74+
#### Input
75+
76+
| Parameter | Type | Required | Description |
77+
| --------- | ---- | -------- | ----------- |
78+
| `awsRegion` | string | Yes | AWS region \(e.g., us-east-1\) |
79+
| `awsAccessKeyId` | string | Yes | AWS access key ID |
80+
| `awsSecretAccessKey` | string | Yes | AWS secret access key |
81+
| `queryExecutionId` | string | Yes | Query execution ID to check |
82+
83+
#### Output
84+
85+
| Parameter | Type | Description |
86+
| --------- | ---- | ----------- |
87+
| `queryExecutionId` | string | Query execution ID |
88+
| `query` | string | SQL query string |
89+
| `state` | string | Query state \(QUEUED, RUNNING, SUCCEEDED, FAILED, CANCELLED\) |
90+
| `stateChangeReason` | string | Reason for state change \(e.g., error message\) |
91+
| `statementType` | string | Statement type \(DDL, DML, UTILITY\) |
92+
| `database` | string | Database name |
93+
| `catalog` | string | Data catalog name |
94+
| `workGroup` | string | Workgroup name |
95+
| `submissionDateTime` | number | Query submission time \(Unix epoch ms\) |
96+
| `completionDateTime` | number | Query completion time \(Unix epoch ms\) |
97+
| `dataScannedInBytes` | number | Amount of data scanned in bytes |
98+
| `engineExecutionTimeInMillis` | number | Engine execution time in milliseconds |
99+
| `queryPlanningTimeInMillis` | number | Query planning time in milliseconds |
100+
| `queryQueueTimeInMillis` | number | Time the query spent in queue in milliseconds |
101+
| `totalExecutionTimeInMillis` | number | Total execution time in milliseconds |
102+
| `outputLocation` | string | S3 location of query results |
103+
104+
### `athena_get_query_results`
105+
106+
Retrieve the results of a completed Athena query execution
107+
108+
#### Input
109+
110+
| Parameter | Type | Required | Description |
111+
| --------- | ---- | -------- | ----------- |
112+
| `awsRegion` | string | Yes | AWS region \(e.g., us-east-1\) |
113+
| `awsAccessKeyId` | string | Yes | AWS access key ID |
114+
| `awsSecretAccessKey` | string | Yes | AWS secret access key |
115+
| `queryExecutionId` | string | Yes | Query execution ID to get results for |
116+
| `maxResults` | number | No | Maximum number of rows to return \(1-1000\) |
117+
| `nextToken` | string | No | Pagination token from a previous request |
118+
119+
#### Output
120+
121+
| Parameter | Type | Description |
122+
| --------- | ---- | ----------- |
123+
| `columns` | array | Column metadata \(name and type\) |
124+
| `rows` | array | Result rows as key-value objects |
125+
| `nextToken` | string | Pagination token for next page of results |
126+
| `updateCount` | number | Number of rows affected \(for INSERT/UPDATE statements\) |
127+
128+
### `athena_stop_query`
129+
130+
Stop a running Athena query execution
131+
132+
#### Input
133+
134+
| Parameter | Type | Required | Description |
135+
| --------- | ---- | -------- | ----------- |
136+
| `awsRegion` | string | Yes | AWS region \(e.g., us-east-1\) |
137+
| `awsAccessKeyId` | string | Yes | AWS access key ID |
138+
| `awsSecretAccessKey` | string | Yes | AWS secret access key |
139+
| `queryExecutionId` | string | Yes | Query execution ID to stop |
140+
141+
#### Output
142+
143+
| Parameter | Type | Description |
144+
| --------- | ---- | ----------- |
145+
| `success` | boolean | Whether the query was successfully stopped |
146+
147+
### `athena_list_query_executions`
148+
149+
List recent Athena query execution IDs
150+
151+
#### Input
152+
153+
| Parameter | Type | Required | Description |
154+
| --------- | ---- | -------- | ----------- |
155+
| `awsRegion` | string | Yes | AWS region \(e.g., us-east-1\) |
156+
| `awsAccessKeyId` | string | Yes | AWS access key ID |
157+
| `awsSecretAccessKey` | string | Yes | AWS secret access key |
158+
| `workGroup` | string | No | Workgroup to list executions for \(default: primary\) |
159+
| `maxResults` | number | No | Maximum number of results \(0-50\) |
160+
| `nextToken` | string | No | Pagination token from a previous request |
161+
162+
#### Output
163+
164+
| Parameter | Type | Description |
165+
| --------- | ---- | ----------- |
166+
| `queryExecutionIds` | array | List of query execution IDs |
167+
| `nextToken` | string | Pagination token for next page |
168+
169+
### `athena_create_named_query`
170+
171+
Create a saved/named query in AWS Athena
172+
173+
#### Input
174+
175+
| Parameter | Type | Required | Description |
176+
| --------- | ---- | -------- | ----------- |
177+
| `awsRegion` | string | Yes | AWS region \(e.g., us-east-1\) |
178+
| `awsAccessKeyId` | string | Yes | AWS access key ID |
179+
| `awsSecretAccessKey` | string | Yes | AWS secret access key |
180+
| `name` | string | Yes | Name for the saved query |
181+
| `database` | string | Yes | Database the query runs against |
182+
| `queryString` | string | Yes | SQL query string to save |
183+
| `description` | string | No | Description of the named query |
184+
| `workGroup` | string | No | Workgroup to create the named query in |
185+
186+
#### Output
187+
188+
| Parameter | Type | Description |
189+
| --------- | ---- | ----------- |
190+
| `namedQueryId` | string | ID of the created named query |
191+
192+
### `athena_get_named_query`
193+
194+
Get details of a saved/named query in AWS Athena
195+
196+
#### Input
197+
198+
| Parameter | Type | Required | Description |
199+
| --------- | ---- | -------- | ----------- |
200+
| `awsRegion` | string | Yes | AWS region \(e.g., us-east-1\) |
201+
| `awsAccessKeyId` | string | Yes | AWS access key ID |
202+
| `awsSecretAccessKey` | string | Yes | AWS secret access key |
203+
| `namedQueryId` | string | Yes | Named query ID to retrieve |
204+
205+
#### Output
206+
207+
| Parameter | Type | Description |
208+
| --------- | ---- | ----------- |
209+
| `namedQueryId` | string | Named query ID |
210+
| `name` | string | Name of the saved query |
211+
| `description` | string | Query description |
212+
| `database` | string | Database the query runs against |
213+
| `queryString` | string | SQL query string |
214+
| `workGroup` | string | Workgroup name |
215+
216+
### `athena_list_named_queries`
217+
218+
List saved/named query IDs in AWS Athena
219+
220+
#### Input
221+
222+
| Parameter | Type | Required | Description |
223+
| --------- | ---- | -------- | ----------- |
224+
| `awsRegion` | string | Yes | AWS region \(e.g., us-east-1\) |
225+
| `awsAccessKeyId` | string | Yes | AWS access key ID |
226+
| `awsSecretAccessKey` | string | Yes | AWS secret access key |
227+
| `workGroup` | string | No | Workgroup to list named queries for |
228+
| `maxResults` | number | No | Maximum number of results \(0-50\) |
229+
| `nextToken` | string | No | Pagination token from a previous request |
230+
231+
#### Output
232+
233+
| Parameter | Type | Description |
234+
| --------- | ---- | ----------- |
235+
| `namedQueryIds` | array | List of named query IDs |
236+
| `nextToken` | string | Pagination token for next page |
237+
238+

‎apps/docs/content/docs/en/tools/meta.json‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
"arxiv",
1414
"asana",
1515
"ashby",
16+
"athena",
1617
"attio",
1718
"box",
1819
"brandfetch",

‎apps/sim/app/(landing)/integrations/data/icon-mapping.ts‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ import {
1616
ArxivIcon,
1717
AsanaIcon,
1818
AshbyIcon,
19+
AthenaIcon,
1920
AttioIcon,
2021
AzureIcon,
2122
BoxCompanyIcon,
@@ -205,6 +206,7 @@ export const blockTypeToIconMap: Record<string, IconComponent> = {
205206
arxiv: ArxivIcon,
206207
asana: AsanaIcon,
207208
ashby: AshbyIcon,
209+
athena: AthenaIcon,
208210
attio: AttioIcon,
209211
box: BoxCompanyIcon,
210212
brandfetch: BrandfetchIcon,

‎apps/sim/app/(landing)/integrations/data/integrations.json‎

Lines changed: 51 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -971,6 +971,57 @@
971971
"integrationType": "hr",
972972
"tags": ["hiring"]
973973
},
974+
{
975+
"type": "athena",
976+
"slug": "athena",
977+
"name": "Athena",
978+
"description": "Run SQL queries on data in Amazon S3 using AWS Athena",
979+
"longDescription": "Integrate AWS Athena into workflows. Execute SQL queries against data in S3, check query status, retrieve results, manage named queries, and list executions. Requires AWS access key and secret access key.",
980+
"bgColor": "linear-gradient(45deg, #4D27A8 0%, #A166FF 100%)",
981+
"iconName": "AthenaIcon",
982+
"docsUrl": "https://docs.sim.ai/tools/athena",
983+
"operations": [
984+
{
985+
"name": "Start Query",
986+
"description": "Start an SQL query execution in AWS Athena"
987+
},
988+
{
989+
"name": "Get Query Execution",
990+
"description": "Get the status and details of an Athena query execution"
991+
},
992+
{
993+
"name": "Get Query Results",
994+
"description": "Retrieve the results of a completed Athena query execution"
995+
},
996+
{
997+
"name": "Stop Query",
998+
"description": "Stop a running Athena query execution"
999+
},
1000+
{
1001+
"name": "List Query Executions",
1002+
"description": "List recent Athena query execution IDs"
1003+
},
1004+
{
1005+
"name": "Create Named Query",
1006+
"description": "Create a saved/named query in AWS Athena"
1007+
},
1008+
{
1009+
"name": "Get Named Query",
1010+
"description": "Get details of a saved/named query in AWS Athena"
1011+
},
1012+
{
1013+
"name": "List Named Queries",
1014+
"description": "List saved/named query IDs in AWS Athena"
1015+
}
1016+
],
1017+
"operationCount": 8,
1018+
"triggers": [],
1019+
"triggerCount": 0,
1020+
"authType": "none",
1021+
"category": "tools",
1022+
"integrationType": "analytics",
1023+
"tags": ["cloud", "data-analytics"]
1024+
},
9741025
{
9751026
"type": "attio",
9761027
"slug": "attio",

0 commit comments

Comments
 (0)