Spaces:
Sleeping
Sleeping
Update plugins/ragllm.py
Browse fileschange default to huggingface
- plugins/ragllm.py +2 -2
plugins/ragllm.py
CHANGED
|
@@ -163,7 +163,7 @@ class RagllmPlugin(Plugin):
|
|
| 163 |
updated_config = {}
|
| 164 |
|
| 165 |
# Load config file only if provider is not huggingface
|
| 166 |
-
current_provider = config.get('provider', '
|
| 167 |
if current_provider != 'huggingface':
|
| 168 |
self.config = self.load_llm_config()
|
| 169 |
|
|
@@ -220,7 +220,7 @@ class RagllmPlugin(Plugin):
|
|
| 220 |
return updated_config
|
| 221 |
|
| 222 |
def get_sidebar_config_ui(self, config: Dict[str, Any]) -> Dict[str, Any]:
|
| 223 |
-
provider = config.get('provider', '
|
| 224 |
available_models = self.get_available_models(provider)
|
| 225 |
default_model = config.get('llm_model', available_models[0] if available_models else None)
|
| 226 |
|
|
|
|
| 163 |
updated_config = {}
|
| 164 |
|
| 165 |
# Load config file only if provider is not huggingface
|
| 166 |
+
current_provider = config.get('provider', 'huggingface')
|
| 167 |
if current_provider != 'huggingface':
|
| 168 |
self.config = self.load_llm_config()
|
| 169 |
|
|
|
|
| 220 |
return updated_config
|
| 221 |
|
| 222 |
def get_sidebar_config_ui(self, config: Dict[str, Any]) -> Dict[str, Any]:
|
| 223 |
+
provider = config.get('provider', 'huggingface')
|
| 224 |
available_models = self.get_available_models(provider)
|
| 225 |
default_model = config.get('llm_model', available_models[0] if available_models else None)
|
| 226 |
|