feat: add some overload on with method

We can now skip the `count` parameter to be less verbose

packages/core/src/builder/builder.ts

@@ -122,8 +122,18 @@ export class Builder<
   /**
    * Apply a relationship
    */
-  public with(name: Relationships, count = 1, callback?: WithCallback) {
-    this.relationshipBuilder.apply(name as string, count, callback)
+  public with(name: Relationships, callback?: WithCallback): this
+  public with(name: Relationships, count: number, callback?: WithCallback): this
+  public with(
+    name: Relationships,
+    countOrCallback?: number | WithCallback,
+    callback?: WithCallback
+  ) {
+    if (typeof countOrCallback === 'function') {
+      this.relationshipBuilder.apply(name as string, 1, countOrCallback)
+    } else {
+      this.relationshipBuilder.apply(name as string, countOrCallback || 1, callback)
+    }
     return this
   }
 

tests/has_one.spec.ts

@@ -29,6 +29,17 @@ test.group('HasOne', (group) => {
     await database.assertHas('profile', { user_id: users[1].id }, 1)
   })
 
+  test('with overloading', async ({ database }) => {
+    const user = await UserFactory.with('profile', (profile) =>
+      profile.merge({ email: 'hey@ok.com' })
+    ).create()
+
+    await database.assertCount('user', 1)
+    await database.assertCount('profile', 1)
+
+    await database.assertHas('profile', { user_id: user.id, email: 'hey@ok.com' }, 1)
+  })
+
   test('returns relationship', async ({ expect }) => {
     const user = await UserFactory.with('profile').create()