robosystems-client 0.2.3__py3-none-any.whl → 0.2.4__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of robosystems-client might be problematic. Click here for more details.

Files changed (103) hide show
  1. robosystems_client/api/agent/auto_select_agent.py +1 -41
  2. robosystems_client/api/agent/batch_process_queries.py +1 -41
  3. robosystems_client/api/agent/execute_specific_agent.py +1 -41
  4. robosystems_client/api/agent/get_agent_metadata.py +5 -49
  5. robosystems_client/api/agent/list_agents.py +4 -42
  6. robosystems_client/api/agent/recommend_agent.py +5 -45
  7. robosystems_client/api/auth/generate_sso_token.py +0 -18
  8. robosystems_client/api/auth/get_current_auth_user.py +14 -74
  9. robosystems_client/api/auth/logout_user.py +14 -50
  10. robosystems_client/api/auth/refresh_auth_session.py +14 -50
  11. robosystems_client/api/auth/resend_verification_email.py +14 -74
  12. robosystems_client/api/backup/create_backup.py +5 -45
  13. robosystems_client/api/backup/get_backup_download_url.py +4 -42
  14. robosystems_client/api/backup/get_backup_stats.py +5 -49
  15. robosystems_client/api/backup/list_backups.py +4 -42
  16. robosystems_client/api/backup/restore_backup.py +5 -45
  17. robosystems_client/api/connections/create_connection.py +5 -45
  18. robosystems_client/api/connections/create_link_token.py +5 -45
  19. robosystems_client/api/connections/delete_connection.py +5 -49
  20. robosystems_client/api/connections/exchange_link_token.py +5 -45
  21. robosystems_client/api/connections/get_connection.py +5 -49
  22. robosystems_client/api/connections/get_connection_options.py +5 -49
  23. robosystems_client/api/connections/init_o_auth.py +5 -45
  24. robosystems_client/api/connections/list_connections.py +4 -42
  25. robosystems_client/api/connections/oauth_callback.py +5 -45
  26. robosystems_client/api/connections/sync_connection.py +5 -45
  27. robosystems_client/api/graph_analytics/get_graph_metrics.py +5 -49
  28. robosystems_client/api/graph_analytics/get_graph_usage_stats.py +4 -42
  29. robosystems_client/api/graph_billing/get_current_graph_bill.py +5 -49
  30. robosystems_client/api/graph_billing/get_graph_billing_history.py +4 -42
  31. robosystems_client/api/graph_billing/get_graph_monthly_bill.py +5 -49
  32. robosystems_client/api/graph_billing/get_graph_usage_details.py +4 -42
  33. robosystems_client/api/graph_credits/check_credit_balance.py +0 -38
  34. robosystems_client/api/graph_credits/check_storage_limits.py +1 -45
  35. robosystems_client/api/graph_credits/get_credit_summary.py +1 -45
  36. robosystems_client/api/graph_credits/get_storage_usage.py +0 -38
  37. robosystems_client/api/graph_credits/list_credit_transactions.py +4 -42
  38. robosystems_client/api/graph_health/get_database_health.py +5 -49
  39. robosystems_client/api/graph_info/get_database_info.py +5 -49
  40. robosystems_client/api/graph_limits/get_graph_limits.py +5 -49
  41. robosystems_client/api/graphs/create_graph.py +21 -57
  42. robosystems_client/api/graphs/get_available_extensions.py +131 -15
  43. robosystems_client/api/graphs/get_graphs.py +154 -79
  44. robosystems_client/api/graphs/select_graph.py +117 -49
  45. robosystems_client/api/mcp/call_mcp_tool.py +24 -47
  46. robosystems_client/api/mcp/list_mcp_tools.py +13 -61
  47. robosystems_client/api/operations/cancel_operation.py +1 -45
  48. robosystems_client/api/operations/get_operation_status.py +1 -45
  49. robosystems_client/api/query/execute_cypher_query.py +69 -53
  50. robosystems_client/api/schema/export_graph_schema.py +223 -65
  51. robosystems_client/api/schema/get_graph_schema.py +137 -79
  52. robosystems_client/api/schema/validate_schema.py +5 -45
  53. robosystems_client/api/subgraphs/create_subgraph.py +5 -45
  54. robosystems_client/api/subgraphs/delete_subgraph.py +5 -45
  55. robosystems_client/api/subgraphs/get_subgraph_info.py +5 -49
  56. robosystems_client/api/subgraphs/get_subgraph_quota.py +5 -49
  57. robosystems_client/api/subgraphs/list_subgraphs.py +5 -49
  58. robosystems_client/api/tables/delete_file.py +181 -301
  59. robosystems_client/api/tables/get_file_info.py +117 -265
  60. robosystems_client/api/tables/get_upload_url.py +193 -389
  61. robosystems_client/api/tables/ingest_tables.py +277 -465
  62. robosystems_client/api/tables/list_table_files.py +193 -373
  63. robosystems_client/api/tables/list_tables.py +189 -361
  64. robosystems_client/api/tables/query_tables.py +85 -141
  65. robosystems_client/api/tables/update_file_status.py +205 -349
  66. robosystems_client/api/user/create_user_api_key.py +1 -41
  67. robosystems_client/api/user/get_all_credit_summaries.py +14 -111
  68. robosystems_client/api/user/get_current_user.py +14 -75
  69. robosystems_client/api/user/list_user_api_keys.py +14 -75
  70. robosystems_client/api/user/revoke_user_api_key.py +1 -45
  71. robosystems_client/api/user/update_user.py +1 -41
  72. robosystems_client/api/user/update_user_api_key.py +1 -41
  73. robosystems_client/api/user/update_user_password.py +1 -41
  74. robosystems_client/api/user_analytics/get_detailed_user_analytics.py +0 -38
  75. robosystems_client/api/user_analytics/get_user_usage_overview.py +14 -75
  76. robosystems_client/api/user_limits/get_all_shared_repository_limits.py +14 -105
  77. robosystems_client/api/user_limits/get_shared_repository_limits.py +1 -45
  78. robosystems_client/api/user_limits/get_user_limits.py +14 -75
  79. robosystems_client/api/user_limits/get_user_usage.py +14 -75
  80. robosystems_client/api/user_subscriptions/cancel_shared_repository_subscription.py +1 -45
  81. robosystems_client/api/user_subscriptions/get_repository_credits.py +1 -45
  82. robosystems_client/api/user_subscriptions/get_shared_repository_credits.py +14 -75
  83. robosystems_client/api/user_subscriptions/get_user_shared_subscriptions.py +0 -38
  84. robosystems_client/api/user_subscriptions/subscribe_to_shared_repository.py +1 -41
  85. robosystems_client/api/user_subscriptions/upgrade_shared_repository_subscription.py +1 -41
  86. robosystems_client/extensions/__init__.py +8 -1
  87. robosystems_client/extensions/auth_integration.py +1 -2
  88. robosystems_client/extensions/query_client.py +3 -2
  89. robosystems_client/extensions/sse_client.py +1 -1
  90. robosystems_client/extensions/table_ingest_client.py +5 -0
  91. robosystems_client/extensions/utils.py +2 -2
  92. robosystems_client/models/__init__.py +4 -4
  93. robosystems_client/models/create_graph_request.py +4 -3
  94. robosystems_client/models/cypher_query_request.py +5 -22
  95. robosystems_client/models/schema_export_response.py +4 -2
  96. robosystems_client/models/schema_info_response.py +77 -0
  97. robosystems_client/models/{get_graph_schema_response_getgraphschema.py → schema_info_response_schema.py} +6 -6
  98. robosystems_client/models/schema_validation_response.py +7 -6
  99. robosystems_client/models/table_query_request.py +37 -2
  100. {robosystems_client-0.2.3.dist-info → robosystems_client-0.2.4.dist-info}/METADATA +2 -4
  101. {robosystems_client-0.2.3.dist-info → robosystems_client-0.2.4.dist-info}/RECORD +103 -102
  102. {robosystems_client-0.2.3.dist-info → robosystems_client-0.2.4.dist-info}/WHEEL +0 -0
  103. {robosystems_client-0.2.3.dist-info → robosystems_client-0.2.4.dist-info}/licenses/LICENSE +0 -0
@@ -8,37 +8,17 @@ from ...client import AuthenticatedClient, Client
8
8
  from ...models.error_response import ErrorResponse
9
9
  from ...models.http_validation_error import HTTPValidationError
10
10
  from ...models.table_list_response import TableListResponse
11
- from ...types import UNSET, Response, Unset
11
+ from ...types import Response
12
12
 
13
13
 
14
14
  def _get_kwargs(
15
15
  graph_id: str,
16
- *,
17
- token: Union[None, Unset, str] = UNSET,
18
- authorization: Union[None, Unset, str] = UNSET,
19
16
  ) -> dict[str, Any]:
20
- headers: dict[str, Any] = {}
21
- if not isinstance(authorization, Unset):
22
- headers["authorization"] = authorization
23
-
24
- params: dict[str, Any] = {}
25
-
26
- json_token: Union[None, Unset, str]
27
- if isinstance(token, Unset):
28
- json_token = UNSET
29
- else:
30
- json_token = token
31
- params["token"] = json_token
32
-
33
- params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
34
-
35
17
  _kwargs: dict[str, Any] = {
36
18
  "method": "get",
37
19
  "url": f"/v1/graphs/{graph_id}/tables",
38
- "params": params,
39
20
  }
40
21
 
41
- _kwargs["headers"] = headers
42
22
  return _kwargs
43
23
 
44
24
 
@@ -94,95 +74,57 @@ def sync_detailed(
94
74
  graph_id: str,
95
75
  *,
96
76
  client: AuthenticatedClient,
97
- token: Union[None, Unset, str] = UNSET,
98
- authorization: Union[None, Unset, str] = UNSET,
99
77
  ) -> Response[Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]]:
100
- r""" List Staging Tables
101
-
102
- List all DuckDB staging tables with comprehensive metrics and status.
103
-
104
- **Purpose:**
105
- Get a complete inventory of all staging tables for a graph, including
106
- file counts, storage sizes, and row estimates. Essential for monitoring
107
- the data pipeline and determining which tables are ready for ingestion.
108
-
109
- **What You Get:**
110
- - Table name and type (node/relationship)
111
- - File count per table
112
- - Total storage size in bytes
113
- - Estimated row count
114
- - S3 location pattern
115
- - Ready-for-ingestion status
116
-
117
- **Use Cases:**
118
- - Monitor data upload progress
119
- - Check which tables have files ready
120
- - Track storage consumption
121
- - Validate pipeline before ingestion
122
- - Capacity planning
123
-
124
- **Workflow:**
125
- 1. List tables to see current state
126
- 2. Upload files to empty tables
127
- 3. Re-list to verify uploads
128
- 4. Check file counts and sizes
129
- 5. Ingest when ready
130
-
131
- **Example Response:**
132
- ```json
133
- {
134
- \"tables\": [
135
- {
136
- \"table_name\": \"Entity\",
137
- \"row_count\": 5000,
138
- \"file_count\": 3,
139
- \"total_size_bytes\": 2457600,
140
- \"s3_location\": \"s3://bucket/user-staging/user123/graph456/Entity/**/*.parquet\"
141
- },
142
- {
143
- \"table_name\": \"Transaction\",
144
- \"row_count\": 15000,
145
- \"file_count\": 5,
146
- \"total_size_bytes\": 8192000,
147
- \"s3_location\": \"s3://bucket/user-staging/user123/graph456/Transaction/**/*.parquet\"
148
- }
149
- ],
150
- \"total_count\": 2
151
- }
152
- ```
153
-
154
- **Example Usage:**
155
- ```bash
156
- curl -H \"Authorization: Bearer YOUR_TOKEN\" \
157
- https://api.robosystems.ai/v1/graphs/kg123/tables
158
- ```
159
-
160
- **Tips:**
161
- - Tables with `file_count > 0` have data ready
162
- - Check `total_size_bytes` for storage monitoring
163
- - Use `s3_location` to verify upload paths
164
- - Empty tables (file_count=0) are skipped during ingestion
165
-
166
- **Note:**
167
- Table queries are included - no credit consumption.
168
-
169
- Args:
170
- graph_id (str): Graph database identifier
171
- token (Union[None, Unset, str]): JWT token for SSE authentication
172
- authorization (Union[None, Unset, str]):
173
-
174
- Raises:
175
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
176
- httpx.TimeoutException: If the request takes longer than Client.timeout.
177
-
178
- Returns:
179
- Response[Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]]
180
- """
78
+ """List Staging Tables
79
+
80
+ List all DuckDB staging tables with comprehensive metrics and status.
81
+
82
+ Get a complete inventory of all staging tables for a graph, including
83
+ file counts, storage sizes, and row estimates. Essential for monitoring
84
+ the data pipeline and determining which tables are ready for ingestion.
85
+
86
+ **Returned Metrics:**
87
+ - Table name and type (node/relationship)
88
+ - File count per table
89
+ - Total storage size in bytes
90
+ - Estimated row count
91
+ - S3 location pattern
92
+ - Ready-for-ingestion status
93
+
94
+ **Use Cases:**
95
+ - Monitor data upload progress
96
+ - Check which tables have files ready
97
+ - Track storage consumption
98
+ - Validate pipeline before ingestion
99
+ - Capacity planning
100
+
101
+ **Workflow:**
102
+ 1. List tables to see current state
103
+ 2. Upload files to empty tables
104
+ 3. Re-list to verify uploads
105
+ 4. Check file counts and sizes
106
+ 5. Ingest when ready
107
+
108
+ **Important Notes:**
109
+ - Tables with `file_count > 0` have data ready
110
+ - Check `total_size_bytes` for storage monitoring
111
+ - Use `s3_location` to verify upload paths
112
+ - Empty tables (file_count=0) are skipped during ingestion
113
+ - Table queries are included - no credit consumption
114
+
115
+ Args:
116
+ graph_id (str):
117
+
118
+ Raises:
119
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
120
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
121
+
122
+ Returns:
123
+ Response[Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]]
124
+ """
181
125
 
182
126
  kwargs = _get_kwargs(
183
127
  graph_id=graph_id,
184
- token=token,
185
- authorization=authorization,
186
128
  )
187
129
 
188
130
  response = client.get_httpx_client().request(
@@ -196,96 +138,58 @@ def sync(
196
138
  graph_id: str,
197
139
  *,
198
140
  client: AuthenticatedClient,
199
- token: Union[None, Unset, str] = UNSET,
200
- authorization: Union[None, Unset, str] = UNSET,
201
141
  ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]]:
202
- r""" List Staging Tables
203
-
204
- List all DuckDB staging tables with comprehensive metrics and status.
205
-
206
- **Purpose:**
207
- Get a complete inventory of all staging tables for a graph, including
208
- file counts, storage sizes, and row estimates. Essential for monitoring
209
- the data pipeline and determining which tables are ready for ingestion.
210
-
211
- **What You Get:**
212
- - Table name and type (node/relationship)
213
- - File count per table
214
- - Total storage size in bytes
215
- - Estimated row count
216
- - S3 location pattern
217
- - Ready-for-ingestion status
218
-
219
- **Use Cases:**
220
- - Monitor data upload progress
221
- - Check which tables have files ready
222
- - Track storage consumption
223
- - Validate pipeline before ingestion
224
- - Capacity planning
225
-
226
- **Workflow:**
227
- 1. List tables to see current state
228
- 2. Upload files to empty tables
229
- 3. Re-list to verify uploads
230
- 4. Check file counts and sizes
231
- 5. Ingest when ready
232
-
233
- **Example Response:**
234
- ```json
235
- {
236
- \"tables\": [
237
- {
238
- \"table_name\": \"Entity\",
239
- \"row_count\": 5000,
240
- \"file_count\": 3,
241
- \"total_size_bytes\": 2457600,
242
- \"s3_location\": \"s3://bucket/user-staging/user123/graph456/Entity/**/*.parquet\"
243
- },
244
- {
245
- \"table_name\": \"Transaction\",
246
- \"row_count\": 15000,
247
- \"file_count\": 5,
248
- \"total_size_bytes\": 8192000,
249
- \"s3_location\": \"s3://bucket/user-staging/user123/graph456/Transaction/**/*.parquet\"
250
- }
251
- ],
252
- \"total_count\": 2
253
- }
254
- ```
255
-
256
- **Example Usage:**
257
- ```bash
258
- curl -H \"Authorization: Bearer YOUR_TOKEN\" \
259
- https://api.robosystems.ai/v1/graphs/kg123/tables
260
- ```
261
-
262
- **Tips:**
263
- - Tables with `file_count > 0` have data ready
264
- - Check `total_size_bytes` for storage monitoring
265
- - Use `s3_location` to verify upload paths
266
- - Empty tables (file_count=0) are skipped during ingestion
267
-
268
- **Note:**
269
- Table queries are included - no credit consumption.
270
-
271
- Args:
272
- graph_id (str): Graph database identifier
273
- token (Union[None, Unset, str]): JWT token for SSE authentication
274
- authorization (Union[None, Unset, str]):
275
-
276
- Raises:
277
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
278
- httpx.TimeoutException: If the request takes longer than Client.timeout.
279
-
280
- Returns:
281
- Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]
282
- """
142
+ """List Staging Tables
143
+
144
+ List all DuckDB staging tables with comprehensive metrics and status.
145
+
146
+ Get a complete inventory of all staging tables for a graph, including
147
+ file counts, storage sizes, and row estimates. Essential for monitoring
148
+ the data pipeline and determining which tables are ready for ingestion.
149
+
150
+ **Returned Metrics:**
151
+ - Table name and type (node/relationship)
152
+ - File count per table
153
+ - Total storage size in bytes
154
+ - Estimated row count
155
+ - S3 location pattern
156
+ - Ready-for-ingestion status
157
+
158
+ **Use Cases:**
159
+ - Monitor data upload progress
160
+ - Check which tables have files ready
161
+ - Track storage consumption
162
+ - Validate pipeline before ingestion
163
+ - Capacity planning
164
+
165
+ **Workflow:**
166
+ 1. List tables to see current state
167
+ 2. Upload files to empty tables
168
+ 3. Re-list to verify uploads
169
+ 4. Check file counts and sizes
170
+ 5. Ingest when ready
171
+
172
+ **Important Notes:**
173
+ - Tables with `file_count > 0` have data ready
174
+ - Check `total_size_bytes` for storage monitoring
175
+ - Use `s3_location` to verify upload paths
176
+ - Empty tables (file_count=0) are skipped during ingestion
177
+ - Table queries are included - no credit consumption
178
+
179
+ Args:
180
+ graph_id (str):
181
+
182
+ Raises:
183
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
184
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
185
+
186
+ Returns:
187
+ Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]
188
+ """
283
189
 
284
190
  return sync_detailed(
285
191
  graph_id=graph_id,
286
192
  client=client,
287
- token=token,
288
- authorization=authorization,
289
193
  ).parsed
290
194
 
291
195
 
@@ -293,95 +197,57 @@ async def asyncio_detailed(
293
197
  graph_id: str,
294
198
  *,
295
199
  client: AuthenticatedClient,
296
- token: Union[None, Unset, str] = UNSET,
297
- authorization: Union[None, Unset, str] = UNSET,
298
200
  ) -> Response[Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]]:
299
- r""" List Staging Tables
300
-
301
- List all DuckDB staging tables with comprehensive metrics and status.
302
-
303
- **Purpose:**
304
- Get a complete inventory of all staging tables for a graph, including
305
- file counts, storage sizes, and row estimates. Essential for monitoring
306
- the data pipeline and determining which tables are ready for ingestion.
307
-
308
- **What You Get:**
309
- - Table name and type (node/relationship)
310
- - File count per table
311
- - Total storage size in bytes
312
- - Estimated row count
313
- - S3 location pattern
314
- - Ready-for-ingestion status
315
-
316
- **Use Cases:**
317
- - Monitor data upload progress
318
- - Check which tables have files ready
319
- - Track storage consumption
320
- - Validate pipeline before ingestion
321
- - Capacity planning
322
-
323
- **Workflow:**
324
- 1. List tables to see current state
325
- 2. Upload files to empty tables
326
- 3. Re-list to verify uploads
327
- 4. Check file counts and sizes
328
- 5. Ingest when ready
329
-
330
- **Example Response:**
331
- ```json
332
- {
333
- \"tables\": [
334
- {
335
- \"table_name\": \"Entity\",
336
- \"row_count\": 5000,
337
- \"file_count\": 3,
338
- \"total_size_bytes\": 2457600,
339
- \"s3_location\": \"s3://bucket/user-staging/user123/graph456/Entity/**/*.parquet\"
340
- },
341
- {
342
- \"table_name\": \"Transaction\",
343
- \"row_count\": 15000,
344
- \"file_count\": 5,
345
- \"total_size_bytes\": 8192000,
346
- \"s3_location\": \"s3://bucket/user-staging/user123/graph456/Transaction/**/*.parquet\"
347
- }
348
- ],
349
- \"total_count\": 2
350
- }
351
- ```
352
-
353
- **Example Usage:**
354
- ```bash
355
- curl -H \"Authorization: Bearer YOUR_TOKEN\" \
356
- https://api.robosystems.ai/v1/graphs/kg123/tables
357
- ```
358
-
359
- **Tips:**
360
- - Tables with `file_count > 0` have data ready
361
- - Check `total_size_bytes` for storage monitoring
362
- - Use `s3_location` to verify upload paths
363
- - Empty tables (file_count=0) are skipped during ingestion
364
-
365
- **Note:**
366
- Table queries are included - no credit consumption.
367
-
368
- Args:
369
- graph_id (str): Graph database identifier
370
- token (Union[None, Unset, str]): JWT token for SSE authentication
371
- authorization (Union[None, Unset, str]):
372
-
373
- Raises:
374
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
375
- httpx.TimeoutException: If the request takes longer than Client.timeout.
376
-
377
- Returns:
378
- Response[Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]]
379
- """
201
+ """List Staging Tables
202
+
203
+ List all DuckDB staging tables with comprehensive metrics and status.
204
+
205
+ Get a complete inventory of all staging tables for a graph, including
206
+ file counts, storage sizes, and row estimates. Essential for monitoring
207
+ the data pipeline and determining which tables are ready for ingestion.
208
+
209
+ **Returned Metrics:**
210
+ - Table name and type (node/relationship)
211
+ - File count per table
212
+ - Total storage size in bytes
213
+ - Estimated row count
214
+ - S3 location pattern
215
+ - Ready-for-ingestion status
216
+
217
+ **Use Cases:**
218
+ - Monitor data upload progress
219
+ - Check which tables have files ready
220
+ - Track storage consumption
221
+ - Validate pipeline before ingestion
222
+ - Capacity planning
223
+
224
+ **Workflow:**
225
+ 1. List tables to see current state
226
+ 2. Upload files to empty tables
227
+ 3. Re-list to verify uploads
228
+ 4. Check file counts and sizes
229
+ 5. Ingest when ready
230
+
231
+ **Important Notes:**
232
+ - Tables with `file_count > 0` have data ready
233
+ - Check `total_size_bytes` for storage monitoring
234
+ - Use `s3_location` to verify upload paths
235
+ - Empty tables (file_count=0) are skipped during ingestion
236
+ - Table queries are included - no credit consumption
237
+
238
+ Args:
239
+ graph_id (str):
240
+
241
+ Raises:
242
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
243
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
244
+
245
+ Returns:
246
+ Response[Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]]
247
+ """
380
248
 
381
249
  kwargs = _get_kwargs(
382
250
  graph_id=graph_id,
383
- token=token,
384
- authorization=authorization,
385
251
  )
386
252
 
387
253
  response = await client.get_async_httpx_client().request(**kwargs)
@@ -393,96 +259,58 @@ async def asyncio(
393
259
  graph_id: str,
394
260
  *,
395
261
  client: AuthenticatedClient,
396
- token: Union[None, Unset, str] = UNSET,
397
- authorization: Union[None, Unset, str] = UNSET,
398
262
  ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]]:
399
- r""" List Staging Tables
400
-
401
- List all DuckDB staging tables with comprehensive metrics and status.
402
-
403
- **Purpose:**
404
- Get a complete inventory of all staging tables for a graph, including
405
- file counts, storage sizes, and row estimates. Essential for monitoring
406
- the data pipeline and determining which tables are ready for ingestion.
407
-
408
- **What You Get:**
409
- - Table name and type (node/relationship)
410
- - File count per table
411
- - Total storage size in bytes
412
- - Estimated row count
413
- - S3 location pattern
414
- - Ready-for-ingestion status
415
-
416
- **Use Cases:**
417
- - Monitor data upload progress
418
- - Check which tables have files ready
419
- - Track storage consumption
420
- - Validate pipeline before ingestion
421
- - Capacity planning
422
-
423
- **Workflow:**
424
- 1. List tables to see current state
425
- 2. Upload files to empty tables
426
- 3. Re-list to verify uploads
427
- 4. Check file counts and sizes
428
- 5. Ingest when ready
429
-
430
- **Example Response:**
431
- ```json
432
- {
433
- \"tables\": [
434
- {
435
- \"table_name\": \"Entity\",
436
- \"row_count\": 5000,
437
- \"file_count\": 3,
438
- \"total_size_bytes\": 2457600,
439
- \"s3_location\": \"s3://bucket/user-staging/user123/graph456/Entity/**/*.parquet\"
440
- },
441
- {
442
- \"table_name\": \"Transaction\",
443
- \"row_count\": 15000,
444
- \"file_count\": 5,
445
- \"total_size_bytes\": 8192000,
446
- \"s3_location\": \"s3://bucket/user-staging/user123/graph456/Transaction/**/*.parquet\"
447
- }
448
- ],
449
- \"total_count\": 2
450
- }
451
- ```
452
-
453
- **Example Usage:**
454
- ```bash
455
- curl -H \"Authorization: Bearer YOUR_TOKEN\" \
456
- https://api.robosystems.ai/v1/graphs/kg123/tables
457
- ```
458
-
459
- **Tips:**
460
- - Tables with `file_count > 0` have data ready
461
- - Check `total_size_bytes` for storage monitoring
462
- - Use `s3_location` to verify upload paths
463
- - Empty tables (file_count=0) are skipped during ingestion
464
-
465
- **Note:**
466
- Table queries are included - no credit consumption.
467
-
468
- Args:
469
- graph_id (str): Graph database identifier
470
- token (Union[None, Unset, str]): JWT token for SSE authentication
471
- authorization (Union[None, Unset, str]):
472
-
473
- Raises:
474
- errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
475
- httpx.TimeoutException: If the request takes longer than Client.timeout.
476
-
477
- Returns:
478
- Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]
479
- """
263
+ """List Staging Tables
264
+
265
+ List all DuckDB staging tables with comprehensive metrics and status.
266
+
267
+ Get a complete inventory of all staging tables for a graph, including
268
+ file counts, storage sizes, and row estimates. Essential for monitoring
269
+ the data pipeline and determining which tables are ready for ingestion.
270
+
271
+ **Returned Metrics:**
272
+ - Table name and type (node/relationship)
273
+ - File count per table
274
+ - Total storage size in bytes
275
+ - Estimated row count
276
+ - S3 location pattern
277
+ - Ready-for-ingestion status
278
+
279
+ **Use Cases:**
280
+ - Monitor data upload progress
281
+ - Check which tables have files ready
282
+ - Track storage consumption
283
+ - Validate pipeline before ingestion
284
+ - Capacity planning
285
+
286
+ **Workflow:**
287
+ 1. List tables to see current state
288
+ 2. Upload files to empty tables
289
+ 3. Re-list to verify uploads
290
+ 4. Check file counts and sizes
291
+ 5. Ingest when ready
292
+
293
+ **Important Notes:**
294
+ - Tables with `file_count > 0` have data ready
295
+ - Check `total_size_bytes` for storage monitoring
296
+ - Use `s3_location` to verify upload paths
297
+ - Empty tables (file_count=0) are skipped during ingestion
298
+ - Table queries are included - no credit consumption
299
+
300
+ Args:
301
+ graph_id (str):
302
+
303
+ Raises:
304
+ errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
305
+ httpx.TimeoutException: If the request takes longer than Client.timeout.
306
+
307
+ Returns:
308
+ Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]
309
+ """
480
310
 
481
311
  return (
482
312
  await asyncio_detailed(
483
313
  graph_id=graph_id,
484
314
  client=client,
485
- token=token,
486
- authorization=authorization,
487
315
  )
488
316
  ).parsed