tfrere commited on
Commit
e097fac
·
1 Parent(s): eee5a9a

improve get available model provider standalone test

Browse files
backend/tasks/get_available_model_provider.py CHANGED
@@ -126,7 +126,7 @@ def get_available_model_provider(model_name, verbose=False):
126
  if verbose:
127
  logger.error(f"Error in get_available_model_provider: {str(e)}")
128
  return None
129
-
130
  if __name__ == "__main__":
131
 
132
  models = [
@@ -143,7 +143,9 @@ if __name__ == "__main__":
143
 
144
  for model in models:
145
  provider = get_available_model_provider(model, verbose=True)
146
- providers.append(provider)
 
 
 
147
 
148
- print(f"Providers {len(providers)}: {providers}")
149
-
 
126
  if verbose:
127
  logger.error(f"Error in get_available_model_provider: {str(e)}")
128
  return None
129
+
130
  if __name__ == "__main__":
131
 
132
  models = [
 
143
 
144
  for model in models:
145
  provider = get_available_model_provider(model, verbose=True)
146
+ providers.append((model, provider))
147
+
148
+ for model, provider in providers:
149
+ print(f"Model: {model}, Provider: {provider}")
150
 
151
+ print(f"Total Providers {len(providers)}: {providers}")