fix import in teraslice-state-storage

terascope · Aug 27, 2024 · 047e9e4 · 047e9e4
1 parent 7c7f783
commit 047e9e4
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/packages/teraslice-state-storage/src/elasticsearch-state-storage/index.ts b/packages/teraslice-state-storage/src/elasticsearch-state-storage/index.ts
@@ -3,7 +3,7 @@ import {
     chunk, pMap
 } from '@terascope/utils';
 import { ClientParams, ClientResponse } from '@terascope/types';
-import esApi, { Client, BulkRecord } from '@terascope/elasticsearch-api';
+import esApi from '@terascope/elasticsearch-api';
 import { ESStateStorageConfig, MGetCacheResponse } from '../interfaces.js';
 import CachedStateStorage from '../cached-state-storage/index.js';
 
@@ -15,11 +15,11 @@ export default class ESCachedStateStorage {
     private chunkSize: number;
     private persist: boolean;
     private metaKey: string;
-    private es: Client;
+    private es: esApi.Client;
     private logger: Logger;
     public cache: CachedStateStorage<DataEntity>;
 
-    constructor(client: Client, logger: Logger, config: ESStateStorageConfig) {
+    constructor(client: esApi.Client, logger: Logger, config: ESStateStorageConfig) {
         this.index = config.index;
         this.type = config.type;
         this.concurrency = config.concurrency;
@@ -228,7 +228,7 @@ export default class ESCachedStateStorage {
         return this.es.mget(request);
     }
 
-    private _esBulkUpdatePrep(dataArray: DataEntity[]): BulkRecord[] {
+    private _esBulkUpdatePrep(dataArray: DataEntity[]): esApi.BulkRecord[] {
         return dataArray.map((doc) => ({
             action: {
                 index: {