Can I use "where" clause on joins ?

const modelId = "dummyModelId";
const userId = "dummyUserId";

const data = await prisma.model.findFirst({
  where: {
    id: modelId,
    task: {
      dataset: {
        userDatasets: {
          some: {
            userId,
          },
        },
      },
    },
  },
});

const modelId = "dummyModelId";
const userId = "dummyUserId";

const data = await db.query.model.findFirst({
  where({id}){
    return eq(id, modelId)
  }
})

const data = await db.query.model.findFirst({
  where({id}){
    return eq(id, modelId)
  },
  with:{
    task:{
      with:{
        dataset:{
          with:{
            userDatasets:{
              // 'where' clause not possible here
            }
          }
        }
      }
    }
  }
})