Niklas Hoepner commited on
Commit
67e59b5
·
1 Parent(s): 32f7c6a

Better error handling in L3Score computation

Browse files
Files changed (1) hide show
  1. L3Score.py +3 -3
L3Score.py CHANGED
@@ -128,19 +128,19 @@ class L3Score(evaluate.Metric):
128
  client = openai.OpenAI(api_key=api_key)
129
  model_names = set([model.id for model in client.models.list()])
130
  if model not in model_names:
131
- raise ValueError(f"Model {model} not found for provider {provider}, available models: {model_names}")
132
 
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
- raise 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
- raise ValueError(f"Model {model} not found for provider {provider}, available models: {model_names}")
144
 
145
  except openai.AuthenticationError as e:
146
  message = e.body["message"]
 
128
  client = openai.OpenAI(api_key=api_key)
129
  model_names = set([model.id for model in client.models.list()])
130
  if model not in model_names:
131
+ return ValueError(f"Model {model} not found for provider {provider}, available models: {model_names}")
132
 
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
 
145
  except openai.AuthenticationError as e:
146
  message = e.body["message"]