refactor(type): Narrow options types

freshsomebody · Jul 5, 2020 · dc802d7 · dc802d7
1 parent 8a046cc
commit dc802d7
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/types.ts b/src/types.ts
@@ -12,11 +12,12 @@ export interface Keypoint {
 };
 
 export interface Options {
-  strategy?: string | Function,
+  strategy?: 'cosineSimilarity' | 'cosineDistance' | 'weightedDistance' | Function,
   customWeight?: WeightOption
 };
 
 export interface WeightOption {
-  mode: string,
+  mode: WeightOptionMode,
   scores: Record<string, number> | number[]
 }
+export type WeightOptionMode = 'multiply' | 'replace' | 'add'