tasks/sql_generation/nyc_taxi_small/schema.sql

-- NYC Taxi Small Dataset Schema
-- This is a simplified version of the NYC taxi dataset for testing

CREATE TABLE trips (
    trip_id INTEGER,
    pickup_datetime TIMESTAMP,
    dropoff_datetime TIMESTAMP,
    passenger_count INTEGER,
    trip_distance DOUBLE,
    pickup_longitude DOUBLE,
    pickup_latitude DOUBLE,
    dropoff_longitude DOUBLE,
    dropoff_latitude DOUBLE,
    fare_amount DOUBLE,
    tip_amount DOUBLE,
    total_amount DOUBLE,
    payment_type VARCHAR(10),
    vendor_id VARCHAR(10)
);

CREATE TABLE zones (
    zone_id INTEGER,
    borough VARCHAR(50),
    zone_name VARCHAR(100),
    service_zone VARCHAR(50)
);