fix: openllm credentials

langgenius · Dec 28, 2023 · f392b3c · f392b3c
1 parent e030458
commit f392b3c
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/api/core/model_runtime/model_providers/openllm/llm/llm.py b/api/core/model_runtime/model_providers/openllm/llm/llm.py
@@ -147,12 +147,6 @@ def get_customizable_model_schema(self, model: str, credentials: dict) -> AIMode
         """
             used to define customizable model schema
         """
-        model_type = None
-        if credentials['model_type'] == 'text-generation':
-            model_type = ModelType.LLM
-        else:
-            raise Exception(f'Large Language Model {credentials["model_type"]} is not supported')
-
         rules = [
             ParameterRule(
                 name='temperature',
@@ -200,7 +194,7 @@ def get_customizable_model_schema(self, model: str, credentials: dict) -> AIMode
                 en_US=model
             ),
             fetch_from=FetchFrom.CUSTOMIZABLE_MODEL,
-            model_type=model_type,
+            model_type=ModelType.LLM,
             model_properties={ 
                 'mode':  LLMMode.COMPLETION,
             },