feat:add support for 'custom' model provider

charSLee013 · Nov 22, 2023 · 1457cab · 1457cab
1 parent 0d67284
commit 1457cab
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/api/core/model_providers/model_provider_factory.py b/api/core/model_providers/model_provider_factory.py
@@ -72,6 +72,9 @@ def get_model_provider_class(cls, provider_name: str) -> Type[BaseModelProvider]
         elif provider_name == 'localai':
             from core.model_providers.providers.localai_provider import LocalAIProvider
             return LocalAIProvider
+        elif provider_name == 'custom':
+            from core.model_providers.providers.custom_provider import CustomProvider
+            return CustomProvider
         else:
             raise NotImplementedError