44import { Guid } from '@dolittle/rudiments' ;
55import { EmbeddingsClient } from '@dolittle/runtime.contracts/Embeddings/Embeddings_grpc_pb' ;
66import { EmbeddingStoreClient } from '@dolittle/runtime.contracts/Embeddings/Store_grpc_pb' ;
7- import { ExecutionContext , TenantId } from '@dolittle/sdk.execution' ;
7+ import { ExecutionContext , TenantIdLike } from '@dolittle/sdk.execution' ;
88import { IProjectionAssociations , ProjectionsToSDKConverter } from '@dolittle/sdk.projections' ;
99import { Logger } from 'winston' ;
1010import { Embedding } from './Embedding' ;
1111import { IEmbedding } from './IEmbedding' ;
12- import { EmbeddingStore , IEmbeddingStore , EmbeddingStoreBuilder } from './Store' ;
12+ import { IEmbeddings } from './IEmbeddings' ;
13+ import { EmbeddingStoreBuilder } from './Store' ;
1314
1415/**
15- * Represents a builder for building an embedding sdk client .
16+ * Represents an implementation of { @link IEmbeddings} .
1617 */
17- export class Embeddings {
18-
18+ export class Embeddings extends IEmbeddings {
1919 /**
2020 * Initializes an instance of {@link EmbeddingStoreBuilder}.
2121 * @param {EmbeddingStoreClient } _embeddingsStoreClient The embedding store client.
@@ -29,14 +29,11 @@ export class Embeddings {
2929 private readonly _executionContext : ExecutionContext ,
3030 private readonly _embeddingAssociations : IProjectionAssociations ,
3131 private readonly _logger : Logger ) {
32+ super ( ) ;
3233 }
3334
34- /**
35- * Build an {@link IEmbeddingStore} for the given tenant.
36- * @param {TenantId | Guid | string } tenantId The tenant id.
37- * @returns {IEmbeddingStore } The embedding store.
38- */
39- forTenant ( tenantId : TenantId | Guid | string ) : IEmbedding {
35+ /** @inheritdoc */
36+ forTenant ( tenantId : TenantIdLike ) : IEmbedding {
4037 const executionContext = this . _executionContext
4138 . forTenant ( tenantId )
4239 . forCorrelation ( Guid . create ( ) ) ;
0 commit comments