Update model_mapping.py

FlagOpen · Jan 4, 2025 · 3c2d321 · 3c2d321
1 parent c2506f5
commit 3c2d321
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/FlagEmbedding/inference/embedder/model_mapping.py b/FlagEmbedding/inference/embedder/model_mapping.py
@@ -1,5 +1,5 @@
 from enum import Enum
-from typing import Type
+from typing import Type, List
 from dataclasses import dataclass
 from collections import OrderedDict
 
@@ -239,8 +239,8 @@ class EmbedderConfig:
 
 # TODO: Add more models, such as Jina, Stella_v5, NV-Embed, etc.
 
-def support_native_bge_model_list():
+def support_native_bge_model_list()->List[str]:
     return list(BGE_MAPPING.keys())
 
-def support_model_list():
-    return (AUTO_EMBEDDER_MAPPING.keys())
+def support_model_list()->List[str]:
+    return list(AUTO_EMBEDDER_MAPPING.keys())