Niklas Hoepner commited on
Commit
36cc152
·
1 Parent(s): 67e59b5

Better error handling in L3Score computation

Browse files
Files changed (1) hide show
  1. L3Score.py +2 -0
L3Score.py CHANGED
@@ -133,12 +133,14 @@ class L3Score(evaluate.Metric):
133
  elif provider == "deepseek":
134
  client = openai.OpenAI(api_key=api_key,base_url="https://api.deepseek.com")
135
  model_names = [model.id for model in client.models.list()]
 
136
  if model not in model_names:
137
  return ValueError(f"Model {model} not found for provider {provider}, available models: {model_names}")
138
 
139
  elif provider == "xai":
140
  client = openai.OpenAI(api_key=api_key, base_url="https://api.xai.com")
141
  model_names = [model.id for model in client.models.list()]
 
142
  if model not in model_names:
143
  return ValueError(f"Model {model} not found for provider {provider}, available models: {model_names}")
144
 
 
133
  elif provider == "deepseek":
134
  client = openai.OpenAI(api_key=api_key,base_url="https://api.deepseek.com")
135
  model_names = [model.id for model in client.models.list()]
136
+ print(model_names)
137
  if model not in model_names:
138
  return ValueError(f"Model {model} not found for provider {provider}, available models: {model_names}")
139
 
140
  elif provider == "xai":
141
  client = openai.OpenAI(api_key=api_key, base_url="https://api.xai.com")
142
  model_names = [model.id for model in client.models.list()]
143
+ print(model_names)
144
  if model not in model_names:
145
  return ValueError(f"Model {model} not found for provider {provider}, available models: {model_names}")
146