ADK changes

PiperOrigin-RevId: 853778465

Author: google-genai-bot

src/app/components/session-tab/session-tab.component.spec.ts

@@ -49,6 +49,7 @@ describe('SessionTabComponent', () => {
     sessionService = new MockSessionService();
     mockUiStateService = new MockUiStateService();
     mockFeatureFlagService = new MockFeatureFlagService();
+    mockFeatureFlagService.isInfinityMessageScrollingEnabledResponse.next(false);
 
     sessionService.listSessionsResponse.next({
       items: [],
@@ -369,6 +370,37 @@ describe('SessionTabComponent', () => {
         });
       });
     });
+
+    describe('when infinity scrolling is enabled', () => {
+      beforeEach(() => {
+        mockFeatureFlagService.isInfinityMessageScrollingEnabledResponse.next(
+            true);
+        sessionService.getSessionResponse.next({} as any);
+        component.reloadSession('session1');
+      });
+
+      it('fetches session with pagination params', () => {
+        expect(sessionService.getSession)
+            .toHaveBeenCalledWith(
+                component.userId, component.appName, 'session1',
+                {pageSize: 100, pageToken: ''});
+      });
+    });
+
+    describe('when infinity scrolling is disabled', () => {
+      beforeEach(() => {
+        mockFeatureFlagService.isInfinityMessageScrollingEnabledResponse.next(
+            false);
+        sessionService.getSessionResponse.next({} as any);
+        component.reloadSession('session1');
+      });
+
+      it('fetches session without pagination params', () => {
+        expect(sessionService.getSession)
+            .toHaveBeenCalledWith(
+                component.userId, component.appName, 'session1', undefined);
+      });
+    });
   });
 
   describe('on refresh', () => {

src/app/components/session-tab/session-tab.component.ts

@@ -25,7 +25,7 @@ import {MatInputModule} from '@angular/material/input';
 import {MatProgressBar} from '@angular/material/progress-bar';
 import {ActivatedRoute} from '@angular/router';
 import {of, Subject} from 'rxjs';
-import {catchError, debounceTime, map, switchMap, tap,} from 'rxjs/operators';
+import {catchError, debounceTime, map, switchMap, tap, withLatestFrom} from 'rxjs/operators';
 
 import {Session} from '../../core/models/Session';
 import {FEATURE_FLAG_SERVICE} from '../../core/services/interfaces/feature-flag';
@@ -171,10 +171,16 @@ export class SessionTabComponent implements OnInit {
 
     this.reloadSessionSubject
         .pipe(
+            withLatestFrom(
+                this.featureFlagService.isInfinityMessageScrollingEnabled()),
             switchMap(
-                (sessionId) =>
+                ([sessionId, isInfinityScrollingEnabled]) =>
                     this.sessionService
-                        .getSession(this.userId, this.appName, sessionId)
+                        .getSession(
+                            this.userId, this.appName, sessionId,
+                            isInfinityScrollingEnabled ?
+                                {pageSize: 100, pageToken: ''} :
+                                undefined)
                         .pipe(catchError(() => of(null)))),
             tap((res) => {
               if (!res) return;