Niklas Hoepner
commited on
Commit
·
67e59b5
1
Parent(s):
32f7c6a
Better error handling in L3Score computation
Browse files- 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 |
-
|
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 |
-
|
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 |
-
|
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"]
|