@aurios/mizzle 1.1.3 → 1.1.5

package/.turbo/turbo-build.log

@@ -1,5 +1,5 @@
 
-> @aurios/mizzle@1.1.3 build /home/runner/work/mizzle/mizzle/packages/mizzle
+> @aurios/mizzle@1.1.5 build /home/runner/work/mizzle/mizzle/packages/mizzle
 > tsup
 
 CLI Building entry: {"index":"src/index.ts","columns":"src/columns/index.ts","table":"src/core/table.ts","snapshot":"src/core/snapshot.ts","diff":"src/core/diff.ts","introspection":"src/core/introspection.ts","db":"src/db.ts"}
@@ -10,22 +10,22 @@
 CLI Cleaning output folder
 ESM Build start
 ESM dist/index.js                2.32 KB
+ESM dist/columns.js              299.00 B
 ESM dist/table.js                144.00 B
 ESM dist/snapshot.js             193.00 B
-ESM dist/chunk-TOYV2M4M.js       866.00 B
 ESM dist/diff.js                 147.00 B
+ESM dist/chunk-TOYV2M4M.js       866.00 B
 ESM dist/chunk-AQVECMXP.js       2.38 KB
 ESM dist/introspection.js        1.47 KB
 ESM dist/db.js                   166.00 B
-ESM dist/chunk-NPPZW6VT.js       9.59 KB
+ESM dist/chunk-WTOBEKFQ.js       9.74 KB
 ESM dist/transaction-RE7LXTGV.js 193.00 B
 ESM dist/chunk-UM3YF5EC.js       15.21 KB
 ESM dist/chunk-GPYZK4WY.js       1.12 KB
 ESM dist/chunk-DU7UPWBW.js       5.17 KB
-ESM dist/columns.js              299.00 B
-ESM ⚡️ Build success in 50ms
+ESM ⚡️ Build success in 76ms
 DTS Build start
-DTS ⚡️ Build success in 5003ms
+DTS ⚡️ Build success in 6614ms
 DTS dist/index.d.ts              3.84 KB
 DTS dist/diff.d.ts               485.00 B
 DTS dist/introspection.d.ts      258.00 B
@@ -33,5 +33,5 @@
 DTS dist/db.d.ts                 169.00 B
 DTS dist/columns.d.ts            250.00 B
 DTS dist/table.d.ts              403.00 B
-DTS dist/db-zHIHBm1E.d.ts        31.72 KB
+DTS dist/db-ZX6HGN1x.d.ts        31.83 KB
 DTS dist/operators-BVreW0ky.d.ts 29.24 KB

package/README.md

@@ -1,8 +1,24 @@
-# @aurios/mizzle
+# 🌧️ mizzle
 
+![NPM Last Update](https://img.shields.io/npm/last-update/%40aurios%2Fmizzle?style=flat&color=0EA5E9)
+![GitHub Actions Workflow Status](https://img.shields.io/github/actions/workflow/status/realfakenerd/mizzle/release.yml?style=flat&color=0EA5E9)
+![GitHub License](https://img.shields.io/github/license/realfakenerd/mizzle?style=flat&color=0EA5E9)
+
+![NPM Downloads](https://img.shields.io/npm/dw/%40aurios%2Fmizzle?style=flat&color=0EA5E9)
 A Drizzle-like ORM for DynamoDB. Mizzle provides a type-safe, fluent API for interacting with DynamoDB, supporting relational queries, batch operations, and transactions.
 
-## Installation
+## Key Features
+
+- **Type-Safe Schema Definition**: Define your tables and entities with strict TypeScript types.
+- **Fluent Query Builder**: precise API for `insert`, `select`, `update`, and `delete` operations.
+- **Relational Queries**: Query related entities with `db.query`.
+- **Batch Operations**: `batchGet` and `batchWrite` support.
+- **Transactions**: Atomic operations using `db.transaction`.
+- **Automatic Type Inference**: `InferSelectModel` and `InferInsertModel` utilities.
+
+> Skip this with the [way less boring docs](https://mizzle-docs.vercel.app)
+
+## 🚀 Installation
 
 ```bash
 npm install @aurios/mizzle
@@ -10,48 +26,141 @@ npm install @aurios/mizzle
 bun add @aurios/mizzle
 ```
 
-## Usage
+## Get Started
+
+### Defining the Table
+
+In DynamoDB, you first need a physical table. Mizzle separates the definition of the physical table structure (PK, SK, Indexes) from the logical entities that live within it and with this making your database well organized and easier to reason about.
+
+```ts
+import { dynamoTable, string } from "@aurios/mizzle";
+
+// This matches your actual DynamoDB table configuration
+export const myTable = dynamoTable("JediOrder", {
+  pk: string("pk"), // The partition key attribute name
+  sk: string("sk"), // The sort key attribute name (optional)
+});
+```
+
+### Defining the Entity
+
+An Entity represents your data model (e.g., an user, an item on an user). You map the Entity to a Physical Table and define how its keys are generated, so every entity looks kinda like an separated table.
+
+```ts
+import { dynamoEntity, string, uuid, number, enum, date, prefixKey, staticKey } from "@aurios/mizzle";
+
+export const jedi = dynamoEntity(
+  myTable,
+  "Jedi",
+  {
+    id: uuid(), // Automatically generates a v7 UUID
+    name: string(),
+    homeworld: string()
+  },
+  (cols) => ({
+    // PK will look like "JEDI#<uuid>"
+    pk: prefixKey("JEDI#", cols.id),
+    // SK will be a static string "PROFILE"
+    sk: staticKey("PROFILE"),
+  }),
+);
+
+export const jediRank = dynamoEntity(
+  myTable,
+  'JediRank',
+  {
+    jediId: uuid(),
+    position: string().default('initiate'),
+    joinedCouncilDate: string(),
+  },
+  (cols) => ({
+    // Same as the jedi so they can be related
+    pk: prefixKey('JEDI#', cols.jediId),
+    // RANK#initiate
+    sk: prefixKey('RANK#', cols.position),
+  })
+)
+```
+
+> The UUID V7 was chosen for better sorting, since the values are time-sortable with 1 millisecond precision.
+
+### Define the Relations
+
+Relationships in Mizzle are established using the defineRelations function. This step creates a logical map of how your entities interact, enabling you to perform powerful relational queries—such as fetching a Jedi along with their entire rank history—in a single operation.
+
+```ts
+import { defineRelations } from "@aurios/mizzle";
+import * as schema from "./schema";
+
+export const relations = defineRelations(schema, (r) => ({
+  jedi: {
+    // A Jedi could've a lot of ranks throughout his year
+    ranks: r.many.jediRank({
+      fields: [r.jedi.id],
+      references: [r.jediRank.jediId],
+    }),
+  },
+  jediRank: {
+    // Every registry points to one Jedi only
+    member: r.one.jedi({
+      fields: [r.jediRank.jediId],
+      references: [r.jedi.id],
+    }),
+  },
+}));
+```
 
 ### Initialization
 
+Initialize the mizzle client by passing it an instance of the standard AWS DynamoDBClient and the relations we just defined.
+
 ```ts
 import { DynamoDBClient } from "@aws-sdk/client-dynamodb";
 import { mizzle } from "@aurios/mizzle";
+import { relations } from "./relations";
 
-const client = new DynamoDBClient({});
-const db = mizzle(client);
+const client = new DynamoDBClient({ region: "us-east-1" });
+export const db = mizzle({ client, relations });
 ```
 
-### Key Features
+### Query
 
-- **Type-Safe Schema Definition**: Define your tables and entities with strict TypeScript types.
-- **Fluent Query Builder**: precise API for `insert`, `select`, `update`, and `delete` operations.
-- **Relational Queries**: Query related entities with `db.query`.
-- **Batch Operations**: `batchGet` and `batchWrite` support.
-- **Transactions**: Atomic operations using `db.transaction`.
-- **Automatic Type Inference**: `InferSelectModel` and `InferInsertModel` utilities.
+Now you can use the fluent API to interact with your data in the best way possible.
 
-### Example
+#### Insert Data
 
-```ts
-// Define your schema (simplified)
-import { dynamoTable, dynamoEntity, string } from "@aurios/mizzle";
+```typescript
+// api/jedi/new.ts
+import { jedi } from "$lib/schema.ts";
 
-const myTable = dynamoTable("my-app-table", {
-  pk: string("pk"),
-  sk: string("sk"),
-});
+const newJedi = await db
+  .insert(jedi)
+  .values({
+    name: "Luke Skywalker",
+    homeworld: "Tatooine",
+  })
+  .returning();
 
-const users = dynamoEntity(myTable, "users", {
-  id: string("id"),
-  name: string("name"),
-  email: string("email"),
-});
+console.log(newJedi.id); // The auto-generated UUID
+```
 
-// Query
-const result = await db.select().from(users).where(eq(users.id, "123"));
+#### Select Data
+
+Mizzle intelligently routes your request to `GetItem`, `Query`, or `Scan` based on the filters you provide.
+
+```typescript
+// /api/jedi/get.ts
+import { jedi } from "$lib/schema.ts";
+import { eq } from "@aurios/mizzle";
+
+// This will use GetItem because both PK and SK are fully resolved
+const user = await db.select().from(jedi).where(eq(jedi.id, "some-uuid")).execute();
 ```
 
-## License
+## Mizzling
+
+This will work if you already has a DynamoDB table with data in it. If you want to create a new table with the schema you defined, you can use the [`mizzling`](https://www.npmjs.com/package/@aurios/mizzling) CLI.
+
+## 📄 License
 
-MIT
+This project is licensed under the MIT License.

package/dist/{chunk-NPPZW6VT.js → chunk-WTOBEKFQ.js}

@@ -1 +1 @@
-import{D as Q,E as U,I as L,J as w,K as H,L as $,N as J,P as X,Q as Z,S as ee,e as P,p as q,q as G,r as z}from"./chunk-UM3YF5EC.js";import{b as D}from"./chunk-GPYZK4WY.js";import{d as E,e as d,f as k,i as v}from"./chunk-DU7UPWBW.js";import{DynamoDBClient as ne}from"@aws-sdk/client-dynamodb";import{DynamoDBDocumentClient as le}from"@aws-sdk/lib-dynamodb";import{GetCommand as ie,QueryCommand as se,ScanCommand as oe}from"@aws-sdk/lib-dynamodb";var O=class{constructor(t,e){this.client=t;this.fields=e}from(t){return new I(t,this.client,this.fields)}},I=class extends w{constructor(e,i,n){super(e,i);this.fields=n}static[d.ENTITY_KIND]="SelectBase";_whereClause;_limitVal;_pageSizeVal;_consistentReadVal;_sortForward=!0;_forcedIndexName;where(e){return this._whereClause=e,this}limit(e){return this._limitVal=e,this}pageSize(e){return this._pageSizeVal=e,this}consistentRead(e=!0){return this._consistentReadVal=e,this}sort(e){return this._sortForward=e,this}index(e){return this._forcedIndexName=e,this}iterator(){return async function*(){let e,i=0;do{let n=await this.fetchPage(e);for(let s of n.items)if(yield s,i++,this._limitVal!==void 0&&i>=this._limitVal)return;e=n.lastEvaluatedKey}while(e)}.bind(this)()}async fetchPage(e){let i=this.resolveKeys(this._whereClause,void 0,this._forcedIndexName);return i.hasPartitionKey&&i.hasSortKey&&!i.indexName&&!e?{items:await this.executeGet(i.keys)}:i.hasPartitionKey||i.indexName?await this.executeQuery(i,e):await this.executeScan(e)}async execute(){let{items:e}=await this.fetchPage();return e}async executeGet(e){let i=new ie({TableName:this.tableName,Key:e,ConsistentRead:this._consistentReadVal}),n=await this.client.send(i);return n.Item?[this.mapToLogical(n.Item)]:[]}async executeQuery(e,i){let{expressionAttributeNames:n,expressionAttributeValues:s,addName:r,addValue:o}=this.createExpressionContext(),a=[],c=new Set;for(let[p,_]of Object.entries(e.keys))a.push(`${r(p)} = ${o(_)}`),c.add(p);let m=a.join(" AND "),u=this._whereClause?z(this._whereClause,r,o,c):void 0,f=new se({TableName:this.tableName,IndexName:e.indexName,KeyConditionExpression:m,FilterExpression:u||void 0,ExpressionAttributeNames:Object.keys(n).length>0?n:void 0,ExpressionAttributeValues:Object.keys(s).length>0?s:void 0,Limit:this._pageSizeVal??this._limitVal,ScanIndexForward:this._sortForward,ConsistentRead:e.indexName?void 0:this._consistentReadVal,ExclusiveStartKey:i}),T=await this.client.send(f);return{items:(T.Items||[]).map(p=>this.mapToLogical(p)),lastEvaluatedKey:T.LastEvaluatedKey}}async executeScan(e){let{expressionAttributeNames:i,expressionAttributeValues:n,addName:s,addValue:r}=this.createExpressionContext(),o=this._whereClause?z(this._whereClause,s,r):void 0,a=new oe({TableName:this.tableName,FilterExpression:o,ExpressionAttributeNames:Object.keys(i).length>0?i:void 0,ExpressionAttributeValues:Object.keys(n).length>0?n:void 0,Limit:this._pageSizeVal??this._limitVal,ConsistentRead:this._consistentReadVal,ExclusiveStartKey:e}),c=await this.client.send(a);return{items:(c.Items||[]).map(m=>this.mapToLogical(m)),lastEvaluatedKey:c.LastEvaluatedKey}}};var N=class{constructor(t){this.schema=t}parse(t,e,i={}){let n=this.schema.entities[e];if(!n)throw new Error(`Root entity ${e} not found in schema`);let s=[],r=[];for(let o of t)this.isEntity(o,n.entity)?s.push({...o}):r.push(o);for(let o of s)for(let[a,c]of Object.entries(i)){if(!c)continue;let m=n.relations[a];if(!m)continue;let u=m.config.to,f=r.filter(T=>this.isEntity(T,u));f.length>0&&(m.type==="many"?o[a]=f:m.type==="one"&&(o[a]=f[0]||null))}return s}isEntity(t,e){let i=e[d.ENTITY_STRATEGY],n=e[d.PHYSICAL_TABLE];if(!n||!n[E.PARTITION_KEY])return!1;let s=n[E.PARTITION_KEY].name,r=n[E.SORT_KEY]?.name,o=this.matchStrategy(t[s],i.pk),a=r?this.matchStrategy(t[r],i.sk):!0;return o&&a}matchStrategy(t,e){if(!e)return!0;if(t==null)return!1;let i=String(t);if(e.type==="static")return i===e.segments[0];if(e.type==="prefix"||e.type==="composite"){let n=e.segments[0];return typeof n=="string"&&i.startsWith(n)}return!1}};var b=class l{constructor(t,e,i,n){this.client=t;this.table=e;this.schema=i;this.entityName=n}async findFirst(t={}){return(await this.findMany({...t,limit:1}))[0]}async findMany(t={}){let e=new I(this.table,this.client,void 0);t.orderBy&&e.sort(t.orderBy==="asc");let i;if(t.where){let r=this.table._?.columns||this.table;typeof t.where=="function"?i=t.where(r,G):i=t.where}let n,s=L(this.table,i);if(this.schema&&this.entityName&&(t.with||t.include)&&s.hasPartitionKey&&!s.indexName){let r=this.table[d.PHYSICAL_TABLE];if(!r)throw new Error("Physical table not found for entity");let o=r[E.PARTITION_KEY].name,a=s.keys[o];e.where(P({name:o},a));let m=(await e).map(f=>v(this.table,f));n=new N(this.schema).parse(m,this.entityName,t.with||t.include),t.limit&&(n=n.slice(0,t.limit))}else t.limit&&e.limit(t.limit),i&&e.where(i),n=await e;if(this.schema&&this.entityName&&(t.with||t.include)){let r=t.with||t.include;await Promise.all(n.map(async o=>{for(let[a,c]of Object.entries(r)){let m=this.schema.entities[this.entityName];if(!m)continue;let u=m.relations[a];if(!u)continue;let f=o[a]!==void 0,T=typeof c=="object"&&(c.with||c.include);if(f&&!T)continue;let p=u.config.to,_=Object.entries(this.schema.entities).find(([R,y])=>y.entity===p)?.[0],V=v(this.table,o);if(u.config.fields&&u.config.references){let R={};u.config.fields.forEach((y,g)=>{let h=u.config.references[g];if(!h)return;let F=Object.entries(p[d.COLUMNS]).find(([K,C])=>C===h||C.name===h.name)?.[0];if(F){let K=Object.entries(this.table[d.COLUMNS]).find(([de,W])=>W===y||W.name===y.name)?.[0],C=K?o[K]??o[y.name]:o[y.name];C!==void 0&&(R[F]=C)}}),V=R}let j=L(p,void 0,V);if(j.hasPartitionKey){let R=new l(this.client,p,this.schema,_),y=[];for(let[g,h]of Object.entries(j.keys))y.push(P({name:g},h));if(y.length>0){let g=y.length===1?y[0]:q(...y),h=typeof c=="object"?c:{};u.type==="one"?o[a]=await R.findFirst({...h,where:g}):o[a]=await R.findMany({...h,where:g})}}}}))}return n}};import{BatchGetCommand as re}from"@aws-sdk/lib-dynamodb";var M=class{constructor(t){this.client=t}items(t,e){return new A(t,this.client,e)}},A=class extends w{constructor(e,i,n){super(e,i);this.keysData=n}static[d.ENTITY_KIND]="BatchGetBase";async execute(){let e=[],i=[],n=this.keysData.map(o=>this.resolveKeys(void 0,o).keys),s=0,r=5;for(;n.length>0&&s<r;){s++;let o=new re({RequestItems:{[this.tableName]:{Keys:n}}}),a=await this.client.send(o);a.Responses?.[this.tableName]&&e.push(...a.Responses[this.tableName]);let c=a.UnprocessedKeys?.[this.tableName]?.Keys;c&&c.length>0?n=c:n=[]}return n.length>0&&i.push(...n),{succeeded:e,failed:i}}};import{BatchWriteCommand as ae}from"@aws-sdk/lib-dynamodb";var B=class{constructor(t){this.client=t}operations(t,e){return new Y(t,this.client,e)}},Y=class extends w{constructor(e,i,n){super(e,i);this.ops=n}static[d.ENTITY_KIND]="BatchWriteBase";async execute(){let e=0,i=[],s=[...this.ops.map(a=>{if(a.type==="put"){let c=a.item;if(H(c)>400*1024)throw new $("Item in batch exceeds the 400KB limit.");return{PutRequest:{Item:c}}}else return{DeleteRequest:{Key:a.keys}}})],r=0,o=5;for(;s.length>0&&r<o;){r++;let a=new ae({RequestItems:{[this.tableName]:s}}),m=(await this.client.send(a)).UnprocessedItems?.[this.tableName]||[];e+=s.length-m.length,m.length>0?s=m:s=[]}return s.length>0&&(i=s),{succeededCount:e,failed:i}}};var x=class{constructor(t,e){this.type=t;this.config=e}};function st(l,t){if(l instanceof D){let i=t({one:(n,s)=>new x("one",{to:n,...s}),many:(n,s)=>new x("many",{to:n,...s})});return{entity:l,config:i,[k.RELATION_CONFIG]:!0}}else{let e=l,i=t,n={one:{},many:{}};for(let[r,o]of Object.entries(e))n.one[r]=a=>new x("one",{to:o,...a}),n.many[r]=a=>new x("many",{to:o,...a}),n[r]=o;let s=i(n);return{schema:e,definitions:s,[k.RELATION_CONFIG]:!0}}}function te(l){let t={entities:{}};for(let[e,i]of Object.entries(l))i instanceof D&&(t.entities[e]={entity:i,relations:{}});for(let[,e]of Object.entries(l))if(e&&typeof e=="object"&&e[k.RELATION_CONFIG]){let i=e;if(i.entity){let n=e,s=Object.entries(t.entities).find(([r,o])=>o.entity===n.entity);if(s){let[,r]=s;r.relations={...r.relations,...n.config}}}else if(i.definitions){let n=e;for(let[s,r]of Object.entries(n.definitions)){let o=n.schema[s],a=Object.entries(t.entities).find(([c,m])=>m.entity===o);if(a&&r){let[,c]=a;c.relations={...c.relations,...r}}}}}return t}var S=class{docClient;schema;retryConfig;query;constructor(t,e,i){this.retryConfig={maxAttempts:i?.maxAttempts??3,baseDelay:i?.baseDelay??100},this.docClient=new U(le.from(t),new Q(this.retryConfig)),e&&(this.schema=te(e)),this.query=new Proxy({},{get:(n,s)=>{if(typeof s!="string")return;if(!this.schema)throw new Error("No relations defined. Initialize mizzle with a relations object to use db.query.");let r=this.schema.entities[s];if(!r)throw new Error(`Entity ${s} not found in relations schema.`);return new b(this.docClient,r.entity,this.schema,s)}})}insert(t){return new J(t,this.docClient)}select(t){return new O(this.docClient,t)}batchGet(t,e){return new M(this.docClient).items(t,e)}batchWrite(t,e){return new B(this.docClient).operations(t,e)}_query(t){return new b(this.docClient,t)}update(t){return new X(t,this.docClient)}delete(t,e){return new Z(t,this.docClient,e)}async transaction(t,e){let i=new ee(this.docClient),n=await e(i);if(n.length===0)return;if(n.length>100)throw new Error("DynamoDB transactions are limited to 100 items.");let{TransactionExecutor:s}=await import("./transaction-RE7LXTGV.js");await new s(this.docClient).execute(t,n)}};function gt(l){return l instanceof ne?new S(l):"client"in l&&l.client instanceof ne?new S(l.client,l.relations,l.retry):"client"in l&&l.client?new S(l.client,l.relations,l.retry):new S(l)}export{O as a,I as b,M as c,A as d,B as e,Y as f,b as g,x as h,st as i,te as j,S as k,gt as l};
+import{D as Q,E as U,I as L,J as w,K as H,L as $,N as J,P as X,Q as Z,S as ee,e as P,p as q,q as G,r as z}from"./chunk-UM3YF5EC.js";import{b as D}from"./chunk-GPYZK4WY.js";import{d as E,e as d,f as C,i as v}from"./chunk-DU7UPWBW.js";import{DynamoDBClient as ne}from"@aws-sdk/client-dynamodb";import{DynamoDBDocumentClient as le}from"@aws-sdk/lib-dynamodb";import{GetCommand as ie,QueryCommand as se,ScanCommand as oe}from"@aws-sdk/lib-dynamodb";var O=class{constructor(t,e){this.client=t;this.fields=e}from(t){return new b(t,this.client,this.fields)}},b=class extends w{constructor(e,i,n){super(e,i);this.fields=n}static[d.ENTITY_KIND]="SelectBase";_whereClause;_limitVal;_pageSizeVal;_consistentReadVal;_sortForward=!0;_forcedIndexName;where(e){return this._whereClause=e,this}limit(e){return this._limitVal=e,this}pageSize(e){return this._pageSizeVal=e,this}consistentRead(e=!0){return this._consistentReadVal=e,this}sort(e){return this._sortForward=e,this}index(e){return this._forcedIndexName=e,this}iterator(){return async function*(){let e,i=0;do{let n=await this.fetchPage(e);for(let s of n.items)if(yield s,i++,this._limitVal!==void 0&&i>=this._limitVal)return;e=n.lastEvaluatedKey}while(e)}.bind(this)()}async fetchPage(e){let i=this.resolveKeys(this._whereClause,void 0,this._forcedIndexName);return i.hasPartitionKey&&i.hasSortKey&&!i.indexName&&!e?{items:await this.executeGet(i.keys)}:i.hasPartitionKey||i.indexName?await this.executeQuery(i,e):await this.executeScan(e)}async execute(){let{items:e}=await this.fetchPage();return e}async executeGet(e){let i=new ie({TableName:this.tableName,Key:e,ConsistentRead:this._consistentReadVal}),n=await this.client.send(i);return n.Item?[this.mapToLogical(n.Item)]:[]}async executeQuery(e,i){let{expressionAttributeNames:n,expressionAttributeValues:s,addName:r,addValue:o}=this.createExpressionContext(),l=[],c=new Set;for(let[p,_]of Object.entries(e.keys))l.push(`${r(p)} = ${o(_)}`),c.add(p);let m=l.join(" AND "),u=this._whereClause?z(this._whereClause,r,o,c):void 0,f=new se({TableName:this.tableName,IndexName:e.indexName,KeyConditionExpression:m,FilterExpression:u||void 0,ExpressionAttributeNames:Object.keys(n).length>0?n:void 0,ExpressionAttributeValues:Object.keys(s).length>0?s:void 0,Limit:this._pageSizeVal??this._limitVal,ScanIndexForward:this._sortForward,ConsistentRead:e.indexName?void 0:this._consistentReadVal,ExclusiveStartKey:i}),T=await this.client.send(f);return{items:(T.Items||[]).map(p=>this.mapToLogical(p)),lastEvaluatedKey:T.LastEvaluatedKey}}async executeScan(e){let{expressionAttributeNames:i,expressionAttributeValues:n,addName:s,addValue:r}=this.createExpressionContext(),o=this._whereClause?z(this._whereClause,s,r):void 0,l=new oe({TableName:this.tableName,FilterExpression:o,ExpressionAttributeNames:Object.keys(i).length>0?i:void 0,ExpressionAttributeValues:Object.keys(n).length>0?n:void 0,Limit:this._pageSizeVal??this._limitVal,ConsistentRead:this._consistentReadVal,ExclusiveStartKey:e}),c=await this.client.send(l);return{items:(c.Items||[]).map(m=>this.mapToLogical(m)),lastEvaluatedKey:c.LastEvaluatedKey}}};var M=class{constructor(t){this.schema=t}parse(t,e,i={}){let n=this.schema.entities[e];if(!n)throw new Error(`Root entity ${e} not found in schema`);let s=[],r=[];for(let o of t)this.isEntity(o,n.entity)?s.push({...o}):r.push(o);for(let o of s)for(let[l,c]of Object.entries(i)){if(!c)continue;let m=n.relations[l];if(!m)continue;let u=m.config.to,f=r.filter(T=>this.isEntity(T,u));f.length>0&&(m.type==="many"?o[l]=f:m.type==="one"&&(o[l]=f[0]||null))}return s}isEntity(t,e){let i=e[d.ENTITY_STRATEGY],n=e[d.PHYSICAL_TABLE];if(!n||!n[E.PARTITION_KEY])return!1;let s=n[E.PARTITION_KEY].name,r=n[E.SORT_KEY]?.name,o=this.matchStrategy(t[s],i.pk),l=r?this.matchStrategy(t[r],i.sk):!0;return o&&l}matchStrategy(t,e){if(!e)return!0;if(t==null)return!1;let i=String(t);if(e.type==="static")return i===e.segments[0];if(e.type==="prefix"||e.type==="composite"){let n=e.segments[0];return typeof n=="string"&&i.startsWith(n)}return!1}};var k=class a{constructor(t,e,i,n){this.client=t;this.table=e;this.schema=i;this.entityName=n}async findFirst(t={}){return(await this.findMany({...t,limit:1}))[0]}async findMany(t={}){let e=new b(this.table,this.client,void 0);t.orderBy&&e.sort(t.orderBy==="asc");let i;if(t.where){let r=this.table._?.columns||this.table;typeof t.where=="function"?i=t.where(r,G):i=t.where}let n,s=L(this.table,i);if(this.schema&&this.entityName&&(t.with||t.include)&&s.hasPartitionKey&&!s.indexName){let r=this.table[d.PHYSICAL_TABLE];if(!r)throw new Error("Physical table not found for entity");let o=r[E.PARTITION_KEY].name,l=s.keys[o];e.where(P({name:o},l));let m=(await e).map(f=>v(this.table,f));n=new M(this.schema).parse(m,this.entityName,t.with||t.include),t.limit&&(n=n.slice(0,t.limit))}else t.limit&&e.limit(t.limit),i&&e.where(i),n=await e;if(this.schema&&this.entityName&&(t.with||t.include)){let r=t.with||t.include;await Promise.all(n.map(async o=>{for(let[l,c]of Object.entries(r)){let m=this.schema.entities[this.entityName];if(!m)continue;let u=m.relations[l];if(!u)continue;let f=o[l]!==void 0,T=typeof c=="object"&&(c.with||c.include);if(f&&!T)continue;let p=u.config.to,_=Object.entries(this.schema.entities).find(([R,y])=>y.entity===p)?.[0],V=v(this.table,o);if(u.config.fields&&u.config.references){let R={};u.config.fields.forEach((y,g)=>{let h=u.config.references[g];if(!h)return;let F=Object.entries(p[d.COLUMNS]).find(([K,I])=>I===h||I.name===h.name)?.[0];if(F){let K=Object.entries(this.table[d.COLUMNS]).find(([de,W])=>W===y||W.name===y.name)?.[0],I=K?o[K]??o[y.name]:o[y.name];I!==void 0&&(R[F]=I)}}),V=R}let j=L(p,void 0,V);if(j.hasPartitionKey){let R=new a(this.client,p,this.schema,_),y=[];for(let[g,h]of Object.entries(j.keys))y.push(P({name:g},h));if(y.length>0){let g=y.length===1?y[0]:q(...y),h=typeof c=="object"?c:{};u.type==="one"?o[l]=await R.findFirst({...h,where:g}):o[l]=await R.findMany({...h,where:g})}}}}))}return n}};import{BatchGetCommand as re}from"@aws-sdk/lib-dynamodb";var N=class{constructor(t){this.client=t}items(t,e){return new A(t,this.client,e)}},A=class extends w{constructor(e,i,n){super(e,i);this.keysData=n}static[d.ENTITY_KIND]="BatchGetBase";async execute(){let e=[],i=[],n=this.keysData.map(o=>this.resolveKeys(void 0,o).keys),s=0,r=5;for(;n.length>0&&s<r;){s++;let o=new re({RequestItems:{[this.tableName]:{Keys:n}}}),l=await this.client.send(o);l.Responses?.[this.tableName]&&e.push(...l.Responses[this.tableName]);let c=l.UnprocessedKeys?.[this.tableName]?.Keys;c&&c.length>0?n=c:n=[]}return n.length>0&&i.push(...n),{succeeded:e,failed:i}}};import{BatchWriteCommand as ae}from"@aws-sdk/lib-dynamodb";var B=class{constructor(t){this.client=t}operations(t,e){return new Y(t,this.client,e)}},Y=class extends w{constructor(e,i,n){super(e,i);this.ops=n}static[d.ENTITY_KIND]="BatchWriteBase";async execute(){let e=0,i=[],s=[...this.ops.map(l=>{if(l.type==="put"){let c=l.item;if(H(c)>400*1024)throw new $("Item in batch exceeds the 400KB limit.");return{PutRequest:{Item:c}}}else return{DeleteRequest:{Key:l.keys}}})],r=0,o=5;for(;s.length>0&&r<o;){r++;let l=new ae({RequestItems:{[this.tableName]:s}}),m=(await this.client.send(l)).UnprocessedItems?.[this.tableName]||[];e+=s.length-m.length,m.length>0?s=m:s=[]}return s.length>0&&(i=s),{succeededCount:e,failed:i}}};var x=class{constructor(t,e){this.type=t;this.config=e}};function st(a,t){if(a instanceof D){let i=t({one:(n,s)=>new x("one",{to:n,...s}),many:(n,s)=>new x("many",{to:n,...s})});return{entity:a,config:i,[C.RELATION_CONFIG]:!0}}else{let e=a,i=t,n={one:{},many:{}};for(let[r,o]of Object.entries(e))n.one[r]=l=>new x("one",{to:o,...l}),n.many[r]=l=>new x("many",{to:o,...l}),n[r]=o;let s=i(n);return{schema:e,definitions:s,[C.RELATION_CONFIG]:!0}}}function te(a){let t={entities:{}};if(a&&a[C.RELATION_CONFIG]&&a.definitions){let e=a;for(let[i,n]of Object.entries(e.schema))t.entities[i]={entity:n,relations:e.definitions[i]||{}};return t}for(let[e,i]of Object.entries(a))i instanceof D&&(t.entities[e]={entity:i,relations:{}});for(let[,e]of Object.entries(a))if(e&&typeof e=="object"&&e[C.RELATION_CONFIG]){let i=e;if(i.entity){let n=e,s=Object.entries(t.entities).find(([r,o])=>o.entity===n.entity);if(s){let[,r]=s;r.relations={...r.relations,...n.config}}}else if(i.definitions){let n=e;for(let[s,r]of Object.entries(n.definitions)){let o=n.schema[s],l=Object.entries(t.entities).find(([c,m])=>m.entity===o);if(l&&r){let[,c]=l;c.relations={...c.relations,...r}}}}}return t}var S=class{docClient;schema;retryConfig;query;constructor(t,e,i){this.retryConfig={maxAttempts:i?.maxAttempts??3,baseDelay:i?.baseDelay??100},this.docClient=new U(le.from(t),new Q(this.retryConfig)),e&&(this.schema=te(e)),this.query=new Proxy({},{get:(n,s)=>{if(typeof s!="string")return;if(!this.schema)throw new Error("No relations defined. Initialize mizzle with a relations object to use db.query.");let r=this.schema.entities[s];if(!r)throw new Error(`Entity ${s} not found in relations schema.`);return new k(this.docClient,r.entity,this.schema,s)}})}insert(t){return new J(t,this.docClient)}select(t){return new O(this.docClient,t)}batchGet(t,e){return new N(this.docClient).items(t,e)}batchWrite(t,e){return new B(this.docClient).operations(t,e)}_query(t){return new k(this.docClient,t)}update(t){return new X(t,this.docClient)}delete(t,e){return new Z(t,this.docClient,e)}async transaction(t,e){let i=new ee(this.docClient),n=await e(i);if(n.length===0)return;if(n.length>100)throw new Error("DynamoDB transactions are limited to 100 items.");let{TransactionExecutor:s}=await import("./transaction-RE7LXTGV.js");await new s(this.docClient).execute(t,n)}};function gt(a){return a instanceof ne?new S(a):"client"in a&&a.client instanceof ne?new S(a.client,a.relations,a.retry):"client"in a&&a.client?new S(a.client,a.relations,a.retry):new S(a)}export{O as a,b,N as c,A as d,B as e,Y as f,k as g,x as h,st as i,te as j,S as k,gt as l};

package/dist/{db-zHIHBm1E.d.ts → db-ZX6HGN1x.d.ts}

@@ -654,7 +654,9 @@ declare class RelationnalQueryBuilder<T extends Entity> {
     findMany<TOptions extends RelationalQueryOptions<T>>(options?: TOptions): Promise<InferRelationalModel<T, TOptions>[]>;
 }
 
-type QuerySchema<TSchema extends Record<string, any>> = {
+type QuerySchema<TSchema extends Record<string, any>> = TSchema extends MultiRelationsDefinition<infer T> ? {
+    [K in keyof T]: RelationnalQueryBuilder<T[K]>;
+} : {
     [K in keyof TSchema as TSchema[K] extends Entity ? K : never]: RelationnalQueryBuilder<TSchema[K] extends Entity ? TSchema[K] : never>;
 };
 /**

package/dist/db.d.ts

@@ -1,3 +1,3 @@
 import './operators-BVreW0ky.js';
-export { i as DynamoDB, M as MizzleConfig, Q as QuerySchema, X as mizzle } from './db-zHIHBm1E.js';
+export { i as DynamoDB, M as MizzleConfig, Q as QuerySchema, X as mizzle } from './db-ZX6HGN1x.js';
 import '@aws-sdk/client-dynamodb';

package/dist/db.js

@@ -1 +1 @@
-import{k as a,l as b}from"./chunk-NPPZW6VT.js";import"./chunk-UM3YF5EC.js";import"./chunk-GPYZK4WY.js";import"./chunk-DU7UPWBW.js";export{a as DynamoDB,b as mizzle};
+import{k as a,l as b}from"./chunk-WTOBEKFQ.js";import"./chunk-UM3YF5EC.js";import"./chunk-GPYZK4WY.js";import"./chunk-DU7UPWBW.js";export{a as DynamoDB,b as mizzle};

package/dist/index.d.ts

@@ -1,6 +1,6 @@
 import { E as Expression } from './operators-BVreW0ky.js';
 export { A as AnyTable, a as Assume, b as AtomicValues, B as BinaryExpression, C as ColumnType, c as ColumnsBuilder, d as Condition, e as ENTITY_SYMBOLS, f as Entity, g as EntityConfig, h as EntityWithColumns, F as FunctionExpression, I as INFER_MODE, i as IndexBuilder, j as IndexColumnConfig, k as InferInsertModel, l as InferModelFromColumns, m as InferSelectModel, n as InferSelectedModel, K as KeyStrategy, o as KeyStrategyType, L as LogicalExpression, M as MapColumnName, N as NULLS, O as ORDER, p as Operators, P as PhysicalTable, q as PhysicalTableConfig, R as RELATION_SYMBOLS, S as Simplify, r as StrategyCallback, s as StrategyResolution, T as TABLE_SYMBOLS, t as TableDefinition, U as Update, u as UpdateTableConfig, v as and, w as attributeExists, x as beginsWith, y as between, z as binary, D as binarySet, G as boolean, H as compositeKey, J as contains, Q as date, V as dynamoEntity, W as dynamoTable, X as eq, Y as getColumnBuilders, Z as getEntityColumns, _ as gsi, $ as gt, a0 as gte, a1 as inList, a2 as json, a3 as list, a4 as lsi, a5 as lt, a6 as lte, a7 as map, a8 as mapToLogical, a9 as number, aa as numberSet, ab as operators, ac as or, ad as prefixKey, ae as resolveStrategies, af as resolveTableName, ag as staticKey, ah as string, ai as stringSet, aj as uuid } from './operators-BVreW0ky.js';
-export { A as ActionType, a as AddAction, B as BatchGetBase, b as BatchGetBuilder, c as BatchGetResult, d as BatchWriteBase, e as BatchWriteBuilder, f as BatchWriteOperation, g as BatchWriteResult, C as ConditionCheckBuilder, D as DeleteAction, h as DeleteBuilder, i as DynamoDB, E as EntityMetadata, I as IncludeOptions, j as InferRelationalModel, k as InsertBase, l as InsertBuilder, m as InternalRelationalSchema, M as MizzleConfig, n as MultiRelationsCallback, o as MultiRelationsDefinition, Q as QuerySchema, R as Relation, p as RelationConfig, q as RelationType, r as RelationalQueryOptions, s as RelationnalQueryBuilder, t as RelationsCallback, u as RelationsDefinition, v as RelationsHelpers, w as RemoveAction, x as RetryConfig, y as RetryHandler, S as SelectBase, z as SelectBuilder, F as SelectedFields, G as SetAction, T as TransactionExecutor, H as TransactionProxy, U as UpdateAction, J as UpdateBuilder, K as add, L as addToSet, N as append, O as defineRelations, P as deleteFromSet, V as extractMetadata, W as ifNotExists, X as mizzle, Y as remove } from './db-zHIHBm1E.js';
+export { A as ActionType, a as AddAction, B as BatchGetBase, b as BatchGetBuilder, c as BatchGetResult, d as BatchWriteBase, e as BatchWriteBuilder, f as BatchWriteOperation, g as BatchWriteResult, C as ConditionCheckBuilder, D as DeleteAction, h as DeleteBuilder, i as DynamoDB, E as EntityMetadata, I as IncludeOptions, j as InferRelationalModel, k as InsertBase, l as InsertBuilder, m as InternalRelationalSchema, M as MizzleConfig, n as MultiRelationsCallback, o as MultiRelationsDefinition, Q as QuerySchema, R as Relation, p as RelationConfig, q as RelationType, r as RelationalQueryOptions, s as RelationnalQueryBuilder, t as RelationsCallback, u as RelationsDefinition, v as RelationsHelpers, w as RemoveAction, x as RetryConfig, y as RetryHandler, S as SelectBase, z as SelectBuilder, F as SelectedFields, G as SetAction, T as TransactionExecutor, H as TransactionProxy, U as UpdateAction, J as UpdateBuilder, K as add, L as addToSet, N as append, O as defineRelations, P as deleteFromSet, V as extractMetadata, W as ifNotExists, X as mizzle, Y as remove } from './db-ZX6HGN1x.js';
 export { SchemaChange, compareSchema } from './diff.js';
 import '@aws-sdk/client-dynamodb';
 import './snapshot.js';

package/dist/index.js

@@ -1 +1 @@
-import{a as Dr}from"./chunk-TOYV2M4M.js";import"./chunk-AQVECMXP.js";import{a as lr,b as kr,c as Ir,d as dr,e as hr,f as jr,g as qr,h as vr,i as zr,j as Ar,k as Er,l as Fr}from"./chunk-NPPZW6VT.js";import{A as O,B as P,C as Q,D as fr,F as xr,G as mr,H as nr,I as ir,L as sr,M as gr,N as cr,O as ur,P as Cr,Q as ar,R as br,S as wr,T as yr,a as u,b as l,c as k,d as C,e as a,f as b,g as w,h as y,i as I,j as d,k as h,l as j,m as q,n as v,o as z,p as A,q as D,r as E,s as F,t as G,u as H,v as J,w as K,x as L,y as M,z as N}from"./chunk-UM3YF5EC.js";import{a as or,b as tr,c as er,d as pr}from"./chunk-GPYZK4WY.js";import{a as p,b as f,c as x,d as m,e as n,f as i,g as s,h as g,i as c,k as R,l as S,m as T,n as U,o as V,p as W,q as X,r as Y,s as Z,t as _,u as $,v as B,w as rr}from"./chunk-DU7UPWBW.js";var o=class{constructor(t,e){this.type=t;this.config=e}};function Gr(r,t){return new o("gsi",{pk:r,sk:t})}function Hr(r){return new o("lsi",{sk:r})}export{H as AddAction,dr as BatchGetBase,Ir as BatchGetBuilder,jr as BatchWriteBase,hr as BatchWriteBuilder,l as BinaryExpression,br as ConditionCheckBuilder,K as DeleteAction,ar as DeleteBuilder,Er as DynamoDB,n as ENTITY_SYMBOLS,tr as Entity,u as Expression,C as FunctionExpression,p as INFER_MODE,o as IndexBuilder,ur as InsertBase,cr as InsertBuilder,sr as ItemSizeExceededError,k as LogicalExpression,x as NULLS,f as ORDER,or as PhysicalTable,i as RELATION_SYMBOLS,vr as Relation,qr as RelationnalQueryBuilder,J as RemoveAction,fr as RetryHandler,kr as SelectBase,lr as SelectBuilder,G as SetAction,m as TABLE_SYMBOLS,yr as TransactionExecutor,gr as TransactionFailedError,wr as TransactionProxy,F as UpdateAction,Cr as UpdateBuilder,L as add,P as addToSet,A as and,M as append,v as attributeExists,j as beginsWith,d as between,R as binary,S as binarySet,T as boolean,E as buildExpression,Dr as compareSchema,nr as compositeKey,q as contains,U as date,zr as defineRelations,Q as deleteFromSet,er as dynamoEntity,pr as dynamoTable,a as eq,Ar as extractMetadata,rr as getColumnBuilders,s as getEntityColumns,Gr as gsi,b as gt,w as gte,N as ifNotExists,h as inList,B as json,V as list,Hr as lsi,y as lt,I as lte,W as map,c as mapToLogical,Fr as mizzle,X as number,Y as numberSet,D as operators,z as or,mr as prefixKey,O as remove,ir as resolveStrategies,g as resolveTableName,xr as staticKey,Z as string,_ as stringSet,$ as uuid};
+import{a as Dr}from"./chunk-TOYV2M4M.js";import"./chunk-AQVECMXP.js";import{a as lr,b as kr,c as Ir,d as dr,e as hr,f as jr,g as qr,h as vr,i as zr,j as Ar,k as Er,l as Fr}from"./chunk-WTOBEKFQ.js";import{A as O,B as P,C as Q,D as fr,F as xr,G as mr,H as nr,I as ir,L as sr,M as gr,N as cr,O as ur,P as Cr,Q as ar,R as br,S as wr,T as yr,a as u,b as l,c as k,d as C,e as a,f as b,g as w,h as y,i as I,j as d,k as h,l as j,m as q,n as v,o as z,p as A,q as D,r as E,s as F,t as G,u as H,v as J,w as K,x as L,y as M,z as N}from"./chunk-UM3YF5EC.js";import{a as or,b as tr,c as er,d as pr}from"./chunk-GPYZK4WY.js";import{a as p,b as f,c as x,d as m,e as n,f as i,g as s,h as g,i as c,k as R,l as S,m as T,n as U,o as V,p as W,q as X,r as Y,s as Z,t as _,u as $,v as B,w as rr}from"./chunk-DU7UPWBW.js";var o=class{constructor(t,e){this.type=t;this.config=e}};function Gr(r,t){return new o("gsi",{pk:r,sk:t})}function Hr(r){return new o("lsi",{sk:r})}export{H as AddAction,dr as BatchGetBase,Ir as BatchGetBuilder,jr as BatchWriteBase,hr as BatchWriteBuilder,l as BinaryExpression,br as ConditionCheckBuilder,K as DeleteAction,ar as DeleteBuilder,Er as DynamoDB,n as ENTITY_SYMBOLS,tr as Entity,u as Expression,C as FunctionExpression,p as INFER_MODE,o as IndexBuilder,ur as InsertBase,cr as InsertBuilder,sr as ItemSizeExceededError,k as LogicalExpression,x as NULLS,f as ORDER,or as PhysicalTable,i as RELATION_SYMBOLS,vr as Relation,qr as RelationnalQueryBuilder,J as RemoveAction,fr as RetryHandler,kr as SelectBase,lr as SelectBuilder,G as SetAction,m as TABLE_SYMBOLS,yr as TransactionExecutor,gr as TransactionFailedError,wr as TransactionProxy,F as UpdateAction,Cr as UpdateBuilder,L as add,P as addToSet,A as and,M as append,v as attributeExists,j as beginsWith,d as between,R as binary,S as binarySet,T as boolean,E as buildExpression,Dr as compareSchema,nr as compositeKey,q as contains,U as date,zr as defineRelations,Q as deleteFromSet,er as dynamoEntity,pr as dynamoTable,a as eq,Ar as extractMetadata,rr as getColumnBuilders,s as getEntityColumns,Gr as gsi,b as gt,w as gte,N as ifNotExists,h as inList,B as json,V as list,Hr as lsi,y as lt,I as lte,W as map,c as mapToLogical,Fr as mizzle,X as number,Y as numberSet,D as operators,z as or,mr as prefixKey,O as remove,ir as resolveStrategies,g as resolveTableName,xr as staticKey,Z as string,_ as stringSet,$ as uuid};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aurios/mizzle",
-  "version": "1.1.3",
+  "version": "1.1.5",
   "description": "a drizzle-like orm for dynamoDB",
   "publishConfig": {
     "access": "public",

package/src/core/relations.ts

@@ -238,6 +238,18 @@ export function extractMetadata(schema: Record<string, unknown>): InternalRelati
     entities: {},
   };
 
+  // If the schema itself is a MultiRelationsDefinition, unwrap it
+  if (schema && (schema as any)[RELATION_SYMBOLS.RELATION_CONFIG] && (schema as any).definitions) {
+    const multiDef = schema as unknown as MultiRelationsDefinition;
+    for (const [key, entity] of Object.entries(multiDef.schema)) {
+      metadata.entities[key] = {
+        entity,
+        relations: multiDef.definitions[key] || {},
+      };
+    }
+    return metadata;
+  }
+
   // First pass: identify entities
   for (const [key, value] of Object.entries(schema)) {
     if (value instanceof Entity) {

package/src/db.ts

@@ -10,15 +10,21 @@ import { DeleteBuilder } from "./builders/delete";
 import { BatchGetBuilder } from "./builders/batch-get";
 import { BatchWriteBuilder, type BatchWriteOperation } from "./builders/batch-write";
 import { TransactionProxy } from "./builders/transaction";
-import { extractMetadata, type InternalRelationalSchema } from "./core/relations";
+import { extractMetadata, type InternalRelationalSchema, type MultiRelationsDefinition } from "./core/relations";
 import { RetryHandler, type RetryConfig } from "./core/retry";
 import { MizzleClient, type IMizzleClient } from "./core/client";
 
-export type QuerySchema<TSchema extends Record<string, any>> = {
-  [K in keyof TSchema as TSchema[K] extends Entity ? K : never]: RelationnalQueryBuilder<
-    TSchema[K] extends Entity ? TSchema[K] : never
-  >;
-};
+export type QuerySchema<TSchema extends Record<string, any>> = TSchema extends MultiRelationsDefinition<
+  infer T
+>
+  ? {
+      [K in keyof T]: RelationnalQueryBuilder<T[K]>;
+    }
+  : {
+      [K in keyof TSchema as TSchema[K] extends Entity ? K : never]: RelationnalQueryBuilder<
+        TSchema[K] extends Entity ? TSchema[K] : never
+      >;
+    };
 
 /**
  * DynamoDB database instance.

package/test/repro-user-issue.test.ts

@@ -0,0 +1,33 @@
+import { describe, it, expect } from "vitest";
+import { DynamoDBClient } from "@aws-sdk/client-dynamodb";
+import { dynamoTable, dynamoEntity } from "../src/core/table";
+import { defineRelations } from "../src/core/relations";
+import { string, uuid } from "../src/columns";
+import { mizzle } from "../src/db";
+
+describe("User Issue Reproduction: MultiRelationsDefinition as schema", () => {
+  const table = dynamoTable("mizzle-test", {
+    pk: string("pk"),
+    sk: string("sk"),
+  });
+
+  const shoppingLists = dynamoEntity(table, "shoppingLists", {
+    id: uuid("id"),
+    name: string("name"),
+  });
+
+  const client = new DynamoDBClient({ region: "us-east-1" });
+
+  it("should allow using MultiRelationsDefinition directly in mizzle config", () => {
+    const relations = defineRelations({ shoppingLists }, () => ({}));
+    
+    const db = mizzle({ 
+      client, 
+      relations 
+    });
+
+    // This is the part that fails in TypeScript for the user
+    // We are testing runtime behavior here, but the types should also be checked
+    expect(db.query.shoppingLists).toBeDefined();
+  });
+});