johannoriel commited on
Commit
dbc6bab
1 Parent(s): 89c9051

Update plugins/ragllm.py

Browse files

change default to huggingface

Files changed (1) hide show
  1. 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', 'ollama')
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', 'ollama')
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