diff --git a/docs/API.md b/docs/API.md
index 16920ac..eea3704 100644
--- a/docs/API.md
+++ b/docs/API.md
@@ -229,6 +229,7 @@
 | ----- | ---- | ----- | ----------- |
 | songs | [tensorbeat.common.File](#tensorbeat.common.File) | repeated |  |
 | next_page_token | [int64](#int64) |  |  |
+| total_size | [int64](#int64) |  |  |
 
 
 
@@ -262,6 +263,7 @@
 | ----- | ---- | ----- | ----------- |
 | songs | [tensorbeat.common.File](#tensorbeat.common.File) | repeated |  |
 | next_page_token | [int64](#int64) |  |  |
+| total_size | [int64](#int64) |  |  |
 
 
 
@@ -314,6 +316,7 @@ Using an * for the value will return any song with that tag set. Using a specifi
 | ----- | ---- | ----- | ----------- |
 | songs | [tensorbeat.common.File](#tensorbeat.common.File) | repeated |  |
 | next_page_token | [int64](#int64) |  |  |
+| total_size | [int64](#int64) |  |  |
 
 
 
diff --git a/tensorbeat/datalake.proto b/tensorbeat/datalake.proto
index 9b97bc7..fad4c77 100644
--- a/tensorbeat/datalake.proto
+++ b/tensorbeat/datalake.proto
@@ -41,6 +41,7 @@ message GetSongsByTagsRequest {
 message GetSongsByTagsResponse {
     repeated tensorbeat.common.File songs = 1;
     int64 next_page_token = 2;
+    int64 total_size = 3;
 }
 
 message AddSongsRequest {
@@ -76,6 +77,7 @@ message GetAllSongsRequest {
 message GetAllSongsResponse {
     repeated tensorbeat.common.File songs = 1;
     int64 next_page_token = 2;
+    int64 total_size = 3;
 }
 
 message GetSongsByIDsRequest {
@@ -86,6 +88,7 @@ message GetSongsByIDsRequest {
 message GetSongsByIDsResponse {
     repeated tensorbeat.common.File songs = 1;
     int64 next_page_token = 2;
+    int64 total_size = 3;
 }
 
 service DatalakeService {