Niklas Hoepner
commited on
Commit
·
2009d97
1
Parent(s):
6d12913
Fixed error handling
Browse files- 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 |
-
|
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 |
|