add models to connection type

stargate · Jan 14, 2025 · 11db183 · 11db183
1 parent aa0f0e9
commit 11db183
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 2 deletions.
diff --git a/dist/src/driver/connection.d.ts b/dist/src/driver/connection.d.ts
@@ -2,7 +2,7 @@ import { Collection, MongooseCollectionOptions } from './collection';
 import { AstraAdmin, CollectionDescriptor, CreateTableDefinition, DataAPIDbAdmin, ListCollectionsOptions, ListTablesOptions, RawDataAPIResponse, TableDescriptor } from '@datastax/astra-db-ts';
 import { CollectionsDb, TablesDb } from './db';
 import { default as MongooseConnection } from 'mongoose/lib/connection';
-import type { ConnectOptions, Mongoose } from 'mongoose';
+import type { ConnectOptions, Mongoose, Model } from 'mongoose';
 import { DataAPIClient } from '@datastax/astra-db-ts';
 interface ConnectOptionsInternal extends ConnectOptions {
     useTables?: boolean;
@@ -25,6 +25,7 @@ export declare class Connection extends MongooseConnection {
     config?: ConnectOptionsInternal;
     baseUrl: string | null;
     baseApiPath: string | null;
+    models: Record<string, Model<unknown>>;
     constructor(base: Mongoose);
     /**
      * Helper borrowed from Mongoose to wait for the connection to finish connecting. Because Mongoose

diff --git a/dist/src/driver/connection.js b/dist/src/driver/connection.js
diff --git a/dist/src/driver/connection.js.map b/dist/src/driver/connection.js.map
diff --git a/src/driver/connection.ts b/src/driver/connection.ts
@@ -47,6 +47,7 @@ export class Connection extends MongooseConnection {
     config?: ConnectOptionsInternal;
     baseUrl: string | null = null;
     baseApiPath: string | null = null;
+    models: Record<string, Model<unknown>> = {};
 
     constructor(base: Mongoose) {
         super(base);