awslabs.s3-tables-mcp-server 0.0.2__tar.gz → 0.0.4__tar.gz

{awslabs_s3_tables_mcp_server-0.0.2 → awslabs_s3_tables_mcp_server-0.0.4}/Dockerfile

@@ -13,7 +13,7 @@
 # limitations under the License.
 
 # dependabot should continue to update this to the latest hash.
-FROM public.ecr.aws/sam/build-python3.13@sha256:0c274ddd44e1d80e4dab3a70c25fe29508f612a045cba7d27840461c12eee86d AS uv
+FROM public.ecr.aws/docker/library/python:3.13-slim-bookworm@sha256:6544e0e002b40ae0f59bc3618b07c1e48064c4faed3a15ae2fbd2e8f663e8283 AS uv
 
 # Install the project into `/app`
 WORKDIR /app
@@ -48,10 +48,10 @@ COPY . /app
 RUN --mount=type=cache,target=/root/.cache/uv \
     uv sync --python 3.13 --frozen --no-dev --no-editable
 
-# Make the directory just in case it doesn't exist
-RUN mkdir -p /root/.local
+# # Make the directory just in case it doesn't exist
+# RUN mkdir -p /root/.local
 
-FROM public.ecr.aws/sam/build-python3.13@sha256:0c274ddd44e1d80e4dab3a70c25fe29508f612a045cba7d27840461c12eee86d
+FROM public.ecr.aws/docker/library/python:3.13-slim-bookworm@sha256:6544e0e002b40ae0f59bc3618b07c1e48064c4faed3a15ae2fbd2e8f663e8283
 
 # Place executables in the environment at the front of the path and include other binaries
 ENV PATH="/app/.venv/bin:$PATH:/usr/sbin" \
@@ -63,7 +63,7 @@ RUN groupadd --force --system app && \
     chmod o+x /root
 
 # Copy application artifacts from build stage
-COPY --from=uv --chown=app:app /root/.local /root/.local
+# COPY --from=uv --chown=app:app /root/.local /root/.local
 COPY --from=uv --chown=app:app /app/.venv /app/.venv
 
 # Get healthcheck script

{awslabs_s3_tables_mcp_server-0.0.2 → awslabs_s3_tables_mcp_server-0.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: awslabs.s3-tables-mcp-server
-Version: 0.0.2
+Version: 0.0.4
 Summary: An AWS Labs Model Context Protocol (MCP) server for awslabs.s3-tables-mcp-server
 Project-URL: homepage, https://awslabs.github.io/mcp/
 Project-URL: docs, https://awslabs.github.io/mcp/servers/s3-tables-mcp-server/

{awslabs_s3_tables_mcp_server-0.0.2 → awslabs_s3_tables_mcp_server-0.0.4}/awslabs/s3_tables_mcp_server/__init__.py

@@ -15,4 +15,4 @@
 # This file is part of the awslabs namespace.
 # It is intentionally minimal to support PEP 420 namespace packages.
 
-__version__ = '0.0.0'
+__version__ = '0.0.4'

{awslabs_s3_tables_mcp_server-0.0.2 → awslabs_s3_tables_mcp_server-0.0.4}/awslabs/s3_tables_mcp_server/engines/pyiceberg.py

@@ -14,32 +14,14 @@
 
 """Engine for interacting with Iceberg tables using pyiceberg and daft (read-only)."""
 
+import io
+import json
 import pyarrow as pa
+import pyarrow.json as pj
 from ..utils import pyiceberg_load_catalog
 from daft import Catalog as DaftCatalog
 from daft.session import Session
-from datetime import date, datetime, time
-from decimal import Decimal
 from pydantic import BaseModel
-from pyiceberg.types import (
-    BinaryType,
-    BooleanType,
-    DateType,
-    DecimalType,
-    DoubleType,
-    FixedType,
-    FloatType,
-    IntegerType,
-    ListType,
-    LongType,
-    MapType,
-    StringType,
-    StructType,
-    TimestampType,
-    TimestamptzType,
-    TimeType,
-    UUIDType,
-)
 
 # pyiceberg and daft imports
 from typing import Any, Dict, Optional
@@ -57,78 +39,6 @@ class PyIcebergConfig(BaseModel):
     rest_sigv4_enabled: str = 'true'
 
 
-def convert_value_for_append(value, iceberg_type):
-    """Convert a value to the appropriate type for appending to an Iceberg table column.
-
-    Args:
-        value: The value to convert. Can be of various types (str, int, float, etc.).
-        iceberg_type: The Iceberg type to convert the value to.
-
-    Returns:
-        The value converted to the appropriate type for the Iceberg column, or None if value is None.
-
-    Raises:
-        NotImplementedError: If the iceberg_type is a complex type (ListType, MapType, StructType).
-        ValueError: If the conversion is unsupported or fails.
-    """
-    if value is None:
-        return None
-    # Already correct type
-    if isinstance(iceberg_type, BooleanType) and isinstance(value, bool):
-        return value
-    if isinstance(iceberg_type, (IntegerType, LongType)) and isinstance(value, int):
-        return value
-    if isinstance(iceberg_type, (FloatType, DoubleType)) and isinstance(value, float):
-        return value
-    if isinstance(iceberg_type, DecimalType) and isinstance(value, Decimal):
-        return value
-    if isinstance(iceberg_type, DateType) and isinstance(value, date):
-        return value
-    if isinstance(iceberg_type, TimeType) and isinstance(value, time):
-        return value
-    if isinstance(iceberg_type, (TimestampType, TimestamptzType)) and isinstance(value, datetime):
-        return value
-    if isinstance(iceberg_type, StringType) and isinstance(value, str):
-        return value
-    # Convert from string
-    if isinstance(value, str):
-        if isinstance(iceberg_type, BooleanType):
-            return value.lower() in ('true', '1', 'yes')
-        if isinstance(iceberg_type, (IntegerType, LongType)):
-            return int(value)
-        if isinstance(iceberg_type, (FloatType, DoubleType)):
-            return float(value)
-        if isinstance(iceberg_type, DecimalType):
-            return Decimal(value)
-        if isinstance(iceberg_type, DateType):
-            return date.fromisoformat(value)
-        if isinstance(iceberg_type, TimeType):
-            return time.fromisoformat(value)
-        if isinstance(iceberg_type, (TimestampType, TimestamptzType)):
-            return datetime.fromisoformat(value)
-        if isinstance(iceberg_type, StringType):
-            return value
-        if isinstance(iceberg_type, UUIDType):
-            import uuid
-
-            return uuid.UUID(value)
-        if isinstance(iceberg_type, (BinaryType, FixedType)):
-            return bytes.fromhex(value)
-    # Convert from number
-    if isinstance(value, (int, float)):
-        if isinstance(iceberg_type, (IntegerType, LongType)):
-            return int(value)
-        if isinstance(iceberg_type, (FloatType, DoubleType)):
-            return float(value)
-        if isinstance(iceberg_type, DecimalType):
-            return Decimal(str(value))
-        if isinstance(iceberg_type, StringType):
-            return str(value)
-    if isinstance(iceberg_type, (ListType, MapType, StructType)):
-        raise NotImplementedError(f'Complex type {iceberg_type} not supported in append_rows')
-    raise ValueError(f'Unsupported conversion from {type(value)} to {iceberg_type}')
-
-
 class PyIcebergEngine:
     """Engine for read-only queries on Iceberg tables using pyiceberg and daft."""
 
@@ -197,7 +107,7 @@ class PyIcebergEngine:
             return False
 
     def append_rows(self, table_name: str, rows: list[dict]) -> None:
-        """Append rows to an Iceberg table using pyiceberg.
+        """Append rows to an Iceberg table using pyiceberg with JSON encoding.
 
         Args:
             table_name: The name of the table (e.g., 'namespace.tablename' or just 'tablename' if namespace is set)
@@ -214,26 +124,31 @@ class PyIcebergEngine:
                 full_table_name = f'{self.config.namespace}.{table_name}'
             else:
                 full_table_name = table_name
+
+            # Load the Iceberg table
             table = self._catalog.load_table(full_table_name)
-            iceberg_schema = table.schema()
-            converted_rows = []
+            # Encode rows as JSON (line-delimited format)
+            json_lines = []
             for row in rows:
-                converted_row = {}
-                for field in iceberg_schema.fields:
-                    field_name = field.name
-                    field_type = field.field_type
-                    value = row.get(field_name)
-                    if field.required and value is None:
-                        raise ValueError(f'Required field {field_name} is missing or None')
-                    try:
-                        converted_row[field_name] = convert_value_for_append(value, field_type)
-                    except (ValueError, TypeError) as e:
-                        raise ValueError(
-                            f'Error converting value for field {field_name}: {str(e)}'
-                        )
-                converted_rows.append(converted_row)
-            schema = iceberg_schema.as_arrow()
-            pa_table = pa.Table.from_pylist(converted_rows, schema=schema)
-            table.append(pa_table)
+                json_lines.append(json.dumps(row))
+            json_data = '\n'.join(json_lines)
+
+            # Create a file-like object from the JSON data
+            json_buffer = io.BytesIO(json_data.encode('utf-8'))
+
+            # Read JSON data into PyArrow Table using pyarrow.json.read_json
+            # This enforces the Iceberg schema and validates the data
+            try:
+                new_data_table = pj.read_json(
+                    json_buffer, read_options=pj.ReadOptions(use_threads=True)
+                )
+            except pa.ArrowInvalid as e:
+                raise ValueError(
+                    f'Schema mismatch detected: {e}. Please ensure your data matches the table schema.'
+                )
+
+            # Append the new data to the Iceberg table
+            table.append(new_data_table)
+
         except Exception as e:
             raise Exception(f'Error appending rows: {str(e)}')

awslabs_s3_tables_mcp_server-0.0.4/awslabs/s3_tables_mcp_server/file_processor/__init__.py

@@ -0,0 +1,24 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""AWS S3 Tables MCP Server file processing module.
+
+This module provides functionality for processing and analyzing uploaded files,
+particularly focusing on CSV and Parquet file handling and import capabilities.
+"""
+
+from .csv import import_csv_to_table
+from .parquet import import_parquet_to_table
+
+__all__ = ['import_csv_to_table', 'import_parquet_to_table']

awslabs_s3_tables_mcp_server-0.0.4/awslabs/s3_tables_mcp_server/file_processor/csv.py

@@ -0,0 +1,123 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""AWS S3 Tables MCP Server file processing module.
+
+This module provides functionality for processing and analyzing uploaded files,
+particularly focusing on CSV file handling and import capabilities.
+"""
+
+import io
+import os
+import pyarrow.csv as pc
+from ..utils import get_s3_client, pyiceberg_load_catalog
+from pyiceberg.exceptions import NoSuchTableError
+from typing import Dict
+from urllib.parse import urlparse
+
+
+async def import_csv_to_table(
+    warehouse: str,
+    region: str,
+    namespace: str,
+    table_name: str,
+    s3_url: str,
+    uri: str,
+    catalog_name: str = 's3tablescatalog',
+    rest_signing_name: str = 's3tables',
+    rest_sigv4_enabled: str = 'true',
+) -> Dict:
+    """Import data from a CSV file into an S3 table.
+
+    This function reads data from a CSV file stored in S3 and imports it into an existing S3 table.
+    If the table doesn't exist, it will be created using the schema inferred from the CSV file.
+
+    Args:
+        warehouse: Warehouse string for Iceberg catalog
+        region: AWS region for S3Tables/Iceberg REST endpoint
+        namespace: The namespace containing the table
+        table_name: The name of the table to import data into
+        s3_url: The S3 URL of the CSV file (format: s3://bucket-name/key)
+        uri: REST URI for Iceberg catalog
+        catalog_name: Catalog name
+        rest_signing_name: REST signing name
+        rest_sigv4_enabled: Enable SigV4 signing
+
+    Returns:
+        A dictionary containing:
+        - status: 'success' or 'error'
+        - message: Success message or error details
+        - rows_processed: Number of rows processed (on success)
+        - file_processed: Name of the processed file
+        - table_created: Boolean indicating if a new table was created (on success)
+    """
+    # Parse S3 URL
+    parsed = urlparse(s3_url)
+    bucket = parsed.netloc
+    key = parsed.path.lstrip('/')
+
+    try:
+        # Load Iceberg catalog
+        catalog = pyiceberg_load_catalog(
+            catalog_name,
+            warehouse,
+            uri,
+            region,
+            rest_signing_name,
+            rest_sigv4_enabled,
+        )
+
+        # Get S3 client and read the CSV file to infer schema
+        s3_client = get_s3_client()
+        response = s3_client.get_object(Bucket=bucket, Key=key)
+        csv_data = response['Body'].read()
+
+        # Read CSV file into PyArrow Table to infer schema
+        # Convert bytes to file-like object for PyArrow
+        csv_buffer = io.BytesIO(csv_data)
+        csv_table = pc.read_csv(csv_buffer)
+        csv_schema = csv_table.schema
+
+        table_created = False
+        try:
+            # Try to load existing table
+            table = catalog.load_table(f'{namespace}.{table_name}')
+        except NoSuchTableError:
+            # Table doesn't exist, create it using the CSV schema
+            try:
+                table = catalog.create_table(
+                    identifier=f'{namespace}.{table_name}',
+                    schema=csv_schema,
+                )
+                table_created = True
+            except Exception as create_error:
+                return {
+                    'status': 'error',
+                    'error': f'Failed to create table: {str(create_error)}',
+                }
+
+        # Append data to Iceberg table
+        table.append(csv_table)
+
+        return {
+            'status': 'success',
+            'message': f'Successfully imported {csv_table.num_rows} rows{" and created new table" if table_created else ""}',
+            'rows_processed': csv_table.num_rows,
+            'file_processed': os.path.basename(key),
+            'table_created': table_created,
+            'table_uuid': table.metadata.table_uuid,
+        }
+
+    except Exception as e:
+        return {'status': 'error', 'error': str(e)}

awslabs_s3_tables_mcp_server-0.0.4/awslabs/s3_tables_mcp_server/file_processor/parquet.py

@@ -0,0 +1,116 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import pyarrow.parquet as pq
+from awslabs.s3_tables_mcp_server.utils import get_s3_client, pyiceberg_load_catalog
+from io import BytesIO
+from pyiceberg.exceptions import NoSuchTableError
+from typing import Dict
+
+
+async def import_parquet_to_table(
+    warehouse: str,
+    region: str,
+    namespace: str,
+    table_name: str,
+    s3_url: str,
+    uri: str,
+    catalog_name: str = 's3tablescatalog',
+    rest_signing_name: str = 's3tables',
+    rest_sigv4_enabled: str = 'true',
+) -> Dict:
+    """Import data from a Parquet file into an S3 table.
+
+    This function reads data from a Parquet file stored in S3 and imports it into an existing Iceberg table.
+    If the table doesn't exist, it will be created using the schema from the Parquet file.
+
+    Args:
+        warehouse: Warehouse string for Iceberg catalog
+        region: AWS region for S3Tables/Iceberg REST endpoint
+        namespace: The namespace containing the table
+        table_name: The name of the table to import data into
+        s3_url: The S3 URL of the Parquet file
+        uri: REST URI for Iceberg catalog
+        catalog_name: Catalog name
+        rest_signing_name: REST signing name
+        rest_sigv4_enabled: Enable SigV4 signing
+
+    Returns:
+        A dictionary containing:
+        - status: 'success' or 'error'
+        - message: Success message or error details
+        - rows_processed: Number of rows processed (on success)
+        - file_processed: Name of the processed file
+        - table_created: Boolean indicating if a new table was created (on success)
+    """
+    import os
+    from urllib.parse import urlparse
+
+    # Parse S3 URL
+    parsed = urlparse(s3_url)
+    bucket = parsed.netloc
+    key = parsed.path.lstrip('/')
+
+    try:
+        # Load Iceberg catalog
+        catalog = pyiceberg_load_catalog(
+            catalog_name,
+            warehouse,
+            uri,
+            region,
+            rest_signing_name,
+            rest_sigv4_enabled,
+        )
+
+        # Get S3 client and read the Parquet file first to get the schema
+        s3_client = get_s3_client()
+        response = s3_client.get_object(Bucket=bucket, Key=key)
+        parquet_data = BytesIO(response['Body'].read())
+
+        # Read Parquet file into PyArrow Table
+        parquet_table = pq.read_table(parquet_data)
+        parquet_schema = parquet_table.schema
+
+        table_created = False
+        try:
+            # Try to load existing table
+            table = catalog.load_table(f'{namespace}.{table_name}')
+        except NoSuchTableError:
+            # Table doesn't exist, create it using the Parquet schema
+            try:
+                table = catalog.create_table(
+                    identifier=f'{namespace}.{table_name}',
+                    schema=parquet_schema,
+                )
+                table_created = True
+            except Exception as create_error:
+                return {
+                    'status': 'error',
+                    'error': f'Failed to create table: {str(create_error)}',
+                }
+
+        # Append data to Iceberg table
+        table.append(parquet_table)
+
+        return {
+            'status': 'success',
+            'message': f'Successfully imported {parquet_table.num_rows} rows{" and created new table" if table_created else ""}',
+            'rows_processed': parquet_table.num_rows,
+            'file_processed': os.path.basename(key),
+            'table_created': table_created,
+            'table_uuid': table.metadata.table_uuid,
+        }
+
+    except Exception as e:
+        return {'status': 'error', 'error': str(e)}

{awslabs_s3_tables_mcp_server-0.0.2 → awslabs_s3_tables_mcp_server-0.0.4}/awslabs/s3_tables_mcp_server/server.py

@@ -32,7 +32,6 @@ from .utils import set_user_agent_mode
 from awslabs.s3_tables_mcp_server import (
     __version__,
     database,
-    file_processor,
     namespaces,
     resources,
     s3_operations,
@@ -48,6 +47,12 @@ from awslabs.s3_tables_mcp_server.constants import (
     TABLE_BUCKET_NAME_PATTERN,
     TABLE_NAME_FIELD,
 )
+from awslabs.s3_tables_mcp_server.file_processor import (
+    import_csv_to_table as import_csv_to_table_func,
+)
+from awslabs.s3_tables_mcp_server.file_processor import (
+    import_parquet_to_table as import_parquet_to_table_func,
+)
 from datetime import datetime, timezone
 from mcp.server.fastmcp import FastMCP
 from pydantic import Field
@@ -567,32 +572,75 @@ async def query_database(
 
 @app.tool()
 @log_tool_call_with_response
-async def preview_csv_file(
+@write_operation
+async def import_csv_to_table(
+    warehouse: Annotated[str, Field(..., description='Warehouse string for Iceberg catalog')],
+    region: Annotated[
+        str, Field(..., description='AWS region for S3Tables/Iceberg REST endpoint')
+    ],
+    namespace: Annotated[str, NAMESPACE_NAME_FIELD],
+    table_name: Annotated[str, TABLE_NAME_FIELD],
     s3_url: Annotated[str, S3_URL_FIELD],
+    uri: Annotated[str, Field(..., description='REST URI for Iceberg catalog')],
+    catalog_name: Annotated[
+        str, Field('s3tablescatalog', description='Catalog name')
+    ] = 's3tablescatalog',
+    rest_signing_name: Annotated[
+        str, Field('s3tables', description='REST signing name')
+    ] = 's3tables',
+    rest_sigv4_enabled: Annotated[str, Field('true', description='Enable SigV4 signing')] = 'true',
 ) -> dict:
-    """Preview the structure of a CSV file stored in S3.
+    """Import data from a CSV file into an S3 table.
 
-    This tool provides a quick preview of a CSV file's structure by reading
-    only the headers and first row of data from an S3 location. It's useful for
-    understanding the schema and data format without downloading the entire file.
-    It can be used before creating an s3 table from a csv file to get the schema and data format.
+    This tool reads data from a CSV file stored in S3 and imports it into an S3 table.
+    If the table doesn't exist, it will be created with a schema inferred from the CSV file.
+    If the table exists, the CSV file schema must be compatible with the table's schema.
+    The tool will validate the schema before attempting to import the data.
 
     Returns error dictionary with status and error message if:
         - URL is not a valid S3 URL
         - File is not a CSV file
         - File cannot be accessed
+        - Table does not exist
+        - CSV headers don't match table schema
         - Any other error occurs
 
+    Example input values:
+        warehouse: 'arn:aws:s3tables:<Region>:<accountID>:bucket/<bucketname>'
+        region: 'us-west-2'
+        namespace: 'retail_data'
+        table_name: 'customers'
+        s3_url: 's3://bucket-name/path/to/file.csv'
+        uri: 'https://s3tables.us-west-2.amazonaws.com/iceberg'
+        catalog_name: 's3tablescatalog'
+        rest_signing_name: 's3tables'
+        rest_sigv4_enabled: 'true'
+
     Permissions:
-    You must have the s3:GetObject permission for the S3 bucket and key.
+    You must have:
+    - s3:GetObject permission for the CSV file
+    - s3tables:GetTable and s3tables:GetTables permissions to access table information
+    - s3tables:PutTableData permission to write to the table
     """
-    return file_processor.preview_csv_structure(s3_url)
+    if uri is None:
+        uri = _default_uri_for_region(region)
+    return await import_csv_to_table_func(
+        warehouse=warehouse,
+        region=region,
+        namespace=namespace,
+        table_name=table_name,
+        s3_url=s3_url,
+        uri=uri,
+        catalog_name=catalog_name,
+        rest_signing_name=rest_signing_name,
+        rest_sigv4_enabled=rest_sigv4_enabled,
+    )
 
 
 @app.tool()
 @log_tool_call_with_response
 @write_operation
-async def import_csv_to_table(
+async def import_parquet_to_table(
     warehouse: Annotated[str, Field(..., description='Warehouse string for Iceberg catalog')],
     region: Annotated[
         str, Field(..., description='AWS region for S3Tables/Iceberg REST endpoint')
@@ -609,29 +657,33 @@ async def import_csv_to_table(
     ] = 's3tables',
     rest_sigv4_enabled: Annotated[str, Field('true', description='Enable SigV4 signing')] = 'true',
 ) -> dict:
-    """Import data from a CSV file into an S3 table.
+    """Import data from a Parquet file into an S3 table.
 
-    This tool reads data from a CSV file stored in S3 and imports it into an existing S3 table.
-    The CSV file must have headers that match the table's schema. The tool will validate the CSV structure
-    before attempting to import the data.
-
-    To create a table, first use the preview_csv_file tool to get the schema and data format.
-    Then use the create_table tool to create the table.
+    This tool reads data from a Parquet file stored in S3 and imports it into an S3 table.
+    If the table doesn't exist, it will be created with a schema inferred from the Parquet file.
+    If the table exists, the Parquet file schema must be compatible with the table's schema.
+    The tool will validate the schema before attempting to import the data.
 
     Returns error dictionary with status and error message if:
         - URL is not a valid S3 URL
-        - File is not a CSV file
+        - File is not a Parquet file
         - File cannot be accessed
-        - Table does not exist
-        - CSV headers don't match table schema
+        - Parquet schema is incompatible with existing table schema
         - Any other error occurs
 
+    Returns success dictionary with:
+        - status: 'success'
+        - message: Success message with row count
+        - rows_processed: Number of rows imported
+        - file_processed: Name of the processed file
+        - table_created: True if a new table was created
+
     Example input values:
         warehouse: 'arn:aws:s3tables:<Region>:<accountID>:bucket/<bucketname>'
         region: 'us-west-2'
         namespace: 'retail_data'
         table_name: 'customers'
-        s3_url: 's3://bucket-name/path/to/file.csv'
+        s3_url: 's3://bucket-name/path/to/file.parquet'
         uri: 'https://s3tables.us-west-2.amazonaws.com/iceberg'
         catalog_name: 's3tablescatalog'
         rest_signing_name: 's3tables'
@@ -639,14 +691,14 @@ async def import_csv_to_table(
 
     Permissions:
     You must have:
-    - s3:GetObject permission for the CSV file
-    - s3tables:GetDatabase and s3tables:GetDatabases permissions to access database information
+    - s3:GetObject permission for the Parquet file
     - s3tables:GetTable and s3tables:GetTables permissions to access table information
     - s3tables:PutTableData permission to write to the table
+    - s3tables:CreateTable permission (if table doesn't exist)
     """
     if uri is None:
         uri = _default_uri_for_region(region)
-    return await file_processor.import_csv_to_table(
+    return await import_parquet_to_table_func(
         warehouse=warehouse,
         region=region,
         namespace=namespace,

{awslabs_s3_tables_mcp_server-0.0.2 → awslabs_s3_tables_mcp_server-0.0.4}/pyproject.toml

@@ -2,7 +2,7 @@
 name = "awslabs.s3-tables-mcp-server"
 
 # NOTE: "Patch"=9223372036854775807 bumps next release to zero.
-version = "0.0.2"
+version = "0.0.4"
 
 description = "An AWS Labs Model Context Protocol (MCP) server for awslabs.s3-tables-mcp-server"
 readme = "README.md"