reduce the number of ppl results to sample

Signed-off-by: Yulong Ruan <ruanyl@amazon.com>

package.json

@@ -133,7 +133,6 @@
   },
   "dependencies": {
     "@aws-crypto/client-node": "^3.1.1",
-    "@aws-sdk/client-bedrock-runtime": "^3.575.0",
     "@elastic/datemath": "5.0.3",
     "@elastic/eui": "npm:@opensearch-project/oui@1.5.1",
     "@elastic/good": "^9.0.1-kibana3",
@@ -237,7 +236,7 @@
     "uuid": "3.3.2",
     "whatwg-fetch": "^3.0.0",
     "yauzl": "^2.10.0"
-    
+
   },
   "devDependencies": {
     "@babel/core": "^7.22.9",

src/plugins/vis_type_vega/public/text_to_vega.ts

@@ -1,5 +1,12 @@
 import { BehaviorSubject, Observable } from 'rxjs';
-import { debounceTime, distinctUntilChanged, switchMap, tap, filter } from 'rxjs/operators';
+import {
+  debounceTime,
+  distinctUntilChanged,
+  switchMap,
+  tap,
+  filter,
+  finalize,
+} from 'rxjs/operators';
 import { HttpSetup } from 'opensearch-dashboards/public';
 
 const topN = (ppl: string, n: number) => `${ppl} | head ${n}`;
@@ -63,7 +70,7 @@ export class Text2Vega {
         if (value instanceof Error) {
           return value;
         }
-        const ppl = topN(value.ppl, 5);
+        const ppl = topN(value.ppl, 2);
         const sample = await this.http.post('/api/ppl/search', {
           body: JSON.stringify({ query: ppl, format: 'jdbc' }),
         });
@@ -83,7 +90,8 @@ export class Text2Vega {
         };
         return result;
       }),
-      tap(() => llmRunning$.next(false))
+      tap(() => llmRunning$.next(false)),
+      finalize(() => llmRunning$.next(false))
     );
   }
 

src/plugins/vis_type_vega/server/plugin.ts

@@ -29,7 +29,6 @@
  */
 
 import { schema } from '@osd/config-schema';
-import { BedrockRuntimeClient, InvokeModelCommand } from '@aws-sdk/client-bedrock-runtime';
 import { PluginInitializerContext, CoreSetup, CoreStart, Plugin } from '../../../core/server';
 import { registerVegaUsageCollector } from './usage_collector';
 import {