Niklas Hoepner commited on
Commit
2009d97
·
1 Parent(s): 6d12913

Fixed error handling

Browse files
Files changed (1) hide show
  1. L3Score.py +1 -5
L3Score.py CHANGED
@@ -122,8 +122,7 @@ class L3Score(evaluate.Metric):
122
  )
123
 
124
  # Check whether the model is available
125
- print(provider,model)
126
-
127
  if provider == "openai":
128
  client = openai.OpenAI(api_key=api_key)
129
  model_names = set([model.id for model in client.models.list()])
@@ -131,17 +130,14 @@ class L3Score(evaluate.Metric):
131
  raise ValueError(f"Model {model} not found for provider {provider}, available models: {model_names}")
132
 
133
  elif provider == "deepseek":
134
- print("Checking DeepSeek model")
135
  client = openai.OpenAI(api_key=api_key,base_url="https://api.deepseek.com")
136
  model_names = [model.id for model in client.models.list()]
137
- print(model_names)
138
  if model not in model_names:
139
  raise ValueError(f"Model {model} not found for provider {provider}, available models: {model_names}")
140
 
141
  elif provider == "xai":
142
  client = openai.OpenAI(api_key=api_key, base_url="https://api.xai.com")
143
  model_names = [model.id for model in client.models.list()]
144
- print(model_names)
145
  if model not in model_names:
146
  raise ValueError(f"Model {model} not found for provider {provider}, available models: {model_names}")
147
 
 
122
  )
123
 
124
  # Check whether the model is available
125
+
 
126
  if provider == "openai":
127
  client = openai.OpenAI(api_key=api_key)
128
  model_names = set([model.id for model in client.models.list()])
 
130
  raise ValueError(f"Model {model} not found for provider {provider}, available models: {model_names}")
131
 
132
  elif provider == "deepseek":
 
133
  client = openai.OpenAI(api_key=api_key,base_url="https://api.deepseek.com")
134
  model_names = [model.id for model in client.models.list()]
 
135
  if model not in model_names:
136
  raise ValueError(f"Model {model} not found for provider {provider}, available models: {model_names}")
137
 
138
  elif provider == "xai":
139
  client = openai.OpenAI(api_key=api_key, base_url="https://api.xai.com")
140
  model_names = [model.id for model in client.models.list()]
 
141
  if model not in model_names:
142
  raise ValueError(f"Model {model} not found for provider {provider}, available models: {model_names}")
143