types: add missing $median pipeline stage to aggregation types

Fix #15209
Automattic · Feb 4, 2025 · 3d67419 · 3d67419
1 parent 6107403
commit 3d67419
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/test/types/PipelineStage.test.ts b/test/types/PipelineStage.test.ts
@@ -575,3 +575,12 @@ const vectorSearchStages: PipelineStage[] = [
     }
   }
 ];
+
+function median() {
+  const medianStage: PipelineStage = {
+    $median: {
+      input: ['$test01', '$test02', '$test03'],
+      method: 'approximate'
+    }
+  };
+}
diff --git a/types/pipelinestage.d.ts b/types/pipelinestage.d.ts
@@ -20,6 +20,7 @@ declare module 'mongoose' {
     | PipelineStage.ListSessions
     | PipelineStage.Lookup
     | PipelineStage.Match
+    | PipelineStage.Median
     | PipelineStage.Merge
     | PipelineStage.Out
     | PipelineStage.PlanCacheStats
@@ -187,6 +188,14 @@ declare module 'mongoose' {
       $match: FilterQuery<any>;
     }
 
+    export interface Median {
+      /** [`$median` reference](https://www.mongodb.com/docs/v7.0/reference/operator/aggregation/median/) */
+      $median: {
+        input: number | Expression,
+        method: 'approximate'
+      }
+    }
+
     export interface Merge {
       /** [`$merge` reference](https://www.mongodb.com/docs/manual/reference/operator/aggregation/merge/) */
       $merge: {